stax 0.0.1 → 0.0.2

data/lib/stax/aws/ecs.rb

@@ -0,0 +1,54 @@
+module Stax
+  module Aws
+    class Ecs < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::ECS::Client.new
+        end
+
+        def clusters(names)
+          client.describe_clusters(clusters: Array(names)).clusters
+        end
+
+        def services(cluster, services)
+          client.describe_services(cluster: cluster, services: services).services
+        end
+
+        def task_definition(name)
+          client.describe_task_definition(task_definition: name).task_definition
+        end
+
+        def list_tasks(cluster, status = :RUNNING)
+          paginate(:task_arns) do |token|
+            client.list_tasks(cluster: cluster, next_token: token, desired_status: status)
+          end
+        end
+
+        def tasks(cluster, status = :RUNNING)
+          client.describe_tasks(cluster: cluster, tasks: list_tasks(cluster, status)).tasks
+        end
+
+        def list_instances(cluster)
+          paginate(:container_instance_arns) do |token|
+            client.list_container_instances(cluster: cluster, next_token: token)
+          end
+        end
+
+        def instances(cluster)
+          client.describe_container_instances(cluster: cluster, container_instances: list_instances(cluster)).container_instances
+        end
+
+        def run(cluster, task)
+          client.run_task(cluster: cluster, task_definition: task).tasks
+        end
+
+        def stop(cluster, task)
+          client.stop_task(cluster: cluster, task: task).task
+        end
+
+      end
+    end
+  end
+end

data/lib/stax/aws/elb.rb

@@ -0,0 +1,30 @@
+module Stax
+  module Aws
+    class Elb < Sdk
+
+      COLORS = {
+        InService:    :green,
+        OutOfService: :red,
+      }
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::ElasticLoadBalancing::Client.new
+        end
+
+        def describe(elb_names)
+          paginate(:load_balancer_descriptions) do |marker|
+            client.describe_load_balancers(load_balancer_names: elb_names, marker: marker)
+          end
+        end
+
+        def instance_health(elb_name)
+          client.describe_instance_health(load_balancer_name: elb_name).instance_states
+        end
+
+      end
+
+    end
+  end
+end

data/lib/stax/aws/emr.rb

@@ -0,0 +1,28 @@
+module Stax
+  module Aws
+    class Emr < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::EMR::Client.new
+        end
+
+        def describe(id)
+          client.describe_cluster(cluster_id: id).cluster
+        end
+
+        def groups(id)
+          ## TODO paginate me
+          client.list_instance_groups(cluster_id: id).instance_groups
+        end
+
+        def instances(id, types = nil)
+          ## TODO paginate me
+          client.list_instances(cluster_id: id, instance_group_types: types).instances
+        end
+
+      end
+    end
+  end
+end

data/lib/stax/aws/iam.rb

@@ -0,0 +1,19 @@
+module Stax
+  module Aws
+    class Iam < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::IAM::Client.new
+        end
+
+        def aliases
+          client.list_account_aliases.account_aliases
+        end
+
+      end
+
+    end
+  end
+end

data/lib/stax/aws/keypair.rb

@@ -0,0 +1,26 @@
+module Stax
+  module Aws
+    class Keypair < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::EC2::Client.new
+        end
+
+        def describe(names = nil)
+          client.describe_key_pairs(key_names: names).key_pairs
+        end
+
+        def create(name)
+          client.create_key_pair(key_name: name)
+        end
+
+        def delete(name)
+          client.delete_key_pair(key_name: name)
+        end
+
+      end
+    end
+  end
+end

data/lib/stax/aws/kms.rb

@@ -0,0 +1,19 @@
+module Stax
+  module Aws
+    class Kms < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::KMS::Client.new
+        end
+
+        def describe(id)
+          client.describe_key(key_id: id).key_metadata
+        end
+
+      end
+
+    end
+  end
+end

data/lib/stax/aws/lambda.rb

@@ -0,0 +1,33 @@
+module Stax
+  module Aws
+    class Lambda < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::Lambda::Client.new
+        end
+
+        def list
+          paginate(:functions) do |marker|
+            client.list_functions(marker: marker)
+          end
+        end
+
+        def configuration(name)
+          client.get_function_configuration(function_name: name)
+        end
+
+        def code(name)
+          client.get_function(function_name: name).code.location
+        end
+
+        def invoke(opt)
+          client.invoke(opt)
+        end
+
+      end
+
+    end
+  end
+end

data/lib/stax/aws/logs.rb

@@ -0,0 +1,25 @@
+module Stax
+  module Aws
+    class Logs < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::CloudWatchLogs::Client.new
+        end
+
+        def groups(prefix = nil)
+          paginate(:log_groups) do |token|
+            client.describe_log_groups(log_group_name_prefix: prefix, next_token: token)
+          end
+        end
+
+        def streams(opt)
+          client.describe_log_streams(opt).log_streams
+        end
+
+      end
+
+    end
+  end
+end

data/lib/stax/aws/s3.rb

@@ -0,0 +1,41 @@
+module Stax
+  module Aws
+    class S3 < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::S3::Client.new
+        end
+
+        def list_buckets
+          client.list_buckets.buckets
+        end
+
+        def bucket_tags(bucket)
+          client.get_bucket_tagging(bucket: bucket).tag_set
+        rescue ::Aws::S3::Errors::NoSuchTagSet
+          []
+        end
+
+        def bucket_region(bucket)
+          client.get_bucket_location(bucket: bucket).location_constraint
+        end
+
+        def put(opt)
+          client.put_object(opt)
+        end
+
+        def get_lifecycle(bucket)
+          client.get_bucket_lifecycle_configuration(bucket: bucket).rules
+        end
+
+        def put_lifecycle(bucket, cfg)
+          client.put_bucket_lifecycle_configuration(bucket: bucket, lifecycle_configuration: cfg)
+        end
+
+      end
+
+    end
+  end
+end

data/lib/stax/aws/sdk.rb

@@ -0,0 +1,21 @@
+module Stax
+  module Aws
+    class Sdk
+
+      ## universal paginator for aws-sdk calls
+      def self.paginate(thing)
+        token = nil
+        things = []
+        loop do
+          resp = yield(token)
+          things += resp.send(thing)
+          ## some apis use marker, some use token
+          token = resp.respond_to?(:next_marker) ? resp.next_marker : resp.next_token
+          break if token.nil?
+        end
+        things
+      end
+    end
+
+  end
+end

data/lib/stax/aws/sg.rb

@@ -0,0 +1,42 @@
+module Stax
+  module Aws
+    class Sg < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::EC2::Client.new
+        end
+
+        def describe(ids)
+          client.describe_security_groups(group_ids: Array(ids)).security_groups
+        end
+
+        def authorize(id, cidr, port = 22)
+          client.authorize_security_group_ingress(
+            group_id:    id,
+            ip_protocol: :tcp,
+            from_port:   port,
+            to_port:     port,
+            cidr_ip:     cidr,
+          )
+        rescue ::Aws::EC2::Errors::InvalidPermissionDuplicate => e
+          warn(e.message)
+        end
+
+        def revoke(id, cidr, port = 22)
+          client.revoke_security_group_ingress(
+            group_id:    id,
+            ip_protocol: :tcp,
+            from_port:   port,
+            to_port:     port,
+            cidr_ip:     cidr,
+          )
+        rescue ::Aws::EC2::Errors::InvalidPermissionNotFound => e
+          warn(e.message)
+        end
+
+      end
+    end
+  end
+end

data/lib/stax/aws/sqs.rb

@@ -0,0 +1,30 @@
+module Stax
+  module Aws
+    class Sqs < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::SQS::Client.new
+        end
+
+        def get(url, attributes = :All)
+          client.get_queue_attributes(queue_url: url, attribute_names: Array(attributes)).attributes
+        end
+
+        def purge(url)
+          client.purge_queue(queue_url: url)
+        end
+
+        def queue_url(name)
+          client.get_queue_url(queue_name: name)&.queue_url
+        end
+
+        def send(opt)
+          client.send_message(opt)
+        end
+      end
+
+    end
+  end
+end

data/lib/stax/aws/ssm.rb

@@ -0,0 +1,49 @@
+module Stax
+  module Aws
+    class Ssm < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::SSM::Client.new
+        end
+
+        def instances(stack)
+          client.describe_instance_information(filters: [{key: 'tag:aws:cloudformation:stack-name', values: [stack]}]).instance_information_list
+        end
+
+        def run(opt)
+          client.send_command(opt).command
+        end
+
+        def commands
+          client.list_commands.commands
+        end
+
+        def invocation(id)
+          client.list_command_invocations(command_id: id, details: true).command_invocations
+        end
+
+        def parameters(opt)
+          paginate(:parameters) do |token|
+            client.get_parameters_by_path(opt.merge(next_token: token))
+          end
+        end
+
+        def get(opt)
+          client.get_parameters(opt).parameters
+        end
+
+        def put(opt)
+          client.put_parameter(opt)
+        end
+
+        def delete(opt)
+          client.delete_parameters(opt).deleted_parameters
+        end
+
+      end
+
+    end
+  end
+end

data/lib/stax/aws/sts.rb

@@ -0,0 +1,31 @@
+module Stax
+  module Aws
+    class Sts < Sdk
+
+      class << self
+
+        def client
+          @_client ||= ::Aws::STS::Client.new
+        end
+
+        def id
+          @_id ||= client.get_caller_identity
+        end
+
+        def account_id
+          id.account
+        end
+
+        def user_id
+          id.user_id
+        end
+
+        def user_arn
+          id.arn
+        end
+
+      end
+
+    end
+  end
+end