From 9d0b0511c5b7dfc0544e79a4b23c6dc28777465c Mon Sep 17 00:00:00 2001
From: Dai MIKURUBE <dmikurube@acm.org>
Date: Tue, 10 Dec 2024 17:18:07 +0900
Subject: [PATCH] Select GitHub token

---
 .ruby-version |  2 +-
 audit.rb      | 22 +++++++++++++++++++++-
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/.ruby-version b/.ruby-version
index e4604e3..be94e6f 100644
--- a/.ruby-version
+++ b/.ruby-version
@@ -1 +1 @@
-3.2.1
+3.2.2
diff --git a/audit.rb b/audit.rb
index 964fb1d..e1bcd15 100644
--- a/audit.rb
+++ b/audit.rb
@@ -1,8 +1,11 @@
 # coding: utf-8
 require 'octokit'
 require 'minitest'
+require 'pathname'
 require 'yaml'
 
+SCRIPT_PATH = Pathname.new(__dir__)
+
 include Minitest::Assertions
 
 class << self
@@ -10,6 +13,23 @@ class << self
 end
 self.assertions = 0
 
+def retrieve_github_token()
+  token_path = SCRIPT_PATH.join('github_token')
+  if File.exist?(token_path)
+    File.open(token_path) do |f|
+      puts "Found GitHub token file: github_token"
+      return f.gets
+    end
+  end
+
+  if ENV.has_key?('GITHUB_TOKEN')
+    puts "Not found GitHub token file. Using the env: GITHUB_TOKEN"
+    return ENV.fetch('GITHUB_TOKEN')
+  end
+
+  raise StandardError, "Not found GitHub token token."
+end
+
 def assert_repos(expected_repos, actual_repos)
   expected_repo_names = expected_repos.map { |node| node[:name] }
   actual_repo_names = actual_repos.map { |node| node[:name] }
@@ -46,7 +66,7 @@ def load_actual_repos(access_token)
 end
 
 expected_repos = load_expected_repos("repos.yaml")
-actual_repos = load_actual_repos(ENV.fetch('GITHUB_TOKEN'))
+actual_repos = load_actual_repos(retrieve_github_token())
 
 puts actual_repos.inspect