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 



59
60
61
62
63
64
65
66
67
68
 
# File 'lib/bolt/pal/yaml_plan/transpiler.rb', line 59

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

  Bolt::PAL::YamlPlan::Loader.from_string(@modulename, file_contents, @plan_path)
end

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

def transpile(relative_path)
  @plan_path = File.expand_path(relative_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 may not behave as expected.\n"
  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 



70
71
72
73
74
 
# File 'lib/bolt/pal/yaml_plan/transpiler.rb', line 70

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 



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

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