summaryrefslogtreecommitdiff
path: root/lib/chef/resource/powershell_package_source.rb
blob: b3bad41f650b187abc3c12a0ea05ec849f4ab1d2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
# Author:: Tor Magnus Rakvåg (tm@intility.no)
# Copyright:: 2018, Intility AS
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

require "chef/resource"
require "chef/json_compat"

class Chef
  class Resource
    class PowershellPackageSource < Chef::Resource
      preview_resource true
      resource_name "powershell_package_source"

      description "Use the powershell_package_source resource to register a PowerShell package repository."
      introduced "14.3"

      property :source_name, String,
               description: "The name of the package source.",
               name_property: true

      property :url, String,
               description: "The url to the package source.",
               required: true

      property :trusted, [TrueClass, FalseClass],
               description: "Whether or not to trust packages from this source.",
               default: false

      property :provider_name, String,
               equal_to: %w{ Programs msi NuGet msu PowerShellGet psl chocolatey },
               validation_message: "The following providers are supported: 'Programs', 'msi', 'NuGet', 'msu', 'PowerShellGet', 'psl' or 'chocolatey'",
               description: "The package management provider for the source. It supports the following providers: 'Programs', 'msi', 'NuGet', 'msu', 'PowerShellGet', 'psl' and 'chocolatey'.",
               default: "NuGet"

      property :publish_location, String,
               description: "The url where modules will be published to for this source. Only valid if the provider is 'PowerShellGet'."

      property :script_source_location, String,
               description: "The url where scripts are located for this source. Only valid if the provider is 'PowerShellGet'."

      property :script_publish_location, String,
               description: "The location where scripts will be published to for this source. Only valid if the provider is 'PowerShellGet'."

      load_current_value do
        cmd = load_resource_state_script(name)
        repo = powershell_out!(cmd)
        status = Chef::JSONCompat.from_json(repo.stdout)
        url status["url"].nil? ? "not_set" : status["url"]
        trusted status["trusted"]
        provider_name status["provider_name"]
        publish_location status["publish_location"]
        script_source_location status["script_source_location"]
        script_publish_location status["script_publish_location"]
      end

      action :register do
        description "Registers and updates the powershell package source."
        # TODO: Ensure package provider is installed?
        if psrepository_cmdlet_appropriate?
          if package_source_exists?
            converge_if_changed :url, :trusted, :publish_location, :script_source_location, :script_publish_location do
              update_cmd = build_ps_repository_command("Set", new_resource)
              res = powershell_out(update_cmd)
              raise "Failed to update #{new_resource.source_name}: #{res.stderr}" unless res.stderr.empty?
            end
          else
            converge_by("register source: #{new_resource.source_name}") do
              register_cmd = build_ps_repository_command("Register", new_resource)
              res = powershell_out(register_cmd)
              raise "Failed to register #{new_resource.source_name}: #{res.stderr}" unless res.stderr.empty?
            end
          end
        else
          if package_source_exists?
            converge_if_changed :url, :trusted, :provider_name do
              update_cmd = build_package_source_command("Set", new_resource)
              res = powershell_out(update_cmd)
              raise "Failed to update #{new_resource.source_name}: #{res.stderr}" unless res.stderr.empty?
            end
          else
            converge_by("register source: #{new_resource.source_name}") do
              register_cmd = build_package_source_command("Register", new_resource)
              res = powershell_out(register_cmd)
              raise "Failed to register #{new_resource.source_name}: #{res.stderr}" unless res.stderr.empty?
            end
          end
        end
      end

      action :unregister do
        description "Unregisters the powershell package source."
        if package_source_exists?
          unregister_cmd = "Get-PackageSource -Name '#{new_resource.source_name}' | Unregister-PackageSource"
          converge_by("unregister source: #{new_resource.source_name}") do
            res = powershell_out(unregister_cmd)
            raise "Failed to unregister #{new_resource.source_name}: #{res.stderr}" unless res.stderr.empty?
          end
        end
      end

      action_class do
        def package_source_exists?
          cmd = powershell_out!("(Get-PackageSource -Name '#{new_resource.source_name}').Name")
          cmd.stdout.downcase.strip == new_resource.source_name.downcase
        end

        def psrepository_cmdlet_appropriate?
          new_resource.provider_name == "PowerShellGet"
        end

        def build_ps_repository_command(cmdlet_type, new_resource)
          cmd = "#{cmdlet_type}-PSRepository -Name '#{new_resource.source_name}'"
          cmd << " -SourceLocation '#{new_resource.url}'" if new_resource.url
          cmd << " -InstallationPolicy '#{new_resource.trusted ? "Trusted" : "Untrusted"}'"
          cmd << " -PublishLocation '#{new_resource.publish_location}'" if new_resource.publish_location
          cmd << " -ScriptSourceLocation '#{new_resource.script_source_location}'" if new_resource.script_source_location
          cmd << " -ScriptPublishLocation '#{new_resource.script_publish_location}'" if new_resource.script_publish_location
          cmd
        end

        def build_package_source_command(cmdlet_type, new_resource)
          cmd = "#{cmdlet_type}-PackageSource -Name '#{new_resource.source_name}'"
          cmd << " -Location '#{new_resource.url}'" if new_resource.url
          cmd << " -Trusted:#{new_resource.trusted ? "$true" : "$false"}"
          cmd << " -ProviderName '#{new_resource.provider_name}'" if new_resource.provider_name
          cmd
        end
      end
    end

    private

    def load_resource_state_script(name)
      <<-EOH
        if(Get-PackageSource -Name '#{name}' -ErrorAction SilentlyContinue) {
            if ((Get-PackageSource -Name '#{name}').ProviderName -eq 'PowerShellGet') {
                (Get-PSRepository -Name '#{name}') | Select @{n='source_name';e={$_.Name}}, @{n='url';e={$_.SourceLocation}},
                @{n='trusted';e={$_.Trusted}}, @{n='provider_name';e={$_.PackageManagementProvider}}, @{n='publish_location';e={$_.PublishLocation}},
                @{n='script_source_location';e={$_.ScriptSourceLocation}}, @{n='script_publish_location';e={$_.ScriptPublishLocation}} | ConvertTo-Json
            }
            else {
                (Get-PackageSource -Name '#{name}') | Select @{n='source_name';e={$_.Name}}, @{n='url';e={$_.Location}},
                @{n='provider_name';e={$_.ProviderName}}, @{n='trusted';e={$_.IsTrusted}} | ConvertTo-Json
            }
        }
        else {
            "" | Select source_name, url, provider_name, trusted | ConvertTo-Json
        }
      EOH
    end
  end
end