update_readme.rb

# partial code from ChatGPT
require 'net/http'
require 'uri'
require 'json'


# update repos
def update_all_repos
  repo_stars = {} # cache stars
  update_repos('## Repos', '**Tip:**', 'README.md', repo_stars)
  update_repos('## 代码库', '**说明:**', 'README.zh-CN.md', repo_stars)
end

# update last date
def update_all_last_update
  last_update_time('# AI TreasureBox', '## Repos', 'README.md')
  last_update_time('# AI 百宝箱', '## 代码库', 'README.zh-CN.md')
end

# update repos content
def update_repos(start_str, end_str, file_name, repo_stars)
  readme = File.read(file_name)
  lines = readme.lines
  start_index = lines.index {|e| e.include?(start_str)}
  end_index = lines.index {|e| e.include?(end_str)}
  repos = []
  Array(lines[start_index...end_index]).each_with_index do |line, index|
    if index > 2 # skip head of table
        _, _, repo_info, desc  = line.split('|')
        next if repo_info.nil?
        repo_info.gsub!('🔥', '') # reset fire
        match = repo_info.scan(/\[(.*?)\]/).flatten
        next if match.empty?
        star_count = if repo_stars[match[0]].nil?
                       get_star_count(match[0])
                     else
                       repo_stars[match[0]]
                     end
        repo_stars[match[0]] = star_count
        change_stars = 0
        date, total_stars, change_stars = sync_today_stars(match[1], star_count)
        star_info = format("%s_%s_%s", date, total_stars, change_stars)
        repo_info.sub!(match[1], star_info)
        repo = { repo_info: repo_info, desc: desc, star_count:  star_count, change_stars: change_stars.to_i, original_index: index - 2 }
        repos << repo
    end
  end

  new_readme = ''
  new_readme << lines[0..(start_index + 2)].join
  repos.sort_by!{ |r| -r[:star_count] }
  repos.each_with_index do |repo, index|
    now_index = index + 1
    line = format("|%s %i|%s%s|%s|\n", 
      arrow_style(file_name, repo[:original_index], now_index), now_index, 
      popularity_style(repo[:change_stars], 200), repo[:repo_info], repo[:desc]
    )
    new_readme << line
  end
  new_readme << "\n"
  new_readme << lines[end_index..-1].join
  File.write(file_name, new_readme)
end

# last update time
def last_update_time(start_str, end_str, file_name)
  time = Time.now.strftime('%H:%M:%S%Z')
  readme = File.read(file_name)
  lines = readme.lines
  start_index = lines.index {|e| e.include?(start_str)}
  end_index = lines.index {|e| e.include?(end_str)}
  new_readme = ''
  new_readme << lines[0...start_index].join

  Array(lines[start_index...end_index]).each_with_index do |line, index|
    if line.include?('last update')
       prev_time = line.match(/update-(.*)-brightgreen/)
       new_readme << (prev_time.nil? ? line : line.sub(prev_time[1], time))
    else
      new_readme << line
    end
  end
  new_readme << lines[end_index..-1].join
  File.write(file_name, new_readme)
end

# cumulate arrow style
def arrow_style(file_name, original_index, now_index)
  return nil if now_index == original_index
  style = ' '
  if file_name == 'README.md'
    style = now_index < original_index ? '![](./images/green-up-arrow.svg)' : '![](./images/red-down-arrow.svg)'
  else
    style = now_index < original_index ? '![](./images/red-up-arrow.svg)' : '![](./images/green-down-arrow.svg)'
  end
  style
end

# code popularity
def popularity_style(change_stars, threshold)
  return ""       if change_stars < threshold
  return "🔥"     if change_stars > threshold && change_stars <= threshold * 2
  return "🔥🔥"   if change_stars > threshold * 2 && change_stars <= threshold * 5
  return "🔥🔥🔥" if change_stars > threshold * 5
end

# cumulate stars changes
def sync_today_stars(info, new_stars)
  today = Time.now.strftime('%Y-%m-%d')
  if info.nil? || !info.include?('_')
    [today, new_stars, 0]
  else
    date, total_stars, change_stars = info.split('_')
    if date != today
      change_stars = 0
    end
    change_stars = change_stars.to_i + (new_stars.to_i - total_stars.to_i)
    [today, new_stars, change_stars]
  end
end

# fetch star count from github api
def get_star_count(repo)
  uri = URI.parse("https://api.github.com/repos/#{repo}")
  http = Net::HTTP.new(uri.host, uri.port)
  http.use_ssl = true

  request = Net::HTTP::Get.new(uri.request_uri)
  request['Authorization'] = "Bearer #{ENV['GITHUB_TOKEN']}"
  request['Accept'] = 'application/vnd.github+json'
  request['User-Agent'] = 'Mozilla/5.0'

  response = http.request(request)
  if response.code == '200'
    result = JSON.parse(response.body)
    result['stargazers_count']
  else
    0
  end
end

# main
if __FILE__ == $0
  update_all_repos
  update_all_last_update
end