miaoShare/server.js

import express from "express";
import cors from "cors";
import Database from "better-sqlite3";
import fs from "fs";
import { WebSocketServer } from "ws";
import http from "http";
// import Config from "./miaoShareConfig.json"

const HOST = "0.0.0.0";
// const PORT = Config.PORT
const PORT = 17057;
const fileDir = "./temp";
const dbPath = "./miaoShare.db";
const PATH = "./dist";

// 上传文件最大大小：200MB
const uploadMaxSize = 200 * 1024 * 1024;
// 一天的毫秒数
const millisecondsOfOneDay = parseInt(24 * 3600 * 1000);
// 文件3天过期
const LifeOfFile = millisecondsOfOneDay * 3;
// const LifeOfFile=0

// 数据库相关
const db = new Database(dbPath, { verbose: null });
db.prepare(
	`CREATE TABLE IF NOT EXISTS id2fileName (
    id CHAR(8) PRIMARY KEY NOT NULL UNIQUE,
    fileName TEXT NOT NULL,
    timestamp INT NOT NULL,
    freeze tinyint(1) NOT NULL,
    downloadCredit INT NOT NULL
);`
).run();
const stmt_getFileName = db.prepare(`SELECT fileName FROM id2fileName WHERE id = ?`);
const stmt_getTimestamp = db.prepare(`SELECT timestamp FROM id2fileName WHERE id = ?`);
const stmt_getFreeze = db.prepare(`SELECT freeze FROM id2fileName WHERE id = ?`);
const stmt_getDownloadCredit = db.prepare(`SELECT downloadCredit FROM id2fileName WHERE id = ?`);
const stmt_getFreezenIDs = db.prepare(`SELECT id FROM id2fileName WHERE freeze = 1`);
const stmt_insertId2FileName = db.prepare(`INSERT INTO id2fileName (id, fileName, timestamp, freeze, downloadCredit) VALUES (?, ?, ?, ?, ?)`);
const stmt_deleteExpireData = db.prepare(`DELETE FROM id2fileName WHERE freeze=1`);
const stmt_freezeExpireData = db.prepare(`UPDATE id2fileName SET freeze=1 WHERE timestamp < ? OR downloadCredit < 1`);
const stmt_updateDownloadCredit = db.prepare(`UPDATE id2fileName SET downloadCredit=downloadCredit-1 WHERE id = ?`);

const db_insert = (id, fileName, downloadCredit) => {
	stmt_insertId2FileName.run(id, fileName, parseInt(Date.now()), 0, downloadCredit);
};
const db_getFileName = id => {
	try {
		if (stmt_getFreeze.get(id)["freeze"] == 1) return null;
		return stmt_getFileName.get(id)["fileName"];
	} catch (e) {
		return null;
	}
};
// 检查是否可下载
const db_downloadable = id => {
	try {
		// 懒
		if (stmt_getDownloadCredit.get(id)["downloadCredit"] < 1) return false;
		if (stmt_getFreeze.get(id)["freeze"] == 1) return false;
		return true;
	} catch (e) {
		return false;
	}
};
const db_updateDownloadCredit = id => {
	try {
		stmt_updateDownloadCredit.run(id);
		return true;
	} catch (e) {
		return false;
	}
};

const db_getTimestamp = id => {
	return stmt_getTimestamp.get(id)["timestamp"];
};

// 更新存储的文件，删除已经被冻结的文件以及相关记录，然后冻结过期文件
const updateDBAndFiles = () => {
	try {
		let rows = stmt_getFreezenIDs.all();
		stmt_deleteExpireData.run();
		for (let row of rows) {
			let fileName = row["id"];
			let path = fileDir + "/" + fileName;
			try {
				fs.unlink(path, e => {
					if (e) throw e;
				});
			} catch (e) {
				console.log(e);
				continue;
			}
		}
		stmt_freezeExpireData.run(parseInt(Date.now()) - LifeOfFile);
	} catch (e) {
		console.log(e);
		return;
	}
};
updateDBAndFiles();
// 3小时更新一次
setInterval(updateDBAndFiles, 1000 * 3600 * 3);
// // 测试环境一分钟更新一次，记得注释掉
// setInterval(updateDBAndFiles, 1000 * 60);

const app = express();

app.use(cors());
app.options("*", cors());
// var tempMap = {}

if (!fs.existsSync(fileDir)) {
	fs.mkdirSync(fileDir);
}

function generateId(isGID) {
	var returnStr = "",
		range = 8,
		charStr = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";

	for (var i = 0; i < range; i++) {
		if(isGID === true && i == 0){
			var index = parseInt(Math.random() * 26);
		}else{
			var index = parseInt(Math.random() * 36);
		}
		returnStr += charStr.substring(index, index + 1);
	}
	return returnStr;
}

// 上传文件接口
function uploadFile(req, res, fileName, downloadCredit) {
	let id = generateId();
	const fileStream = fs.createWriteStream(fileDir + "/" + id);
	db_insert(id, fileName, downloadCredit);
	// tempMap[id] = fileName;
	// console.log(tempMap);
	req.pipe(fileStream);
	res.on("data", c => {
		console.log(c);
	});
	req.on("end", function () {
		fileStream.end();
		res.statusCode = 200;
		res.end(id);
	});
	req.on("error", err => {
		fileStream.end();
		res.statusCode = 500;
		res.end("File save failed: " + err.message);
	});
}

function downloadFile(req, res, fileId) {
	try {
		if (db_downloadable(fileId) == false) {
			res.writeHead(404);
			res.end(null);
			return;
		}
		db_updateDownloadCredit(fileId);
		var fileName = db_getFileName(fileId);
		if (fileName == null) {
			res.writeHead(404);
			res.end(null);
			return;
		}
		let data = fs.readFileSync(fileDir + "/" + fileId);
		res.writeHead(200, {
			"Content-Type": "application/octet-stream ",
			"File-Name": fileName,
			"Content-Length": data.length
		});
		res.end(data);
	} catch (err) {
		console.log(err);
		res.writeHead(404);
		res.end(null);
	}
}

app.use(express.static(PATH));
// app.all()
app.post("/file/upload", function (req, res) {
	res.header("Access-Control-Allow-Origin", "*");
	res.header("Access-Control-Allow-Headers", "*");
	res.header("Access-Control-Allow-Methods", "POST");
	res.header("Access-Control-Expose-Headers", "File-Name");
	var fileSize = parseInt(req.get("Content-Length"));
	var fileName = req.query.fileName;
	var downloadCredit = parseInt(req.query.downloadCredit);
	if (downloadCredit == -1) downloadCredit = 2147483647;
	if (!fileName) {
		res.statusCode = 500;
		res.end("FileName Not Found");
	} else if (fileSize > uploadMaxSize) {
		res.statusCode = 500;
		res.end("Maximum size limit exceeded");
	} else {
		uploadFile(req, res, fileName, downloadCredit);
	}
});

app.get("/file/download", function (req, res) {
	res.header("Access-Control-Allow-Origin", "*");
	res.header("Access-Control-Allow-Headers", "*");
	res.header("Access-Control-Allow-Methods", "GET");
	res.header("Access-Control-Expose-Headers", "File-Name");
	var fileId = req.query.fileName;
	if (!fileId) {
		res.statusCode = 500;
		res.end("FileName Not Found");
	} else {
		downloadFile(req, res, fileId);
	}
});

// // 启动服务器
// app.listen(PORT, HOST, function(){
//     console.log(app)
//     // console.log(`miaoShare server listening on ${}`)
// });

const server = http.createServer(app);
const wss = new WebSocketServer({ server });
// 先实现最低限度功能，暂时不考虑身份认证
const groupsMap = {};
// 三天不活跃的群组将被清除
const LifeOfGroup = millisecondsOfOneDay * 3;
// const LifeOfGroup = 0;


db.exec(
	`CREATE TABLE IF NOT EXISTS GROUPS (
	gID CHAR(8) PRIMARY KEY UNIQUE NOT NULL,
	verify TEXT NOT NULL,
	activeTimestamp INT NOT NULL
);`)


const stmt_getExpiredGroups = db.prepare(`SELECT gID FROM GROUPS WHERE activeTimestamp < ? `);

const stmt_getGroupVerify = db.prepare(`SELECT verify FROM GROUPS WHERE gID = ? `)

const stmt_insertGROUPS = db.prepare(`INSERT INTO GROUPS (gID,verify, activeTimestamp) VALUES (?, ?, ?)`);

const stmt_updateGROUPS_ActiveTimestamp = db.prepare(`UPDATE GROUPS SET activeTimestamp=? WHERE gID=?`);

const stmt_deleteGroup = db.prepare(`DELETE FROM GROUPS WHERE gID=?`);

const generateStmt_insertMessage_map = {}


const updateGROUPS = () => {
	// console.log(1)
	try {
		let gIDs = stmt_getExpiredGroups.all(parseInt(Date.now()) - LifeOfGroup);
		for (let row of gIDs) {
			let gID = row.gID
			db.exec(`DROP TABLE IF EXISTS ${gID}`)
			stmt_deleteGroup.run(gID)
		}
	} catch (e) {
		console.log(e);
		return;
	}
};
updateGROUPS()
setInterval(updateGROUPS,1000*3600*3)

// 创建群组的数据库
function createGroupTableDB(gID,verify){
	db.exec(
		`CREATE TABLE IF NOT EXISTS ${gID} (
		message TEXT NOT NULL,
		type TEXT NOT NULL,
		sender TEXT NOT NULL,
		timestamp INT NOT NULL
	);`)
	stmt_insertGROUPS.run(gID,verify,parseInt(Date.now()))
}

// 插入信息
function insertMessageByGID(gID,msg,sender){
	if(!generateStmt_insertMessage_map[gID]){
		generateStmt_insertMessage_map[gID] = db.prepare(`INSERT INTO ${gID} (message,type,sender, timestamp) VALUES (?, ?, ?, ?)`);
	}
	generateStmt_insertMessage_map[gID].run(msg,'gMsg',sender,parseInt(Date.now()))
}

function updateActiveTimestamp(gID){
	stmt_updateGROUPS_ActiveTimestamp.run(parseInt(Date.now()),gID)
}

function createGroup(verify) {
	// 暂时不考虑id随机生成时重复的问题
	let gID = generateId(true);
	groupsMap[gID] = { verify, member: [], msgHistory: [] };
	createGroupTableDB(gID,verify)
	return gID;
}

function getGroupVerify(gID) {
	if(!groupsMap[gID]){
		let verify = stmt_getGroupVerify.get(gID).verify
		// 暂时这么写
		let msgHistory = db.prepare(`SELECT message,type FROM ${gID}`).all().map(v=>{
			return {msg: v.message,type:v.type}
		})
		// console.log(msgHistory)
		groupsMap[gID] = { verify, member: [], msgHistory};
	}
	return groupsMap[gID].verify;
}

// 正式加入群组
function joinGroup(socket, gID, callback) {
	groupsMap[gID].member.push(socket);
	updateActiveTimestamp(gID)
	callback();
}

// 暂时这么写
function getHistory(gID) {
	return groupsMap[gID].msgHistory;
}

function broadcastMsg(gID, msg) {
	updateActiveTimestamp(gID)
	insertMessageByGID(gID,msg,'NULL')

	let bmsg = JSON.stringify({ type: "gMsg", gID, msg })
	groupsMap[gID].msgHistory.push({ type: "gMsg", msg });
	for (let socket of groupsMap[gID].member) {
		socket.send(bmsg);
	}
}

wss.on("connection", socket => {
	console.log("新的连接已建立");
	const no = wss.clients.size;
	const joinedGroup = []
	socket.on("message", message => {
		try {
			console.log(message.toString());
			let msg = JSON.parse(message.toString());
			if (msg.type) {
				let type = msg.type;
				// 创建群组
				if (type === "create") {
					socket.send(JSON.stringify({ type: "sMsg", type1: "create", gID: createGroup(msg.verify) }));
				} else if (type === "join") {
					// 加入群组——客端验证key
					if (msg.type1 === "verify") {
						// 已经加入则return
						if(joinedGroup.includes(msg.gID)){
							return
						}
						socket.send(JSON.stringify({ type: "sMsg", type1: "joinVerify", verify: getGroupVerify(msg.gID), gID: msg.gID }));
					} else if (msg.type1 === "success") {
						if(joinedGroup.includes(msg.gID)){
							return
						}
						// 成功加入
						joinGroup(socket, msg.gID, () => {
							joinedGroup.push(msg.gID)
							// 不考虑失败属于是
							socket.send(JSON.stringify({ type: "sMsg",gID:msg.gID, type1: "history", history: JSON.stringify(getHistory(msg.gID)) }));
						});
					}
				} else if (type === "message") {
					let gID = msg.gID;
					broadcastMsg(gID, msg.msg);
				}
			}
		} catch (e) {
			
			console.log(e);
		}
	});

	socket.on("close", () => {
		for(let gID of joinedGroup){
			groupsMap[gID].member = groupsMap[gID].member.filter(v=>v!==socket)
		}
		console.log(`No.${no} exit.`);
		console.log(groupsMap)
	});
});

server.listen(PORT, HOST, () => {});