viaduct-toolkit 0.1.0 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e74b426a21bcd09ed6d55df3caeccb100ac0e83b
-  data.tar.gz: 659ef1f517c899b2ebf5f69e80f221aecad3af75
+  metadata.gz: 3de7f4fcc1044f4d5ef119608fcb1dab981382c8
+  data.tar.gz: 9dff6dd8840043b0d91c8c080d2946fa466568d3
 SHA512:
-  metadata.gz: 218b9c6b7bba9286de733666a97bde7c0b4f0d10ab65e23ab28534771db1f402284ede76c7c1621061d038730d4e12561c613575c999c9ce6cbbc64b84509539
-  data.tar.gz: c3ba43711e4928386dfe4fd9fa1d7d1af1f43b601259fffb5e3e71c2c84d0a4133d5b66457fe9fd679a7ddfed399d5d89ce2bf115ca50e3bba5e75dc8c69a646
+  metadata.gz: 8c95ca7b91a199502f5b9a8032e2722ac30850d41d10a72a10bd4ac4727d25d1c6a1aadaf46e3dafedf34ea696a80cf97a94a67f599b92a7a336a18c8688d058
+  data.tar.gz: db2ed5d0b4da511570cbbbc0d7919164d1768f28cac0c12fe214f9952826590d7d52f4f62443664a96a0fc3e5d0e63df58473ee42e1f86e1bd205a626ee01068

data/bin/vdt

@@ -2,4 +2,9 @@
 $:.unshift(File.expand_path(File.join('..', '..', 'lib'), __FILE__))
 require 'viaduct/toolkit'
 Viaduct::Toolkit.add_commands
-Viaduct::Toolkit.cli.run!
+begin
+  Viaduct::Toolkit.cli.run!
+rescue Interrupt
+  puts
+  exit
+end

data/lib/viaduct/toolkit/commands/app_deploy.rb

@@ -0,0 +1,117 @@
+Viaduct::Toolkit.cli.command "app:deploy" do |c|
+  c.syntax = "app:deploy [VERSION]"
+  c.description = "Deploy the given version (or latest) for an application"
+  c.option "-y", "Always deploy the latest commit without asking"
+  c.option "-f", "--force-build", "Build the application if even it isn't needed"
+  c.option "-i", "--no-inherit-build-env", "Do not inherit the build environment for this deployment"
+  c.option "-b", "--branch BRANCH", "The branch/commit you wish to deploy (defaults to application default)"
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+
+    puts "Getting latest commit information...".magenta
+    if app = find_application
+
+      #
+      # Get the latest commits for the remote repository
+      #
+      commits = Viaduct::Toolkit.api.repositories.commits(:application => app['subdomain'], :limit => 1, :update => true, :branch => opts.branch)
+      unless commits.success? && commit = commits.data['commits'][0]
+        puts
+        puts "We couldn't find any commits on `#{commits.data['branch']}` branch.".red
+        puts "Please check that you have pushed your code to your repository."
+        puts
+        puts "  Your repository is hosted at: #{commits.data['repository']['url'].underline}"
+        puts
+        exit 1
+      end
+
+      details do
+        heading "Latest commit for #{app['subdomain']}"
+        field "Repository", commits.data['repository']['url']
+        field "Branch", commits.data['branch']
+        field "Commit", commit['ref'][0, 10]
+        field "Message", commit['message']
+      end
+
+      unless opts.y || agree("Do you wish to deploy this commit?".blue)
+        puts "OK then. We won't do anything.".green
+        exit 0
+      end
+
+      deployment_params = {
+        :application => app['subdomain'],
+        :version => commit['ref'],
+        :hold => true,
+        :force_build => opts.force_build,
+        :inherit_build_environment => opts.no_inherit_build_env == false ? false : true
+      }
+      deployment_response = Viaduct::Toolkit.api.deployments.start(deployment_params)
+      if deployment_response.success?
+
+        puts
+        puts "Deployment will now start. You can view the full log in your web browser:"
+        puts "https://my.viaduct.io/applications/#{app['subdomain']}/deployments/#{deployment_response.data['number']}".underline
+        puts
+        print "Waiting for capacity to deploy".ljust(40)
+        $stdout.flush
+
+        statuses_seen = []
+        require 'eventmachine'
+        EM.run do
+          Viaduct::Toolkit.api.push.connected do |event|
+            Viaduct::Toolkit.api.push.subscribe :deployment_status, deployment_response.data['id'].to_i
+          end
+
+          Viaduct::Toolkit.api.push.receive do |exchange, process_id, data|
+            if exchange == :deployment_status
+              if statuses_seen.include?(data['status'])
+                next
+              end
+              statuses_seen << data['status']
+              case data['status']
+              when 'building'
+                puts "[  OK  ]".green
+                print "Building package".ljust(40)
+                $stdout.flush
+              when 'starting'
+                puts "[  OK  ]".green
+                print "Starting application".ljust(40)
+                $stdout.flush
+              when 'activating'
+                puts "[  OK  ]".green
+                print "Activating deployment".ljust(40)
+                $stdout.flush
+              when 'deployed'
+                puts "[  OK  ]".green
+                puts "Deployment completed successfully!".green
+                puts
+                puts "You can view your application at:"
+                if app['main_domain']
+                  puts "http://#{app['main_domain']['name']}".underline
+                else
+                  puts "http://#{app['viaduct_domain']}".underline
+                end
+                puts
+                exit 0
+              when 'failed'
+                puts "[ FAIL ]".red
+                puts
+                puts "Deployment did not completed successfully.".red
+                puts "You can review information about this failure at the URL provided above."
+                puts
+                exit 0
+              end
+            elsif exchange == :global && data['type'] == 'subscribed'
+              Viaduct::Toolkit.api.push.send(:remove_deployment_hold, {'deployment_id' => deployment_response.data['id'].to_i})
+            end
+          end
+        end
+      else
+        error "We couldn't start this deployment at this time."
+      end
+    end
+  end
+end
+
+Viaduct::Toolkit.cli.alias_command "deploy", "app:deploy"

data/lib/viaduct/toolkit/commands/app_status.rb

@@ -2,7 +2,7 @@ Viaduct::Toolkit.cli.command "app:status" do |c|
   c.syntax = "app:status"
   c.description = "Return current status of an application"
   c.option "--process PROCESS_TYPE", String, "A process to show expanded information for"
-  c.option "--database DATABASE_ID", String, "The ID of the database to display further information for"
+  c.option "--database FRUIT", String, "The ID of the database to display further information for"
 
   c.action do |args, opts|
     include Commander::Methods
@@ -29,57 +29,66 @@ Viaduct::Toolkit.cli.command "app:status" do |c|
 
         elsif opts.database
 
-          if db = response.data['databases'][(opts.database.to_i - 1).abs]
+          if db = response.data['databases'].select { |d| d['fruit'] == opts.database }.first
             details do
               heading "Database Details"
+              field "Fruit", db['fruit']
               field "Label", db['label']
-              field "Database", db['stack']['name']
+              field "Engine", db['stack']['name']
               process_details(db['process'])
             end
           else
-            puts "No database found with ID '#{opts.database}'".red
+            puts "No database found matching '#{opts.database}'".red
             exit 1
           end
 
         else
 
-          puts "Deployment Details".yellow
-          details do
-            puts "+" + '-' * 78 + '+'
-            field "Number", response.data['deployment']['number']
-            field 'Version', response.data['deployment']['version']['id']
-            field "Description", response.data['deployment']['version']['description']
-            field "Source", response.data['deployment']['triggered_from']
-            field "Deployer", response.data['deployment']['user'] ? response.data['deployment']['user']['name'] : response.data['deployment']['triggerer']
-            field "Started", time(response.data['deployment']['timing']['started_at'])
-          end
+          if response.data['deployment']
+            puts "Deployment Details".yellow
+            details do
+              puts "+" + '-' * 78 + '+'
+              field "Number", response.data['deployment']['number']
+              field 'Version', response.data['deployment']['version']['id']
+              field "Description", response.data['deployment']['version']['description']
+              field "Source", response.data['deployment']['triggered_from']
+              field "Deployer", response.data['deployment']['user'] ? response.data['deployment']['user']['name'] : response.data['deployment']['triggerer']
+              field "Started", time(response.data['deployment']['timing']['started_at'])
+            end
 
-          processes = response.data['processes'].map do |p|
-            memory_used = (p['process']['resources']['memory_usage'] / 1024 / 1024).to_s + "MB"
-            memory_available = (p['process']['resources']['max_memory'] / 1024 / 1024).to_s + "MB"
-            memory = "#{memory_used}/#{memory_available}"
+            processes = response.data['processes'].map do |p|
+              memory_used = (p['process']['resources']['memory_usage'] / 1024 / 1024).to_s + "MB"
+              memory_available = (p['process']['resources']['max_memory'] / 1024 / 1024).to_s + "MB"
+              memory = "#{memory_used}/#{memory_available}"
 
-            respawns = "#{p['process']['respawning']['current']}/#{p['process']['respawning']['maximum']}"
-            [p['label'], p['process']['status'].capitalize, memory, p['container']['command'], respawns]
+              respawns = "#{p['process']['respawning']['current']}/#{p['process']['respawning']['maximum']}"
+              [p['label'], p['process']['status'].capitalize, memory, p['container']['command'], respawns]
+            end
+            puts
+            puts "Processes".yellow
+            table ['Label', 'Status', 'Memory', 'Command', 'Respawns'], processes
+          else
+            puts "No process info available. Not deployed yet.".yellow
           end
-          puts
-          puts "Processes".yellow
-          table ['Label', 'Status', 'Memory', 'Command', 'Respawns'], processes
 
           count = 0
           databases = response.data['databases'].map do |d|
             count += 1
-            memory_used = (d['process']['resources']['memory_usage'] / 1024 / 1024).to_s + "MB"
-            memory_available = (d['process']['resources']['max_memory'] / 1024 / 1024).to_s + "MB"
-            memory = "#{memory_used}/#{memory_available}"
+            if d['process']['resources']['memory_usage']
+              memory_used = (d['process']['resources']['memory_usage'] / 1024 / 1024).to_s + "MB"
+              memory_available = (d['process']['resources']['max_memory'] / 1024 / 1024).to_s + "MB"
+              memory = "#{memory_used}/#{memory_available}"
+            else
+              memory = ""
+            end
             respawns = "#{d['process']['respawning']['current']}/#{d['process']['respawning']['maximum']}"
 
-            [count, d['label'], d['process']['status'].capitalize, d['stack']['name'], memory, respawns, d['process']['networking']['ip_address']]
+            [d['fruit'], d['label'], d['process']['status'].capitalize, d['stack']['name'], memory, respawns, d['process']['networking']['ip_address']]
           end
           unless databases.empty?
             puts
             puts "Dedicated Databases".yellow
-            table ['#', 'Label', 'Status', 'Type', 'Memory', 'Respawns', 'IP Address'], databases
+            table ['Fruit', 'Label', 'Status', 'Type', 'Memory', 'Respawns', 'IP Address'], databases
           end
         end
 

data/lib/viaduct/toolkit/commands/db_console.rb

@@ -2,31 +2,22 @@ Viaduct::Toolkit.cli.command "db:console" do |c|
 
   c.syntax = "db:console"
   c.description = "Open a console session for the given database"
+  c.option "-d FRUIT", "--database FRUIT", String, "The name of the database to import into (uses the main DB by default)"
 
   c.action do |args, opts|
     include Commander::Methods
     ensure_logged_in!
 
-    if args[0].nil?
-      error "You need to specify a database name"
-    end
-    if app = find_application
-      database = Viaduct::Toolkit.api.applications.database(:application => app['subdomain'], :database => args[0])
-      if database.status == 'not-found'
-        puts "No database found named '#{args[0]}'".red
-        exit 1
-      end
+    app = find_application
+    database = find_database(app, opts.database)
 
-      error "Couldn't get database details" unless database.success?
-      if command = database.data['database']['console_command']
-        chech_ssh_key_presence
-        console = get_application_console_port_forward(app)
-        puts command
-        exec_console_command(console, "ssh -t vdt@#{console['ip_address']} -p #{console['port']} '#{command}'")
-      else
-        puts "Console access is not available for this database type.".red
-        exit 1
-      end
+    if command = database['database']['console_command']
+      chech_ssh_key_presence
+      console = get_application_console_port_forward(app)
+      exec_console_command(console, "ssh -t vdt@#{console['ip_address']} -p #{console['port']} '#{command}'")
+    else
+      puts "Console access is not available for this database type.".red
+      exit 1
     end
   end
 

data/lib/viaduct/toolkit/commands/db_export.rb

@@ -0,0 +1,79 @@
+Viaduct::Toolkit.cli.command "db:export" do |c|
+
+  c.syntax = "db:export [PATH]"
+  c.description = "Download an export for a given database"
+  c.option "-d FRUIT", "--database FRUIT", String, "The name of the database to export (uses the main DB by default)"
+
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+
+    # Get the application
+    app = find_application
+    database = find_database(app, opts.database)
+
+    # Set some parameters to create our new  export with
+    export_params = {}
+    export_params[:application] = app['subdomain']
+    export_params[:database]    = database['database']['fruit']
+
+    # If this is e-mail only, set that falg
+    if args[0].nil?
+      export_params[:email] = true
+    end
+
+    # Create our new export
+    export = Viaduct::Toolkit.api.database_exports.start(export_params)
+
+    # Is an export in progress?
+    if export.is_a?(MoonropeClient::Responses::ValidationError) && (export.errors['base'] || []).any? { |e| e =~ /already in progress/}
+      puts "An export is already in progress for this database. Please try again later.".red
+      exit 1
+    end
+
+    # Any other export starting errors?
+    unless export.success?
+      puts "The database export could not be started for this database.".red
+      exit 1
+    end
+
+    # If this is an email, we're done now. Just let the user know and exit.
+    if args[0].nil?
+      puts "Export of #{database['database']['fruit']} database has been started.".green
+      puts "You'll receive an email shortly.".green
+      exit 0
+    end
+
+    # Keep checking the export status and download the file when it's ready to
+    # be downloaded.
+    puts "Exporting #{database['database']['fruit']} database...".magenta
+    checks = 0
+    loop do
+      checks += 1
+      export = Viaduct::Toolkit.api.database_exports.info(:id => export.data['id'])
+      if export.success?
+        if export.data['status'] == 'complete' && export.data['can_download?']
+          break
+        end
+      end
+
+      if checks > 600
+        puts "Export did not build in time. Please try again through the web interface"
+        puts "and request that the result be e-mailed to you."
+        exit 1
+      else
+        sleep 2
+      end
+    end
+
+    puts "Export build successfully. Downloading...".magenta
+    if system("curl #{export.data['download_url']} -o #{args[0]}")
+      puts "Export downloaded successfully to #{args[0]}".green
+    else
+      puts "Could not download using `curl`. Ensure you have curl installed to".red
+      puts "download file using the Viaduct Toolkit.".red
+      exit 1
+    end
+  end
+
+end

data/lib/viaduct/toolkit/commands/db_import.rb

@@ -0,0 +1,68 @@
+Viaduct::Toolkit.cli.command "db:import" do |c|
+
+  c.syntax = "db:import [PATH]"
+  c.description = "Import a dump file to a database"
+  c.option "--database FRUIT", String, "The name of the database to import into (uses the main DB by default)"
+
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+
+    if args[0].nil? || !File.file?(args[0])
+      puts "You must provide a path to the dump file to import".red
+      exit 1
+    end
+
+    # Get the application
+    app = find_application
+
+    # Get the database
+    database = find_database(app, opts.database)
+
+    # Set some parameters to create our new  export with
+    import_params = {}
+    import_params[:application] = app['subdomain']
+    import_params[:database]    = database['database']['fruit']
+    import_params[:file]        = Base64.encode64(File.read(args[0]))
+
+    # Start the import
+    puts "Uploading #{args[0]} to Viaduct...".magenta
+    import = Viaduct::Toolkit.api.database_imports.import(import_params)
+    unless import.success?
+      puts "Couldn't start this import. Ensure that the type of database supports importing.".red
+      exit 1
+    end
+
+    puts "Importing #{args[0]} into #{database['database']['fruit']} database...".magenta
+
+    # Keep checking in on the status
+    count = 0
+    loop do
+      count += 1
+      import = Viaduct::Toolkit.api.database_imports.info(:id => import.data['id'])
+      if import.success? && ['complete', 'failed'].include?(import.data['status'])
+        case import.data['status']
+        when 'complete'
+          puts "Database has been imported successfully.".green
+        else
+          puts "The database import failed. You can use the console to try and".red
+          puts "import your file directly into your database. See documentation".red
+          puts "for further details".red
+        end
+        exit 0
+      end
+
+      if count > 600
+        break
+      else
+        sleep 2
+      end
+    end
+
+    # Oh no, it timed out.
+    puts "The database import did not complete in a timely manner. Please try".red
+    puts "again later or contact support.".red
+    exit 1
+  end
+
+end

data/lib/viaduct/toolkit/helpers.rb

@@ -87,7 +87,28 @@ module Viaduct
         end
 
         puts "Couldn't determine a Viaduct application from command.".red
-        exit(1)
+        exit 1
+      end
+
+      def find_database(app, database_name = nil)
+        # Is there a DB?
+        if database_name
+          database_fruit = database_name
+        elsif app['main_database']
+          database_fruit = app['main_database']['fruit']
+        else
+          puts "There is no database specified and you don't have a main database for".red
+          puts "this application. Try again using the --database option.".red
+          exit 1
+        end
+
+        database = Viaduct::Toolkit.api.applications.database(:application => app['subdomain'], :database => database_fruit)
+        if database.success?
+          database.data
+        else
+          puts "No database found matching '#{database_fruit}'".red
+          exit 1
+        end
       end
 
       def heading(title)

data/lib/viaduct/toolkit/version.rb

@@ -1,5 +1,5 @@
 module Viaduct
   module Toolkit
-    VERSION = '0.1.0'
+    VERSION = '1.0.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: viaduct-toolkit
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 1.0.0
 platform: ruby
 authors:
 - Adam Cooke
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-04 00:00:00.000000000 Z
+date: 2014-11-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: commander
@@ -29,6 +29,9 @@ dependencies:
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+    - - ">="
       - !ruby/object:Gem::Version
         version: 1.0.5
   type: :runtime
@@ -36,6 +39,9 @@ dependencies:
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+    - - ">="
       - !ruby/object:Gem::Version
         version: 1.0.5
 - !ruby/object:Gem::Dependency
@@ -103,6 +109,7 @@ files:
 - bin/vdt
 - lib/viaduct/toolkit.rb
 - lib/viaduct/toolkit/commands/app_console.rb
+- lib/viaduct/toolkit/commands/app_deploy.rb
 - lib/viaduct/toolkit/commands/app_deployments.rb
 - lib/viaduct/toolkit/commands/app_download.rb
 - lib/viaduct/toolkit/commands/app_env.rb
@@ -115,6 +122,8 @@ files:
 - lib/viaduct/toolkit/commands/app_upload.rb
 - lib/viaduct/toolkit/commands/app_visit.rb
 - lib/viaduct/toolkit/commands/db_console.rb
+- lib/viaduct/toolkit/commands/db_export.rb
+- lib/viaduct/toolkit/commands/db_import.rb
 - lib/viaduct/toolkit/commands/db_list.rb
 - lib/viaduct/toolkit/commands/domain_list.rb
 - lib/viaduct/toolkit/commands/login.rb