lib/jets/resource/function.rb in jets-0.10.4 vs lib/jets/resource/function.rb in jets-1.0.0

@@ -36,22 +36,29 @@
       {environment: { variables: variables }}
     end
 
     def environment
       env = Jets.config.environment ? Jets.config.environment.to_h : {}
-      jets_env_options = {JETS_ENV: Jets.env.to_s}
-      jets_env_options[:JETS_ENV_EXTRA] = Jets.config.env_extra if Jets.config.env_extra
-      env.deep_merge(jets_env_options)
+      env.deep_merge(jets_env)
     end
 
+    # These jets env variables are always included
+    def jets_env
+      env = {}
+      env[:JETS_ENV] = Jets.env.to_s
+      env[:JETS_ENV_EXTRA] = Jets.config.env_extra if Jets.config.env_extra
+      env[:JETS_STAGE] = Jets::Resource::ApiGateway::Deployment.stage_name
+      env
+    end
+
     # Global properties example:
     # jets defaults are in jets/default/application.rb.
     # Your application's default config/application.rb then get used. Example:
     #
     #   Jets.application.configure do
     #     config.function = ActiveSupport::OrderedOptions.new
-    #     config.function.timeout = 10
+    #     config.function.timeout = 30
     #     config.function.runtime = "nodejs8.10"
     #     config.function.memory_size = 1536
     #   end
     def global_properties
       baseline = {
@@ -180,10 +187,11 @@
         default_handler
       end
     end
 
     def code_s3_key
-      Jets::Naming.code_s3_key
+      checksum = Jets::Builders::Md5.checksums["stage/code"]
+      "jets/code/code-#{checksum}.zip" # s3_key
     end
 
     # Examples:
     #   "#{Jets.config.project_namespace}-sleep_job-perform"
     #   "demo-dev-sleep_job-perform"
\ No newline at end of file