0.5

Author: zmh-program

cache.js

@@ -1,5 +1,6 @@
 const logger = (require("log4js")).getLogger("Cache");
 const axios = require("axios");
+const conf = require("./config");
 
 logger.level = "debug";
 axios.defaults.baseURL = "https://api.github.com";
@@ -36,9 +37,10 @@ class LightCache {
 }
 
 class ApiCache extends LightCache {
-  constructor(token) {
+  constructor() {
     super();
-    this.token = token;
+    this.token = conf.token;
+    this.expiration = conf.expiration;
   }
   async syncAxios(url) {
     logger.debug("Request GitHub API address:", url);
@@ -50,12 +52,12 @@ class ApiCache extends LightCache {
     })).data;
   }
 
-  async requestWithCache(url, expiration=3600) {
+  async requestWithCache(url) {
     // 类似于 Service Worker 缓存机制
     return this.asyncGetOrSet(
       url,
       async () => (await this.syncAxios(url)),
-      expiration,
+      this.expiration,
     );
   }
 }

config.js

@@ -0,0 +1,5 @@
+module.exports = {
+  token: process.env.CODE_STATISTIC, // GitHub Access Token (Minimum permissions). Increase QPS of GitHub APIs.
+  expiration: 3600,  // expiration second
+  requires: ["*"], // CODE STATISTIC can only be parsed for allowed users. ( * indicates that all users are allowed )
+}

index.js

@@ -1,7 +1,13 @@
 const express = require('express');
-const cache = new (require('./cache').ApiCache)(process.env.CODE_STATISTIC || "");
+const cache = new (require('./cache').ApiCache)();
+const conf = require('./config');
 const app = express();
 
+
+const isAvailableUser = (username) => {
+    return ( !! username ) && ( conf.requires.includes("*") || conf.requires.includes(username));
+}
+
 async function getLanguage(user, repo) {
     return await cache.requestWithCache(`/repos/${user}/${repo}/languages`);
 }
@@ -22,17 +28,26 @@ async function langStatistics(queue) {
     return res;
 }
 
-app.get('/user/:user/', async function (req, res) {
-    const username = req.params.user;
+app.get('/:user/', async function (req, res) {
+    const username = req.params['user'];
+    if ( ! isAvailableUser(username) ) {
+        res.send('permission denied');
+        return;
+    }
     const response = await cache.requestWithCache(`/users/${username}/repos`);
     const result = await langStatistics(Object.values(response).map(async (resp) => {
         return await getLanguage(username, resp['name']);
     }));
     res.send(result);
 })
 
-app.get('/repo/:user/:repo/', function (req, res) {
-    res.send(getLanguage(req.params['user'], req.params['repo']));
+app.get('/:user/:repo/', async function (req, res) {
+    const username = req.params['user'], repo = req.params['repo'];
+    if ( ! isAvailableUser(username) ) {
+        res.send('permission denied');
+        return;
+    }
+    res.send(await getLanguage(username, repo));
 })
 
 app.listen(3000);