legion-cli 0.2.0

data/lib/legion/cli/task.rb

@@ -0,0 +1,49 @@
+module Legion
+  class Cli
+    class Task < Thor
+      package_name 'Legion'
+
+      option :limit, type: :numeric, required: true, default: 10, desc: 'how many tasks to return'
+      desc 'show', 'show'
+      option :status, type: :string, required: false, desc: 'search for specific status'
+      def show
+        Legion::Service.new(cache: false, crypt: false, extensions: false, log_level: 'error')
+        rows = [%w[id relationship function status]]
+        Legion::Data::Model::Task.limit(options[:limit]).order(:id).reverse_each do |row|
+          rows.push([row.values[:id], row.values[:relationship_id], row.values[:function_id], row.values[:status]])
+        end
+
+        print_table rows
+      end
+
+      desc 'test', 'test'
+      def status(id)
+        Legion::Service.new(cache: false, crypt: false, extensions: false, log_level: 'error')
+        say Legion::Data::Model::Task[id].values
+      end
+
+      desc 'logs', 'logs'
+      option :limit, type: :numeric, required: true, default: 10, desc: 'how many tasks to return'
+      def logs(id)
+        Legion::Service.new(cache: false, crypt: false, extensions: false, log_level: 'error')
+        rows = [%w[id node_id created entry]]
+        Legion::Data::Model::TaskLog.where(task_id: id).limit(options[:limit]).each do |row|
+          rows.push([row.values[:id], row.values[:node_id], row.values[:created], row.values[:entry]])
+        end
+        print_table rows
+      end
+
+      desc 'purge', 'purge'
+      def purge
+        Legion::Service.new(cache: false, crypt: false, extensions: false, log_level: 'error')
+        days = ask 'how many days do you want to keep?', default: 7
+        dataset = Legion::Data::Model::Task.where { created < DateTime.now - days.to_i }
+        yes? "This will delete #{dataset.count} tasks, continue?", :red
+        dataset.delete
+        say 'Done!'
+      end
+
+      default_task :show
+    end
+  end
+end

data/lib/legion/cli/trigger.rb

@@ -0,0 +1,81 @@
+module Legion
+  class Cli
+    class Trigger < Thor
+      desc 'queue', 'used to send a job directly to a worker via Legion::Transport'
+      option :extension, type: :string, required: false, desc: 'extension short name'
+      option :runner, type: :string, required: false, desc: 'runner short name'
+      option :function, type: :string, required: false, desc: 'function short name'
+      option :delay, type: :numeric, default: 0, desc: 'how long to wait before running the task'
+      def queue(*args)
+        Legion::Service.new(cache: false, crypt: false, extensions: false, log_level: 'error')
+        include Legion::Extensions::Helpers::Task
+        response = if options['extension'].is_a? String
+                     options[:extension]
+                   else
+                     ask 'trigger extension?', limited_to: Legion::Data::Model::Extension.map(:name)
+                   end
+        trigger_extension = Legion::Data::Model::Extension.where(name: response).first
+        runners = Legion::Data::Model::Runner.where(extension_id: trigger_extension.values[:id])
+        if runners.count === 1
+          trigger_runner = runners.first
+          say "Auto selecting #{trigger_runner.values[:name]} since it is the only option for runners"
+        else
+          response = options[:runner].is_a?(String) ? options[:runner] : ask('trigger runner?', limited_to: runners.map(:name))
+          trigger_runner = Legion::Data::Model::Runner.where(name: response).where(extension_id: trigger_extension.values[:id]).first
+        end
+
+        functions = Legion::Data::Model::Function.where(runner_id: trigger_runner.values[:id])
+
+        if functions.count === 1
+          trigger_function = functions.first
+          say "Auto selecting #{trigger_function.values[:name]} since it is the only option for functions"
+        else
+          response = options[:function].is_a?(String) ? options[:function] : ask('trigger function?', limited_to: Legion::Data::Model::Function.where(runner_id: trigger_runner.values[:id]).map(:name))
+          trigger_function = Legion::Data::Model::Function.where(runner_id: trigger_runner.values[:id]).where(name: response).first
+        end
+        say "#{trigger_runner.values[:namespace]}.#{trigger_function.values[:name]} selected as trigger", :green, :italicized
+        payload = {}
+        auto_opts = {}
+        unless args.count.zero?
+          args.each do |arg|
+            test = arg.split(':')
+            auto_opts[test[0].to_sym] = test[1]
+          end
+        end
+
+        Legion::JSON.load(trigger_function.values[:args]).each do |arg, required|
+          next if %w[args payload opts options].include? arg.to_s
+
+          if auto_opts.key? arg
+            payload[arg.to_sym] = auto_opts[arg]
+            next
+          end
+          response = ask "#{required == 'keyreq' ? '[required]' : '[optional]'} #{arg} value:"
+          if response.empty? && required == 'keyreq'
+            say "Error! #{arg} is required and cannot be empty", :red
+            redo
+          end
+          payload[arg.to_sym] = response unless response.empty?
+        end
+
+        status = options[:delay].zero? ? 'task.queued' : 'task.delayed'
+        task = generate_task_id(function_id: trigger_function.values[:id], status: status, runner_id: trigger_runner.values[:id], args: payload, delay: options[:delay])
+
+        unless options[:delay].zero?
+          say "Task: #{task[:task_id]} is queued and will be run in #{options[:delay]}s"
+          return true
+        end
+
+        routing_key = "#{trigger_extension.values[:exchange]}.#{trigger_runner.values[:queue]}.#{trigger_function.values[:name]}"
+        exchange = Legion::Transport::Messages::Dynamic.new(function: trigger_function.values[:name], function_id: trigger_function.values[:id], routing_key: routing_key, args: payload)
+        exchange.options[:task_id] = task[:task_id]
+        exchange.publish if options[:delay].zero?
+
+        say "Task: #{task[:task_id]} was queued"
+      end
+      remove_command :generate_task_id
+
+      default_task :queue
+    end
+  end
+end

data/lib/legion/cli/version.rb

@@ -0,0 +1,5 @@
+module Legion
+  class Cli
+    VERSION = '0.2.0'.freeze
+  end
+end

data/lib/legion/lex.rb

@@ -0,0 +1,91 @@
+require 'thor'
+require 'legion/cli/version'
+require 'legion/cli/lex/actor'
+require 'legion/cli/lex/exchange'
+require 'legion/cli/lex/message'
+require 'legion/cli/lex/queue'
+require 'legion/cli/lex/runner'
+
+module Legion
+  class Cli
+    class LexBuilder < Thor
+      check_unknown_options!
+      include Thor::Actions
+
+      no_commands do
+        def lex
+          Dir.pwd.split('/').last.split('-').last
+        end
+      end
+
+      def self.exit_on_failure?
+        true
+      end
+
+      def self.source_root
+        File.dirname(__FILE__)
+      end
+
+      desc 'actor', 'creates and manages actors'
+      subcommand 'actor', Legion::Cli::Lex::Actor
+
+      desc 'exchange', 'creates and manages exchanges'
+      subcommand 'exchange', Legion::Cli::Lex::Exchange
+
+      desc 'messages', 'creates and manages messages'
+      subcommand 'message',  Legion::Cli::Lex::Message
+
+      desc 'queue', 'creates and manages queues'
+      subcommand 'queue', Legion::Cli::Lex::Queue
+
+      desc 'runner', 'creates and manages runners'
+      subcommand 'runner', Legion::Cli::Lex::Runner
+
+      desc 'version', 'Display Version'
+      map %w[-v --version] => :version
+      def version
+        say "Legion::CLI #{Legion::Cli::VERSION}"
+      end
+
+      method_option rspec: true
+      method_option pipeline: true
+      method_option git_init: true
+      method_option bundle_install: true
+      desc 'create :name', 'creates a new lex'
+      def create(name)
+        if Dir.pwd.include?('lex-')
+          say('already inside a lex_gen, try moving to a different directory', :red)
+          return nil
+        end
+
+        vars = { filename: "lex-#{name}", class_name: name.capitalize, lex: name }
+        filename = vars[:filename]
+        template('lex/templates/base/gemspec.erb', "#{filename}/#{filename}.gemspec", vars)
+        template('lex/templates/base/gemfile.erb', "#{filename}/Gemfile", vars)
+        template('lex/templates/base/gitignore.erb', "#{filename}/.gitignore", vars)
+        template('lex/templates/base/lic.erb', "#{filename}/LICENSE.txt", vars)
+        template('lex/templates/base/rakefile.erb', "#{filename}/Rakefile", vars)
+        template('lex/templates/base/rubocop.yml.erb', "#{filename}/.rubocop.yml", vars)
+        template('lex/templates/base/readme.md.erb', "#{filename}/README.md", **vars)
+        template('lex/templates/base/lex.erb', "#{filename}/lib/legion/extensions/#{name}.rb", vars)
+        template('lex/templates/base/version.erb', "#{filename}/lib/legion/extensions/#{name}/version.rb", vars)
+        if options[:pipeline]
+          template('lex/templates/base/bitbucket.yml.erb', "#{filename}/bitbucket-pipelines.yml", vars)
+        end
+        template('lex/templates/base/spec_helper.rb.erb', "#{filename}/spec/spec_helper.rb", vars)
+        template('lex/templates/base/lex_spec.erb', "#{filename}/spec/legion/#{name}_spec.rb", vars)
+
+        return if !options[:git_init] && !options[:bundle_install]
+
+        run("cd lex_gen-#{filename}")
+        if options[:git_init]
+          run('git init')
+          run('git add .')
+          run('git commit -m \'Initial commit\'')
+        end
+
+        run('bundle update') if options[:bundle_install]
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,176 @@
+--- !ruby/object:Gem::Specification
+name: legion-cli
+version: !ruby/object:Gem::Version
+  version: 0.2.0
+platform: ruby
+authors:
+- Esity
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2020-11-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: legionio
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Legion::CLI
+email:
+- matthewdiverson@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".circleci/config.yml"
+- ".gitignore"
+- ".rspec"
+- ".rubocop.yml"
+- Gemfile
+- README.md
+- Rakefile
+- bitbucket-pipelines.yml
+- exe/legionio
+- exe/lex_gen
+- legion-cli.gemspec
+- lib/legion/cli.rb
+- lib/legion/cli/chain.rb
+- lib/legion/cli/cohort.rb
+- lib/legion/cli/function.rb
+- lib/legion/cli/lex/actor.rb
+- lib/legion/cli/lex/exchange.rb
+- lib/legion/cli/lex/message.rb
+- lib/legion/cli/lex/queue.rb
+- lib/legion/cli/lex/runner.rb
+- lib/legion/cli/lex/templates/actor.erb
+- lib/legion/cli/lex/templates/actor_spec.erb
+- lib/legion/cli/lex/templates/base/bitbucket.yml.erb
+- lib/legion/cli/lex/templates/base/gemfile.erb
+- lib/legion/cli/lex/templates/base/gemspec.erb
+- lib/legion/cli/lex/templates/base/gitignore.erb
+- lib/legion/cli/lex/templates/base/lex.erb
+- lib/legion/cli/lex/templates/base/lex_spec.erb
+- lib/legion/cli/lex/templates/base/lic.erb
+- lib/legion/cli/lex/templates/base/rakefile.erb
+- lib/legion/cli/lex/templates/base/readme.md.erb
+- lib/legion/cli/lex/templates/base/rubocop.yml.erb
+- lib/legion/cli/lex/templates/base/spec_helper.rb.erb
+- lib/legion/cli/lex/templates/base/version.erb
+- lib/legion/cli/lex/templates/exchange.erb
+- lib/legion/cli/lex/templates/exchange_spec.erb
+- lib/legion/cli/lex/templates/message.erb
+- lib/legion/cli/lex/templates/message_spec.erb
+- lib/legion/cli/lex/templates/queue.erb
+- lib/legion/cli/lex/templates/queue_spec.erb
+- lib/legion/cli/lex/templates/runner.erb
+- lib/legion/cli/lex/templates/runner_spec.erb
+- lib/legion/cli/relationship.rb
+- lib/legion/cli/task.rb
+- lib/legion/cli/trigger.rb
+- lib/legion/cli/version.rb
+- lib/legion/lex.rb
+homepage: https://bitbucket.org/legion-io/legion-cli
+licenses: []
+metadata:
+  homepage_uri: https://bitbucket.org/legion-io/legion-cli
+  source_code_uri: https://bitbucket.org/legion-io/legion-cli
+  changelog_uri: https://bitbucket.org/legion-io/legion-cli
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.5.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.2
+signing_key:
+specification_version: 4
+summary: Legion::CLI
+test_files: []