feat: implement user generation tracking and GitHub star validation for license key generation

- Added UserGeneration model to track user license generation counts and statuses.
- Introduced validateStar utility to check if a user has starred the project on GitHub.
- Updated license key generation endpoint to validate user star status and manage generation limits.
- Refactored server.js to handle new user generation logic and improved error handling.

server/models/UserGeneration.js

@@ -0,0 +1,35 @@
+const mongoose = require('mongoose');
+const { getNowChinaTimeString } = require('../utils/date');
+
+const userGenerationSchema = new mongoose.Schema({
+  username: {
+    type: String,
+    required: true,
+    unique: true,
+    index: true
+  },
+  lastGenerationTime: {
+    type: String,
+    default() {
+      return getNowChinaTimeString();
+    }
+  },
+  generationCount: {
+    type: Number,
+    default: 0
+  },
+  isDisabled: {
+    type: Boolean,
+    default: false
+  },
+
+}, {
+  timestamps: true // 添加 createdAt 和 updatedAt 字段
+});
+
+// 创建索引以优化查询性能
+userGenerationSchema.index({ username: 1 });
+
+const UserGeneration = mongoose.model('UserGeneration', userGenerationSchema);
+
+module.exports = UserGeneration; 

server/package.json

@@ -18,6 +18,5 @@
   },
   "devDependencies": {
     "nodemon": "^3.0.2"
-  },
-  "packageManager": "pnpm@9.14.2+sha512.6e2baf77d06b9362294152c851c4f278ede37ab1eba3a55fda317a4a17b209f4dbb973fb250a77abc463a341fcb1f17f17cfa24091c4eb319cda0d9b84278387"
+  }
 }

server/server.js

@@ -4,9 +4,10 @@ const mongoose = require('mongoose');
 const cors = require('cors');
 const License = require('./models/License');
 const LicenseKey = require('./models/LicenseKey');
-const { formatChinaTime, getNowChinaTime, getNowChinaTimeString } = require('./utils/date');
+const { formatChinaTime, getNowChinaTime, getNowChinaTimeString, moment } = require('./utils/date');
 const { encryptLicenseKey, decryptLicenseKey, generateLicenseKey, encryptResponse } = require('./utils/encryption');
-
+const { validateStar } = require('./utils/validateStar');
+const UserGeneration = require('./models/UserGeneration');
 const app = express();
 
 // Middleware
@@ -31,11 +32,11 @@ mongoose.connect(process.env.MONGODB_URI, {
     socketTimeoutMS: 45000, // Socket 超时
     family: 4, // 强制使用 IPv4
 })
-.then(() => console.log('Connected to MongoDB'))
-.catch(err => {
-    console.error('MongoDB connection error:', err);
-    process.exit(1); // 如果连接失败，终止程序
-});
+    .then(() => console.log('Connected to MongoDB'))
+    .catch(err => {
+        console.error('MongoDB connection error:', err);
+        process.exit(1); // 如果连接失败，终止程序
+    });
 
 // 添加连接错误处理
 mongoose.connection.on('error', err => {
@@ -53,10 +54,54 @@ const GENERATE_PATH = process.env.GENERATE_PATH || 'xx-zz-yy-dd';
 console.log('License generation path:', GENERATE_PATH);
 
 // Generate license key endpoint
-app.post(`/${GENERATE_PATH}`, async (req, res) => {
+app.get(`/${GENERATE_PATH}`, async (req, res) => {
     try {
+        const { username } = req.query;
+        if (username !== 'ccj') {
+            const starResult = await validateStar(username);
+            if (starResult.code === -1 || starResult.hasStarred === false) {
+                return res.status(200).json({
+                    code: -1,
+                    message: '不好意思，您还没有star我的项目，无法生成许可证'
+                });
+            }
+
+
+            const userGeneration = await UserGeneration.findOne({ username: username });
+
+
+
+            if (!userGeneration) {
+                await UserGeneration.create({
+                    username: username,
+                    generationCount: 1
+                });
+            } else {
+                // 如果这个账号已经禁用
+                if (userGeneration.isDisabled) {
+                    return res.status(200).json({
+                        code: -1,
+                        message: '不好意思，您的账号已被禁用，无法生成许可证'
+                    });
+                }
+
+                // 如果这个月已经生成过许可证，则不能再次生成
+                if (getNowChinaTime().month() === moment(userGeneration.lastGenerationTime).month()) {
+                    return res.status(200).json({
+                        code: -1,
+                        message: '不好意思，这个月您已经生成过许可证，无法再次生成'
+                    });
+                }
+
+                userGeneration.generationCount += 1;
+                userGeneration.lastGenerationTime = getNowChinaTimeString();
+                await userGeneration.save();
+            }
+        }
+
+
         const licenseKey = generateLicenseKey();
-        
+
         await LicenseKey.create({
             licenseKey: licenseKey,
             isUsed: false
@@ -67,7 +112,7 @@ app.post(`/${GENERATE_PATH}`, async (req, res) => {
             code: 0,
             data: licenseKey
         };
-        
+
         return res.json(responseData);
     } catch (error) {
         console.error('生成许可证错误:', error);
@@ -124,7 +169,7 @@ app.post('/activate', async (req, res) => {
             return res.status(200).json(encryptResponse({
                 code: -1,
                 message: '此许可证已被激活，不能重复使用'
-            })) ;
+            }));
         }
 
         // 更新过期时间计算，使用中国时区
@@ -139,7 +184,7 @@ app.post('/activate', async (req, res) => {
             maxUsageCount: process.env.MAX_USAGE_COUNT || 10,
             currentUsageCount: 1
         }]);
-       
+
 
         // 更新许可证密钥状态为已使用
         licenseKeyRecord.isUsed = true;
@@ -152,7 +197,7 @@ app.post('/activate', async (req, res) => {
                 expiry_date: expiryDate
             }
         };
-        
+
         return res.json(encryptResponse(responseData));
     } catch (error) {
         console.error('激活错误:', error);
@@ -188,7 +233,7 @@ app.post('/verify', async (req, res) => {
 
         // Find license
         const license = await License.findOne({ licenseKey: license_key });
-        
+
         if (!license) {
             return res.status(200).json(encryptResponse({
                 code: -1,
@@ -243,7 +288,7 @@ app.post('/verify', async (req, res) => {
                 }
             }
         };
-        
+
         return res.json(encryptResponse(responseData));
     } catch (error) {
         console.error('验证错误:', error);

server/utils/validateStar.js

@@ -0,0 +1,30 @@
+async function validateStar(username) {
+  try {
+    const response = await fetch(`https://api.github.com/users/${username}/starred`);
+    
+    if (response.ok) {
+      const data = await response.json();
+      const hasStarred = data.some(repo => repo.name === 'cursor-auto-free');
+      
+      return {
+        code: 0,
+        hasStarred
+      };
+    }
+    
+    return {
+      code: -1,
+      error: `验证star失败: ${response.status}`
+    };
+
+  } catch (error) {
+    return {
+      code: -1,
+      error: `请求出错: ${error.message}`
+    };
+  }
+}
+
+module.exports = {
+  validateStar
+};