enfcli 4.0.0 → 4.1.0.pre.alpha

data/lib/enfcli.rb

@@ -13,26 +13,29 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-require 'openssl'
-require 'enfapi'
-require 'enfthor'
-require 'readline'
-require 'singleton'
-require 'ipaddr'
-
-require 'rubygems/commands/update_command'
-require 'rubygems/commands/search_command'
-
-
-require 'enfcli/version'
-require 'enfcli/commands/xcr'
-require 'enfcli/commands/xiam'
-require 'enfcli/commands/xfw'
-require 'enfcli/commands/user'
-require 'enfcli/commands/captive'
-require 'enfcli/commands/xdns'
+require "openssl"
+require "enfapi"
+require "enfthor"
+require "readline"
+require "singleton"
+require "ipaddr"
+require "clipboard"
+require "json"
+
+require "rubygems/commands/update_command"
+require "rubygems/commands/search_command"
+
+require "enfcli/version"
+require "enfcli/commands/xcr"
+require "enfcli/commands/xiam"
+require "enfcli/commands/xfw"
+require "enfcli/commands/user"
+require "enfcli/commands/captive"
+require "enfcli/commands/xdns"
 
 module EnfCli
+  CONFIG_FILE = "#{Dir.home() + "/.xaptum_config.json"}"
+
   FIREWALL_CMD = "firewall"
   IAM_CMD = "iam"
   NETWORK_CMD = "network"
@@ -63,13 +66,13 @@ module EnfCli
       @ip.hton
     end
 
-    def self.ntoh ipv6_bytes
-      ip = IPAddr::new_ntoh( ipv6_bytes )
+    def self.ntoh(ipv6_bytes)
+      ip = IPAddr::new_ntoh(ipv6_bytes)
       EnfCli::IPV6.new ip.to_s
     end
 
     private
-    
+
     attr_accessor :ip
   end
 
@@ -84,10 +87,9 @@ module EnfCli
       # store in prefix/len
       @prefix = EnfCli::IPV6.new tokens[0]
       @len = tokens[1].to_i
-      
+
       # raise if len is not 0
       raise EnfCli::ERROR, "#{ipv6cidr} is not a valid CIDR notation." unless len > 2
-      
     end
 
     def prefix_bytes
@@ -101,47 +103,47 @@ module EnfCli
     def to_s
       "#{@prefix.to_s}/#{@len}"
     end
-      
+
     private
+
     attr_accessor :prefix, :len
   end
-  
+
   def self.ask_password(prompt = nil)
     begin
-      prompt = 'Enter Password:' unless prompt
+      prompt = "Enter Password:" unless prompt
       print prompt
       # We hide the entered characters before to ask for the password
-      system 'stty -echo'
+      system "stty -echo"
       password = $stdin.gets.chomp
-      system 'stty echo'
+      system "stty echo"
       puts ""
       return password
-      
     rescue NoMethodError, Interrupt
       # When the process is exited, we display the characters again
       # And we exit
-      system 'stty echo'
+      system "stty echo"
       exit
     end
   end
 
   def self.generate_ec_cert(key, ipv6)
-    # monkey patch 
+    # monkey patch
     OpenSSL::PKey::EC.send(:alias_method, :private?, :private_key?)
-    
+
     # Generate cert
     cert = OpenSSL::X509::Certificate.new
-    cert.subject = cert.issuer = OpenSSL::X509::Name.new([['CN', "#{ipv6}"]])
+    cert.subject = cert.issuer = OpenSSL::X509::Name.new([["CN", "#{ipv6}"]])
     cert.not_before = Time.now
     cert.not_after = Time.now + 365 * 24 * 60 * 60
     cert.public_key = key
     cert.serial = 0x0
     cert.version = 2
-    
+
     cert.sign key, OpenSSL::Digest::SHA1.new
     cert
-  end  
-  
+  end
+
   def self.expand_path(file)
     new_file = File.expand_path(EnfCli::expand_env(file))
   end
@@ -154,7 +156,7 @@ module EnfCli
     ip = EnfCli::IPV6.new ipv6
     ip.to_s
   end
-  
+
   ##
   # EnfCli error
   #
@@ -174,7 +176,16 @@ module EnfCli
 
     def initialize
       @prompt = "enfcli"
-      @host = ''
+      @host = ""
+      @user = ""
+      @session = nil
+    end
+
+    def init(host, user, session)
+      @host = host
+      @user = user
+      @session = session
+      @prompt = "enfcli-#{user}" if user
     end
 
     def xaptum_admin?
@@ -189,23 +200,17 @@ module EnfCli
       "#{@host}"
     end
 
-    def host=(value)
-      @host = value
+    def auth_token
+      @session[:token]
     end
 
-    def prompt
-      "\001\033[1;32m\002#{@prompt}>\001\033[0m\002 "
+    def user
+      @user
     end
 
-    def prompt=(value)
-      if value
-        @prompt = "enfcli-#{value}"
-      else
-        @prompt= "enfcli"
-      end
-      
+    def prompt
+      "\001\033[1;32m\002#{@prompt}>\001\033[0m\002 "
     end
-    
   end
 
   ##
@@ -213,75 +218,112 @@ module EnfCli
   #
   class CLI < EnfThor
     no_commands {
-      def execute_gem_cmd cmd
-        begin 
+      def execute_gem_cmd(cmd)
+        begin
           cmd.execute
         rescue Gem::SystemExitException => e
           say "Unable to execute commnad. Please try again!", :red
         end
       end
     }
-    
+
     desc "connect", "Connect to ENF Controller"
-    method_option :host, :type => :string, :required => true
-    method_option :user, :type => :string, :required => true
-    method_option :password, :type => :string
+    method_option :host, :type => :string
+    method_option :user, :type => :string
+
     def connect(*names)
-      host = options[:host]
-      user = options[:user]
+      host = ""
+      user = ""
 
       try_with_rescue do
+        ## first check for command options
+        if options[:host] && options[:user]
+          host = options[:host]
+          user = options[:user]
+        elsif File.file?(CONFIG_FILE) # then check for config file
+          config_json = JSON.load(File.open(CONFIG_FILE))
+          host = config_json["host"]
+          user = config_json["user"]
+        else
+          raise EnfCli::ERROR, "You must either specify the --host and --user parameters or create a Xaptum config file to connect."
+        end
+
         # Make sure to use https as default
         host = "https://#{host}" unless host =~ /^(http|https):\/\//
-        
+
         # Ask for password
         say "Connecting to '#{host}'.....", :bold
         password = EnfCli::ask_password()
-        
+
         # Authenticate
         resp = EnfApi::API.instance.authenticate(host, user, password)
 
-        # set session
-        EnfCli::CTX.instance.session = resp[:data][0]
+        # initialize CTX
+        EnfCli::CTX.instance.init host, user, resp[:data][0]
 
         # launch shell/exec cmd
         if names.empty?
-          EnfCli::Shell::Console::start host, user
+          EnfCli::Shell::Console::start
         else
           EnfCli::Shell::CLI.start names
         end
       end
-    end    
+    end
 
     map %w[--version -v] => :__print_version
     desc "--version, -v", "print the version"
+
     def __print_version
       puts EnfCli::VERSION
     end
 
     desc "update", "", :hide => true
+
     def update
       cmd = Gem::Commands::UpdateCommand.new
-      cmd.handle_options ['enfcli']
+      cmd.handle_options ["enfcli"]
       execute_gem_cmd cmd
     end
 
     desc "search", "", :hide => true
+
     def search
       cmd = Gem::Commands::SearchCommand.new
       cmd.handle_options ["enfcli"]
       execute_gem_cmd cmd
     end
 
+    desc "create-config-file", "Create a Xaptum configuration file in your home directory"
+    method_option :host, :type => :string, :required => true
+    method_option :user, :type => :string, :required => true
+
+    def create_config_file
+      host = options[:host]
+      user = options[:user]
+      config_file = File.new(CONFIG_FILE, "w+")
+      config_file.puts({ :host => host, :user => user }.to_json)
+      config_file.close
+      say "Config file created successfully at #{CONFIG_FILE}!", :green
+    end
+
+    desc "display-config-file", "Displays your Xaptum configuraton file, if it exists"
+
+    def display_config_file
+      file = CONFIG_FILE
+
+      ## return if file not found
+      raise EnfCli::ERROR, "#{file} not found!" unless File.exists?(file)
+
+      say File.readlines(file).join
+    end
+
     default_task :connect
-    
   end
 
   ##
   # Shell Module
   #
   module Shell
-    
     class Console
       class << self
         def execute(input)
@@ -289,61 +331,57 @@ module EnfCli
           argv = input.split(/[\s=](?=(?:[^"]|"[^"]*")*$)/)
           # now remove quotes.
           argv.each do |arg|
-            arg.gsub!(/\A"|"\Z/, '')
+            arg.gsub!(/\A"|"\Z/, "")
           end
           EnfCli::Shell::CLI.start(argv)
         end
 
-        def start(host, user)
+        def start
           $stdout.sync = true
-          # Set prompt
-          EnfCli::CTX.instance.prompt = user
 
-          # Set host
-          EnfCli::CTX.instance.host = host
-          
           # Read each line
-          comp = proc { |s| Readline::HISTORY.grep(/^#{Regexp.escape(s)}/) }        
+          comp = proc { |s| Readline::HISTORY.grep(/^#{Regexp.escape(s)}/) }
           Readline.completion_append_character = " "
           Readline.completion_proc = comp
 
           stty_save = `stty -g`.chomp
-          trap('INT') { system('stty', stty_save); exit }
-          
+          trap("INT") { system("stty", stty_save); exit }
+
           while input = Readline.readline(EnfCli::CTX.instance.prompt, true)
             break if input == "exit" or input == "\\q" or input == "quit"
-            
+
             # Remove blank lines from history
             Readline::HISTORY.pop if input == ""
-            
+
             execute(input) unless input == ""
           end
-        end  
-      end    
+        end
+      end
     end
 
     # Shell CLI class
-    class CLI < EnfThor
-
+    class CLI < EnfCli::EnfThor
       desc "ls [<dir>]", "List files in a directory", :hide => true
       method_option :dir, :type => :string, :required => false
+
       def ls(dir = nil)
         try_with_rescue do
           dir = "." unless dir
-          dir = EnfCli::expand_path( dir )
-          
-          Dir.entries( dir ).each{ |f|
-            puts f unless f.start_with?('.')
+          dir = EnfCli::expand_path(dir)
+
+          Dir.entries(dir).each { |f|
+            puts f unless f.start_with?(".")
           }
         end
       end
 
       desc "cat <file>", "Display contents of a file", :hide => true
+
       def cat(file)
         try_with_rescue do
           # expand path
           file = EnfCli::expand_path(file)
-          
+
           ## return if keyfile not found
           raise EnfCli::ERROR, "#{file} not found!" unless File.exists?(file)
 
@@ -352,6 +390,7 @@ module EnfCli
       end
 
       desc "pwd", "Current Working Directory", :hide => true
+
       def pwd
         try_with_rescue do
           say Dir.pwd
@@ -359,35 +398,69 @@ module EnfCli
       end
 
       desc "cd [<dir>]", "Change working directory", :hide => true
+
       def cd(dir = "~")
         try_with_rescue do
-          dir = EnfCli::expand_path( dir )
+          dir = EnfCli::expand_path(dir)
           raise EnfCli::ERROR, "No such directory #{dir}" unless Dir.exist?(dir)
           Dir.chdir(dir)
         end
       end
 
       desc "host", "Display ENF Controller host", :hide => true
+
       def host
         try_with_rescue do
           say EnfCli::CTX.instance.host, :bold
         end
       end
-      
+
       desc "clear", "Clear Terminal Screen", :hide => true
+
       def clear
         try_with_rescue do
           clear_code = %x{clear}
           print clear_code or system("cls")
         end
       end
-      
+
+      desc "display-session-token", "Gets the current session token"
+
+      def display_session_token
+        try_with_rescue_in_session do
+          say "#{EnfCli::CTX.instance.auth_token}"
+        end
+      end
+
+      desc "refresh-session-token", "Refreshes the current session token"
+
+      def refresh_session_token
+        try_with_rescue_in_session do
+          # Get user and host
+          host = EnfCli::CTX.instance.host
+          user = EnfCli::CTX.instance.user
+
+          # Ask for password
+          say "Refreshing session token.....", :bold
+          password = EnfCli::ask_password()
+
+          # Authenticate
+          resp = EnfApi::API.instance.authenticate(host, user, password)
+
+          # update session
+          EnfCli::CTX.instance.session = resp[:data][0]
+
+          # display success
+          say "Refreshed session token!", :green
+        end
+      end
+
       desc "#{EnfCli::NETWORK_CMD} COMMANDS", "#{EnfCli::NETWORK_CMD} commands"
       subcommand EnfCli::NETWORK_CMD, EnfCli::Cmd::Xcr
-      
+
       desc "#{EnfCli::IAM_CMD} COMMANDS", "#{EnfCli::IAM_CMD} commands"
       subcommand EnfCli::IAM_CMD, EnfCli::Cmd::Xiam
-      
+
       desc "#{EnfCli::FIREWALL_CMD} COMMANDS", "#{EnfCli::FIREWALL_CMD} commands"
       subcommand EnfCli::FIREWALL_CMD, EnfCli::Cmd::Xfw
 
@@ -401,5 +474,4 @@ module EnfCli
       subcommand EnfCli::DNS_CMD, EnfCli::Cmd::Xdns
     end
   end
-  
 end