ocean_kit 0.1.1 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3a7c18376eb5ecea1afb1556c36e5737408e62fe0f56f748e3854a2e505256ef
-  data.tar.gz: 4b8dfd49bd3ccebd047ce80413735eb7b3904d94e8764a5dd8520d889d34e79b
+  metadata.gz: f64bff4db9951b545ab33726c4cb0ad20ccaa0d55006737759d023352ccb0f9f
+  data.tar.gz: e1bf61b1b1ec8d36ed813ec00681f5a8ae8b06de9046eee2e7f4785edabc4a27
 SHA512:
-  metadata.gz: 10736e6687f8776e27d85c02efc41f07f0662161f99f46eb74e1b8f177d86f68f87f4d10ea4660711cfeb3da7b6093e801ee8f7ec0a185c576cdbdf0ced31d85
-  data.tar.gz: 8f8454d175bddb8b5933f5d4a5cfc7afbfe75058f6f3811d69f1d41bb4778272f34db9e1da7c76afe13248bdbe02b87a3629c633b8544732d45385526ecd7b41
+  metadata.gz: 68d98b0f3681e5cab5ebb2f6e5635cd06993382406a7f3ce0d881820416ecf35ee6b98196e426553752e9a6724fdcfcae3605e87af8a3f0613c3541c5d2f8c37
+  data.tar.gz: 27d79bfe3f6241dc8a07a26476b2148bdb705c88c25aee38cb5f7361dca878337c5a94570229c53f2399d6d66f99e578526e410fdaa0a6fb131d329527a54b00

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ocean_kit (0.1.0)
+    ocean_kit (0.1.3)
       droplet_kit (~> 3.16, >= 3.16.1)
       pastel
       thor (~> 1.2)

data/README.md

@@ -55,7 +55,7 @@ Commands:
   ocean_kit firewalls enable_all_ssh                 # Enables SSH on all firewalls
   ocean_kit firewalls enable_ssh [firewall_number]   # Enable SSH on given firewall
   ocean_kit firewalls help [COMMAND]                 # Describe subcommands or one specific subcommand
-  ocean_kit firewalls list                           # Lists all firewalls for account.
+  ocean_kit firewalls list                           # Lists all firewalls.
 ```
 
 ## Development

data/exe/ocean_kit

@@ -2,6 +2,6 @@
 
 require "ocean_kit"
 
-require_relative "../lib/helpers"
+check_credentials_file
 
 OceanKit::Client.start(ARGV)

data/lib/ocean_kit/helpers/base.rb

@@ -0,0 +1,3 @@
+require_relative "./client"
+require_relative "./console"
+require_relative "./firewalls"

data/lib/ocean_kit/helpers/client.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+def do_client
+  DropletKit::Client.new(access_token: access_token)
+end
+
+def access_token
+  credentials_file["digital_ocean_token"]
+rescue => e
+  puts pastel.red.bold("Error: #{e.message}")
+end
+
+def credentials_file
+  YAML.load(File.read(File.expand_path("~/.ocean_kit/credentials.yml")))
+end
+
+def check_credentials_file
+  if credentials_file.nil?
+    puts pastel.red.bold("Error: credentials file not found. Please run `ocean_kit setup` first.")
+    exit 1
+  end
+end

data/lib/ocean_kit/helpers/console.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+def pastel
+  Pastel.new
+end
+
+def default_text(text)
+  pastel.white(text)
+end
+
+def underline_text(text)
+  pastel.white.bold.underline(text)
+end
+
+def bold_text(text)
+  pastel.white.bold(text)
+end

data/lib/ocean_kit/helpers/firewalls.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+def fetch_firewall(number)
+  do_client.firewalls.all.each_with_index.filter { |firewall, index| index == number.to_i }.flatten.first
+end
+
+def update_firewall(firewall)
+  new_firewall = DropletKit::Firewall.new(
+    name: firewall.name,
+    inbound_rules: firewall.inbound_rules.map { |rule| new_inbound_rule(rule) },
+    outbound_rules: firewall.outbound_rules,
+    droplet_ids: firewall.droplet_ids,
+    tags: firewall.tags
+  )
+  do_client.firewalls.update(new_firewall, id: firewall.id)
+end
+
+def firewall_inbound_rules(firewall)
+  firewall.inbound_rules.map(&:to_h)
+end
+
+def new_inbound_rule(rule)
+  DropletKit::FirewallInboundRule.new(
+    protocol: rule[:protocol],
+    ports: rule[:ports],
+    sources: rule[:sources]
+  )
+end
+
+def remove_ssh_rule(rules_array)
+  rules_array.delete_if { |r| r[:ports] == "22" }
+end
+
+def add_ssh_rule(rules_array)
+  rules_array << {protocol: "tcp", ports: "22", sources: {addresses: ["0.0.0.0/0", "::/0"]}}
+end

data/lib/ocean_kit/resources/base.rb

@@ -0,0 +1,3 @@
+require_relative "./firewalls"
+require_relative "./droplets"
+require_relative "./config"

data/lib/ocean_kit/resources/config.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module OceanKit
+  class Config < Thor
+    desc "setup", "Setup OceanKit environment"
+    def setup
+      # First create the ~/.ocean_kit directory if it doesn't exist
+      Dir.mkdir(File.expand_path("~/.ocean_kit")) unless File.directory?(File.expand_path("~/.ocean_kit"))
+      # Next create the credentials.yml file if it doesn't exist
+      unless File.file?(File.expand_path("~/.ocean_kit/credentials.yml"))
+        File.write(File.expand_path("~/.ocean_kit/credentials.yml"), <<~YAML)
+          ---
+          digital_ocean_token: <YOUR_DIGITAL_OCEAN_TOKEN>
+
+        YAML
+      end
+      puts pastel.green.bold("Successfully setup OceanKit environment.")
+    end
+  end
+end

data/lib/ocean_kit/resources/droplets.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module OceanKit
+  class Droplets < Thor
+    desc "list", "Lists all droplers."
+    def list
+      puts underline_text("Droplets:")
+      puts pastel.blue.bold "🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳\n"
+      do_client.droplets.all.each_with_index do |droplet, index|
+        puts pastel.white.bold "Name: ", pastel.clear.white(droplet.name)
+        puts pastel.white.bold "ID: ", pastel.clear.white(droplet.id)
+        puts pastel.white.bold "Public IP: ", pastel.clear.white(droplet.networks.v4.first.ip_address)
+        puts pastel.white.bold "Region: ", pastel.clear.white(droplet.region.slug)
+        puts pastel.white.bold "Status: ", pastel.clear.white(droplet.status)
+        puts pastel.white.bold "Created: ", pastel.clear.white(droplet.created_at)
+        puts pastel.blue.bold "\n🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳🐳\n"
+      end
+    end
+  end
+end

data/lib/ocean_kit/{firewalls.rb → resources/firewalls.rb}

@@ -2,7 +2,7 @@
 
 module OceanKit
   class Firewalls < Thor
-    desc "list", "Lists all firewalls for account."
+    desc "list", "Lists all firewalls."
     def list
       puts pastel.white.bold.underline("Firewalls:\n")
       do_client.firewalls.all.each_with_index do |firewall, index|
@@ -45,7 +45,7 @@ module OceanKit
 
     desc "enable_ssh [firewall_number]", "Enable SSH on given firewall"
     def enable_ssh(number)
-      firewall = do_client.firewalls.find(id: fetch_firewall_id(number))
+      firewall = fetch_firewall(number)
       inbound_rules = firewall_inbound_rules(firewall)
       firewall.inbound_rules = add_ssh_rule(inbound_rules)
       begin
@@ -58,7 +58,7 @@ module OceanKit
 
     desc "disable_ssh [firewall_number]", "Disable SSH on given firewall"
     def disable_ssh(number)
-      firewall = do_client.firewalls.find(id: fetch_firewall_id(number))
+      firewall = fetch_firewall(number)
       inbound_rules = firewall_inbound_rules(firewall)
       firewall.inbound_rules = remove_ssh_rule(inbound_rules)
       begin
@@ -68,50 +68,5 @@ module OceanKit
         puts pastel.red.bold("Error: #{e.message}")
       end
     end
-
-    private
-
-    def fetch_firewall_id(number)
-      do_client.firewalls.all.each_with_index.filter { |firewall, index| index == number.to_i }.flatten.first.id
-    end
-
-    def update_firewall(firewall)
-      new_firewall = DropletKit::Firewall.new(
-        name: firewall.name,
-        inbound_rules: firewall.inbound_rules.map { |rule| new_inbound_rule(rule) },
-        outbound_rules: firewall.outbound_rules,
-        droplet_ids: firewall.droplet_ids,
-        tags: firewall.tags
-      )
-      do_client.firewalls.update(new_firewall, id: firewall.id)
-    end
-
-    def firewall_inbound_rules(firewall)
-      firewall.inbound_rules.map(&:to_h)
-    end
-
-    def new_inbound_rule(rule)
-      DropletKit::FirewallInboundRule.new(
-        protocol: rule[:protocol],
-        ports: rule[:ports],
-        sources: rule[:sources]
-      )
-    end
-
-    def remove_ssh_rule(rules_array)
-      rules_array.delete_if { |r| r[:ports] == "22" }
-    end
-
-    def add_ssh_rule(rules_array)
-      rules_array << {protocol: "tcp", ports: "22", sources: {addresses: ["0.0.0.0/0", "::/0"]}}
-    end
-
-    def pastel
-      Pastel.new
-    end
-
-    def find_firewall_by_id(id)
-      do_client.firewalls.find(id:)
-    end
   end
 end

data/lib/ocean_kit/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OceanKit
-  VERSION = "0.1.1"
+  VERSION = "0.1.4"
 end

data/lib/ocean_kit.rb

@@ -6,23 +6,19 @@ require "thor"
 require "yaml"
 
 require_relative "ocean_kit/version"
-require_relative "./ocean_kit/firewalls"
+require_relative "./ocean_kit/resources/base"
+require_relative "./ocean_kit/helpers/base"
 
 module OceanKit
   class Client < Thor
-    desc "firewalls SUBCOMMAND ...ARGS", "manage your DO firewall"
-    subcommand "firewalls", Firewalls
+    desc "config SUBCOMMAND ...ARGS", "Create OceanKit folder and credentials.yml file"
+    subcommand "config", Config
 
-    no_commands {
-      def client
-        DropletKit::Client.new(access_token: access_token)
-      end
+    desc "droplets SUBCOMMAND ...ARGS", "manage your Digital Ocean droplets"
+    subcommand "droplets", Droplets
 
-      def access_token
-        credentials_file = YAML.load(File.read(File.expand_path("~/.ocean_kit/credentials.yml")))
-        credentials_file["digital_ocean_token"]
-      end
-    }
+    desc "firewalls SUBCOMMAND ...ARGS", "manage your Digital Ocean firewalls"
+    subcommand "firewalls", Firewalls
   end
 
   class Error < StandardError; end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ocean_kit
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.4
 platform: ruby
 authors:
 - Leo Policastro
@@ -77,9 +77,15 @@ files:
 - README.md
 - Rakefile
 - exe/ocean_kit
-- lib/helpers.rb
 - lib/ocean_kit.rb
-- lib/ocean_kit/firewalls.rb
+- lib/ocean_kit/helpers/base.rb
+- lib/ocean_kit/helpers/client.rb
+- lib/ocean_kit/helpers/console.rb
+- lib/ocean_kit/helpers/firewalls.rb
+- lib/ocean_kit/resources/base.rb
+- lib/ocean_kit/resources/config.rb
+- lib/ocean_kit/resources/droplets.rb
+- lib/ocean_kit/resources/firewalls.rb
 - lib/ocean_kit/version.rb
 - ocean_kit.gemspec
 - sig/ocean_kit.rbs

data/lib/helpers.rb

@@ -1,6 +0,0 @@
-# frozen_string_literal: true
-
-def do_client
-  ocean = OceanKit::Client.new
-  ocean.client
-end