init repo

Author: yjl9903

.gitignore

@@ -0,0 +1,2 @@
+node_modules/**
+dist/**

.prettierrc

@@ -0,0 +1,4 @@
+{
+  "semi": true,
+  "singleQuote": true
+}

.travis.yml

@@ -0,0 +1,13 @@
+language: node_js
+node_js: 10
+
+cache:
+  yarn: true
+  directories:
+    - node_modules
+
+install:
+  - yarn
+
+script:
+  - yarn run build

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "editor.tabSize": 2
+}

.vscode/tasks.json

@@ -0,0 +1,18 @@
+{
+// 有关 tasks.json 格式的文档，请参见
+  // https://go.microsoft.com/fwlink/?LinkId=733558
+  "version": "2.0.0",
+  "tasks": [
+    {
+      "type": "typescript",
+      "tsconfig": "tsconfig.json",
+      "problemMatcher": [
+        "$tsc"
+      ],
+      "group": {
+        "kind": "build",
+        "isDefault": true
+      }
+    }
+  ]
+}

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2019 XLor
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -0,0 +1,23 @@
+# Codeforces Analytics
+
+## Install
+
+```bash
+yarn
+```
+
+## Usage
+
+### Start Daemon
+
+```bash
+yarn start
+```
+
+### command line
+
+See cli help.
+
+```bash
+yarn cli --help
+```

package.json

@@ -0,0 +1,28 @@
+{
+  "name": "codeforces-analytics",
+  "version": "1.0.0",
+  "main": "index.js",
+  "author": "XLor",
+  "license": "MIT",
+  "scripts": {
+    "start": "yarn tsc && node dist/index.js",
+    "build": "yarn tsc",
+    "server": "node dist/index.js",
+    "cli": "node dist/cli/index.js",
+    "lint": "yarn prettier --write src/**/*.ts"
+  },
+  "dependencies": {
+    "axios": "^0.19.0",
+    "body-parser": "^1.19.0",
+    "express": "^4.17.1",
+    "yargs": "^14.2.0"
+  },
+  "devDependencies": {
+    "@types/express": "^4.17.1",
+    "@types/node": "^12.11.1",
+    "@types/yargs": "^13.0.3",
+    "cross-env": "^6.0.3",
+    "prettier": "^1.18.2",
+    "typescript": "^3.6.4"
+  }
+}

src/analytics/api.ts

@@ -0,0 +1,26 @@
+import axios from 'axios';
+
+const api = axios.create({
+  baseURL: 'https://codeforces.com/api/'
+});
+
+export default api;
+
+const userInfoCache = new Map<string, object>();
+
+export async function getUserInfo(cfid: string) {
+  if (userInfoCache.has(cfid)) {
+    return userInfoCache.get(cfid);
+  }
+  const {
+    data: {
+      result: [data]
+    }
+  } = await api.get(`user.info`, {
+    params: {
+      handles: cfid
+    }
+  });
+  userInfoCache.set(cfid, data);
+  return data;
+}

src/analytics/index.ts

@@ -0,0 +1,33 @@
+import { Router } from 'express';
+
+import { getUserInfo } from './api';
+import { User } from '../type';
+import { insert, query } from './store';
+
+const router = Router();
+
+router.get('/', (req, res) => {
+  if ('name' in req.query) {
+    try {
+      res.send(query(req.query.name));
+    } catch (err) {
+      res.status(404).send();
+    }
+  } else {
+    res.send(query());
+  }
+});
+
+router.post('/', async (req, res) => {
+  const { name, cfid } = req.body;
+  try {
+    const info = await getUserInfo(cfid);
+    const handle = new User(info);
+    insert(name, handle);
+    res.send(info);
+  } catch (err) {
+    res.status(500).send('');
+  }
+});
+
+export default router;

src/analytics/store.ts

@@ -0,0 +1,38 @@
+import { User } from '../type';
+import assert from 'assert';
+
+const store = new Map<string, User>();
+
+export function insert(name: string, handle: User) {
+  if (store.has(name)) {
+    const old: User = store.get(name) as User;
+    if (old.maxRating > handle.maxRating) {
+      handle.maxRating = old.maxRating;
+      handle.maxRank = old.maxRank;
+    }
+    if (old.rating > handle.rating) {
+      handle.rating = old.rating;
+      handle.rank = old.rank;
+    }
+    handle.cfids = [...new Set([...old.cfids, ...handle.cfids])];
+    store.set(name, handle);
+  } else {
+    store.set(name, handle);
+  }
+}
+
+export function query(): Array<User & { name: string }>;
+export function query(name: string): User;
+
+export function query(name?: string): any {
+  if (typeof name === 'undefined') {
+    const arr: Array<any> = [];
+    store.forEach((value: User, key: string) => {
+      arr.push({ name: key, ...value });
+    });
+    return arr;
+  } else if (typeof name === 'string') {
+    assert(store.has(name));
+    return store.get(name);
+  }
+}

src/app.ts

@@ -0,0 +1,17 @@
+import express from 'express';
+import bodyParser from 'body-parser';
+
+import { DEFAULT_HOST, DEFAULT_PORT } from './config';
+
+import anyRouter from './analytics';
+
+const app = express();
+
+app.set('host', process.env.HOST ? process.env.HOST : DEFAULT_HOST);
+app.set('port', process.env.PORT ? process.env.PORT : DEFAULT_PORT);
+
+app.use(bodyParser.json());
+
+app.use('/', anyRouter);
+
+export default app;

src/cli/api.ts

@@ -0,0 +1,47 @@
+import axios from 'axios';
+
+import { DAEMON_URL } from '../config';
+import { User } from '../type';
+
+const api = axios.create({
+  baseURL: DAEMON_URL,
+  timeout: 60 * 1000
+});
+
+export default api;
+
+export async function checkDaemon() {
+  try {
+    await api.get('');
+    return true;
+  } catch (err) {
+    return false;
+  }
+}
+
+export async function insert(name: string, cfid: string) {
+  console.log(`开始查询 "${cfid}" ...`);
+  try {
+    await api.post('', { name, cfid });
+    console.log(`${name}的 Codeforces ID "${cfid}" 插入成功`);
+    return true;
+  } catch (err) {
+    console.log(`插入失败`);
+    return false;
+  }
+}
+
+export async function query(): Promise<Array<User & { name: string }>>;
+export async function query(name: string): Promise<User>;
+
+export async function query(name?: string): Promise<any> {
+  if (typeof name === 'undefined') {
+    const { data } = await api.get('');
+    return data;
+  } else if (typeof name === 'string') {
+    const { data } = await api.get('', {
+      params: { name }
+    });
+    return data;
+  }
+}

src/cli/index.ts

@@ -0,0 +1,62 @@
+import { writeFile } from 'fs';
+
+import yargs from 'yargs';
+import { checkDaemon, insert, query } from './api';
+
+(async function() {
+  if (!(await checkDaemon())) {
+    console.log('未检测到守护进程的运行');
+    process.exit(1);
+  } else {
+    yargs
+      .command(
+        ['add <name> <cfid>', '$0'],
+        '添加一条 Codeforces ID',
+        yargs => {
+          return yargs.options({
+            name: {
+              type: 'string',
+              describe: '你的姓名',
+              demandOption: true
+            },
+            cfid: {
+              type: 'string',
+              describe: '你的 Codeforces ID',
+              demandOption: true
+            }
+          });
+        },
+        argv => {
+          insert(argv.name, argv.cfid);
+        }
+      )
+      .command(
+        'query [file]',
+        '查询所有用户信息 / 姓名为 name 的用户信息',
+        yargs => {
+          return yargs.options({
+            name: {
+              type: 'string',
+              describe: '查询对象姓名'
+            },
+            file: {
+              type: 'string',
+              describe: '输出文件名'
+            }
+          });
+        },
+        async argv => {
+          const res =
+            typeof argv.name === 'undefined'
+              ? await query()
+              : await query(argv.name as string);
+          if (typeof argv.file === 'undefined') {
+            console.log(res);
+          } else {
+            writeFile(argv.file, JSON.stringify(res), () => {});
+          }
+        }
+      )
+      .help().argv;
+  }
+})();

src/config.ts

@@ -0,0 +1,3 @@
+export const DEFAULT_HOST = 'localhost';
+export const DEFAULT_PORT = 2333;
+export const DAEMON_URL = `http://${DEFAULT_HOST}:${DEFAULT_PORT}`;

src/index.ts

@@ -0,0 +1,9 @@
+import app from './app';
+
+const server = app.listen(app.get('port'), app.get('host'), () => {
+  console.log(
+    `\n  Server starts on http://${app.get('host')}:${app.get('port')}\n`
+  );
+});
+
+export default server;