sumomo 0.1.0 → 0.1.1

data/lib/sumomo/stack.rb

@@ -0,0 +1,152 @@
+
+module Sumomo
+	module Stack
+
+		def hidden_value(value)
+			name = make_default_resource_name("HiddenValue")
+			if !@hidden_values
+				@hidden_values = []
+			end
+
+			@hidden_values << {
+				parameter_key: name,
+				parameter_value: value
+			}
+
+			param name, type: :string
+		end
+
+		def upload_file(name,content)
+			@store.set_raw("uploads/#{name}", content)
+		end
+
+		def make_lambda(name: nil, files:[{name:"index.js", code:""}],
+			description: "Lambda Function in #{@bucket_name}", 
+			function_key: "cloudformation/lambda/function_#{name}",
+			handler: "index.handler",
+			runtime: "nodejs4.3",
+			memory_size: 128,
+			timeout: 30,
+			role: nil)
+
+			name ||= make_default_resource_name("Lambda")
+
+			stringio = Zip::OutputStream.write_buffer do |zio|
+				files.each do |file|
+					zio.put_next_entry(file[:name])
+					if file[:code]
+						zio.write file[:code]
+					elsif file[:path]
+						zio.write File.read(file[:path])
+					else
+						raise "Files needs to be an array of objects with :name and :code or :path members"
+					end
+				end
+			end
+
+			@store.set_raw(function_key, stringio.string)
+
+			code_location = {"S3Bucket": @bucket_name, "S3Key": function_key}
+			fun = make "AWS::Lambda::Function", name: name do
+				Code code_location
+				Description description
+				MemorySize memory_size
+				Handler handler
+				Runtime runtime
+				Timeout timeout
+				Role role || exec_role.Arn
+			end
+		end
+
+		def define_custom_resource(name: nil,code:)
+
+			name ||= make_default_resource_name("CustomResource")
+
+			func = make_lambda(
+				name: name, 
+				files:[
+					{
+						name: "index.js", 
+						code: File.read( File.join(Gem.datadir("sumomo"), "custom_resource_utils.js") ).sub("{{ CODE }}", code)
+					}
+				],
+				description: "CF Resource Custom::#{name}",
+				function_key: "cloudformation/custom_resources/function_#{name}")
+
+			@custom_resources["Custom::#{name}"] = func
+		end
+
+		def make_custom(custom_resource, options = {}, &block)
+			bucket_name = @bucket_name
+			stack_make "Custom::#{custom_resource.name}", options do
+				ServiceToken custom_resource.Arn
+				Region ref("AWS::Region")
+				Bucket bucket_name
+				instance_eval(&block) if block
+			end
+		end
+
+		def make(type, options = {}, &block)
+			match = /^Custom\:\:(?<name>[a-z0-9]+)/i.match(type)
+			if match
+				if !@custom_resources[type]
+
+					resource_function_source = File.join(Gem.datadir("sumomo"), "custom_resources", "#{match[:name]}.js")
+					
+					if File.exists? resource_function_source
+						define_custom_resource(name: match[:name], code: File.read(resource_function_source))
+					else
+						throw "#{resource_function_source} does not exist"
+
+					end
+				end
+				make_custom(@custom_resources[type], options, &block)
+			else
+				stack_make(type, options, &block)
+			end
+		end
+
+		def make_exec_role
+			if @exec_role == nil
+				role_policy_doc = {
+					"Version" => "2012-10-17",
+					"Statement" => [{
+						"Effect" => "Allow",
+						"Principal" => {"Service" => ["lambda.amazonaws.com"]},
+						"Action" => ["sts:AssumeRole"]
+					}]
+				}
+				bucket_name = @bucket_name
+				@exec_role = make "AWS::IAM::Role", name: "LambdaFunctionExecutionRole" do
+					AssumeRolePolicyDocument role_policy_doc
+					Path "/"
+					Policies [
+						{
+							"PolicyName" => "lambdapolicy",
+							"PolicyDocument" => {
+								"Version" => "2012-10-17",
+								"Statement" => [{
+									"Effect" => "Allow",
+									"Action" => ["logs:CreateLogGroup","logs:CreateLogStream","logs:PutLogEvents"],
+									"Resource" => "arn:aws:logs:*:*:*"
+								},
+								{
+									"Effect" => "Allow",
+									"Action" => ["cloudformation:DescribeStacks", "ec2:Describe*", ],
+									"Resource" => "*"
+								},
+								{
+									"Effect" => "Allow",
+									"Action" => ["s3:DeleteObject", "s3:GetObject", "s3:PutObject"],
+									"Resource" => "arn:aws:s3:::#{bucket_name}/*"
+								}]
+							}
+						}
+					]
+				end
+			end
+			@exec_role
+		end
+
+	end
+end

data/lib/sumomo/version.rb

@@ -1,3 +1,3 @@
 module Sumomo
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/sumomo.gemspec

@@ -34,4 +34,6 @@ Gem::Specification.new do |spec|
   spec.add_dependency "s3cabinet"
   spec.add_dependency "aws-sdk"
   spec.add_dependency "rubyzip"
+  spec.add_dependency "activesupport"
+  spec.add_dependency "hashie"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sumomo
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - David Siaw
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-07-28 00:00:00.000000000 Z
+date: 2016-08-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -108,6 +108,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: hashie
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Sumomo
 email:
 - davidsiaw@gmail.com
@@ -123,7 +151,20 @@ files:
 - Rakefile
 - bin/console
 - bin/setup
+- data/sumomo/custom_resource_utils.js
+- data/sumomo/custom_resources/AMILookup.js
+- data/sumomo/custom_resources/AvailabilityZones.js
+- data/sumomo/custom_resources/CloudflareCNAME.js
+- data/sumomo/custom_resources/DeployTime.js
+- data/sumomo/custom_resources/SelectSpot.js
+- data/sumomo/sources/spot-watcher-poller.sh
+- data/sumomo/sources/spot-watcher.sh
 - lib/sumomo.rb
+- lib/sumomo/ec2.rb
+- lib/sumomo/momo_extensions/resource.rb
+- lib/sumomo/momo_extensions/stack.rb
+- lib/sumomo/network.rb
+- lib/sumomo/stack.rb
 - lib/sumomo/version.rb
 - sumomo.gemspec
 homepage: https://github.com/davidsiaw/sumomo
@@ -146,7 +187,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.4.6
+rubygems_version: 2.4.8
 signing_key: 
 specification_version: 4
 summary: An advanced infrastructure description language for AWS