kumogata-template 0.0.1

data/template/redshift-cluster-parameter-group.rb

@@ -0,0 +1,19 @@
+#
+# Redshift ClusterParameterGroup resource
+# http://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-resource-redshift-clusterparametergroup.html
+#
+require 'kumogata/template/helper'
+
+name = _resource_name(args[:name], "redshift cluster parameter group")
+description = args[:description] || "#{args[:name]} redshift cluster parameter group description"
+family = args[:family] || "redshift-1.0"
+parameters = args[:parameters] || []
+
+_(name) do
+  Type "AWS::Redshift::ClusterParameterGroup"
+  Properties do
+    Description description
+    ParameterGroupFamily family
+    Parameters parameters
+  end
+end

data/template/redshift-cluster-subnet-group.rb

@@ -0,0 +1,17 @@
+#
+# Redshift ClusterSubnetGroup resource
+# http://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-resource-redshift-clustersubnetgroup.html
+#
+require 'kumogata/template/helper'
+
+name = _resource_name(args[:name], "redshift cluster subnet group")
+description = args[:description] || "#{args[:name]} redshift cluster subnet group description"
+subnets = _ref_array("subnets", args, "subnet")
+
+_(name) do
+  Type "AWS::Redshift::ClusterSubnetGroup"
+  Properties do
+    Description description
+    SubnetIds subnets
+  end
+end

data/template/redshift-cluster.rb

@@ -0,0 +1,60 @@
+#
+# Redshift cluster resource
+# http://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-resource-redshift-cluster.html
+#
+require 'kumogata/template/helper'
+
+name = _resource_name(args[:name], "redshift cluster")
+allow = _bool("allow", args, true)
+snapshot_retention = args[:snapshot_retention] || DEFAULT_SNAPSHOT_NUM
+az = _availability_zone(args, false)
+parameter = _ref_string("parameter", args, "redshift cluster parameter group")
+subnet = _ref_string("subnet", args, "redshift cluster subnet group")
+type = _valid_values(args[:type], [ "single-node", "multi-node" ], "single-node")
+version = args[:version] || ""
+db_name = _ref_string("db_name", args, "db name")
+elastic = args[:elastic] || ""
+encrypted = _bool("encrypted", args, false)
+user_name = _ref_string("user_name", args, "cluster master user name")
+user_password = _ref_string("user_password", args, "cluster master user password")
+node = _valid_values(args[:node], REDSHIFT_NODE_TYPES, REDSHIFT_DEFAULT_NODE_TYPE)
+num = args[:num] || 1
+owner = args[:owner] || ""
+port = _ref_string("port", args, "cluster port")
+port = PORT[:redshift] if port.empty?
+maintenance = _maintenance_window("redshift", args[:maintenance] || DEFAULT_MAINTENANCE_TIME[:redshift])
+publicly = _bool("publicly", args, false)
+snapshot_cluster = args[:snapshot_cluster] || ""
+snapshot_id = args[:snapshot_id] || ""
+security_groups = _ref_array("security_groups", args, "security group")
+
+_(name) do
+  Type "AWS::Redshift::Cluster"
+  Properties do
+    AllowVersionUpgrade allow
+    AutomatedSnapshotRetentionPeriod snapshot_retention
+    AvailabilityZone az unless az.empty?
+    ClusterParameterGroupName parameter unless parameter.empty?
+    #ClusterSecurityGroups
+    ClusterSubnetGroupName subnet unless subnet.empty?
+    ClusterType type
+    ClusterVersion version unless version.empty?
+    DBName db_name
+    ElasticIp elastic unless elastic.empty?
+    Encrypted encrypted if encrypted == true
+    #HsmClientCertificateIdentifie
+    #HsmConfigurationIdentifier
+    #KmsKeyId
+    MasterUsername user_name
+    MasterUserPassword user_password
+    NodeType node
+    NumberOfNodes num if type == "multi-node"
+    OwnerAccount owner unless owner.empty?
+    Port port
+    PreferredMaintenanceWindow maintenance
+    PubliclyAccessible publicly
+    SnapshotClusterIdentifier  snapshot_cluster unless snapshot_cluster.empty?
+    SnapshotIdentifier snapshot_id unless snapshot_id.empty?
+    VpcSecurityGroupIds security_groups unless security_groups.empty?
+  end
+end

data/template/s3-bucket-policy.rb

@@ -0,0 +1,20 @@
+#
+# S3 bucket policy resource
+# http://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-s3-policy.html
+#
+require 'kumogata/template/helper'
+require 'kumogata/template/iam'
+
+name = _resource_name(args[:name], "bucket policy")
+bucket = _ref_name("bucket", args)
+
+_(name) do
+  Type "AWS::S3::BucketPolicy"
+  Properties do
+    BucketName bucket
+    PolicyDocument do
+      Version "2012-10-17"
+      Statement _iam_policy_document("policy_document", args)
+    end
+  end
+end

data/template/s3-bucket.rb

@@ -0,0 +1,42 @@
+#
+# S3 bucket resource
+# https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-s3-bucket.html
+#
+require 'kumogata/template/helper'
+require 'kumogata/template/s3'
+
+name = _resource_name(args[:name], "bucket")
+access = _valid_values(args[:access],
+                       %w( AuthenticatedRead AwsExecRead BucketOwnerRead
+                           BucketOwnerFullControl LogDeliveryWrite
+                           Private PublicRead PublicReadWrite ),
+                       "Private")
+access = "PublicRead" if args.key? :website
+bucket = _ref_name("bucket", args)
+cors = _s3_cors(args)
+lifecycle = _s3_lifecycle(args)
+logging = _s3_logging(args)
+notification = _s3_notification(args)
+replication = _s3_replication(args)
+tags = _tags(args)
+versioning = _s3_versioning(args)
+website = _s3_website(args)
+deletion_policy = _valid_values(args[:deletion_policy],
+                                %w( Delete Retain Snapshot ), "Retain")
+
+_(name) do
+  Type "AWS::S3::Bucket"
+  Properties do
+    AccessControl access
+    BucketName bucket
+    CorsConfiguration cors unless cors.empty?
+    LifecycleConfiguration lifecycle unless lifecycle.empty?
+    LoggingConfiguration logging unless logging.empty?
+    NotificationConfiguration notification unless notification.empty?
+    ReplicationConfiguration replication unless replication.empty?
+    Tags tags
+    VersioningConfiguration versioning unless versioning.empty?
+    WebsiteConfiguration website unless website.empty?
+  end
+  DeletionPolicy deletion_policy
+end

data/template/sns-topic.rb

@@ -0,0 +1,20 @@
+#
+# SNS Topic resource
+# http://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-sns-topic.html
+#
+require 'kumogata/template/helper'
+require 'kumogata/template/sns'
+
+name = _resource_name(args[:name], "topic")
+display = args[:display] || ""
+subscription = _sns_subscription(args)
+topic = _ref_name("topic", args)
+
+_(name) do
+  Type "AWS::SNS::Topic"
+  Properties do
+    DisplayName display
+    Subscription subscription
+    TopicName topic
+  end
+end

data/template/sqs-queue.rb

@@ -0,0 +1,27 @@
+#
+# SQS Queue resource
+# http://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-sqs-queues.html
+#
+require 'kumogata/template/helper'
+
+name = _resource_name(args[:name], "queue")
+deplay = args[:deplay] || 0
+max = args[:max] || 262144   # default 256KiB
+retention = args[:retention] || 345600   # default 4 days
+queue = _ref_name("queue", args)
+receive = args[:receive] || 0
+redrive = args[:redrive] || ""
+visibility = args[:visibility] || 30   # default 30 seconds
+
+_(name) do
+  Type "AWS::SQS::Queue"
+  Properties do
+    DelaySeconds deplay
+    MaximumMessageSize max
+    MessageRetentionPeriod retention
+    QueueName queue
+    ReceiveMessageWaitTimeSeconds receive if 1 <= receive.to_i and receive.to_i <= 20
+    RedrivePolicy redrive unless redrive.empty?
+    VisibilityTimeout visibility
+  end
+end

data/test/_template.rb

@@ -0,0 +1,26 @@
+require 'abstract_unit'
+
+class Test < Minitest::Test
+  def test_normal
+    template = <<-EOS
+_ "test", { }
+    EOS
+    act_template = run_client_as_json(template)
+    exp_template = <<-EOS
+{
+  "Test": {
+    "Type": "AWS::::",
+    "Properties": {
+      "": "",
+      "": [
+        {
+          "": ""
+        }
+      ]
+    }
+  }
+}
+    EOS
+    assert_equal exp_template.chomp, act_template
+  end
+end

data/test/abstract_unit.rb

@@ -0,0 +1,101 @@
+require 'minitest/autorun'
+require 'kumogata'
+require 'kumogata/argument_parser'
+require 'json'
+require 'tempfile'
+
+# for only test
+ENV['TZ'] = 'Asia/Tokyo'
+
+class Kumogata::Client
+  def define_template_func(scope, path_or_url)
+    functions = ""
+    Dir.glob("template/*.rb").all? do |file|
+      functions << include_func(path_or_url, file)
+      functions << "\n\n"
+    end
+
+    scope.instance_eval(<<-EOS)
+#{functions}
+
+      def _include(file, args = {})
+        path = file.dup
+
+        unless path =~ %r|\\A/| or path =~ %r|\\A\\w+://|
+          path = File.expand_path(File.join(File.dirname(#{path_or_url.inspect}), path))
+        end
+
+        open(path) {|f| instance_eval(f.read) }
+      end
+
+      def _path(path, value = nil, &block)
+        if block
+          value = Dslh::ScopeBlock.nest(binding, 'block')
+        end
+
+        @__hash__[path] = value
+      end
+
+      def _outputs_filter(&block)
+        @__hash__[:_outputs_filter] = block
+      end
+
+      def _post(options = {}, &block)
+        commands = Dslh::ScopeBlock.nest(binding, 'block')
+
+        @__hash__[:_post] = {
+          :options  => options,
+          :commands => commands,
+        }
+      end
+    EOS
+  end
+
+
+  private  ###########################################################
+
+  def include_func(path_or_url, file)
+    <<-EOS
+      def _#{get_funcname(file)}(name, args = {})
+        args[:name] = name unless args.key? :name
+        open("#{file}") {|f| instance_eval(f.read) }
+      end
+    EOS
+  end
+
+  def get_funcname(file)
+    File.basename(file, '.rb').gsub('-', '_')
+  end
+end
+
+
+def tempfile(content, template_ext = nil)
+  basename = "#{File.basename __FILE__}.#{$$}"
+  basename = [basename, template_ext] if template_ext
+
+  Tempfile.open(basename) do |f|
+    f << content
+    f.flush
+    f.rewind
+    yield(f)
+  end
+end
+
+def run_client(template)
+  $stdout = open('/dev/null', 'w') unless ENV['DEBUG']
+
+  kumogata_options = Kumogata::ArgumentParser::DEFAULT_OPTIONS
+  kumogata_options[:output_format] = 'json'
+  kumogata_options[:result_log] = '/dev/null'
+  kumogata_options[:command_result_log] = '/dev/null'
+  template_ext = '.rb'
+
+  template = tempfile(template, template_ext) do |f|
+    Kumogata::Client.new(kumogata_options).send(:evaluate_template, f, f.path)
+  end
+end
+
+def run_client_as_json(template)
+  eval_template = run_client(template)
+  JSON.pretty_generate(eval_template)
+end

data/test/autoscaling_test.rb

@@ -0,0 +1,109 @@
+require 'abstract_unit'
+require 'kumogata/template/autoscaling'
+
+class AutoscalingTest < Minitest::Test
+  def test_autoscaling_metrics
+    template = <<-EOS
+Test _autoscaling_metrics
+    EOS
+    act_template = run_client_as_json(template)
+    exp_template = <<-EOS
+{
+  "Test": {
+    "Granularity": "1Minute",
+    "Metrics": [
+      "GroupMinSize",
+      "GroupMaxSize",
+      "GroupDesiredCapacity",
+      "GroupInServiceInstances",
+      "GroupPendingInstances",
+      "GroupStandbyInstances",
+      "GroupTerminatingInstances",
+      "GroupTotalInstances"
+    ]
+  }
+}
+    EOS
+    assert_equal exp_template.chomp, act_template
+  end
+
+  def test_autoscaling_notification
+    template = <<-EOS
+Test _autoscaling_notification(topic_arn: "test")
+    EOS
+    act_template = run_client_as_json(template)
+    exp_template = <<-EOS
+{
+  "Test": {
+    "NotificationTypes": [
+      "autoscaling:EC2_INSTANCE_LAUNCH",
+      "autoscaling:EC2_INSTANCE_LAUNCH_ERROR",
+      "autoscaling:EC2_INSTANCE_TERMINATE",
+      "autoscaling:EC2_INSTANCE_TERMINATE_ERROR",
+      "autoscaling:TEST_NOTIFICATION"
+    ],
+    "TopicARN": "test"
+  }
+}
+    EOS
+    assert_equal exp_template.chomp, act_template
+  end
+
+  def test_autoscaling_step
+    template = <<-EOS
+Test _autoscaling_step(scaling: 10)
+    EOS
+    act_template = run_client_as_json(template)
+    exp_template = <<-EOS
+{
+  "Test": {
+    "ScalingAdjustment": "10"
+  }
+}
+    EOS
+    assert_equal exp_template.chomp, act_template
+  end
+
+  def test_autoscaling_tags
+    template = <<-EOS
+Test _autoscaling_tags(name: "test")
+    EOS
+    act_template = run_client_as_json(template)
+    exp_template = <<-EOS
+{
+  "Test": [
+    {
+      "Key": "Name",
+      "Value": {
+        "Fn::Join": [
+          "-",
+          [
+            {
+              "Ref": "Service"
+            },
+            "test"
+          ]
+        ]
+      },
+      "PropagateAtLaunch": "true"
+    },
+    {
+      "Key": "Service",
+      "Value": {
+        "Ref": "Service"
+      },
+      "PropagateAtLaunch": "true"
+    },
+    {
+      "Key": "Version",
+      "Value": {
+        "Ref": "Version"
+      },
+      "PropagateAtLaunch": "true"
+    }
+  ]
+}
+    EOS
+    assert_equal exp_template.chomp, act_template
+  end
+end