mirror of
https://github.com/vastxie/99AI.git
synced 2025-11-10 10:43:41 +08:00
v4.1.0
This commit is contained in:
26
AIWebQuickDeploy/dist/modules/redisCache/dto/redis.dto.js
vendored
Normal file
26
AIWebQuickDeploy/dist/modules/redisCache/dto/redis.dto.js
vendored
Normal file
@@ -0,0 +1,26 @@
|
||||
"use strict";
|
||||
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
||||
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
||||
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
||||
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
||||
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
||||
};
|
||||
var __metadata = (this && this.__metadata) || function (k, v) {
|
||||
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
||||
};
|
||||
Object.defineProperty(exports, "__esModule", { value: true });
|
||||
exports.RedisDto = void 0;
|
||||
const class_validator_1 = require("class-validator");
|
||||
const swagger_1 = require("@nestjs/swagger");
|
||||
class RedisDto {
|
||||
}
|
||||
__decorate([
|
||||
(0, swagger_1.ApiProperty)({ example: 'name', description: '邮箱' }),
|
||||
__metadata("design:type", String)
|
||||
], RedisDto.prototype, "key", void 0);
|
||||
__decorate([
|
||||
(0, swagger_1.ApiProperty)({ example: '123456', description: '密码' }),
|
||||
(0, class_validator_1.IsNotEmpty)({ message: '用户密码不能为空!' }),
|
||||
__metadata("design:type", String)
|
||||
], RedisDto.prototype, "val", void 0);
|
||||
exports.RedisDto = RedisDto;
|
||||
48
AIWebQuickDeploy/dist/modules/redisCache/redisCache.controller.js
vendored
Normal file
48
AIWebQuickDeploy/dist/modules/redisCache/redisCache.controller.js
vendored
Normal file
@@ -0,0 +1,48 @@
|
||||
"use strict";
|
||||
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
||||
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
||||
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
||||
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
||||
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
||||
};
|
||||
var __metadata = (this && this.__metadata) || function (k, v) {
|
||||
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
||||
};
|
||||
var __param = (this && this.__param) || function (paramIndex, decorator) {
|
||||
return function (target, key) { decorator(target, key, paramIndex); }
|
||||
};
|
||||
Object.defineProperty(exports, "__esModule", { value: true });
|
||||
exports.RedisCacheController = void 0;
|
||||
const redis_dto_1 = require("./dto/redis.dto");
|
||||
const redisCache_service_1 = require("./redisCache.service");
|
||||
const common_1 = require("@nestjs/common");
|
||||
let RedisCacheController = class RedisCacheController {
|
||||
constructor(redisCacheService) {
|
||||
this.redisCacheService = redisCacheService;
|
||||
}
|
||||
set(body) {
|
||||
return this.redisCacheService.set(body);
|
||||
}
|
||||
get(body) {
|
||||
return this.redisCacheService.get(body);
|
||||
}
|
||||
};
|
||||
__decorate([
|
||||
(0, common_1.Post)('set'),
|
||||
__param(0, (0, common_1.Body)()),
|
||||
__metadata("design:type", Function),
|
||||
__metadata("design:paramtypes", [redis_dto_1.RedisDto]),
|
||||
__metadata("design:returntype", void 0)
|
||||
], RedisCacheController.prototype, "set", null);
|
||||
__decorate([
|
||||
(0, common_1.Get)('get'),
|
||||
__param(0, (0, common_1.Query)()),
|
||||
__metadata("design:type", Function),
|
||||
__metadata("design:paramtypes", [redis_dto_1.RedisDto]),
|
||||
__metadata("design:returntype", void 0)
|
||||
], RedisCacheController.prototype, "get", null);
|
||||
RedisCacheController = __decorate([
|
||||
(0, common_1.Controller)('redisCache'),
|
||||
__metadata("design:paramtypes", [redisCache_service_1.RedisCacheService])
|
||||
], RedisCacheController);
|
||||
exports.RedisCacheController = RedisCacheController;
|
||||
58
AIWebQuickDeploy/dist/modules/redisCache/redisCache.module.js
vendored
Normal file
58
AIWebQuickDeploy/dist/modules/redisCache/redisCache.module.js
vendored
Normal file
@@ -0,0 +1,58 @@
|
||||
"use strict";
|
||||
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
||||
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
||||
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
||||
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
||||
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
||||
};
|
||||
Object.defineProperty(exports, "__esModule", { value: true });
|
||||
exports.RedisCacheModule = void 0;
|
||||
const common_1 = require("@nestjs/common");
|
||||
const redis_1 = require("redis");
|
||||
const redisCache_controller_1 = require("./redisCache.controller");
|
||||
const redisCache_service_1 = require("./redisCache.service");
|
||||
let RedisCacheModule = class RedisCacheModule {
|
||||
};
|
||||
RedisCacheModule = __decorate([
|
||||
(0, common_1.Global)(),
|
||||
(0, common_1.Module)({
|
||||
imports: [],
|
||||
controllers: [redisCache_controller_1.RedisCacheController],
|
||||
providers: [
|
||||
{
|
||||
provide: 'REDIS_CLIENT',
|
||||
useFactory: async () => {
|
||||
const host = process.env.REDIS_HOST;
|
||||
const port = parseInt(process.env.REDIS_PORT, 10);
|
||||
const password = process.env.REDIS_PASSWORD;
|
||||
const username = process.env.REDIS_USER;
|
||||
const database = parseInt(process.env.REDIS_DB, 10) || 0;
|
||||
if (!host || !port) {
|
||||
common_1.Logger.error(`Please configure Redis config | 未配置 Redis 配置信息,请确认配置 Redis 服务以获得更好的体验`, 'RedisCacheModule');
|
||||
return;
|
||||
}
|
||||
const client = (0, redis_1.createClient)({
|
||||
socket: {
|
||||
host,
|
||||
port,
|
||||
},
|
||||
username,
|
||||
password,
|
||||
database,
|
||||
});
|
||||
client.on('ready', () => {
|
||||
common_1.Logger.log(`Redis connection successful`, 'RedisCacheModule');
|
||||
});
|
||||
client.on('error', (err) => {
|
||||
common_1.Logger.error(`Redis connection failed: ${err}`, 'RedisCacheModule');
|
||||
});
|
||||
await client.connect();
|
||||
return client;
|
||||
},
|
||||
},
|
||||
redisCache_service_1.RedisCacheService,
|
||||
],
|
||||
exports: ['REDIS_CLIENT', redisCache_service_1.RedisCacheService],
|
||||
})
|
||||
], RedisCacheModule);
|
||||
exports.RedisCacheModule = RedisCacheModule;
|
||||
83
AIWebQuickDeploy/dist/modules/redisCache/redisCache.service.js
vendored
Normal file
83
AIWebQuickDeploy/dist/modules/redisCache/redisCache.service.js
vendored
Normal file
@@ -0,0 +1,83 @@
|
||||
"use strict";
|
||||
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
||||
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
||||
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
||||
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
||||
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
||||
};
|
||||
var __metadata = (this && this.__metadata) || function (k, v) {
|
||||
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
||||
};
|
||||
var __param = (this && this.__param) || function (paramIndex, decorator) {
|
||||
return function (target, key) { decorator(target, key, paramIndex); }
|
||||
};
|
||||
Object.defineProperty(exports, "__esModule", { value: true });
|
||||
exports.RedisCacheService = void 0;
|
||||
const common_1 = require("@nestjs/common");
|
||||
let RedisCacheService = class RedisCacheService {
|
||||
constructor(redisClient) {
|
||||
this.redisClient = redisClient;
|
||||
}
|
||||
async get(body) {
|
||||
const { key } = body;
|
||||
const res = await this.redisClient.get(key);
|
||||
return await this.redisClient.get(key);
|
||||
}
|
||||
async set(body, time) {
|
||||
try {
|
||||
const { key, val } = body;
|
||||
await this.redisClient.set(key, val);
|
||||
time && (await this.redisClient.expire(key, time));
|
||||
return;
|
||||
}
|
||||
catch (error) {
|
||||
throw new common_1.HttpException(error, common_1.HttpStatus.BAD_REQUEST);
|
||||
}
|
||||
}
|
||||
async getJwtSecret() {
|
||||
const secret = await this.redisClient.get('JWT_SECRET');
|
||||
if (!secret) {
|
||||
throw new Error('JWT secret not found in Redis');
|
||||
}
|
||||
return secret;
|
||||
}
|
||||
async ttl(key) {
|
||||
return await this.redisClient.ttl(key);
|
||||
}
|
||||
async del(body) {
|
||||
const { key } = body;
|
||||
await this.redisClient.del(key);
|
||||
return;
|
||||
}
|
||||
async saveToken(userId, token) {
|
||||
const tokens = await this.redisClient.zRange(`tokens:${userId}`, 0, -1);
|
||||
await this.invalidateTokens(userId, tokens);
|
||||
this.redisClient.set(`token:${userId}`, token);
|
||||
}
|
||||
async invalidateTokens(userId, tokens) {
|
||||
tokens.forEach((token) => {
|
||||
this.redisClient.del(`token:${userId}:${token}`);
|
||||
});
|
||||
}
|
||||
async checkTokenAuth(token, req) {
|
||||
const { id: userId, role } = req.user;
|
||||
if (role === 'visitor')
|
||||
return true;
|
||||
const storedToken = await this.redisClient.get(`token:${userId}`);
|
||||
if (storedToken === null) {
|
||||
await this.redisClient.set(`token:${userId}`, token);
|
||||
return true;
|
||||
}
|
||||
if (storedToken !== token) {
|
||||
if (['super', 'admin'].includes(role))
|
||||
return true;
|
||||
throw new common_1.HttpException('您已在其他设备覆盖登录、请您重新登录!', common_1.HttpStatus.UNAUTHORIZED);
|
||||
}
|
||||
}
|
||||
};
|
||||
RedisCacheService = __decorate([
|
||||
(0, common_1.Injectable)(),
|
||||
__param(0, (0, common_1.Inject)('REDIS_CLIENT')),
|
||||
__metadata("design:paramtypes", [Object])
|
||||
], RedisCacheService);
|
||||
exports.RedisCacheService = RedisCacheService;
|
||||
Reference in New Issue
Block a user