smdev 0.3.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3cb7ef8e326aefb0cec817caef086d1079b26404c1cc6d04ebc91c8db5ccf7b5
-  data.tar.gz: 12b8cc2b5899723602ae6347c31f5cffb9ecf6026ebc39f595d7bc670abe0346
+  metadata.gz: 61964b5c32328c5e2265a32622cb85c366a70780eeaab4b95e154577942602b9
+  data.tar.gz: 39c56ddd43405021c465494bf439e174681e431146e59a808937f0c6592a2f97
 SHA512:
-  metadata.gz: b8e0db249fa4c0f91eee94c4ba0408ee6ea7517a58d6c0b8e91b12f700786ee0a8e504b0fe05bf25b4ba76f7a03b4f5297d755648ae429ee071cc73c221967b7
-  data.tar.gz: 0d31f663ac690004e11f3a3a05c45f995dc4f7c9cf51e5f193b08e73398dddfc36f5718c4602f3a053ed320a0c4892a82bae2be1a0434edc4e43c7cd68c4b105
+  metadata.gz: 54d844cb8292bea2cf364c473809f385bba2375d4bedede0927192c620e33bc685289831b1f7973a17a19d7b8471acfda9cf46de845367b7f2d2ab15ee3994e5
+  data.tar.gz: 123be6e3958715b82e1c0ae92c25e9a63f580ad0cbf55f43a2369ee1dbea782fe6d3b7dbb71e551ce4b66ef767ec817097ea0a8d6cd39ade8d3818c14482d13d

data/lib/smdev/ecs_exec.rb

@@ -0,0 +1,92 @@
+require 'open3'
+
+module Smdev
+  module EcsExec
+    def self.ssh(options)
+      cluster_name = self.cluster_name(options)
+      return if cluster_name == nil
+      puts "Found cluster #{cluster_name}"
+
+      service_name = self.service_name(options, cluster_name)
+      return if service_name == nil
+      puts "Found service #{service_name}"
+
+      task_arn = self.task_arn(cluster_name, service_name)
+      puts "Found first task #{task_arn}"
+      command = options[:command]
+
+      execute_command_cmd = %W[
+          aws ecs execute-command
+          --cluster #{cluster_name}
+          --task #{task_arn}
+          --interactive
+          --command \"#{command}\"
+        ].join(' ')
+
+      puts "Running \"#{command}\""
+      exec(execute_command_cmd)
+    end
+
+    def self.cluster_name(options)
+      cluster = options[:cluster]
+      command = "aws ecs list-clusters | grep #{cluster}"
+      stdout, stderr, status = Open3.capture3(command)
+
+      if status.success?
+        match = stdout.match(/cluster\/([^"]+)/)
+        cluster_name = match[1] if match
+
+        return cluster_name
+      else
+        if stderr != ""
+          puts "Error executing command: #{stderr}"
+        else
+          puts "Could not find cluster whose name contains \"#{cluster}\""
+        end
+      end
+    end
+
+    def self.service_name(options, cluster_name)
+      service = options[:service]
+      command = "aws ecs list-services --cluster #{cluster_name} | grep #{service}"
+      stdout, stderr, status = Open3.capture3(command)
+
+      if status.success?
+        match = stdout.match(/#{cluster_name}\/([^"]+)/)
+        service_name = match[1] if match
+
+        return service_name
+      else
+        if stderr != ""
+          puts "Error executing command: #{stderr}"
+        else
+          puts "Could not find service whose name contains \"#{service}\""
+        end
+      end
+    end
+
+    def self.task_arn(cluster_name, service_name)
+      list_tasks_cmd = "aws ecs list-tasks --cluster #{cluster_name} --service-name #{service_name}"
+      stdout, stderr, status = Open3.capture3(list_tasks_cmd)
+
+      if status.success?
+        begin
+          tasks = JSON.parse(stdout)
+          task_arn = tasks['taskArns'][0]
+
+          if task_arn.nil? || task_arn.empty?
+            puts 'No task ARN found.'
+            exit 1
+          end
+          return task_arn
+        rescue JSON::ParserError
+          puts 'Error parsing JSON'
+          exit 1
+        end
+      else
+        puts "Error executing command: #{stderr}"
+        exit 1
+      end
+    end
+  end
+end

data/lib/smdev.rb

@@ -2,6 +2,7 @@
 require 'io/console'
 require 'octokit'
 require 'optparse'
+require 'smdev/ecs_exec'
 
 module Smdev
   class CLI
@@ -9,7 +10,10 @@ module Smdev
       options = {}
       commands = ["local_app_setup : Install local requirements for application. (Rails Only Currently)\n",
                   "system_install : Install Homebrew, Ruby, PostgreSQL, and Ruby on Rails\n",
-                  "checkout_repos : Checkout all repositories for StrongMind\n"]
+                  "checkout_repos : Checkout all repositories for StrongMind\n",
+                  "console : Open a rails console on ECS\n",
+                  "ssh : Open a bash shell on ECS\n"
+      ]
 
       OptionParser.new do |opts|
         opts.banner = "Usage: smdev.rb [options] <command>"
@@ -22,133 +26,26 @@ module Smdev
           options[:https] = true
         end
 
-        opts.on("-h", "--help", "Prints this help message") do
-          puts opts
-          puts "\nCommands:"
-          commands.each { |cmd| puts "  #{cmd}" }
-          exit
+        opts.on("-r", "--cluster CLUSTER", "Lookup this cluster for the ECS execute (defaults to the name of your current folder)") do |cluster|
+          options[:cluster] = cluster
         end
 
-      end.parse!
-
-      def system_install
-        puts "Installing Homebrew..."
-        system({ 'SHELL' => '/bin/bash' }, '/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"')
-        # system('/bin/bash', '-c', "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)")
-
-        # Add Homebrew to PATH and run initialization script
-        puts "adding homebrew to .zshrc"
-        File.open("#{ENV['HOME']}/.zshrc", "a") do |file|
-          file.write("\n")
-          file.write('eval "$(/opt/homebrew/bin/brew shellenv)"')
-        end
-
-        # Update PATH to include Homebrew
-        ENV['PATH'] = "#{ENV['HOME']}/.homebrew/bin:#{ENV['HOME']}/.homebrew/sbin:#{ENV['PATH']}"
-
-        # Install PostgreSQL
-        puts "Installing PostgreSQL..."
-        system('brew', 'install', 'libpq')
-
-        puts "Installing lib-pq..."
-        system('brew', 'install', 'postgresql@15')
-
-        # Install Code Climate
-        puts "Preparing Code Climate Formulae..."
-        system('brew', 'tap', 'codeclimate/formulae')
-
-        puts "Installing Code Climate..."
-        system('brew', 'install', 'codeclimate')
-
-        # TODO: need to add the path & compiler flags
-        # echo 'export PATH="/opt/homebrew/opt/libpq/bin:$PATH"' >> ~/.zshrc
-        # export LDFLAGS="-L/opt/homebrew/opt/libpq/lib"
-        # export CPPFLAGS="-I/opt/homebrew/opt/libpq/include"
-
-        # Install RBenv
-        puts "Installing RBenv..."
-        system('brew', 'install', 'rbenv')
-        puts "Initialize RBenv..."
-        File.open("#{ENV['HOME']}/.zshrc", "a") do |file|
-          file.write("\n")
-          file.write('eval "$(rbenv init - zsh)"')
+        opts.on("-v", "--service SERVICE", "Lookup this service for the ECS execute (defaults to \"prod-worker\")") do |svc|
+          options[:service] = svc
         end
-        # Load the contents of .zshrc into a string
-        zshrc_contents = File.read(File.expand_path('~/.zshrc'))
-
-        # Execute a new shell session with the .zshrc contents
-        exec('zsh', '-c', zshrc_contents)
 
-        # Install Ruby
-        puts "Installing Ruby via RBenv..."
-        system('rbenv', 'install', '3.2.2')
-        puts "Making ruby 3.2.2 global ..."
-        system('rbenv', 'global', '3.2.2')
-
-        # Install Ruby on Rails
-        puts "Installing Ruby on Rails..."
-        system('sudo', 'gem', 'install', 'rails')
-      end
-      def checkout_repos(options)
-        token = ENV['GITHUB_TOKEN']
-        if token.nil?
-          # Prompt the user for their GitHub personal access token
-          print "GitHub personal access token: "
-          token = STDIN.noecho(&:gets).chomp
-          puts ""
-        end
-        # Authenticate with the GitHub API using the user's credentials
-        client = Octokit::Client.new(access_token: token)
-
-        # Specify the organization name
-        org_name = "StrongMind"
-
-        if options[:team]
-          # A team name was provided as a command-line argument
-          team_name = options[:team]
-          # Get a list of all teams in the organization
-          teams = client.organization_teams(org_name)
-          # Find the team with the specified name
-          team = teams.find { |t| t.name == team_name }
-          if team.nil?
-            puts "Error: Could not find team with name '#{team_name}' in organization '#{org_name}'"
-            exit
-          end
-          # Get a list of all repositories in the organization that are assigned to the specified team
-          #
-          repos = client.org_repos(org_name, { :affiliation => 'organization_member', :team_id => team.id })
-          puts "Team Name #{team_name} - Team ID: #{team.id}"
-        else
-          repos = []
-          page_number = 1
-          per_page = 100
-          loop do
-            repos_page = client.organization_repositories(org_name, { :affiliation => 'owner,organization_member', :per_page => per_page, :page => page_number })
-            repos.concat(repos_page)
-            break if repos_page.length < per_page
-            page_number += 1
-          end
-
-          # # No team name was provided, so get a list of all teams in the organization
-          # teams = client.organization_teams(org_name)
-          # # Get a list of all repositories in the organization that are assigned to each team
-          # repos = []
-          # teams.each do |team|
-          #   team_repos = client.team_repos(team.id)
-          #   repos.concat(team_repos)
-          # end
+        opts.on("-c", "--command COMMAND", "Set the command for the ECS execute (defaults to a bash shell when you use 'ssh' and rails console when you use 'console')") do |ecs_command|
+          options[:command] = ecs_command
         end
 
-        # Clone each repository to a local directory
-        repos.each do |repo|
-          clone_url = repo.clone_url
-          ssh_url = repo.ssh_url
-          repo_name = repo.name
-          url = options[:https] ? clone_url : ssh_url
-          `git clone #{url} #{repo_name}`
+        opts.on("-h", "--help", "Prints this help message") do
+          puts opts
+          puts "\nCommands:"
+          commands.each { |cmd| puts "  #{cmd}" }
+          exit
         end
 
-      end
+      end.parse!
 
       command = args.shift
 
@@ -169,10 +66,150 @@ module Smdev
         system_install
       when "checkout_repos"
         checkout_repos(options)
+      when "console"
+        options[:command] = "rails c"
+        open_ssh(options)
+      when "ssh"
+        options[:command] ||= "/bin/sh"
+        open_ssh(options)
       else
         puts "Invalid command: #{command}.  Use --help for a list of commands."
       end
 
     end
+
+    def system_install
+      puts "Installing Homebrew..."
+      system({ 'SHELL' => '/bin/bash' }, '/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"')
+      # system('/bin/bash', '-c', "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)")
+
+      # Add Homebrew to PATH and run initialization script
+      puts "adding homebrew to .zshrc"
+      File.open("#{ENV['HOME']}/.zshrc", "a") do |file|
+        file.write("\n")
+        file.write('eval "$(/opt/homebrew/bin/brew shellenv)"')
+      end
+
+      # Update PATH to include Homebrew
+      ENV['PATH'] = "#{ENV['HOME']}/.homebrew/bin:#{ENV['HOME']}/.homebrew/sbin:#{ENV['PATH']}"
+
+      # Install PostgreSQL
+      puts "Installing PostgreSQL..."
+      system('brew', 'install', 'libpq')
+
+      puts "Installing lib-pq..."
+      system('brew', 'install', 'postgresql@15')
+
+      # Install Code Climate
+      puts "Preparing Code Climate Formulae..."
+      system('brew', 'tap', 'codeclimate/formulae')
+
+      puts "Installing Code Climate..."
+      system('brew', 'install', 'codeclimate')
+
+      puts "Installing AWS CLI..."
+      system('brew', 'install', 'awscli')
+
+      puts "Installing AWS Session Manager Plugin..."
+      system('brew', 'install', '--cask', 'session-manager-plugin')
+
+      # TODO: need to add the path & compiler flags
+      # echo 'export PATH="/opt/homebrew/opt/libpq/bin:$PATH"' >> ~/.zshrc
+      # export LDFLAGS="-L/opt/homebrew/opt/libpq/lib"
+      # export CPPFLAGS="-I/opt/homebrew/opt/libpq/include"
+
+      # Install RBenv
+      puts "Installing RBenv..."
+      system('brew', 'install', 'rbenv')
+      puts "Initialize RBenv..."
+      File.open("#{ENV['HOME']}/.zshrc", "a") do |file|
+        file.write("\n")
+        file.write('eval "$(rbenv init - zsh)"')
+      end
+      # Load the contents of .zshrc into a string
+      zshrc_contents = File.read(File.expand_path('~/.zshrc'))
+
+      # Execute a new shell session with the .zshrc contents
+      exec('zsh', '-c', zshrc_contents)
+
+      # Install Ruby
+      puts "Installing Ruby via RBenv..."
+      system('rbenv', 'install', '3.2.2')
+      puts "Making ruby 3.2.2 global ..."
+      system('rbenv', 'global', '3.2.2')
+
+      # Install Ruby on Rails
+      puts "Installing Ruby on Rails..."
+      system('sudo', 'gem', 'install', 'rails')
+    end
+
+    def checkout_repos(options)
+      token = ENV['GITHUB_TOKEN']
+      if token.nil?
+        # Prompt the user for their GitHub personal access token
+        print "GitHub personal access token: "
+        token = STDIN.noecho(&:gets).chomp
+        puts ""
+      end
+      # Authenticate with the GitHub API using the user's credentials
+      client = Octokit::Client.new(access_token: token)
+
+      # Specify the organization name
+      org_name = "StrongMind"
+
+      if options[:team]
+        # A team name was provided as a command-line argument
+        team_name = options[:team]
+        # Get a list of all teams in the organization
+        teams = client.organization_teams(org_name)
+        # Find the team with the specified name
+        team = teams.find { |t| t.name == team_name }
+        if team.nil?
+          puts "Error: Could not find team with name '#{team_name}' in organization '#{org_name}'"
+          exit
+        end
+        # Get a list of all repositories in the organization that are assigned to the specified team
+        #
+        repos = client.org_repos(org_name, { :affiliation => 'organization_member', :team_id => team.id })
+        puts "Team Name #{team_name} - Team ID: #{team.id}"
+      else
+        repos = []
+        page_number = 1
+        per_page = 100
+        loop do
+          repos_page = client.organization_repositories(org_name, { :affiliation => 'owner,organization_member', :per_page => per_page, :page => page_number })
+          repos.concat(repos_page)
+          break if repos_page.length < per_page
+          page_number += 1
+        end
+
+        # # No team name was provided, so get a list of all teams in the organization
+        # teams = client.organization_teams(org_name)
+        # # Get a list of all repositories in the organization that are assigned to each team
+        # repos = []
+        # teams.each do |team|
+        #   team_repos = client.team_repos(team.id)
+        #   repos.concat(team_repos)
+        # end
+      end
+
+      # Clone each repository to a local directory
+      repos.each do |repo|
+        clone_url = repo.clone_url
+        ssh_url = repo.ssh_url
+        repo_name = repo.name
+        url = options[:https] ? clone_url : ssh_url
+        `git clone #{url} #{repo_name}`
+      end
+
+    end
+
+    private
+
+    def open_ssh(options)
+      options[:service] ||= "prod-worker"
+      options[:cluster] ||= Dir.pwd.split('/').last
+      Smdev::EcsExec.ssh(options)
+    end
   end
-end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: smdev
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Derek Neighbors
-autorequire:
+autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-09-11 00:00:00.000000000 Z
+date: 2023-10-12 00:00:00.000000000 Z
 dependencies: []
 description: StrongMind Development Tool
 email: derek.neighbors@strongmind.com
@@ -19,11 +19,12 @@ extra_rdoc_files: []
 files:
 - bin/smdev
 - lib/smdev.rb
+- lib/smdev/ecs_exec.rb
 homepage: https://github.com/StrongMind/Helpers/tree/main/smdev
 licenses:
 - MIT
 metadata: {}
-post_install_message:
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -38,8 +39,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.13
-signing_key:
+rubygems_version: 3.3.5
+signing_key: 
 specification_version: 4
 summary: StrongMind Development Tool
 test_files: []