ocean_kit 0.1.2 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 62d343d2e641d29b1d725b8fa28cc31ae1f315c6066301d259690044e5bdec0c
-  data.tar.gz: fe8ea0a906be855815be19084fdba93c37cdb39ddafc70fb3d963e56b1294541
+  metadata.gz: 84dcf8fe7afd5f888192ce09e6ec3c7153052a812142ae4726d7cbf15b7a43dd
+  data.tar.gz: 930fa733a585ea61efe68c924b66596bf05b80f63282e535f4435f5e1e8adb2c
 SHA512:
-  metadata.gz: 18da4133b0419fec8d780d1eeaa5fa7a97e271cc4a99daeb8329d2ca0045757bfd34b9fea68cf346fcac4feca10ab66edbcd264404245bbfa7aa9809d9f73197
-  data.tar.gz: 71c1d8033773066e2ae9d6418bcfb2e4ceb274483131cddbe6677b3c9093250276dacba52a12bfb0fdb94e8d65f74d0bac6942938cfa7bed8eb7cbacc72d686a
+  metadata.gz: 1189f7a21bd24d3bddbf4d0b2144a3b67e1f16df14c02b6b38189cbf921ef2c4bbf5bec976640860ce3a4a7dd9acbefbb23a94ed6f9b52fd38d3db48cff844f1
+  data.tar.gz: 073dc845350bea427160ec66556ffcb06bde23b95cacac2b28b0065f5f58e2cb9543edf372cdfe640bbef32661c49095e3544644bfab1d2f78f8493b8348c9f6

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ocean_kit (0.1.1)
+    ocean_kit (0.1.4)
       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,20 @@
+# frozen_string_literal: true
+
+def do_client
+  DropletKit::Client.new(access_token: access_token)
+end
+
+def access_token
+  credentials_file["digital_ocean_token"]
+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,21 @@
+# 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.")
+      puts pastel.white.bold("Please add your Digital Ocean personal access token your credentials in ~/.ocean_kit/credentials.yml")
+    end
+  end
+end

data/lib/ocean_kit/resources/droplets.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module OceanKit
+  class Droplets < Thor
+    desc "list", "Lists all droplers."
+    def list
+      puts underline_text("Droplets:\n")
+      do_client.droplets.all.each_with_index do |droplet, index|
+        puts "------------------------------\n"
+
+        puts pastel.white.bold "#: ", pastel.clear.white(index + 1)
+        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)
+      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|
@@ -68,50 +68,5 @@ module OceanKit
         puts pastel.red.bold("Error: #{e.message}")
       end
     end
-
-    private
-
-    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
-
-    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.2"
+  VERSION = "0.1.5"
 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.2
+  version: 0.1.5
 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