kybus-cli 0.1.0

data/lib/kybus/cli/bot/deployers/telegram_configurator.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require_relative 'deployer_base'
+
+module Kybus
+  class CLI < Thor
+    class BotDeployerTelegramConfigurator < BotDeployerBase
+      attr_reader :url
+      attr_writer :url
+
+      def initialize(url, config)
+        @url = url
+        super(config)
+      end
+
+      def create_or_update!
+        raise 'Missing Token' if @config['secret_token'].nil?
+
+        uri = URI("https://api.telegram.org/bot#{@config['bot_token']}/setWebhook")
+        params = { url: @url, secret_token: @config['secret_token'] }
+        res = Net::HTTP.post_form(uri, params)
+        puts res.body
+      end
+
+      def destroy!
+        uri = URI("https://api.telegram.org/bot#{@config['bot_token']}/setWebhook")
+        params = { url: '', secret_token: '' }
+        res = Net::HTTP.post_form(uri, params)
+        puts res.body
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_provider.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      class FileProvider
+        def initialize(name, config)
+          @file_writer = Kybus::CLI::FileWriter.new(name)
+          @config = config
+          @name = name
+        end
+
+        def skip_file?
+          false
+        end
+
+        def keep_files
+          []
+        end
+
+        def generate
+          @file_writer.write(saving_path, make_contents) unless skip_file?
+          keep_files.each do |file|
+            @file_writer.write(file, '')
+          end
+        end
+
+        def self.autoregister!
+          Kybus::CLI::Bot::ProjectGenerator.register_file_provider(self)
+        end
+
+        def bot_name
+          @name
+        end
+
+        def bot_name_class
+          @name.split('_').map(&:capitalize).join
+        end
+
+        def bot_name_constantize
+          bot_name_snake_case.upcase
+        end
+
+        def bot_name_snake_case
+          @name.gsub(' ', '_').downcase
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/autoconfig_generator.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      module Config
+        class AutoconfigGenerator < FileProvider
+          autoregister!
+
+          def saving_path
+            'config/autoconfig.yaml'
+          end
+
+          def make_contents
+            <<~YAML
+              autoconfig:
+                env_prefix: #{bot_name_constantize}
+                default_files:
+                  - ./config/config.default.yaml
+                files:
+                  - ./config/config.yaml
+            YAML
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/autoconfig_loader_generator.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      module Config
+        class AutoconfigLoaderGenerator < FileProvider
+          autoregister!
+
+          def saving_path
+            'config_loaders/autoconfig.rb'
+          end
+
+          private
+
+          def make_contents
+            <<~RUBY
+              require 'kybus/bot'
+              require 'kybus/configs'
+
+
+              Dir[File.join(__dir__, './models', '*.rb')].each { |file| require file }
+
+              require_relative '../bot'
+
+              CONF_MANAGER = Kybus::Configuration.auto_load!
+              APP_CONF = CONF_MANAGER.configs
+              require_relative 'db'
+
+              require_relative "bot_builder"
+            RUBY
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/bot_builder_generator.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      module Config
+        class BotBuilderGenerator < FileProvider
+          autoregister!
+
+          def saving_path
+            'config_loaders/bot_builder.rb'
+          end
+
+          def make_contents
+            <<~RUBY
+              # frozen_string_literal: true
+
+              BOT = #{bot_name_class}.new(APP_CONF['bots']['main'])
+            RUBY
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/bot_generator.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      class BotGenerator < FileProvider
+        autoregister!
+        def saving_path
+          'bot.rb'
+        end
+
+        def make_contents
+          <<~RUBY
+            # frozen_string_literal: true
+
+            class #{bot_name_class} < Kybus::Bot::Base
+              def initialize(configs)
+                super(configs)
+                register_command('/hello') do
+                  send_message('Hi human')
+                end
+              end
+            end
+          RUBY
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/composefile_generator.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      class ComposefileGenerator < FileProvider
+        autoregister!
+
+        def skip_file?
+          !@config[:with_docker_compose]
+        end
+
+        def saving_path
+          'docker-compose.yml'
+        end
+
+        def make_contents
+          content = <<~DOCKERCOMPOSE
+            version: '3'
+            services:
+              app:
+                build: .
+                volumes:
+                  - .:/app
+          DOCKERCOMPOSE
+
+          if @config[:db_adapter] == 'dynamoid'
+            content << <<-LOCALSTACK
+  localstack:
+    image: localstack/localstack
+    ports:
+      - "4566:4566"
+    environment:
+      - SERVICES=dynamodb
+            LOCALSTACK
+          elsif @config[:db_adapter] == 'sequel'
+            content << <<-DATABASE
+  db:
+    image: postgres
+    ports:
+      - "5432:5432"
+    environment:
+      POSTGRES_DB: app_development
+      POSTGRES_USER: user
+      POSTGRES_PASSWORD: password
+            DATABASE
+          end
+
+          content
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/config_default_generator.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      module Config
+        class ConfigDefaultGenerator < FileProvider
+          autoregister!
+          def saving_path
+            'config/config.default.yaml'
+          end
+
+          def keep_files
+            ['models/migrations/.keep']
+          end
+
+          def make_contents
+            content = <<~YAML
+              logger:
+                stdout: yes
+                severity: debug
+              bots:
+                main:
+                  pool_size: 1
+                  inline_args: true
+                  provider:#{' '}
+                    name: REPLACE_ME
+                    token: REPLACE_ME
+                    debug: true
+                  state_repository:
+            YAML
+
+            if @config[:db_adapter] == 'sequel'
+              content << <<~SEQUEL
+                        name: sequel
+                        endpoint: 'sqlite://#{bot_name_snake_case}_botmeta.db'
+                database: 'sqlite://#{bot_name_snake_case}.db'
+              SEQUEL
+            elsif @config[:db_adapter] == 'dynamoid'
+              content << <<-DYNAMOID
+        name: json
+        storage: ./storage
+              DYNAMOID
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/config_generator.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      module Config
+        class ConfigGenerator < FileProvider
+          autoregister!
+          def saving_path
+            'config/config.yaml'
+          end
+
+          def make_contents
+            content = <<~YAML
+              bots:
+                main:
+                  provider:
+                    name: #{@config[:bot_provider]}
+                    token: #{@config[:bot_token]}
+                    mode: #{@config[:with_deployment_file] && @config[:cloud_provider] == 'aws' ? 'webhook_lambda' : 'polling'}
+                    debug: true
+                  state_repository:
+            YAML
+
+            if @config[:db_adapter] == 'sequel'
+              content << <<-SEQUEL
+        name: sequel
+        endpoint: 'sqlite://#{bot_name_snake_case}_botmeta.db'
+              SEQUEL
+            elsif @config[:db_adapter] == 'dynamoid'
+              content << <<-DYNAMOID
+        name: dynamoid
+        dynamoid_config: true
+        region:  'us-east-1'
+        namespace: '#{bot_name_snake_case}'
+        read_capacity: 3
+        write_capacity: 3
+              DYNAMOID
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/db_generator.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      module Config
+        class DBGenerator < FileProvider
+          autoregister!
+
+          def saving_path
+            'config_loaders/db.rb'
+          end
+
+          def make_contents
+            case @config[:db_adapter]
+            when 'sequel'
+              <<~RUBY
+                # frozen_string_literal: true
+
+                require 'sequel'
+
+                DB = Sequel.connect(APP_CONF['database'])
+
+                def run_migrations!
+                  require 'kybus/bot/migrator'
+                  require 'sequel/core'
+                  Kybus::Bot::Migrator.run_migrations!(APP_CONF['bots']['main']['state_repository'])
+                  Sequel.extension :migration
+                  Sequel::Migrator.run(DB, 'models/migrations')
+                end
+              RUBY
+            when 'activerecord'
+              <<~RUBY
+                # frozen_string_literal: true
+
+                require 'active_record'
+
+                ActiveRecord::Base.establish_connection(APP_CONF['database'])
+              RUBY
+            when 'dynamoid'
+              <<~RUBY
+                def run_migrations!
+                  require 'kybus/bot/migrator'
+                  Kybus::Bot::Migrator.run_migrations!(APP_CONF['bots']['main']['state_repository'])
+                end
+              RUBY
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/deployment_file_provider.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require 'securerandom'
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      module Config
+        class DeploymentFileProvide < FileProvider
+          autoregister!
+          def saving_path
+            './kybusbot.yaml'
+          end
+
+          def skip_file?
+            !@config[:with_deployment_file]
+          end
+
+          def make_contents
+            <<~YAML
+              name: #{bot_name_snake_case}
+              cloud_provider: #{@config[:cloud_provider]}
+              dynamo:
+                capacity: #{@config[:dynamo_capacity]}
+                table_name: #{@config[:dynamo_table]}
+              chat_provider: #{@config[:bot_provider]}
+              bot_token: #{@config[:bot_token]}
+              secret_token: #{generate_secret_token}
+            YAML
+          end
+
+          def generate_secret_token
+            SecureRandom.alphanumeric(64)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/dockerfile_generator.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      class DockerfileGenerator
+        def initialize(name)
+          @name = name
+          @file_writer = Kybus::CLI::FileWriter.new(name)
+        end
+
+        def generate
+          @file_writer.write('Dockerfile', dockerfile_content)
+        end
+
+        private
+
+        def dockerfile_content
+          <<~DOCKERFILE
+            FROM ruby:#{RUBY_VERSION}
+
+            WORKDIR /app
+            COPY . /app
+
+            RUN bundle install
+
+            CMD ["ruby", "main.rb"]
+          DOCKERFILE
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/gemfile_generator.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      module Config
+        class GemfileGenerator < FileProvider
+          autoregister!
+          def saving_path
+            './Gemfile'
+          end
+
+          private
+
+          def make_contents
+            <<~GEMFILE
+              source 'https://rubygems.org'
+
+              gem 'kybus-bot'
+              gem '#{@config[:db_adapter]}'
+              gem 'minitest'
+              gem 'kybus-storage'
+              gem 'kybus-logger'
+              gem 'kybus-configs'
+              gem 'rake'
+              group :telegram do
+                gem 'telegram-bot-ruby'
+              end
+              group :discord do
+                gem 'discordrb'
+              end
+              group :development do
+                gem 'sqlite3'
+              end
+            GEMFILE
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/lambda_handler_generator.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      module Config
+        class LambdaHandlerGenerator < FileProvider
+          autoregister!
+          def saving_path
+            './handler.rb'
+          end
+
+          private
+
+          def make_contents
+            <<~AWSLAMBDA
+              require './main'
+
+              def lambda_handler(event:, context:)
+                secret_token = ENV['SECRET_TOKEN']
+                header_token = event.dig('headers', 'x-telegram-bot-api-secret-token')
+
+                unless header_token == secret_token
+                  return { statusCode: 403, body: JSON.generate('Forbidden') }
+                end
+
+                body = JSON.parse(event['body'])
+
+                BOT.handle_message(body)
+                { statusCode: 200, body: '' }
+              end
+            AWSLAMBDA
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/rakefile_generator.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      module Config
+        class RakefileGenerator < FileProvider
+          autoregister!
+          def saving_path
+            './Rakefile'
+          end
+
+          private
+
+          def make_contents
+            <<~RAKEFILE
+              require 'rake/testtask'
+              task default: :test
+
+              Rake::TestTask.new do |t|
+                t.libs << 'test'
+                t.warning = false
+                t.pattern = 'test/**/test_*.rb'
+                t.warning = false
+              end
+
+              namespace :db do
+                desc 'Run database migrations'
+                task :migrate do
+                  require_relative 'config_loaders/autoconfig'
+                  run_migrations!
+                end
+              end
+            RAKEFILE
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kybus/cli/bot/file_providers/test_helper_generator.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module Kybus
+  class CLI < Thor
+    class Bot < Thor
+      module Config
+        class TestHelperGenerator
+          def initialize(name, with_simplecov)
+            @file_writer = Kybus::CLI::FileWriter.new(name)
+            @with_simplecov = with_simplecov
+          end
+
+          def generate
+            @file_writer.write('test/test_helper.rb', test_helper_rb_content)
+          end
+
+          private
+
+          def test_helper_rb_content
+            content = <<~RUBY
+              # frozen_string_literal: true
+
+              require 'minitest/autorun'
+            RUBY
+            content << "\nrequire 'simplecov'\nSimpleCov.start" if @with_simplecov
+            content
+          end
+        end
+      end
+    end
+  end
+end