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 



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

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_function_template(path, vars = {}) ⇒ Object

Copy code from internal folder to materialized app code



111
112
113
114
115
116
117
118
 
# File 'lib/jets/builders/handler_generator.rb', line 111

def copy_function_template(path, vars={})
  internal = File.expand_path("../internal", File.dirname(__FILE__))
  src = "#{internal}/app/#{path}"
  result = Jets::Erb.result(src, vars)
  dest = "#{tmp_code}/handlers/#{path}"
  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



121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
 
# File 'lib/jets/builders/handler_generator.rb', line 121

def copy_source_as_handler(fun)
  return if fun.internal?

  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".color(: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 



176
177
178
179
 
# File 'lib/jets/builders/handler_generator.rb', line 176

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 



169
170
171
172
173
174
 
# File 'lib/jets/builders/handler_generator.rb', line 169

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 



137
138
139
140
141
142
 
# File 'lib/jets/builders/handler_generator.rb', line 137

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
100
 
# File 'lib/jets/builders/handler_generator.rb', line 97

def internal_shims
  jets_base_path if Jets.custom_domain?
  s3_bucket_config if Jets.s3_event?
end

#jets_base_path ⇒ Object 



102
103
104
 
# File 'lib/jets/builders/handler_generator.rb', line 102

def jets_base_path
  copy_function_template("functions/jets/base_path.rb", stage_name: Jets::Resource::ApiGateway::Deployment.stage_name)
end

#native_function(original_path, task) ⇒ Object

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



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

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".color(:red)
      puts missing
      exit 1
    end
  end
end

#s3_bucket_config ⇒ Object 



106
107
108
 
# File 'lib/jets/builders/handler_generator.rb', line 106

def s3_bucket_config
  copy_function_template("shared/functions/jets/s3_bucket_config.rb")
end

#shared_ruby_shim(fun) ⇒ Object 



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

def shared_ruby_shim(fun)
  # Cant use native_function because that requires task. Just re-implement
  dest_path = fun.handler_dest
  source_path = dest_path.sub(/^handlers/,'app')
  FileUtils.mkdir_p(File.dirname(dest_path))
  FileUtils.cp(source_path, dest_path)
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



182
183
184
 
# File 'lib/jets/builders/handler_generator.rb', line 182

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