faastruby 0.2.0

data/lib/faastruby/cli/commands/function/run.rb

@@ -0,0 +1,135 @@
+module FaaStRuby
+  module Command
+    module Function
+      class Run < FunctionBaseCommand
+        def initialize(args)
+          @args = args
+          @missing_args = []
+          FaaStRuby::CLI.error(@missing_args, color: nil) if missing_args.any?
+          @options = {}
+          @options['workspace_name'] = @args.shift
+          load_yaml
+          @function_name = @yaml_config['name']
+          parse_options
+          @options['query'] = "?#{@options['query'].join('&')}" if @options['query']&.any?
+        end
+
+        def run
+          return curl if @options['curl']
+          function = FaaStRuby::Function.new(name: @function_name)
+          response = function.run(@options)
+          puts response.body
+        end
+
+        def curl
+          command = ["curl"]
+          command << "-X #{@options['method'].upcase}" if @options['method']
+          @options['headers']&.each {|h,v| command << "-H '#{h}: #{v}'"}
+          command << "-d '#{@options['body']}'" if @options['body']
+          command << "'#{HOST}/#{@options['workspace_name']}/#{@function_name}#{@options['query']}'"
+          puts command.join(" ")
+        end
+
+        def self.help
+          'run'.blue + ' WORKSPACE_NAME [OPTIONS]' +
+          <<-EOS
+
+    Options:
+    -b, --body 'DATA'              # The request body
+    --stdin                        # Read the request body from STDIN
+    -m, --method METHOD            # The request method
+    -h, --header 'Header: Value'   # Set a header. Can be used multiple times.
+    -f, --form 'a=1&b=2'           # Send form data and set header 'Content-Type: application/x-www-form-urlencoded'
+    -j, --json '{"a":"1"}'         # Send JSON data and set header 'Content-Type: application/json'
+    -t, --time                     # Return function run time in the response
+    -q, --query 'foo=bar'          # Set a query parameter for the request. Can be used multiple times.
+    --curl                         # Return the CURL command equivalent for the request
+EOS
+        end
+
+        def usage
+          "Usage: faastruby #{self.class.help}"
+        end
+
+        private
+
+        def parse_options
+          while @args.any?
+            option = @args.shift
+            case option
+            when '-b', '--body'
+              set_body
+            when '-j', '--json'
+              set_json
+            when '-f', '--form'
+              set_form
+            when '--stdin'
+              set_stdin
+            when '-m', '--method'
+              @options['method'] = @args.shift.downcase
+            when '-h', '--header'
+              set_header
+            when '-t', '--time'
+              @options['time'] = true
+            when '--curl'
+              @options['curl'] = true
+            when '-q', '--query'
+              set_query
+            else
+              FaaStRuby::CLI.error(["Unknown argument: #{option}".red, usage], color: nil)
+            end
+          end
+        end
+
+        def set_body
+          @options['method'] ||= 'post'
+          @options['body'] = @args.shift
+          default_content_type = 'text/plain'
+          @options['headers'] ? @options['headers']['Content-Type'] = default_content_type : @options['headers'] = {'Content-Type' => default_content_type}
+        end
+
+        def set_json
+          content_type = 'application/json'
+          @options['method'] ||= 'post'
+          @options['body'] = @args.shift
+          @options['headers'] ? @options['headers']['Content-Type'] = content_type : @options['headers'] = {'Content-Type' => content_type}
+        end
+
+        def set_form
+          content_type = 'application/x-www-form-urlencoded'
+          @options['method'] ||= 'post'
+          @options['body'] = @args.shift
+          @options['headers'] ? @options['headers']['Content-Type'] = content_type : @options['headers'] = {'Content-Type' => content_type}
+        end
+
+        def set_stdin
+          @options['method'] ||= 'post'
+          @options['body'] = STDIN.gets.chomp
+          content_type = 'text/plain'
+          @options['headers'] ? @options['headers']['Content-Type'] = content_type : @options['headers'] = {'Content-Type' => content_type}
+        end
+
+        def set_query
+          query = "#{@args.shift}"
+          @options['query'] ? @options['query'] << query : @options['query'] = [query]
+        end
+
+        def set_header
+          header = @args.shift
+          key, value = header.split(":")
+          value.strip!
+          @options['headers'] ? @options['headers'][key] = value : @options['headers'] = {key => value}
+        end
+
+        def missing_args
+          if @args.empty?
+            @missing_args << "Missing argument: WORKSPACE_NAME".red
+            @missing_args << usage
+          end
+          FaaStRuby::CLI.error(["'#{@args.first}' is not a valid workspace name.".red, usage], color: nil) if @args.first =~ /^-.*/
+          @missing_args
+        end
+      end
+    end
+  end
+end

data/lib/faastruby/cli/commands/function/test.rb

@@ -0,0 +1,42 @@
+require 'open3'
+module FaaStRuby
+  module Command
+    module Function
+      class Test < FunctionBaseCommand
+        def initialize(args)
+          @args = args
+          load_yaml
+          @function_name = @yaml_config['name']
+          @test_command = @yaml_config['test_command']
+        end
+
+        def run(do_not_exit: false)
+          unless @test_command
+            puts "[skipped] You have no 'test_command' key/value in 'faastruby.yml'. Please consider using rspec!".yellow
+            return true
+          end
+          spinner = spin("Running tests...")
+          output, status = Open3.capture2e(@test_command)
+          if status == 0
+            spinner.stop('Passed!')
+            puts output
+            return true
+          else
+            spinner.stop('Failed :(')
+            FaaStRuby::CLI.error(output, color: nil) unless do_not_exit
+            puts output if do_not_exit
+            return false
+          end
+        end
+
+        def self.help
+          'test'.blue
+        end
+
+        def usage
+          "Usage: faastruby #{self.class.help}"
+        end
+      end
+    end
+  end
+end

data/lib/faastruby/cli/commands/function/update_context.rb

@@ -0,0 +1,67 @@
+module FaaStRuby
+  module Command
+    module Function
+      class UpdateContext < FunctionBaseCommand
+        def initialize(args)
+          @args = args
+          @missing_args = []
+          FaaStRuby::CLI.error(@missing_args, color: nil) if missing_args.any?
+          @workspace_name = @args.shift
+          load_yaml
+          @function_name = @yaml_config['name']
+          FaaStRuby::Credentials.load_for(@workspace_name)
+          parse_options(require_options: {'data' => 'context data'} )
+        end
+
+        def run
+          spinner = spin("Uploading context data to '#{@workspace_name}'...")
+          workspace = FaaStRuby::Workspace.new(name: @workspace_name)
+          function = FaaStRuby::Function.new(name: @function_name, workspace: workspace, context: @options['data'])
+          function.update
+          if function.errors.any?
+            spinner.stop('Failed :(')
+            FaaStRuby::CLI.error(function.errors)
+          end
+          spinner.stop('Done!')
+        end
+
+        def self.help
+          "update-context".blue + " WORKSPACE_NAME [-d, --data 'STRING'] [--stdin]"
+        end
+
+        def usage
+          "Usage: faastruby #{self.class.help}"
+        end
+
+        private
+
+        def missing_args
+          if @args.empty?
+            @missing_args << "Missing argument: WORKSPACE_NAME".red
+            @missing_args << usage
+          end
+          FaaStRuby::CLI.error(["'#{@args.first}' is not a valid workspace name.".red, usage], color: nil) if @args.first =~ /^-.*/
+          @missing_args
+        end
+
+        def parse_options(require_options: {})
+          @options = {}
+          while @args.any?
+            option = @args.shift
+            case option
+            when '-d', '--data'
+              @options['data'] = @args.shift
+            when '--stdin'
+              @options['data'] = STDIN.gets.chomp
+            else
+              FaaStRuby::CLI.error("Unknown argument: #{option}")
+            end
+          end
+          require_options.keys.each do |option|
+            FaaStRuby::CLI.error(["Missing #{require_options[option]}".red, usage], color: nil) unless @options[option]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/faastruby/cli/commands/function/upgrade.rb

@@ -0,0 +1,69 @@
+module FaaStRuby
+  module Command
+    module Function
+      class Upgrade < FunctionBaseCommand
+        def initialize(args)
+          @args = args
+        end
+
+        def run
+          check_function_name
+          set_yaml_content
+          write_yaml
+          write_gemfile
+          copy_spec_folder
+          puts "Upgrade complete. Please read the documentation at https://faastruby.io/tutorial.html to learn about the changes."
+          puts "To deploy, run 'faastruby deploy-to WORKSPACE_NAME'."
+        end
+
+        def set_yaml_content
+          @yaml_content = {
+            'name' => @function_name,
+            'test_command' => nil,
+            'abort_build_when_tests_fail' => false,
+            'abort_deploy_when_tests_fail' => false
+          }
+        end
+
+        def check_function_name
+          FaaStRuby::CLI.error('You need to run this command from the function directory') unless File.file?('handler.rb')
+          @function_name = Dir.pwd.split('/').last
+          unless @function_name
+            puts "What's the name of this function? (accepts only letters, numbers, - or _)"
+            print "Enter function name: "
+            @function_name = STDIN.gets.chomp
+          end
+          FaaStRuby::CLI.error('You need to provide a valid function name (letters, numbers, - or _)') unless @function_name.match(/^[a-zA-Z\-_0-9]{1,}$/)
+        end
+
+        def write_yaml
+          write_file("./faastruby.yml", @yaml_content.to_yaml)
+        end
+
+        def write_gemfile
+          content = "\n# Added by 'faastruby upgrade'\ngroup :test do\n  gem 'rspec'\nend\n"
+          write_file("./Gemfile", content, 'a')
+        end
+
+        def copy_spec_folder
+          @base_dir = '.'
+          source = "#{Gem::Specification.find_by_name("faastruby").gem_dir}/example-blank"
+          FileUtils.cp_r("#{source}/spec", "#{@base_dir}/")
+          puts "+ d #{@base_dir}/spec".green
+          puts "+ d #{@base_dir}/spec/helpers".green
+          puts "+ f #{@base_dir}/spec/helpers/faastruby.rb".green
+          puts "+ f #{@base_dir}/spec/handler_spec.rb".green
+          puts "+ f #{@base_dir}/spec/spec_helper.rb".green
+        end
+
+        def self.help
+          "upgrade".blue
+        end
+
+        def usage
+          "Usage: faastruby #{self.class.help}"
+        end
+      end
+    end
+  end
+end

data/lib/faastruby/cli/commands/help.rb

@@ -0,0 +1,31 @@
+module FaaStRuby
+  module Command
+    class Help < BaseCommand
+      def initialize(args)
+        @args = args
+      end
+
+      def run
+        puts "FaaStRuby CLI - Manage workspaces and functions hosted at faastruby.io"
+        puts
+        puts 'help, -h, --help     # Displays this help'
+        puts '-v                   # Print version and exit'
+        puts
+        workspaces = ["Workspaces:"]
+        functions = ["Functions:"]
+        FaaStRuby::Command::COMMANDS.each do |command, klass|
+          next if command == 'upgrade'
+          next if klass.to_s.match(/.+Command::Help$/)
+          next if klass.to_s.match(/.+Command::Version$/)
+          section = functions if klass.to_s.match(/.+::Function::.+/)
+          section = workspaces if klass.to_s.match(/.+::Workspace::.+/)
+          section << "  #{klass.help}"
+        end
+        puts workspaces
+        puts
+        puts functions
+        puts
+      end
+    end
+  end
+end

data/lib/faastruby/cli/commands/version.rb

@@ -0,0 +1,13 @@
+module FaaStRuby
+  module Command
+    class Version < BaseCommand
+      def initialize(args)
+        @args = args
+      end
+
+      def run
+        puts FaaStRuby::VERSION
+      end
+    end
+  end
+end

data/lib/faastruby/cli/commands/workspace.rb

@@ -0,0 +1,12 @@
+module FaaStRuby
+  module Command
+    module Workspace
+      class WorkspaceBaseCommand < BaseCommand
+      end
+    end
+  end
+end
+
+require 'faastruby/cli/commands/workspace/create'
+require 'faastruby/cli/commands/workspace/destroy'
+require 'faastruby/cli/commands/workspace/list'

data/lib/faastruby/cli/commands/workspace/create.rb

@@ -0,0 +1,70 @@
+module FaaStRuby
+  module Command
+    module Workspace
+      class Create < WorkspaceBaseCommand
+        def initialize(args)
+          @args = args
+          @missing_args = []
+          FaaStRuby::CLI.error(@missing_args, color: nil) if missing_args.any?
+          @workspace_name = @args.shift
+          parse_options
+          @options['credentials_file'] ||= FAASTRUBY_CREDENTIALS
+        end
+
+        def run
+          spinner = spin("Requesting credentials...")
+          workspace = FaaStRuby::Workspace.create(name: @workspace_name, email: @options['email'])
+          spinner.stop("Done!")
+          FaaStRuby::CLI.error(workspace.errors) if workspace.errors.any?
+          if @options['stdout']
+            puts "IMPORTANT: Please store the credentials below in a safe place. If you lose them you will not be able to manage your workspace.".yellow
+            puts "API_KEY: #{workspace.credentials['api_key']}"
+            puts "API_SECRET: #{workspace.credentials['api_secret']}"
+          else
+            puts "Writing credentials to #{@options['credentials_file']}"
+            FaaStRuby::Credentials.add(@workspace_name, workspace.credentials, @options['credentials_file'])
+          end
+          puts "Workspace '#{@workspace_name}' created"
+        end
+
+        def self.help
+          "create-workspace".blue + " WORKSPACE_NAME [--stdout|-o, --output-file] [-e, --email YOUR_EMAIL_ADDRESS]"
+        end
+
+        def usage
+          "Usage: faastruby #{self.class.help}"
+        end
+
+        private
+
+        def missing_args
+          if @args.empty?
+            @missing_args << "Missing argument: WORKSPACE_NAME".red
+            @missing_args << "Usage: faastruby create-workspace WORKSPACE_NAME"
+          end
+          FaaStRuby::CLI.error(["'#{@args.first}' is not a valid workspace name.".red, usage], color: nil) if @args.first =~ /^-.*/
+          @missing_args
+        end
+
+        def parse_options
+          @options = {}
+          while @args.any?
+            option = @args.shift
+            case option
+            # when '--file'
+            #   @options['credentials_file'] = @args.shift
+            when '--stdout'
+              @options['stdout'] = true
+            when '-o', '--output-file'
+              @options['credentials_file'] = @args.shift
+            when '-e', '--email'
+              @options['email'] = @args.shift
+            else
+              FaaStRuby.error("Unknown argument: #{option}")
+            end
+          end
+        end
+      end
+    end
+  end
+end