feat: daily check

Author: robot

.github/workflows/schedule.yaml

@@ -15,7 +15,7 @@ jobs:
         with:
           node-version: "14"
       - run: npm install
-      - name: daily problem
+      - name: daily problem and daily check
         env:
           issueToken: ${{secrets.ISSUETOKEN}}
         run: npm run daily-schedule

database.js

@@ -0,0 +1,44 @@
+const MongoClient = require("mongodb").MongoClient;
+const { getDay } = require("./utils/day");
+
+const uri = process.env.collectionString;
+let collection = null;
+const client = new MongoClient(uri, {
+  useNewUrlParser: true,
+  useUnifiedTopology: true,
+});
+client.connect((err, data) => {
+  collection = client.db("91algo").collection("my-solution");
+});
+
+const findSolution = function (name) {
+  return new Promise((resolve, reject) => {
+    collection.find({ name }, function (err, result) {
+      if (err) reject(err);
+      resolve(result);
+    });
+  });
+};
+
+const dailyCheck = async function (name, data, callback) {
+  try {
+    const solution = await findSolution();
+    console.log(solution);
+    solution[getDay() - 1] = data;
+    collection.updateOne({ name }, { $set: solution });
+    return data;
+  } catch (_) {
+    const solution = Array(91);
+    solution[getDay() - 1] = data;
+    return new Promise((resolve, reject) => {
+      collection.insert({ name: solution }, function (err, result) {
+        if (err) reject(err);
+        resolve(result);
+      });
+    });
+  }
+};
+
+module.exports = {
+  dailyCheck,
+};

middleware/mockUserInfo.js

@@ -4,7 +4,8 @@ module.exports = ({ whitelist = [] }) =>
       ctx.session = {};
     }
     ctx.session.user = {
-      login: "lilyzhaoyilu",
+      // login: "lilyzhaoyilu",
+      login: "azl397985856",
       avatar_url: "https://avatars.githubusercontent.com/u/12479470?v=4",
       name: "lucifer",
       pay: true,

package.json

@@ -7,7 +7,7 @@
     "dev": "NODE_ENV=development ./node_modules/.bin/nodemon bin/www",
     "prd": "pm2 start bin/www",
     "test": "echo \"Error: no test specified\" && exit 1",
-    "daily-schedule": "node ./schedule/daily-problem.js",
+    "daily-schedule": "node ./schedule/daily-problem.js && node ./schedule/daily-check.js",
     "generate": "node ./static/users/generate.js && node ./static/solution/generate.js && node ./static/lectures/generate.js"
   },
   "dependencies": {
@@ -23,6 +23,7 @@
     "koa-router": "^7.4.0",
     "koa-static": "^5.0.0",
     "koa-views": "^6.2.0",
+    "mongodb": "^3.6.6",
     "node-fetch": "^2.6.1",
     "pug": "^2.0.3"
   },

routes/github.js

@@ -1,13 +1,10 @@
 const router = require("koa-router")();
 const fetch = require("node-fetch");
-const fs = require("fs");
-const path = require("path");
-const process = require("child_process");
 
 const { getDay } = require("../utils/day");
 const { success, fail } = require("../utils/request");
-const mySolutions = require("../static/my/solutions.json");
 const solutions = require("../static/solution/solutions.json");
+const mySolutions = require("../database");
 
 router.all("/api/v1/github/content", async (ctx) => {
   const { url, ...params } = ctx.query;
@@ -38,24 +35,8 @@ router.all("/api/v1/github/webhook", async (ctx) => {
     return;
   }
   if (action === "created" && comment.body.length > 20) {
-    mySolutions[comment.user.login][getDay() - 1] = {
-      // title: problem.title,
-      url: comment.html_url,
-      body: comment.body,
-    };
-
-    fs.writeFileSync(
-      path.resolve(__dirname, "../static/my/solutions.json"),
-      JSON.stringify(mySolutions)
-    );
-
-    process.exec(
-      "sh " + path.resolve(__dirname, "../scripts/commit.sh"),
-      console.log
-    );
+    ctx.body = success("收到！");
   }
-
-  ctx.body = success(mySolutions[comment.user.login]);
 });
 
 module.exports = router;