faastruby 0.2.0

data/lib/faastruby/base.rb

@@ -0,0 +1,53 @@
+module FaaStRuby
+  HOST = ENV['FAASTRUBY_HOST'] || 'https://api.faastruby.io'
+
+  class << self
+    attr_accessor :configuration
+  end
+
+  def self.api_key
+    configuration&.api_key || ENV['FAASTRUBY_API_KEY']
+  end
+
+  def self.api_secret
+    configuration&.api_secret || ENV['FAASTRUBY_API_SECRET']
+  end
+
+  def self.configure
+    self.configuration ||= Configuration.new
+    yield(configuration)
+  end
+
+  def self.credentials
+    {api_key: api_key, api_secret: api_secret}
+  end
+
+  class Configuration
+    attr_accessor :api_key, :api_secret
+  end
+
+
+  class BaseObject
+    def initialize(params = {}, &block)
+      @errors = []
+      @api = API.new
+      self.mass_assign(params) if params
+      yield self if block_given?
+    end
+
+    def assign_attributes(params = {}, &block)
+      self.mass_assign(params) if params
+      yield self if block_given?
+    end
+
+    def attributes=(params)
+      assign_attributes(params)
+    end
+
+    def mass_assign(attrs)
+      attrs.each do |key, value|
+        self.public_send("#{key}=", value)
+      end
+    end
+  end
+end

data/lib/faastruby/cli.rb

@@ -0,0 +1,42 @@
+require 'yaml'
+require 'tty-spinner'
+require 'tty-table'
+require 'zip'
+require 'colorize'
+require 'faastruby/cli/commands'
+require 'faastruby/cli/package'
+
+module FaaStRuby
+  FAASTRUBY_YAML = 'faastruby.yml'
+  FAASTRUBY_CREDENTIALS = File.expand_path(ENV['FAASTRUBY_CREDENTIALS'] || '~/.faastruby')
+  SPINNER_FORMAT = :spin_2
+
+  class CLI
+    def self.error(message, color: :red)
+      message.each {|m| STDERR.puts m.colorize(color)} if message.is_a?(Array)
+      STDERR.puts message.colorize(color) if message.is_a?(String)
+      exit 1
+    end
+
+    def self.run(command, args)
+      if command.nil?
+        FaaStRuby::Command::Help.new(args).run
+        return
+      end
+      check_version
+      error("Unknown command: #{command}") unless FaaStRuby::Command::COMMANDS.has_key?(command)
+      FaaStRuby::Command::COMMANDS[command].new(args).run
+    end
+
+    def self.check_version
+      latest = RestClient.get('https://faastruby.io/gem/minimum.txt').body rescue '0.0.1'
+      if Gem::Version.new(FaaStRuby::VERSION) < Gem::Version.new(latest)
+        FaaStRuby.error([
+          "You are using an old version of the gem. Please run 'gem update faastruby'.".red,
+          "Installed version: #{FaaStRuby::VERSION}",
+          "Latest version: #{latest}"
+        ], color: nil)
+      end
+    end
+  end
+end

data/lib/faastruby/cli/commands.rb

@@ -0,0 +1,61 @@
+module FaaStRuby
+  module Command
+    class BaseCommand
+      def self.spin(message)
+        spinner = TTY::Spinner.new(":spinner #{message}", format: SPINNER_FORMAT)
+        spinner.auto_spin
+        spinner
+      end
+
+      def write_file(path, content, mode = 'w')
+        File.open(path, mode){|f| f.write(content) }
+        message = "#{mode == 'w' ? '+' : '~'} f #{path}"
+        puts message.green if mode == 'w'
+        puts message.yellow if mode == 'w+' || mode == 'a'
+      end
+
+      def spin(message)
+        spinner = TTY::Spinner.new(":spinner #{message}", format: SPINNER_FORMAT)
+        spinner.auto_spin
+        spinner
+      end
+
+      def load_yaml
+        return YAML.load(File.read(FAASTRUBY_YAML)) if File.file?(FAASTRUBY_YAML)
+        FaaStRuby.error("Could not find file #{FAASTRUBY_YAML}")
+      end
+
+      def spin(message)
+        self.class.spin(message)
+      end
+    end
+  end
+end
+
+require 'faastruby/cli/credentials'
+require 'faastruby/cli/commands/function'
+require 'faastruby/cli/commands/workspace'
+require 'faastruby/cli/commands/help'
+require 'faastruby/cli/commands/version'
+
+module FaaStRuby
+  module Command
+    COMMANDS = {
+      'new' => FaaStRuby::Command::Function::New,
+      'deploy-to' => FaaStRuby::Command::Function::Deploy,
+      'remove-from' => FaaStRuby::Command::Function::RemoveFrom,
+      'update-context' => FaaStRuby::Command::Function::UpdateContext,
+      'upgrade' => FaaStRuby::Command::Function::Upgrade,
+      'build' => FaaStRuby::Command::Function::Build,
+      'create-workspace' => FaaStRuby::Command::Workspace::Create,
+      'destroy-workspace' => FaaStRuby::Command::Workspace::Destroy,
+      'list-workspace' => FaaStRuby::Command::Workspace::List,
+      'test' => FaaStRuby::Command::Function::Test,
+      'run' => FaaStRuby::Command::Function::Run,
+      'help' => FaaStRuby::Command::Help,
+      '-h' => FaaStRuby::Command::Help,
+      '--help' => FaaStRuby::Command::Help,
+      '-v' => FaaStRuby::Command::Version
+    }
+  end
+end

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

@@ -0,0 +1,31 @@
+require 'ostruct'
+module FaaStRuby
+  module Command
+    module Function
+      class FunctionBaseCommand < BaseCommand
+        # --- Example YAML
+        # name: example
+        # test:
+        #   command: "rspec"
+        #   on_fail:
+        #     build: false
+        #     deploy: false
+        def load_yaml
+          FaaStRuby::CLI.error("It looks like you created this function with an old version of faastruby. Please run 'faastruby upgrade'.") if File.file?('handler.rb') && !File.file?('faastruby.yml')
+          FaaStRuby::CLI.error("Could not find file 'faastruby.yml' in the current directory") unless File.file?('faastruby.yml')
+          @yaml_config = YAML.load(File.read('./faastruby.yml'))
+          FaaStRuby::CLI.error("Could read function name from 'faastruby.yml'. Make sure you have a key 'name: FUNCTION_NAME' in that file!") unless @yaml_config['name']
+        end
+      end
+    end
+  end
+end
+
+require 'faastruby/cli/commands/function/build'
+require 'faastruby/cli/commands/function/deploy'
+require 'faastruby/cli/commands/function/new'
+require 'faastruby/cli/commands/function/remove_from'
+require 'faastruby/cli/commands/function/test'
+require 'faastruby/cli/commands/function/update_context'
+require 'faastruby/cli/commands/function/upgrade'
+require 'faastruby/cli/commands/function/run'

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

@@ -0,0 +1,65 @@
+module FaaStRuby
+  module Command
+    module Function
+      class Build < FunctionBaseCommand
+
+        def self.build(source, output_file, quiet = false)
+          spinner = spin("Building package...")
+          FaaStRuby::Package.new(source, output_file).build
+          spinner.stop('Done!')
+          puts "+ f #{output_file}".green unless quiet
+        end
+
+        def initialize(args)
+          @args = args
+          load_yaml
+          @function_name = @yaml_config['name']
+          @abort_when_tests_fail = @yaml_config['abort_build_when_tests_fail']
+          parse_options
+          @options['source'] ||= '.'
+          @options['output_file'] ||= "#{@function_name}.zip"
+        end
+
+        def run
+          tests_passed = run_tests
+          FaaStRuby::CLI.error("Build aborted because tests failed and you have 'abort_build_when_tests_fail: true' in 'faastruby.yml'") unless tests_passed || !@abort_when_tests_fail
+          puts "Warning: Ignoring failed tests because you have 'abort_build_when_tests_fail: false' in 'faastruby.yml'".yellow if !tests_passed && !@abort_when_tests_fail
+          build(@options['source'], @options['output_file'])
+        end
+
+        def self.help
+          "build".blue + " [-s, --source SOURCE_DIR] [-o, --output-file OUTPUT_FILE]"
+        end
+
+        def usage
+          "Usage: faastruby #{self.class.help}"
+        end
+
+        private
+
+        def build(source, output_file)
+          self.class.build(source, output_file)
+        end
+
+        def run_tests
+          FaaStRuby::Command::Function::Test.new(true).run(do_not_exit: true)
+        end
+
+        def parse_options
+          @options = {}
+          while @args.any?
+            option = @args.shift
+            case option
+            when '-s', '--source-dir'
+              @options['source'] = @args.shift
+            when '-o', '--output-file'
+              @options['output_file'] = @args.shift
+            else
+              FaaStRuby::CLI.error(["Unknown argument: #{option}".red, usage], color: nil)
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,62 @@
+module FaaStRuby
+  module Command
+    module Function
+      class Deploy < 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']
+          @abort_when_tests_fail = @yaml_config['abort_deploy_when_tests_fail']
+          FaaStRuby::Credentials.load_for(@workspace_name)
+        end
+
+        def run
+          tests_passed = run_tests
+          FaaStRuby::CLI.error("Deploy aborted because tests failed and you have 'abort_deploy_when_tests_fail: true' in 'faastruby.yml'") unless tests_passed || !@abort_when_tests_fail
+          puts "Warning: Ignoring failed tests because you have 'abort_deploy_when_tests_fail: false' in 'faastruby.yml'".yellow if !tests_passed && !@abort_when_tests_fail
+          package_file_name = build_package
+          spinner = spin("Deploying to workspace '#{@workspace_name}'...")
+          workspace = FaaStRuby::Workspace.new(name: @workspace_name).deploy(package_file_name)
+          if workspace.errors.any?
+            spinner.stop('Failed :(')
+            FaaStRuby::CLI.error(workspace.errors)
+          end
+          spinner.stop('Done!')
+        end
+
+        def self.help
+          "deploy-to".blue + " WORKSPACE_NAME"
+        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 run_tests
+          FaaStRuby::Command::Function::Test.new(true).run(do_not_exit: true)
+        end
+
+        def build_package
+          source = '.'
+          output_file = "#{@function_name}.zip"
+          FaaStRuby::Command::Function::Build.build(source, output_file, true)
+          output_file
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,99 @@
+module FaaStRuby
+  module Command
+    module Function
+      class New < FunctionBaseCommand
+        def initialize(args)
+          @args = args
+          @missing_args = []
+          FaaStRuby::CLI.error(@missing_args, color: nil) if missing_args.any?
+          @function_name = @args.shift
+          @base_dir = "./#{@function_name}"
+          @yaml_content = {
+            'name' => @function_name,
+            'test_command' => 'rspec',
+            'abort_build_when_tests_fail' => true,
+            'abort_deploy_when_tests_fail' => true
+          }
+          parse_options
+          @options['template'] ||= 'example'
+        end
+
+        def run
+          dir_exists? unless @options['force']
+          copy_template
+          write_yaml
+          bundle_install
+        end
+
+        def self.help
+          "new".blue + " FUNCTION_NAME [--blank] [--force]" +
+          <<-EOS
+
+    --blank              # Create a blank function
+    --force              # Continue if directory already exists and overwrite files
+EOS
+        end
+
+        def usage
+          "Usage: faastruby #{self.class.help}"
+        end
+
+        private
+
+        def parse_options
+          @options = {}
+          while @args.any?
+            option = @args.shift
+            case option
+            when '-f', '--force'
+              @options['force'] = true
+            when '--blank'
+              @options['template'] = 'example-blank'
+            else
+              FaaStRuby::CLI.error(["Unknown argument: #{option}".red, usage], color: nil)
+            end
+          end
+        end
+
+        def dir_exists?
+          return unless File.directory?(@base_dir)
+          print "The directory '#{@function_name}' already exists. Overwrite files? [y/N] "
+          response = STDIN.gets.chomp
+          FaaStRuby::CLI.error("Cancelled", color: nil) unless response == 'y'
+        end
+
+        def missing_args
+          if @args.empty?
+            @missing_args << "Missing argument: FUNCTION_NAME".red
+            @missing_args << usage
+          end
+          @missing_args
+        end
+
+        def copy_template
+          source = "#{Gem::Specification.find_by_name("faastruby").gem_dir}/#{@options['template']}"
+          FileUtils.mkdir_p(@base_dir)
+          FileUtils.cp_r("#{source}/.", "#{@base_dir}/")
+          puts "+ d #{@base_dir}".green
+          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
+          puts "+ f #{@base_dir}/Gemfile".green
+          puts "+ f #{@base_dir}/handler.rb".green
+        end
+
+        def write_yaml
+          write_file("#{@base_dir}/faastruby.yml", @yaml_content.to_yaml)
+        end
+
+        def bundle_install
+          spinner = spin("Installing gems...")
+          system("bundle install --gemfile=#{@base_dir}/Gemfile > /dev/null")
+          spinner.stop('Done!')
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,71 @@
+module FaaStRuby
+  module Command
+    module Function
+      class RemoveFrom < 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
+        end
+
+        def run
+          warning unless @options['force']
+          FaaStRuby::CLI.error("Cancelled") unless @options['force'] == 'y'
+          spinner = spin("Removing function from workspace '#{@workspace_name}'...")
+          workspace = FaaStRuby::Workspace.new(name: @workspace_name)
+          function = FaaStRuby::Function.new(name: @function_name, workspace: workspace)
+          function.destroy
+          if function.errors.any?
+            spinner.stop('Failed :(')
+            FaaStRuby::CLI.error(function.errors)
+          end
+          spinner.stop('Done!')
+        end
+
+        def self.help
+          "remove-from".blue + " WORKSPACE_NAME [-y, --yes]"
+        end
+
+        def usage
+          "Usage: faastruby #{self.class.help}"
+        end
+
+        private
+
+        def warning
+          print "WARNING: ".red
+          puts "This action will permanently remove the function '#{@function_name}' from the workspace '#{@workspace_name}'."
+          print "Are you sure? [y/N] "
+          @options['force'] = STDIN.gets.chomp
+        end
+
+        def parse_options
+          @options = {}
+          while @args.any?
+            option = @args.shift
+            case option
+            when '-y', '--yes'
+              @options['force'] = 'y'
+            else
+              FaaStRuby::CLI.error(["Unknown argument: #{option}".red, usage], color: nil)
+            end
+          end
+        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