File: modules.rb

package info (click to toggle)
puppet-agent 8.10.0-6
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 27,404 kB
  • sloc: ruby: 286,820; sh: 492; xml: 116; makefile: 88; cs: 68
file content (82 lines) | stat: -rw-r--r-- 2,555 bytes parent folder | download | duplicates (3)
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
module PuppetSpec::Modules
  class << self
    def create(name, dir, options = {})
      module_dir = File.join(dir, name)
      FileUtils.mkdir_p(module_dir)

      environment = options[:environment]

      metadata = options[:metadata]
      if metadata
        metadata[:source]  ||= 'github'
        metadata[:author]  ||= 'puppetlabs'
        metadata[:version] ||= '9.9.9'
        metadata[:license] ||= 'to kill'
        metadata[:dependencies] ||= []

        metadata[:name] = "#{metadata[:author]}/#{name}"

        File.open(File.join(module_dir, 'metadata.json'), 'w') do |f|
          f.write(metadata.to_json)
        end
      end

      tasks = options[:tasks]
      if tasks
        tasks_dir = File.join(module_dir, 'tasks')
        FileUtils.mkdir_p(tasks_dir)
        tasks.each do |task_files|
          task_files.each do |task_file|
            if task_file.is_a?(String)
              # default content to acceptable metadata
              task_file = { :name => task_file, :content => "{}" }
            end
            File.write(File.join(tasks_dir, task_file[:name]), task_file[:content])
          end
        end
      end

      if (plans = options[:plans])
        plans_dir = File.join(module_dir, 'plans')
        FileUtils.mkdir_p(plans_dir)
        plans.each do |plan_file|
          if plan_file.is_a?(String)
            # default content to acceptable metadata
            plan_file = { :name => plan_file, :content => "{}" }
          end
          File.write(File.join(plans_dir, plan_file[:name]), plan_file[:content])
        end
      end

      if (scripts = options[:scripts])
        scripts_dir = File.join(module_dir, 'scripts')
        FileUtils.mkdir_p(scripts_dir)
        scripts.each do |script_file|
          if script_file.is_a?(String)
            script_file = { :name => script_file, :content => "" }
          end
          File.write(File.join(scripts_dir, script_file[:name]), script_file[:content])
        end
      end

      (options[:files] || {}).each do |fname, content|
        path = File.join(module_dir, fname)
        FileUtils.mkdir_p(File.dirname(path))
        File.write(path, content)
      end

      Puppet::Module.new(name, module_dir, environment)
    end

    def generate_files(name, dir, options = {})
      module_dir = File.join(dir, name)
      FileUtils.mkdir_p(module_dir)

      if (metadata = options[:metadata])
        File.open(File.join(module_dir, 'metadata.json'), 'w') do |f|
          f.write(metadata.to_json)
        end
      end
    end
  end
end