lscontests 1.0.1 → 1.3.0

package/README.md

@@ -13,9 +13,10 @@ A CLI tool to get information about the contests on each OJ
 lsct [options]
 
 Options:
-  -V, --version  output the version number
-  -a, --all      List all contests
-  -d, --days     Only list contests within a given number of days
+  -V, --version      output the version number
+  -d, --days, <day>  Number of days to get contests information (default: "3")
+  -l, --list         List all supported OJ
+  -o, --oj <ojs...>  OJs to get contests information (choices: "cf", "lg")
 ```
 
 ## License

package/bin/cli.js

@@ -8,44 +8,77 @@ function _interopDefaultLegacy (e) { return e && typeof e === 'object' && 'defau
 
 var axios__default = /*#__PURE__*/_interopDefaultLegacy(axios);
 
-var version = "1.0.1";
+var version = "1.3.0";
 
-async function get() {
+const ruleRecord$1 = {
+  "CF": "Codeforces",
+  "ICPC": "ICPC"
+};
+async function get$1() {
   const result = (await axios__default["default"].get("https://codeforces.com/api/contest.list")).data.result;
   return result.map(contest => {
     return {
       oj: "Codeforces",
       name: contest.name,
-      rule: contest.type,
+      rule: ruleRecord$1[contest.type],
       startTime: new Date(contest.startTimeSeconds * 1000),
-      durationHours: contest.durationSeconds / 60 / 60,
+      endTime: new Date((contest.startTimeSeconds + contest.durationSeconds) * 1000),
       url: `https://codeforces.com/contests/${contest.id}`
     };
   }).filter(contest => contest.startTime >= new Date());
 }
 
-const getters = {
-  "codeforces": get
+const ruleRecord = {
+  1: "OI",
+  2: "ICPC",
+  3: "LeDuo",
+  4: "IOI",
+  5: "Codeforces"
 };
+const headers = {
+  "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.4951.54 Safari/537.36",
+  "x-luogu-type": "content-only"
+};
+async function get() {
+  const result = (await axios__default["default"].get("https://www.luogu.com.cn/contest/list", {
+    headers
+  })).data.currentData.contests.result;
+  return result.map(contest => {
+    return {
+      oj: "Luogu",
+      name: contest.name,
+      rule: ruleRecord[contest.ruleType],
+      startTime: new Date(contest.startTime * 1000),
+      endTime: new Date(contest.endTime * 1000),
+      url: `https://www.luogu.com.cn/contest/${contest.id}`
+    };
+  }).filter(contest => contest.startTime >= new Date());
+}
 
-async function list(getters, days) {
-  const cts = [];
-
-  for (const get of getters) {
-    const res = await get();
-    cts.push(...res.filter(ct => ct.startTime <= new Date(Date.now() + days * 86400000)));
+const alloj = {
+  "cf": {
+    name: "Codeforces",
+    getter: get$1
+  },
+  "lg": {
+    name: "Luogu",
+    getter: get
   }
+};
+function getGetterList(ojs) {
+  if (!ojs) ojs = Object.keys(alloj);
+  return ojs.map(oj => alloj[oj].getter);
+}
 
-  return cts;
+async function list(ojs, days) {
+  return (await Promise.all(getGetterList(ojs).map(async get => (await get()).filter(ct => ct.startTime <= new Date(Date.now() + days * 86400000))))).reduce((ls1, ls2) => ls1.concat(ls2));
 }
 
-commander.program.name("lsct").version(version).option("-a, --all", "List all contests").option("-d, --days", "Only list contests within a given number of days", "3").parse();
+commander.program.name("lsct").version(version).option("-d, --days, <day>", "Number of days to get contests information", "3").option("-l, --list", "List all supported OJ").addOption(new commander.Option("-o, --oj <ojs...>", "OJs to get contests information").choices(Object.keys(alloj))).parse();
 const opts = commander.program.opts();
 
 async function main() {
-  const cts = [];
-  if (commander.program.args.length === 0 || opts.all) cts.push(...(await list(Object.values(getters), opts.days)));
-  console.log(cts);
+  if (opts.list) console.log(Object.values(alloj).map(ojd => ojd.name));else console.log(await list(opts.oj, opts.days));
 }
 
 main();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lscontests",
-  "version": "1.0.1",
+  "version": "1.3.0",
   "description": "A CLI tool to get information about the contests on each OJ",
   "keywords": [
     "cli",