kevisual/code-center
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

temp

Browse Source
This commit is contained in:
熊潇
2026-01-17 21:17:50 +08:00
parent dec540d055
commit d6da542989
2 changed files with 17 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
package.json
View File

@@ -23,12 +23,12 @@
"build": "NODE_ENV=production bun bun.config.mjs", "build": "NODE_ENV=production bun bun.config.mjs",
"postbuild": "ev pack", "postbuild": "ev pack",
"deploy": "rsync -avz --delete ./pack-dist/ light:/root/kevisual/assistant-app/apps/root/code-center", "deploy": "rsync -avz --delete ./pack-dist/ light:/root/kevisual/assistant-app/apps/root/code-center",
"deploy:envision": "rsync -avz --delete ./pack-dist/ envision:~/kevisual/assistant-app/apps/root/code-center", "deploy:kevisual": "rsync -avz --delete ./pack-dist/ kevisual:~/kevisual/assistant-app/apps/root/code-center",
"clean": "rm -rf dist", "clean": "rm -rf dist",
"reload": "ssh light pm2 restart root/code-center", "reload": "ssh light pm2 restart root/code-center",
"reload:envision": "ssh envision pm2 restart root/code-center", "reload:kevisual": "ssh kevisual pm2 restart root/code-center",
"pub:me": "npm run build && npm run deploy && npm run reload", "pub:me": "npm run build && npm run deploy && npm run reload",
"pub:envision": "npm run build && npm run deploy:envision && npm run reload:envision", "pub:kevisual": "npm run build && npm run deploy:kevisual && npm run reload:kevisual",
"start": "pm2 start dist/app.js --name code-center", "start": "pm2 start dist/app.js --name code-center",
"client:start": "pm2 start apps/code-center/dist/app.js --name code-center", "client:start": "pm2 start apps/code-center/dist/app.js --name code-center",
"ssl": "ssh -L 5432:localhost:5432 light", "ssl": "ssh -L 5432:localhost:5432 light",

36
src/modules/auth.ts
View File

@@ -5,14 +5,9 @@ import { logger } from './logger.ts';
export const error = (msg: string, code = 500) => { export const error = (msg: string, code = 500) => {
return JSON.stringify({ code, message: msg }); return JSON.stringify({ code, message: msg });
}; };
export const checkAuth = async (req: http.IncomingMessage, res: http.ServerResponse) => { export const getTokenFromRequest = (req: http.IncomingMessage) => {
let token = (req.headers?.['authorization'] as string) || (req.headers?.['Authorization'] as string) || ''; let token = (req.headers?.['authorization'] as string) || (req.headers?.['Authorization'] as string) || '';
const url = new URL(req.url || '', 'http://localhost'); const url = new URL(req.url || '', 'http://localhost');
const resNoPermission = () => {
res.statusCode = 401;
res.end(error('Invalid authorization'));
return { tokenUser: null, token: null };
};
if (!token) { if (!token) {
token = url.searchParams.get('token') || ''; token = url.searchParams.get('token') || '';
} }
@@ -20,12 +15,21 @@ export const checkAuth = async (req: http.IncomingMessage, res: http.ServerRespo
const parsedCookies = cookie.parse(req.headers.cookie || ''); const parsedCookies = cookie.parse(req.headers.cookie || '');
token = parsedCookies.token || ''; token = parsedCookies.token || '';
} }
if (!token) {
return resNoPermission();
}
if (token) { if (token) {
token = token.replace('Bearer ', ''); token = token.replace('Bearer ', '');
} }
return token;
}
export const checkAuth = async (req: http.IncomingMessage, res: http.ServerResponse) => {
const token = getTokenFromRequest(req);
const resNoPermission = () => {
res.statusCode = 401;
res.end(error('Invalid authorization'));
return { tokenUser: null, token: null };
};
if (!token) {
return resNoPermission();
}
let tokenUser; let tokenUser;
try { try {
tokenUser = await User.verifyToken(token); tokenUser = await User.verifyToken(token);
@@ -54,19 +58,7 @@ export const getLoginUserByToken = async (token: string) => {
} }
} }
export const getLoginUser = async (req: http.IncomingMessage) => { export const getLoginUser = async (req: http.IncomingMessage) => {
let token = (req.headers?.['authorization'] as string) || (req.headers?.['Authorization'] as string) || ''; let token = getTokenFromRequest(req);
const url = new URL(req.url || '', 'http://localhost');
if (!token) {
token = url.searchParams.get('token') || '';
}
if (!token) {
const parsedCookies = cookie.parse(req.headers.cookie || '');
token = parsedCookies.token || '';
}
if (token) {
token = token.replace('Bearer ', '');
}
if (!token) { if (!token) {
return null; return null;
} }