Better atcoder friend problemset

Show friend's accepted submissions.

您需要先安装一个扩展,例如 篡改猴Greasemonkey暴力猴,之后才能安装此脚本。

您需要先安装一个扩展,例如 篡改猴暴力猴,之后才能安装此脚本。

您需要先安装一个扩展,例如 篡改猴暴力猴,之后才能安装此脚本。

您需要先安装一个扩展,例如 篡改猴Userscripts ,之后才能安装此脚本。

您需要先安装一款用户脚本管理器扩展,例如 Tampermonkey,才能安装此脚本。

您需要先安装用户脚本管理器扩展后才能安装此脚本。

(我已经安装了用户脚本管理器,让我安装!)

您需要先安装一款用户样式管理器扩展,比如 Stylus,才能安装此样式。

您需要先安装一款用户样式管理器扩展,比如 Stylus,才能安装此样式。

您需要先安装一款用户样式管理器扩展,比如 Stylus,才能安装此样式。

您需要先安装一款用户样式管理器扩展后才能安装此样式。

您需要先安装一款用户样式管理器扩展后才能安装此样式。

您需要先安装一款用户样式管理器扩展后才能安装此样式。

(我已经安装了用户样式管理器,让我安装!)

// ==UserScript==
// @name         Better atcoder friend problemset
// @namespace    https://atcoder.jp
// @version      0.4
// @description  Show friend's accepted submissions.
// @author       Ftt2333
// @match        https://atcoder.jp/*
// @connect      kenkoooo.com
// @connect      atcoder.jp
// @icon         data:image/gif;base64,R0lGODlhAQABAAAAACH5BAEKAAEALAAAAAABAAEAAAICTAEAOw==
// @grant        GM_xmlhttpRequest
// @license      MIT
// ==/UserScript==

var userList
var contestId, problemId

let GetColorByRating = (rating) => {
  if (rating < 400) return 'user-gray';
  if (400 <= rating && rating < 800) return 'user-brown'
  if (800 <= rating && rating < 1200) return 'user-green'
  if (1200 <= rating && rating < 1600) return 'user-cyan'
  if (1600 <= rating && rating < 2000) return 'user-blue'
  if (2000 <= rating && rating < 2400) return 'user-yellow'
  if (2400 <= rating && rating < 2800) return 'user-orange'
  if (2800 <= rating) return 'user-red'
}

var result = [], base, node, counter

let Display = () => {
  console.log(result)
  let str1 = "Solved by " + result.length.toString() + " friend" + (result.length > 1 ? "s. " : ". ")
  let str2 = ""
  for (let i = 0; i < result.length; i++) {
    str2 += `<span class = "` + result[i].color + `">` + result[i].userId + "</span>"
    if (i + 1 != result.length) str2 += ", "
  }
  let str = `<br><span style = "font-size: 16px; color: grey;">` + str1 + str2 + (counter == 0 ? "(finished)" : "") + ".</span>"
  node.innerHTML = base + str
}

let GetUserColor = async (userId) => {
  let tmp = {}, cnt = 10
  let url = "https://atcoder.jp/users/" + userId
  while (cnt >= 0 && tmp.status != 200) tmp = await fetch(url), cnt--
  tmp.text().then((data) => {
    let tmp = /<tr><th class="no-break">Rating<\/th><td><span class='user-(\w*)'>(\d*)<\/span>/
    let arr = tmp.exec(data)
    if (arr != null) result.push({userId : userId, color : GetColorByRating(parseInt(arr[2]))})
    else result.push({userId : userId, color : "user-unrated"})
    Display()
  })
}

let CheckIfAvailable = async (userId) => {
  let tmp = {}, cnt = 10
  let url = "https://atcoder.jp/contests/" + contestId + "/submissions?f.Task=" + problemId + "&f.LanguageName=&f.Status=AC&f.User=" + userId
  while (cnt >= 0 && tmp.status != 200) tmp = await fetch(url), cnt--
  tmp.text().then((data) => {
    let tmp = /<span class='label label-success' data-toggle='tooltip' data-placement='top' title="Accepted">AC<\/span>/
    let arr = tmp.exec(data)
    if (arr != null) GetUserColor(userId)
    counter--
  })
}
let GetAvailableUsers = async () => {
  counter = userList.length
  Display()
  for (let i = 0; i < userList.length; i++) await CheckIfAvailable(userList[i])
}

let Init = () => {
  let tmp1 = /atcoder.jp\/contests\/([\w-]*)\/tasks\/([\w-]*)/
  let tmp2 = tmp1.exec(location.href)
  if (tmp2 == null) return
  contestId = tmp2[1], problemId = tmp2[2]
  userList = JSON.parse(localStorage.fav)
  node = document.querySelector(`#main-div > div.container > div.row > div:nth-child\(2\) > span`)
  base = node.innerHTML
  GetAvailableUsers()
}

Init()