Class: Jets::Builders::HandlerGenerator

Inherits:
Object
  • Object
show all
Defined in:
lib/jets/builders/handler_generator.rb

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.build! ⇒ Object 



12
13
14
 
# File 'lib/jets/builders/handler_generator.rb', line 12

def self.build!
  new.build
end

Instance Method Details

#app_files ⇒ Object 



58
59
60
 
# File 'lib/jets/builders/handler_generator.rb', line 58

def app_files
  Jets::Commands::Build.app_files
end

#app_ruby_shims ⇒ Object 



38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/jets/builders/handler_generator.rb', line 38

def app_ruby_shims
  app_files.each do |path|
    # Generates one shim for each app class: controller, job, etc
    vars = Jets::Builders::ShimVars::App.new(path)
    if path.include?('app/functions')
      copy_simple_function(path)
    else
      generate_handler(vars)
    end
  end
end

#build ⇒ Object 



16
17
18
19
20
21
22
 
# File 'lib/jets/builders/handler_generator.rb', line 16

def build
  generate_data_yaml
  app_ruby_shims
  poly_shims
  shared_shims
  internal_shims
end

#common_base_shim ⇒ Object 



147
148
149
150
151
152
153
 
# File 'lib/jets/builders/handler_generator.rb', line 147

def common_base_shim
  vars = Jets::Builders::ShimVars::Base.new
  result = evaluate_template("shim.js", vars)
  dest = "#{tmp_code}/handlers/shim.js"
  FileUtils.mkdir_p(File.dirname(dest))
  IO.write(dest, result)
end

#copy_simple_function(source_path) ⇒ Object

source_path: app/functions/simple.rb



51
52
53
54
55
56
 
# File 'lib/jets/builders/handler_generator.rb', line 51

def copy_simple_function(source_path)
  # Handler: handlers/controllers/posts_controller.handle
  dest_path = source_path.sub('app/functions', 'handlers/functions')
  FileUtils.mkdir_p(File.dirname(dest_path))
  FileUtils.cp(source_path, dest_path)
end

#copy_source_as_handler(fun) ⇒ Object

app/shared/functions/kevin.py => /tmp/jets/demo/app_root/handlers/shared/functions/kevin.py



112
113
114
115
116
117
118
119
120
121
122
123
124
 
# File 'lib/jets/builders/handler_generator.rb', line 112

def copy_source_as_handler(fun)
  source_path = fun.source_file
  unless source_path
    attributes = fun.template.values.first
    function_name = attributes['Properties']['FunctionName']
    puts "WARN: missing source file for: '#{function_name}' function".colorize(:yellow)
    return
  end

  dest_path = "#{tmp_code}/#{fun.handler_dest}"
  FileUtils.mkdir_p(File.dirname(dest_path))
  FileUtils.cp(source_path, dest_path)
end

#evaluate_template(template_file, vars) ⇒ Object 



162
163
164
165
 
# File 'lib/jets/builders/handler_generator.rb', line 162

def evaluate_template(template_file, vars)
  template_path = File.expand_path("../templates/#{template_file}", __FILE__)
  Jets::Erb.result(template_path, vars: vars)
end

#generate_data_yaml ⇒ Object

The handlers/data.yml is used by the shims



25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/jets/builders/handler_generator.rb', line 25

def generate_data_yaml
  vars = Jets::Builders::ShimVars::Base.new
  data = {
    "s3_bucket" => vars.s3_bucket
  }
  data["rack_zip"] = vars.rack_zip if vars.rack_zip

  content = YAML.dump(data)
  path = "#{tmp_code}/handlers/data.yml"
  FileUtils.mkdir_p(File.dirname(path))
  IO.write(path, content)
end

#generate_handler(vars) ⇒ Object 



155
156
157
158
159
160
 
# File 'lib/jets/builders/handler_generator.rb', line 155

def generate_handler(vars)
  result = evaluate_template("handler.rb", vars)
  dest = "#{tmp_code}/#{vars.dest_path}"
  FileUtils.mkdir_p(File.dirname(dest))
  IO.write(dest, result)
end

#get_source_path(original_path, task) ⇒ Object 



126
127
128
129
130
131
 
# File 'lib/jets/builders/handler_generator.rb', line 126

def get_source_path(original_path, task)
  folder = original_path.sub(/\.rb$/,'')
  lang_folder = "#{folder}/#{task.lang}"
  root = Jets.root unless original_path.include?("lib/jets/internal")
  "#{root}#{lang_folder}/#{task.meth}#{task.lang_ext}"
end

#internal_shims ⇒ Object 



97
98
99
 
# File 'lib/jets/builders/handler_generator.rb', line 97

def internal_shims
  jets_base_path if Jets.custom_domain?
end

#jets_base_path ⇒ Object 



101
102
103
104
105
106
107
108
109
 
# File 'lib/jets/builders/handler_generator.rb', line 101

def jets_base_path
  path = "jets/base_path.rb"
  internal = File.expand_path("../internal", File.dirname(__FILE__))
  src = "#{internal}/app/functions/#{path}"
  result = Jets::Erb.result(src, stage_name: Jets::Resource::ApiGateway::Deployment.stage_name)
  dest = "#{tmp_code}/handlers/functions/#{path}"
  FileUtils.mkdir_p(File.dirname(dest))
  IO.write(dest, result)
end

#native_function(original_path, task) ⇒ Object

Builds and copies over the native source code: python or node



134
135
136
137
138
139
140
 
# File 'lib/jets/builders/handler_generator.rb', line 134

def native_function(original_path, task)
  source_path = get_source_path(original_path, task)
  # Handler: handlers/controllers/posts_controller.handle
  dest_path = "#{tmp_code}/#{task.handler_path}"
  FileUtils.mkdir_p(File.dirname(dest_path))
  FileUtils.cp(source_path, dest_path)
end

#poly_shims ⇒ Object 



62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
 
# File 'lib/jets/builders/handler_generator.rb', line 62

def poly_shims
  missing = []

  app_files.each do |path|
    vars = Jets::Builders::ShimVars::App.new(path)
    poly_tasks = vars.klass.tasks.select { |t| t.lang != :ruby }
    poly_tasks.each do |task|
      source_path = get_source_path(path, task)
      if File.exist?(source_path)
        native_function(path, task)
      else
        missing << source_path
      end
    end

    unless missing.empty?
      puts "ERROR: Missing source files. Please make sure these source files exist or remove their declarations".colorize(:red)
      puts missing
      exit 1
    end
  end
end

#shared_ruby_shim(fun) ⇒ Object 



142
143
144
145
 
# File 'lib/jets/builders/handler_generator.rb', line 142

def shared_ruby_shim(fun)
  vars = Jets::Builders::ShimVars::Shared.new(fun)
  generate_handler(vars)
end

#shared_shims ⇒ Object 



85
86
87
88
89
90
91
92
93
94
95
 
# File 'lib/jets/builders/handler_generator.rb', line 85

def shared_shims
  Jets::Stack.subclasses.each do |subclass|
    subclass.functions.each do |fun|
      if fun.lang.to_s == "ruby"
        shared_ruby_shim(fun)
      else
        copy_source_as_handler(fun)
      end
    end
  end
end

#tmp_code ⇒ Object

TODO: move CodeBuilder.tmp_code to a common level for HandlerGenerator and CodeBuilder



168
169
170
 
# File 'lib/jets/builders/handler_generator.rb', line 168

def tmp_code
  "#{Jets.build_root}/#{CodeBuilder.tmp_code}"
end