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 



51
52
53
54
55
56
57
58
59
60
 
# File 'lib/bolt/pal/yaml_plan/transpiler.rb', line 51

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
 
# 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

  plan_string = String.new("# WARNING: This is an autogenerated plan. " \
                           "It may not behave as expected.\n" \
                           "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 



62
63
64
65
66
 
# File 'lib/bolt/pal/yaml_plan/transpiler.rb', line 62

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 



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

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