dynamocli 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ff41269be712ae8e16cab262ee39da16da6eba5adc8efaa2d6316a357133bc91
-  data.tar.gz: 94f3af39b5c2a3373a7adb4fae76967302170a3a74b01068184671896dbb9f0b
+  metadata.gz: d472e7b32f0ee64582e39cb00bcab25ab4070b025806793fd8595e2acba9bedf
+  data.tar.gz: e7838ac7811278a88629e17fd521fcce36b60b96b37c3ad8349a3ccbdcafc4d1
 SHA512:
-  metadata.gz: a4040a8df21120f3b5e41baf4a3d5b50afe8f95983fc35fcad3f468afc4142ecebff8629e6478d53dc9c7fd420b222fd5531eeb664b3ca7b1e931cd31adbf382
-  data.tar.gz: 1ab22e6e4abc793cfdfc25dcee422530e13209f6b9de8dbca5118d35c09842876f26da1b05eddd75cf90f995924d668f34479e74a2ad93865ec62e15c36ee5d0
+  metadata.gz: 47efb4fe6a7a2325f2517c2686c66e5945e7c1953f1477e2a4947fd9bb3b452ae95d5edf1b4fae5555089a460353df0cbf2de84b5e3b2712bac6ff0465aa3e75
+  data.tar.gz: 56a7caa27afe9f7289bf3389d17403d0c98f4d7a93ce9d3ea0d0bba831518bb524309938e9da5cc85872fde57bac73ae5ce6b123833d6507e50d9ed82112bef1

data/CHANGELOG.md

@@ -2,6 +2,13 @@
 
 All notable changes to this project will be documented in this file.
 
+## [0.1.5] - 2020-01-13
+### Added
+- Option to import data from a CSV exported from AWS.
+
+### Changed
+- Big refactoring in erase functionality.
+
 ## [0.1.5] - 2019-08-08
 ### Fixed
 - Fix erase table without GSIs.
@@ -29,6 +36,7 @@ All notable changes to this project will be documented in this file.
 ### Added
 - Command to import data from a CSV file to a DynamoDB table.
 
+[0.1.6]: https://github.com/matheussilvasantos/dynamocli/compare/v0.1.5...v0.1.6
 [0.1.5]: https://github.com/matheussilvasantos/dynamocli/compare/v0.1.4...v0.1.5
 [0.1.4]: https://github.com/matheussilvasantos/dynamocli/compare/v0.1.3...v0.1.4
 [0.1.3]: https://github.com/matheussilvasantos/dynamocli/compare/v0.1.2...v0.1.3

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    dynamocli (0.1.5)
+    dynamocli (0.1.6)
       aws-sdk-cloudformation (~> 1.23)
       aws-sdk-dynamodb (~> 1.28)
       thor (~> 0.20)
@@ -11,17 +11,17 @@ GEM
   remote: https://rubygems.org/
   specs:
     aws-eventstream (1.0.3)
-    aws-partitions (1.197.0)
-    aws-sdk-cloudformation (1.25.0)
-      aws-sdk-core (~> 3, >= 3.61.1)
+    aws-partitions (1.263.0)
+    aws-sdk-cloudformation (1.29.0)
+      aws-sdk-core (~> 3, >= 3.71.0)
       aws-sigv4 (~> 1.1)
-    aws-sdk-core (3.62.0)
+    aws-sdk-core (3.89.0)
       aws-eventstream (~> 1.0, >= 1.0.2)
-      aws-partitions (~> 1.0)
+      aws-partitions (~> 1, >= 1.239.0)
       aws-sigv4 (~> 1.1)
       jmespath (~> 1.0)
-    aws-sdk-dynamodb (1.34.0)
-      aws-sdk-core (~> 3, >= 3.61.1)
+    aws-sdk-dynamodb (1.41.0)
+      aws-sdk-core (~> 3, >= 3.71.0)
       aws-sigv4 (~> 1.1)
     aws-sigv4 (1.1.0)
       aws-eventstream (~> 1.0, >= 1.0.2)
@@ -33,19 +33,19 @@ GEM
       equatable (~> 0.6)
       tty-color (~> 0.5)
     rake (10.5.0)
-    rspec (3.8.0)
-      rspec-core (~> 3.8.0)
-      rspec-expectations (~> 3.8.0)
-      rspec-mocks (~> 3.8.0)
-    rspec-core (3.8.0)
-      rspec-support (~> 3.8.0)
-    rspec-expectations (3.8.3)
+    rspec (3.9.0)
+      rspec-core (~> 3.9.0)
+      rspec-expectations (~> 3.9.0)
+      rspec-mocks (~> 3.9.0)
+    rspec-core (3.9.1)
+      rspec-support (~> 3.9.1)
+    rspec-expectations (3.9.0)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.8.0)
-    rspec-mocks (3.8.0)
+      rspec-support (~> 3.9.0)
+    rspec-mocks (3.9.1)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.8.0)
-    rspec-support (3.8.0)
+      rspec-support (~> 3.9.0)
+    rspec-support (3.9.2)
     thor (0.20.3)
     tty-color (0.5.0)
     tty-logger (0.1.0)

data/README.md

@@ -15,12 +15,15 @@ You have to configure AWS in your computer first. The program will use the AWS c
 
 - Import data from a CSV file to a DynamoDB table
 
+If you have exported the CSV file you want to import from AWS DynamoDB console, you probaly want to modify the headers before importing the CSV file, because AWS exports the CSV file with a symbol indicating the type of the field in the header. You can pass the option `--exported-from-aws` to do that, the default is false.
+
 ```
 Usage:
   dynamocli import FILE -t, --table, --to=TABLE
 
 Options:
-  -t, --table, --to=TABLE  # table you want to import the data
+  -t, --table, --to=TABLE                                  # table you want to import the data
+          [--exported-from-aws], [--no-exported-from-aws]  # modify the headers before importing the csv
 
 Description:
   `dynamocli import` will import the data in from a file to a table specified.
@@ -56,6 +59,16 @@ From the DynamoDB Guidelines for Working with Tables documentation:
 
 > Deleting an entire table is significantly more efficient than removing items one-by-one, which essentially doubles the write throughput as you do as many delete operations as put operations.
 
+## Known Issues
+
+### Importing a CSV file with arrays and objects as values in it
+
+Unfortunately, at this moment, this library cannot properly import array and objects. These values will appear as strings in the DynamoDB table.
+
+## Cross account or multiple profiles usage
+
+You can run `dynamocli` passing the `AWS_PROFILE` environment variable with the profile you want to use, for example: `AWS_PROFILE=nondefaultprofile dynamocli erase users`.
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/dynamocli.gemspec

@@ -18,7 +18,7 @@ Gem::Specification.new do |spec|
     `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
   end
   spec.bindir        = "bin"
-  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.executables   = spec.files.grep(%r{^bin/dynamocli}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
   spec.add_dependency "thor", "~> 0.20"

data/lib/dynamocli.rb

@@ -13,8 +13,9 @@ module Dynamocli
       > $ dynamo import users.csv --to users
     LONGDESC
     option :to, required: true, desc: "table you want to import the data", banner: "TABLE", aliases: ["-t", "--table"]
+    option "exported-from-aws", desc: "modify the headers before importing the csv", type: :boolean
     def import(file)
-      Dynamocli::Import.new(file: file, table: options[:to]).start
+      Dynamocli::Import.new(file: file, table: options[:to], exported_from_aws: options["exported-from-aws"]).start
     end
 
     desc "erase TABLE", "erase all the data from the DynamoDB TABLE"

data/lib/dynamocli/aws/stack.rb

@@ -0,0 +1,98 @@
+# frozen_string_literal: true
+
+require "forwardable"
+require "json"
+require "yaml"
+require "aws-sdk-cloudformation"
+require "tty-logger"
+
+module Dynamocli::AWS
+  class Stack
+    attr_reader :name, :resources, :template_body, :original_template, :template_without_table, :policy_body
+
+    extend Forwardable
+    def_delegators :stack_on_aws,
+                   :parameters, :capabilities, :role_arn, :rollback_configuration, :notification_arns, :tags
+
+    def initialize(table_name:, table_resource:, cloudformation: nil, logger: nil)
+      @table_name = table_name
+      @table_resource = table_resource
+      @cloudformation = cloudformation || CLOUDFORMARTION.new
+      @logger = logger || LOGGER.new
+
+      set_attributes_now_because_they_will_change
+    end
+
+    def deploying?
+      current_status != DEPLOY_COMPLETED_KEY
+    end
+
+    private
+
+    CLOUDFORMARTION = Aws::CloudFormation::Client
+    LOGGER = TTY::Logger
+    DEPLOY_COMPLETED_KEY = "UPDATE_COMPLETE"
+    private_constant :CLOUDFORMARTION, :LOGGER, :DEPLOY_COMPLETED_KEY
+
+    attr_reader :table_name, :table_resource, :cloudformation, :logger, :stack_on_aws
+
+    def set_attributes_now_because_they_will_change
+      set_name
+      set_stack_on_aws
+      set_resources
+      set_template_body
+      set_original_template
+      set_template_without_table
+      set_policy_body
+    end
+
+    def set_name
+      @name ||= table_resource[:stack_name]
+    end
+
+    def set_stack_on_aws
+      @stack_on_aws ||= cloudformation.describe_stacks(stack_name: name)[0][0]
+    end
+
+    def set_resources
+      @resources ||= cloudformation.describe_stack_resources(physical_resource_id: table_name).to_h
+    end
+
+    def set_template_body
+      @template_body ||= cloudformation.get_template(stack_name: name).to_h[:template_body]
+    end
+
+    def set_original_template
+      @original_template ||= parse_template(template_body)
+    end
+
+    def set_template_without_table
+      @template_without_table ||= parse_template(template_body).tap do |template_without_table|
+        tables = original_template["Resources"].select { |_, v| v["Type"] == "AWS::DynamoDB::Table" }
+        table = tables.find { |_, v| v["Properties"]["TableName"] == table_name }
+
+        if tables.nil?
+          logger.error("table #{table_name} not found in the #{@name} stack")
+          exit(42)
+        end
+
+        logical_resource_id = table.first
+        template_without_table["Resources"].delete(logical_resource_id)
+      end
+    end
+
+    def parse_template(template)
+      JSON.parse(template)
+    rescue JSON::ParserError
+      YAML.load(template)
+    end
+
+    def set_policy_body
+      @policy_body ||= cloudformation.get_stack_policy(stack_name: name).stack_policy_body
+    end
+
+    def current_status
+      cloudformation.describe_stacks(stack_name: name)[0][0].stack_status
+    end
+  end
+end

data/lib/dynamocli/aws/table.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+require "forwardable"
+require "aws-sdk-dynamodb"
+
+module Dynamocli::AWS
+  class Table
+    attr_reader :schema
+
+    extend Forwardable
+    def_delegators :table_on_aws, :delete
+
+    def initialize(table_name:, table_on_aws:, dynamodb: nil)
+      @table_name = table_name
+      @table_on_aws = table_on_aws
+      @dynamodb = dynamodb || DYNAMODB.new
+
+      set_schema_before_we_delete_the_table
+    end
+
+    def deleting?
+      status == DELETION_IN_PROCESSING_KEY
+    rescue Aws::DynamoDB::Errors::ResourceNotFoundException
+      false
+    end
+
+    private
+
+    DYNAMODB = Aws::DynamoDB::Client
+    DELETION_IN_PROCESSING_KEY = "DELETING"
+    private_constant :DYNAMODB, :DELETION_IN_PROCESSING_KEY
+
+    attr_reader :table_name, :table_on_aws, :dynamodb
+
+    def status
+      dynamodb.describe_table(table_name: table_name).table.table_status
+    end
+
+    def set_schema_before_we_delete_the_table
+      @schema ||= dynamodb.describe_table(table_name: table_name).to_h[:table].tap do |schema|
+        schema.delete(:table_status)
+        schema.delete(:creation_date_time)
+        schema.delete(:table_size_bytes)
+        schema.delete(:item_count)
+        schema.delete(:table_arn)
+        schema.delete(:table_id)
+        schema[:provisioned_throughput]&.delete(:number_of_decreases_today)
+        schema[:local_secondary_indexes]&.each do |lsi|
+          lsi.delete(:index_status)
+          lsi.delete(:index_size_bytes)
+          lsi.delete(:item_count)
+          lsi.delete(:index_arn)
+        end
+        schema[:global_secondary_indexes]&.each do |gsi|
+          gsi.delete(:index_status)
+          gsi.delete(:index_size_bytes)
+          gsi.delete(:item_count)
+          gsi.delete(:index_arn)
+          gsi[:provisioned_throughput].delete(:number_of_decreases_today)
+        end
+      end
+    end
+  end
+end

data/lib/dynamocli/erase.rb

@@ -1,127 +1,52 @@
 # frozen_string_literal: true
 
-require "json"
-require "yaml"
 require "tty-logger"
 require "aws-sdk-dynamodb"
 require "aws-sdk-cloudformation"
+require "dynamocli/table/cloudformation_table"
+require "dynamocli/table/standalone_table"
+require "dynamocli/aws/stack"
+require "dynamocli/aws/table"
 
 class Dynamocli::Erase
-  LOGGER = TTY::Logger.new
-
   def initialize(table_name:, with_drift: false)
     @with_drift = with_drift
     @table_name = table_name
 
     @dynamodb = Aws::DynamoDB::Client.new
     @cloudformation = Aws::CloudFormation::Client.new
-    @table = Aws::DynamoDB::Table.new(@table_name)
-
-    set_schema
+    @table_on_aws = Aws::DynamoDB::Table.new(@table_name)
 
     @stack_resources = @cloudformation.describe_stack_resources(physical_resource_id: @table_name).to_h
-
-    set_stack_information
-  rescue Aws::DynamoDB::Errors::ResourceNotFoundException => e
-    LOGGER.error(e.message)
-    exit(42)
   rescue Aws::CloudFormation::Errors::ValidationError
     @stack_resources = nil
   end
 
   def start
     erase_table
-  rescue Aws::CloudFormation::Errors::ValidationError => e
+  rescue Aws::CloudFormation::Errors::ValidationError,
+         Aws::DynamoDB::Errors::ValidationException,
+         Aws::DynamoDB::Errors::ResourceNotFoundException => e
     LOGGER.error(e.message)
     exit(42)
   end
 
   private
 
-  def set_schema
-    @schema = @dynamodb.describe_table(table_name: @table_name).to_h[:table].tap do |schema|
-      schema.delete(:table_status)
-      schema.delete(:creation_date_time)
-      schema.delete(:table_size_bytes)
-      schema.delete(:item_count)
-      schema.delete(:table_arn)
-      schema.delete(:table_id)
-      schema[:provisioned_throughput].delete(:number_of_decreases_today)
-      schema[:local_secondary_indexes]&.each do |lsi|
-        lsi.delete(:index_status)
-        lsi.delete(:index_size_bytes)
-        lsi.delete(:item_count)
-        lsi.delete(:index_arn)
-      end
-      schema[:global_secondary_indexes]&.each do |gsi|
-        gsi.delete(:index_status)
-        gsi.delete(:index_size_bytes)
-        gsi.delete(:item_count)
-        gsi.delete(:index_arn)
-        gsi[:provisioned_throughput].delete(:number_of_decreases_today)
-      end
-    end
-  end
-  
-  def set_stack_information
-    return if @stack_resources.nil?
-
-    set_stack_name
-    set_stack
-    set_templates
-  rescue Aws::CloudFormation::Errors::ValidationError => e
-    LOGGER.error(e.message)
-    exit(42)
-  end
-
-  def set_stack_name
-    table_resource = @stack_resources[:stack_resources].find do |resource|
-      resource[:physical_resource_id] == @table_name
-    end
-    @stack_name = table_resource[:stack_name]
-  end
-
-  def set_stack
-    @stack = @cloudformation.describe_stacks(stack_name: @stack_name)[0][0]
-  end
-
-  def set_templates
-    template_body = @cloudformation.get_template(stack_name: @stack_name).to_h[:template_body]
-    @original_template = parse_template(template_body)
-    @template_without_table = parse_template(template_body)
-
-    tables = @original_template["Resources"].select { |_, v| v["Type"] == "AWS::DynamoDB::Table" }
-    table = tables.find { |_, v| v["Properties"]["TableName"] == @table_name }
-
-    if tables.nil?
-      LOGGER.error("table #{@table_name} not found in the #{@stack_name} stack")
-      exit(42)
-    end
-
-    logical_resource_id = table.first
-    @template_without_table["Resources"].delete(logical_resource_id)
-  end
+  LOGGER = TTY::Logger.new
+  private_constant :LOGGER
 
-  def parse_template(template)
-    JSON.parse(template)
-  rescue JSON::ParserError
-    YAML.load(template)
-  end
+  attr_reader :table_name, :table_on_aws, :stack_resources
 
   def erase_table
-    if @stack_resources.nil? || @with_drift
-      check_if_user_wants_to_continue_with_recreation
-      delete_and_recreate_the_table
-    else
-      check_if_user_wants_to_continue_with_deployment
-      erase_table_through_cloudformation
-    end
+    check_if_user_wants_to_continue
+    dynamocli_table.erase
   end
 
-  def check_if_user_wants_to_continue_with_recreation
+  def check_if_user_wants_to_continue
     LOGGER.warn(
-      "You're going to drop and recreate your #{@table_name} table,\n" \
-      "do you really want to continue?"
+      "#{dynamocli_table.alert_message_before_continue} " \
+      "Do you really want to continue?"
     )
     STDOUT.print("(anything other than 'y' will cancel) > ")
 
@@ -136,111 +61,24 @@ class Dynamocli::Erase
     "Erase of #{@table_name} table canceled"
   end
 
-  def delete_and_recreate_the_table
-    delete_table
-    wait_for_deletion_to_complete
-    create_table
-  end
-
-  def delete_table
-    LOGGER.info("Deleting the #{@table_name} table")
-
-    @table.delete
-
-    LOGGER.success("#{@table_name} table deleted")
-  end
-
-  def wait_for_deletion_to_complete
-    waiting_seconds = 0
-    while get_table_status == "DELETING"
-      LOGGER.info("Waiting for deletion to complete")
-      sleep waiting_seconds += 1
-    end
-  rescue Aws::DynamoDB::Errors::ResourceNotFoundException
-    true
-  end
-
-  def get_table_status
-    @dynamodb.describe_table(table_name: @table_name).table.table_status
-  end
-
-  def create_table
-    LOGGER.info("Creating the #{@table_name} table")
-
-    @dynamodb.create_table(@schema)
-
-    LOGGER.success("#{@table_name} table created")
-  end
-  
-  def check_if_user_wants_to_continue_with_deployment
-    LOGGER.warn(
-      "You are going to deploy and redeploy your #{@stack_name} stack\n" \
-      "to drop and recreate the #{@table_name} table, do you really want to continue?"
-    )
-    STDOUT.print("(anything other than 'y' will cancel) > ")
-
-    confirmation = STDIN.gets.strip
-    return if confirmation == "y"
-
-    LOGGER.info(abort_message)
-    exit(0)
-  end
-
-  def erase_table_through_cloudformation
-    deploy_stack_without_the_table
-    wait_for_deployment_to_complete
-    deploy_stack_with_the_original_template
-  end
-
-  def deploy_stack_without_the_table
-    LOGGER.info("Deploying the stack without the #{@table_name} table")
-
-    @cloudformation.update_stack(
-      stack_name: @stack_name,
-      template_body: @template_without_table.to_json,
-      parameters: @stack.parameters.map(&:to_h),
-      capabilities: @stack.capabilities,
-      role_arn: @stack.role_arn,
-      rollback_configuration: @stack.rollback_configuration.to_h,
-      stack_policy_body: get_stack_policy_body,
-      notification_arns: @stack.notification_arns,
-      tags: @stack.tags.map(&:to_h)
-    )
-
-    LOGGER.success("Stack deployed without the #{@table_name} table")
+  def dynamocli_table
+    @dynamocli_table ||=
+      if stack_resources.nil? || with_drift?
+        table = Dynamocli::AWS::Table.new(table_name: table_name, table_on_aws: table_on_aws)
+        Dynamocli::Table::StandaloneTable.new(table_name: table_name, table: table)
+      else
+        stack = Dynamocli::AWS::Stack.new(table_name: table_name, table_resource: table_resource)
+        Dynamocli::Table::CloudformationTable.new(table_name: table_name, stack: stack)
+      end
   end
 
-  def get_stack_policy_body
-    @cloudformation.get_stack_policy(stack_name: @stack_name).stack_policy_body
+  def with_drift?
+    @with_drift
   end
 
-  def wait_for_deployment_to_complete
-    waiting_seconds = 0
-    while get_stack_status != "UPDATE_COMPLETE"
-      LOGGER.info("Waiting for deployment to complete")
-      sleep waiting_seconds += 1
+  def table_resource
+    @table_resource ||= stack_resources[:stack_resources].find do |resource|
+      resource[:physical_resource_id] == @table_name
     end
   end
-
-  def get_stack_status
-    @cloudformation.describe_stacks(stack_name: @stack_name)[0][0].stack_status
-  end
-
-  def deploy_stack_with_the_original_template
-    LOGGER.info("Deploying the stack with the #{@table_name} table")
-
-    @cloudformation.update_stack(
-      stack_name: @stack_name,
-      template_body: @original_template.to_json,
-      parameters: @stack.parameters.map(&:to_h),
-      capabilities: @stack.capabilities,
-      role_arn: @stack.role_arn,
-      rollback_configuration: @stack.rollback_configuration.to_h,
-      stack_policy_body: get_stack_policy_body,
-      notification_arns: @stack.notification_arns,
-      tags: @stack.tags.map(&:to_h)
-    )
-
-    LOGGER.success("Stack deployed with the #{@table_name} table")
-  end
 end

data/lib/dynamocli/import.rb

@@ -8,25 +8,32 @@ class Dynamocli::Import
   LOGGER = TTY::Logger.new
   SUPPORTED_FILE_FORMATS = ["CSV"]
 
-  def initialize(file:, table:)
+  def initialize(file:, table:, exported_from_aws: false)
     @file = file
     @table = table
+    @exported_from_aws = exported_from_aws
     @dynamodb = Aws::DynamoDB::Client.new
   end
 
   def start
     records = get_records
     write_records_to_dynamodb_table(records)
+    LOGGER.success("#{records.size} record#{"s" if records.size != 1} imported to #{table}")
+  rescue Aws::DynamoDB::Errors::ValidationException => e
+    LOGGER.error(e.message)
+    exit(42)
   end
 
   private
 
+  attr_reader :file, :table, :dynamodb
+
   def get_records
-    extension = File.extname(@file)
+    extension = File.extname(file)
 
     case extension
     when ".csv"
-      records_from_csv(@file)
+      records_from_csv(file)
     else
       LOGGER.error("Not supported file format. Only supported file formats are: #{SUPPORTED_FILE_FORMATS}")
       exit(42)
@@ -37,7 +44,30 @@ class Dynamocli::Import
     set_custom_converter_for_csv
     csv_options = { encoding: "UTF-8", headers: true, converters: :attribute_definitions }
     records_csv = CSV.read(csv, csv_options)
-    records_csv.map(&:to_hash)
+    if exported_from_aws?
+      transform_records_csv_from_aws(records_csv)
+    else
+      records_csv.map(&:to_hash)
+    end
+  end
+
+  def exported_from_aws?
+    @exported_from_aws
+  end
+
+  # When the CSV comes from AWS, the header of the CSV is like this: "email (S)".
+  # However, we cannot import the CSV with the header like this, we have to remove
+  # the part that specifies the type of the field before importing it.
+  RANGE_TO_REMOVE_TYPE_FROM_HEADER = (0..-5)
+  private_constant :RANGE_TO_REMOVE_TYPE_FROM_HEADER
+  def transform_records_csv_from_aws(records_csv)
+    records_csv.map do |record_csv|
+      record = record_csv.to_h
+
+      record.each_with_object({}) do |(key, value), records|
+        records[key[RANGE_TO_REMOVE_TYPE_FROM_HEADER]] = value
+      end
+    end
   end
 
   ATTRIBUTE_TYPES_CONVERTERS = {
@@ -46,7 +76,7 @@ class Dynamocli::Import
     "B" => Proc.new(&StringIO.method(:new))
   }
   def set_custom_converter_for_csv
-    attribute_definitions = @dynamodb.describe_table(table_name: @table).table.attribute_definitions
+    attribute_definitions = dynamodb.describe_table(table_name: table).table.attribute_definitions
     CSV::Converters[:attribute_definitions] = lambda do |value, info|
       attribute_definition = attribute_definitions.find { |it| it.attribute_name == info.header }
       return value if attribute_definition.nil?
@@ -56,7 +86,7 @@ class Dynamocli::Import
 
   def write_records_to_dynamodb_table(records)
     slice_items_to_attend_batch_write_limit(records).each do |items|
-      @dynamodb.batch_write_item(request_items: format_request_items(items))
+      dynamodb.batch_write_item(request_items: format_request_items(items))
     end
   end
 
@@ -66,6 +96,6 @@ class Dynamocli::Import
   end
 
   def format_request_items(items)
-    { @table => items.map { |item| { put_request: { item: item } } } }
+    { table => items.map { |item| { put_request: { item: item } } } }
   end
 end

data/lib/dynamocli/table/cloudformation_table.rb

@@ -0,0 +1,77 @@
+# frozen_string_literal: true
+
+require "aws-sdk-cloudformation"
+require "tty-logger"
+
+module Dynamocli::Table
+  class CloudformationTable
+    def initialize(table_name:, stack:, cloudformation: nil, logger: nil)
+      @table_name = table_name
+      @stack = stack
+      @cloudformation = cloudformation || CLOUDFORMARTION.new
+      @logger = logger || LOGGER.new
+    end
+
+    def alert_message_before_continue
+      "You're going to deploy and redeploy your #{stack.name} stack to drop and recreate the #{@table_name} table!"
+    end
+
+    def erase
+      deploy_stack_without_the_table
+      wait_for_deployment_to_complete
+      deploy_stack_with_the_original_template
+    end
+
+    private
+
+    CLOUDFORMARTION = Aws::CloudFormation::Client
+    LOGGER = TTY::Logger
+    private_constant :CLOUDFORMARTION, :LOGGER
+
+    attr_reader :table_name, :stack, :cloudformation, :logger
+
+    def deploy_stack_without_the_table
+      logger.info("Deploying the stack without the #{table_name} table")
+
+      cloudformation.update_stack(
+        stack_name: stack.name,
+        template_body: stack.template_without_table.to_json,
+        parameters: stack.parameters.map(&:to_h),
+        capabilities: stack.capabilities,
+        role_arn: stack.role_arn,
+        rollback_configuration: stack.rollback_configuration.to_h,
+        stack_policy_body: stack.policy_body,
+        notification_arns: stack.notification_arns,
+        tags: stack.tags.map(&:to_h)
+      )
+
+      logger.success("Stack deployed without the #{table_name} table")
+    end
+
+    def wait_for_deployment_to_complete
+      waiting_seconds = 0
+      while stack.deploying?
+        logger.info("Waiting for deployment to complete")
+        sleep waiting_seconds += 1
+      end
+    end
+
+    def deploy_stack_with_the_original_template
+      logger.info("Deploying the stack with the #{table_name} table")
+
+      cloudformation.update_stack(
+        stack_name: stack.name,
+        template_body: stack.original_template.to_json,
+        parameters: stack.parameters.map(&:to_h),
+        capabilities: stack.capabilities,
+        role_arn: stack.role_arn,
+        rollback_configuration: stack.rollback_configuration.to_h,
+        stack_policy_body: stack.policy_body,
+        notification_arns: stack.notification_arns,
+        tags: stack.tags.map(&:to_h)
+      )
+
+      logger.success("Stack deployed with the #{table_name} table")
+    end
+  end
+end

data/lib/dynamocli/table/standalone_table.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require "aws-sdk-dynamodb"
+require "tty-logger"
+
+module Dynamocli::Table
+  class StandaloneTable
+    def initialize(table_name:, table:, dynamodb: nil, logger: nil)
+      @table_name = table_name
+      @table = table
+      @dynamodb = dynamodb || DYNAMODB.new
+      @logger = logger || LOGGER.new
+    end
+
+    def alert_message_before_continue
+      "You're going to drop and recreate your #{@table_name} table!"
+    end
+
+    def erase
+      delete_table
+      wait_for_deletion_to_complete
+      create_table
+    end
+
+    private
+
+    LOGGER = TTY::Logger
+    DYNAMODB = Aws::DynamoDB::Client
+    private_constant :LOGGER, :DYNAMODB
+
+    attr_reader :table_name, :table, :dynamodb, :logger
+
+    def delete_table
+      logger.info("Deleting the #{table_name} table")
+
+      table.delete
+
+      logger.success("#{table_name} table deleted")
+    end
+
+    def wait_for_deletion_to_complete
+      waiting_seconds = 0
+      while table.deleting?
+        logger.info("Waiting for deletion to complete")
+        sleep waiting_seconds += 1
+      end
+    end
+
+    def create_table
+      logger.info("Creating the #{table_name} table")
+
+      dynamodb.create_table(table.schema)
+
+      logger.success("#{table_name} table created")
+    end
+  end
+end

data/lib/dynamocli/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Dynamocli
-  VERSION = "0.1.5"
+  VERSION = "0.1.6"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dynamocli
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - Matheus Silva Santos de Oliveira
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-08-08 00:00:00.000000000 Z
+date: 2020-01-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -126,9 +126,7 @@ description:
 email:
 - oliveira.matheussilvasantos@gmail.com
 executables:
-- console
 - dynamocli
-- setup
 extensions: []
 extra_rdoc_files: []
 files:
@@ -147,8 +145,12 @@ files:
 - bin/setup
 - dynamocli.gemspec
 - lib/dynamocli.rb
+- lib/dynamocli/aws/stack.rb
+- lib/dynamocli/aws/table.rb
 - lib/dynamocli/erase.rb
 - lib/dynamocli/import.rb
+- lib/dynamocli/table/cloudformation_table.rb
+- lib/dynamocli/table/standalone_table.rb
 - lib/dynamocli/version.rb
 homepage: https://github.com/matheussilvasantos/dynamocli
 licenses:
@@ -169,7 +171,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
+rubygems_version: 3.0.6
 signing_key: 
 specification_version: 4
 summary: Utilities for interaction with AWS DynamoDB