Console - GitHub / Slack

Adding details about slack account

Signed-off-by: Rodrigo Nardi <rnardi@netdef.org>

Author: RodrigoMNardi

bin/organization_console bin/consolebin/organization_console renamed to bin/console

@@ -1,7 +1,7 @@
 #!/usr/bin/env ruby
 #  SPDX-License-Identifier: BSD-2-Clause
 #
-#  organization_console.rb
+#  console
 #  Part of NetDEF CI System
 #
 #  Copyright (c) 2023 by
@@ -10,9 +10,14 @@
 # frozen_string_literal: true
 
 require 'irb'
+
+ENV['RAILS_ENV'] = ARGV.shift || 'production'
+
+puts "Starting console: #{ENV.fetch('RAILS_ENV', nil)}"
+
 require_relative '../config/setup'
+require_relative '../config/delayed_job'
 
-# Defina métodos para buscar e criar empresas
 def find_organization(name)
   organization = Organization.find_by(name: name)
   if organization
@@ -97,7 +102,25 @@ def remove_user_from_organization(github_login)
   end
 end
 
-def help
+def add_github_user_slack_user(github_login, slack_user)
+  user = GithubUser.find_by(github_login: github_login)
+
+  if user.nil?
+    puts 'Github user not found'
+    return
+  end
+
+  user.update(slack_username: slack_user)
+  SlackUsername2Id.fetch_id(github_login, slack_user)
+
+  if user.persisted?
+    puts "Slack user linked to github user: #{user.inspect}"
+  else
+    puts "Failed to link slack user to github user: #{user.errors.full_messages.join(', ')}"
+  end
+end
+
+def help?
   puts <<~HELP
     Available commands:
     - find_organization(name)
@@ -106,6 +129,7 @@ def help
     - find_github_user(login)
     - add_user_in_organization(login, organization_name)
     - remove_user_from_organization(login)
+    - add_github_user_slack_user(github_login, slack_user)
 
     create_organization / edit_organization attributes:
     - contact_email: string
@@ -122,5 +146,4 @@ def help
   HELP
 end
 
-# Inicie o console IRB
 IRB.start

bin/console.rb

@@ -0,0 +1,16 @@
+#  SPDX-License-Identifier: BSD-2-Clause
+#
+#  20240617121935_create_delayed_jobs.rb
+#  Part of NetDEF CI System
+#
+#  Copyright (c) 2024 by
+#  Network Device Education Foundation, Inc. ("NetDEF")
+#
+#  frozen_string_literal: true
+
+class AddGithubUserSlackId < ActiveRecord::Migration[6.0]
+  def change
+    add_column :github_users, :slack_username, :string
+    add_column :github_users, :slack_id, :string
+  end
+end

db/migrate/20241014134659_add_github_user_slack_id.rb

@@ -10,7 +10,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema[7.2].define(version: 2024_10_09_102350) do
+ActiveRecord::Schema[7.2].define(version: 2024_10_14_134659) do
   # These are extensions that must be enabled in order to support this database
   enable_extension "plpgsql"
 
@@ -108,6 +108,8 @@
     t.datetime "created_at", null: false
     t.datetime "updated_at", null: false
     t.bigint "organization_id"
+    t.string "slack_username"
+    t.string "slack_id"
     t.index ["github_id"], name: "index_github_users_on_github_id", unique: true
     t.index ["organization_id"], name: "index_github_users_on_organization_id"
   end

db/schema.rb

@@ -42,6 +42,7 @@
 require_relative '../workers/ci_job_status'
 require_relative '../workers/timeout_execution'
 require_relative '../workers/ci_job_fetch_topotest_failures'
+require_relative '../workers/slack_username2_id'
 
 # Slack libs
 require_relative 'slack/slack'

lib/github_ci_app.rb

@@ -28,7 +28,7 @@ def download(uri, machine: 'ci1.netdef.org')
       http.request(req).body
     end
 
-    def get_request(uri, machine: 'ci1.netdef.org')
+    def get_request(uri, machine: 'ci1.netdef.org', json: true)
       user, passwd = fetch_user_pass(machine)
       http = create_http(uri)
 
@@ -40,7 +40,7 @@ def get_request(uri, machine: 'ci1.netdef.org')
       # Add JSON request header
       req.add_field 'Accept', 'application/json'
 
-      JSON.parse(http.request(req).body)
+      json ? JSON.parse(http.request(req).body) : http.request(req).body
     rescue StandardError => e
       logger(Logger::ERROR, "HTTP GET Request failed (#{e.message}) for #{uri.host}")
     end

lib/helpers/request.rb

@@ -19,6 +19,11 @@ def initialize
     @logger_manager << GithubLogger.instance.create('github_retry.log', Logger::INFO)
   end
 
+  def find_user_id_by_name(username)
+    url = "#{GitHubApp::Configuration.instance.config['slack_bot_url']}/github/translate/#{username}"
+    get_request(URI(url), json: false)
+  end
+
   def invalid_rerun_group(job)
     return unless current_execution?(job.check_suite)
 

lib/slack_bot/slack_bot.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+#  SPDX-License-Identifier: BSD-2-Clause
+#
+#  slack_username2_id.rb
+#  Part of NetDEF CI System
+#
+#  Copyright (c) 2024 by
+#  Network Device Education Foundation, Inc. ("NetDEF")
+#
+#  frozen_string_literal: true
+
+class SlackUsername2Id
+  class << self
+    def fetch_id(username, slack_name)
+      @logger = GithubLogger.instance.create('slack_username_to_id.log', Logger::INFO)
+
+      user = GithubUser.find_by(github_login: username)
+
+      return false unless user
+
+      id = SlackBot.instance.find_user_id_by_name(slack_name)
+
+      @logger.info("Username: '#{username}' -> '#{id}'")
+
+      user.slack_id = id
+      user.save
+
+      true
+    end
+  end
+end