Class: Bolt::PAL::YamlPlan::Transpiler

Inherits:
Object
  • Object
show all
Defined in:
lib/bolt/pal/yaml_plan/transpiler.rb

Defined Under Namespace

Classes: ConvertError

Instance Method Summary collapse

Instance Method Details

#parse_plan ⇒ Object 



60
61
62
63
64
65
66
67
68
69
70
71
72
73
 
# File 'lib/bolt/pal/yaml_plan/transpiler.rb', line 60

def parse_plan
  begin
    file_contents = File.read(@plan_path)
  rescue Errno::ENOENT
    msg = "Could not read yaml plan file: #{@plan_path}"
    raise Bolt::FileError.new(msg, @plan_path)
  end

  begin
    Bolt::PAL::YamlPlan::Loader.from_string(@modulename, file_contents, @plan_path)
  rescue Puppet::PreformattedError, StandardError => e
    raise PALError.from_preformatted_error(e)
  end
end

#transpile(plan_path) ⇒ Object 



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
 
# File 'lib/bolt/pal/yaml_plan/transpiler.rb', line 17

def transpile(plan_path)
  @plan_path = plan_path
  @modulename = Bolt::Util.module_name(@plan_path)
  @filename = @plan_path.split(File::SEPARATOR)[-1]
  validate_path

  plan_object = parse_plan
  param_descriptions = plan_object.parameters.map do |param|
    str = String.new("# @param #{param.name}")
    str << " #{param.description}" if param.description
    str
  end.join("\n")

  plan_string = String.new('')
  plan_string << "# #{plan_object.description}\n" if plan_object.description
  plan_string << "# WARNING: This is an autogenerated plan. It might not behave as expected.\n"
  plan_string << "# @api #{plan_object.private ? 'private' : 'public'}\n" unless plan_object.private.nil?
  plan_string << "#{param_descriptions}\n" unless param_descriptions.empty?

  plan_string << "plan #{plan_object.name}("
  # Parameters are Bolt::PAL::YamlPlan::Parameter
  plan_object.parameters&.each_with_index do |param, i|
    plan_string << param.transpile

    # If it's the last parameter add a newline and no comma
    last = i + 1 == plan_object.parameters.length ? "\n" : ","
    # This encodes strangely if we << directly to plan_string
    plan_string << last
  end
  plan_string << ") {\n"

  plan_object.steps&.each do |step|
    plan_string << step.transpile
  end

  plan_string << "\n  return #{Bolt::Util.to_code(plan_object.return)}\n" if plan_object.return
  plan_string << "}"
  # We always print the plan, even if there's an error
  puts plan_string
  validate_plan(plan_string)
  plan_string
end

#validate_path ⇒ Object 



75
76
77
78
79
 
# File 'lib/bolt/pal/yaml_plan/transpiler.rb', line 75

def validate_path
  unless File.extname(@filename) == ".yaml"
    raise ConvertError.new("You can only convert plans written in yaml", @plan_path)
  end
end

#validate_plan(plan) ⇒ Object 



81
82
83
84
85
86
 
# File 'lib/bolt/pal/yaml_plan/transpiler.rb', line 81

def validate_plan(plan)
  Puppet::Pops::Parser::EvaluatingParser.new.parse_string(plan)
rescue Puppet::Error => e
  $stderr.puts "The converted puppet plan contains invalid puppet code: #{e.message}"
  exit 1
end