feat(BUG): Azure CosmosDB Integration
This commit is contained in:
10
package.json
10
package.json
@@ -6,6 +6,7 @@
|
||||
"scripts": {
|
||||
"dev": "vite",
|
||||
"build": "vite build",
|
||||
"test": "jest",
|
||||
"preview": "vite preview",
|
||||
"check": "svelte-check --tsconfig ./tsconfig.json",
|
||||
"tauri": "tauri",
|
||||
@@ -18,6 +19,7 @@
|
||||
"dependencies": {
|
||||
"@adobe/css-tools": "4.3.2",
|
||||
"@aws-crypto/sha256-js": "^5.2.0",
|
||||
"@azure/cosmos": "^4.4.1",
|
||||
"@breezystack/lamejs": "^1.2.7",
|
||||
"@browsermt/bergamot-translator": "^0.4.9",
|
||||
"@capacitor/android": "^5.6.0",
|
||||
@@ -91,7 +93,8 @@
|
||||
"uuid": "^9.0.1",
|
||||
"wasmoon": "^1.16.0",
|
||||
"wavefile": "^11.0.0",
|
||||
"web-streams-polyfill": "^3.3.2"
|
||||
"web-streams-polyfill": "^3.3.2",
|
||||
"yup": "^1.6.1"
|
||||
},
|
||||
"devDependencies": {
|
||||
"@capacitor/assets": "^3.0.4",
|
||||
@@ -105,6 +108,7 @@
|
||||
"@types/blueimp-md5": "^2.18.2",
|
||||
"@types/codemirror": "^5.60.15",
|
||||
"@types/diff": "^6.0.0",
|
||||
"@types/jest": "^29.5.14",
|
||||
"@types/libsodium-wrappers-sumo": "^0.7.8",
|
||||
"@types/lodash": "^4.14.202",
|
||||
"@types/lodash.isequal": "^4.5.8",
|
||||
@@ -117,6 +121,7 @@
|
||||
"@types/wicg-file-system-access": "^2020.9.8",
|
||||
"autoprefixer": "^10.4.16",
|
||||
"internal-ip": "^7.0.0",
|
||||
"jest": "^29.7.0",
|
||||
"postcss": "^8.4.33",
|
||||
"svelte": "^5.1.9",
|
||||
"svelte-check": "^4.0.5",
|
||||
@@ -127,5 +132,6 @@
|
||||
"vite": "^5.4.4",
|
||||
"vite-plugin-top-level-await": "1.4.1",
|
||||
"vite-plugin-wasm": "3.3.0"
|
||||
}
|
||||
},
|
||||
"packageManager": "pnpm@10.11.0+sha512.6540583f41cc5f628eb3d9773ecee802f4f9ef9923cc45b69890fb47991d4b092964694ec3a4f738a420c918a333062c8b925d312f42e4f0c263eb603551f977"
|
||||
}
|
||||
|
||||
2076
pnpm-lock.yaml
generated
2076
pnpm-lock.yaml
generated
File diff suppressed because it is too large
Load Diff
107
server/node/db.cjs
Normal file
107
server/node/db.cjs
Normal file
@@ -0,0 +1,107 @@
|
||||
const { Container, ContainerResponse, CosmosClient } = require("@azure/cosmos");
|
||||
const { array, mixed, object, string } = require("yup");
|
||||
|
||||
class CosmosStore {
|
||||
/**
|
||||
* @constructor
|
||||
* @param {string} endpoint - Azure Cosmos Endpoint
|
||||
* @param {string} key - Azure Key
|
||||
* @param {string} database - Cosmos Database ID
|
||||
* @param {string} container - Cosmos Container ID
|
||||
* @throws {Error}
|
||||
*/
|
||||
constructor(endpoint, key, database, container) {
|
||||
if (typeof endpoint != "string") throw new Error("Invalid endpoint type");
|
||||
if (typeof key != "string") throw new Error("Invalid key type");
|
||||
if (typeof database != "string") throw new Error("Invalid database type");
|
||||
if (typeof container != "string") throw new Error("Invalid container type");
|
||||
|
||||
this.$endpoint = endpoint;
|
||||
this.$key = key;
|
||||
this.$databaseName = database;
|
||||
this.$containerName = container;
|
||||
|
||||
this.$containerRun = null;
|
||||
this.$client = new CosmosClient({ endpoint, key });
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets a container, create if not exists.
|
||||
* @returns {Promise<Container>}
|
||||
* @throws {Error}
|
||||
*/
|
||||
async resolveContainer() {
|
||||
if (this.$containerRun) {
|
||||
return this.$containerRun;
|
||||
} else {
|
||||
const { database } = await this.$client.databases.createIfNotExists({
|
||||
id: this.$databaseName,
|
||||
});
|
||||
|
||||
const { container } = await database.containers.createIfNotExists({
|
||||
id: this.$containerName,
|
||||
partitionKey: {
|
||||
paths: ["/id"],
|
||||
kind: "Hash",
|
||||
version: 2,
|
||||
},
|
||||
});
|
||||
|
||||
this.$containerRun = container;
|
||||
return container;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* List all data in container.
|
||||
* @returns {Promise<Array<{ name: string, content: Buffer }>>}
|
||||
* @throws {Error}
|
||||
*/
|
||||
async listData() {
|
||||
const c = await this.resolveContainer();
|
||||
const list = await c.items.readAll().fetchAll();
|
||||
return list.resources;
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a data in container.
|
||||
* @param {string} name - Name of the data
|
||||
* @param {Buffer} content - Content
|
||||
* @throws {Error}
|
||||
*/
|
||||
async createData(name, content) {
|
||||
const c = await this.resolveContainer();
|
||||
|
||||
const item = {
|
||||
id: name,
|
||||
content: content,
|
||||
};
|
||||
|
||||
await c.items.create(item);
|
||||
}
|
||||
|
||||
/**
|
||||
* Removes a data in container by name.
|
||||
* @param {string} name - Name of the data to remove
|
||||
* @throws {Error}
|
||||
*/
|
||||
async removeData(name) {
|
||||
const c = await this.resolveContainer();
|
||||
await c.item(name).delete();
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets a data in container by name.
|
||||
* @param {string} name - Name of the data to get
|
||||
* @throws {Error}
|
||||
*/
|
||||
async getData(name) {
|
||||
const c = await this.resolveContainer();
|
||||
const data = await c.item(name).read();
|
||||
|
||||
console.log(data);
|
||||
return data.content;
|
||||
}
|
||||
}
|
||||
|
||||
exports.CosmosStore = CosmosStore;
|
||||
22
server/node/db.spec.js
Normal file
22
server/node/db.spec.js
Normal file
@@ -0,0 +1,22 @@
|
||||
const { CosmosStore } = require("./db.cjs");
|
||||
|
||||
describe("Azure Cosmos DB", () => {
|
||||
test("should able to create", async () => {
|
||||
const c = new CosmosStore(
|
||||
"http://127.0.0.1:8081",
|
||||
"C2y6yDjf5/R+ob0N8A7Cgv30VRDJIWEHLM+4QDU5DE2nQ9nDuVTqobD4b8mGGyPMbIZnqyMsEcaGQy67XIw/Jw==",
|
||||
"db",
|
||||
"container",
|
||||
);
|
||||
|
||||
const data = Buffer.from("sfsdf");
|
||||
await c.createData("hello2", data);
|
||||
|
||||
const l = await c.listData();
|
||||
expect(l.length).toBeGreaterThanOrEqual(1);
|
||||
|
||||
const d = await c.getData("hello2");
|
||||
console.log(d);
|
||||
expect(d).toBe(data);
|
||||
});
|
||||
});
|
||||
@@ -1,65 +1,83 @@
|
||||
const express = require('express');
|
||||
const { CosmosStore } = require("./db.cjs");
|
||||
const express = require("express");
|
||||
const app = express();
|
||||
const path = require('path');
|
||||
const htmlparser = require('node-html-parser');
|
||||
const { existsSync, mkdirSync, readFileSync, writeFileSync } = require('fs');
|
||||
const fs = require('fs/promises')
|
||||
const crypto = require('crypto')
|
||||
app.use(express.static(path.join(process.cwd(), 'dist'), {index: false}));
|
||||
app.use(express.json({ limit: '50mb' }));
|
||||
app.use(express.raw({ type: 'application/octet-stream', limit: '50mb' }));
|
||||
const {pipeline} = require('stream/promises')
|
||||
const https = require('https');
|
||||
const sslPath = path.join(process.cwd(), 'server/node/ssl/certificate');
|
||||
const hubURL = 'https://sv.risuai.xyz';
|
||||
const path = require("path");
|
||||
const htmlparser = require("node-html-parser");
|
||||
const { existsSync, mkdirSync, readFileSync, writeFileSync } = require("fs");
|
||||
const fs = require("fs/promises");
|
||||
const crypto = require("crypto");
|
||||
app.use(express.static(path.join(process.cwd(), "dist"), { index: false }));
|
||||
app.use(express.json({ limit: "50mb" }));
|
||||
app.use(express.raw({ type: "application/octet-stream", limit: "50mb" }));
|
||||
const { pipeline } = require("stream/promises");
|
||||
const https = require("https");
|
||||
const sslPath = path.join(process.cwd(), "server/node/ssl/certificate");
|
||||
const hubURL = "https://sv.risuai.xyz";
|
||||
|
||||
let password = ''
|
||||
let password = "";
|
||||
|
||||
const savePath = path.join(process.cwd(), "save")
|
||||
if(!existsSync(savePath)){
|
||||
mkdirSync(savePath)
|
||||
const cosmosStore = new CosmosStore(
|
||||
"http://127.0.0.1:8081",
|
||||
"C2y6yDjf5/R+ob0N8A7Cgv30VRDJIWEHLM+4QDU5DE2nQ9nDuVTqobD4b8mGGyPMbIZnqyMsEcaGQy67XIw/Jw==",
|
||||
"db",
|
||||
"container",
|
||||
);
|
||||
|
||||
const savePath = path.join(process.cwd(), "save");
|
||||
if (!existsSync(savePath)) {
|
||||
mkdirSync(savePath);
|
||||
}
|
||||
|
||||
const passwordPath = path.join(process.cwd(), 'save', '__password')
|
||||
if(existsSync(passwordPath)){
|
||||
password = readFileSync(passwordPath, 'utf-8')
|
||||
const passwordPath = path.join(process.cwd(), "save", "__password");
|
||||
if (existsSync(passwordPath)) {
|
||||
password = readFileSync(passwordPath, "utf-8");
|
||||
}
|
||||
const hexRegex = /^[0-9a-fA-F]+$/;
|
||||
function isHex(str) {
|
||||
return hexRegex.test(str.toUpperCase().trim()) || str === '__password';
|
||||
return hexRegex.test(str.toUpperCase().trim()) || str === "__password";
|
||||
}
|
||||
|
||||
app.get('/', async (req, res, next) => {
|
||||
|
||||
const clientIP = req.headers['x-forwarded-for'] || req.ip || req.socket.remoteAddress || 'Unknown IP';
|
||||
app.get("/", async (req, res, next) => {
|
||||
const clientIP =
|
||||
req.headers["x-forwarded-for"] ||
|
||||
req.ip ||
|
||||
req.socket.remoteAddress ||
|
||||
"Unknown IP";
|
||||
const timestamp = new Date().toISOString();
|
||||
console.log(`[Server] ${timestamp} | Connection from: ${clientIP}`);
|
||||
|
||||
try {
|
||||
const mainIndex = await fs.readFile(path.join(process.cwd(), 'dist', 'index.html'))
|
||||
const root = htmlparser.parse(mainIndex)
|
||||
const head = root.querySelector('head')
|
||||
head.innerHTML = `<script>globalThis.__NODE__ = true</script>` + head.innerHTML
|
||||
const mainIndex = await fs.readFile(
|
||||
path.join(process.cwd(), "dist", "index.html"),
|
||||
);
|
||||
const root = htmlparser.parse(mainIndex);
|
||||
const head = root.querySelector("head");
|
||||
head.innerHTML =
|
||||
`<script>globalThis.__NODE__ = true</script>` + head.innerHTML;
|
||||
|
||||
res.send(root.toString())
|
||||
res.send(root.toString());
|
||||
} catch (error) {
|
||||
console.log(error)
|
||||
next(error)
|
||||
console.log(error);
|
||||
next(error);
|
||||
}
|
||||
})
|
||||
});
|
||||
|
||||
const reverseProxyFunc = async (req, res, next) => {
|
||||
const urlParam = req.headers['risu-url'] ? decodeURIComponent(req.headers['risu-url']) : req.query.url;
|
||||
const urlParam = req.headers["risu-url"]
|
||||
? decodeURIComponent(req.headers["risu-url"])
|
||||
: req.query.url;
|
||||
|
||||
if (!urlParam) {
|
||||
res.status(400).send({
|
||||
error:'URL has no param'
|
||||
error: "URL has no param",
|
||||
});
|
||||
return;
|
||||
}
|
||||
const header = req.headers['risu-header'] ? JSON.parse(decodeURIComponent(req.headers['risu-header'])) : req.headers;
|
||||
if(!header['x-forwarded-for']){
|
||||
header['x-forwarded-for'] = req.ip
|
||||
const header = req.headers["risu-header"]
|
||||
? JSON.parse(decodeURIComponent(req.headers["risu-header"]))
|
||||
: req.headers;
|
||||
if (!header["x-forwarded-for"]) {
|
||||
header["x-forwarded-for"] = req.ip;
|
||||
}
|
||||
let originalResponse;
|
||||
try {
|
||||
@@ -67,17 +85,17 @@ const reverseProxyFunc = async (req, res, next) => {
|
||||
originalResponse = await fetch(urlParam, {
|
||||
method: req.method,
|
||||
headers: header,
|
||||
body: JSON.stringify(req.body)
|
||||
body: JSON.stringify(req.body),
|
||||
});
|
||||
// get response body as stream
|
||||
const originalBody = originalResponse.body;
|
||||
// get response headers
|
||||
const head = new Headers(originalResponse.headers);
|
||||
head.delete('content-security-policy');
|
||||
head.delete('content-security-policy-report-only');
|
||||
head.delete('clear-site-data');
|
||||
head.delete('Cache-Control');
|
||||
head.delete('Content-Encoding');
|
||||
head.delete("content-security-policy");
|
||||
head.delete("content-security-policy-report-only");
|
||||
head.delete("clear-site-data");
|
||||
head.delete("Cache-Control");
|
||||
head.delete("Content-Encoding");
|
||||
const headObj = {};
|
||||
for (let [k, v] of head) {
|
||||
headObj[k] = v;
|
||||
@@ -88,44 +106,45 @@ const reverseProxyFunc = async (req, res, next) => {
|
||||
res.status(originalResponse.status);
|
||||
// send response body to client
|
||||
await pipeline(originalResponse.body, res);
|
||||
|
||||
|
||||
}
|
||||
catch (err) {
|
||||
} catch (err) {
|
||||
next(err);
|
||||
return;
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
const reverseProxyFunc_get = async (req, res, next) => {
|
||||
const urlParam = req.headers['risu-url'] ? decodeURIComponent(req.headers['risu-url']) : req.query.url;
|
||||
const urlParam = req.headers["risu-url"]
|
||||
? decodeURIComponent(req.headers["risu-url"])
|
||||
: req.query.url;
|
||||
|
||||
if (!urlParam) {
|
||||
res.status(400).send({
|
||||
error:'URL has no param'
|
||||
error: "URL has no param",
|
||||
});
|
||||
return;
|
||||
}
|
||||
const header = req.headers['risu-header'] ? JSON.parse(decodeURIComponent(req.headers['risu-header'])) : req.headers;
|
||||
if(!header['x-forwarded-for']){
|
||||
header['x-forwarded-for'] = req.ip
|
||||
const header = req.headers["risu-header"]
|
||||
? JSON.parse(decodeURIComponent(req.headers["risu-header"]))
|
||||
: req.headers;
|
||||
if (!header["x-forwarded-for"]) {
|
||||
header["x-forwarded-for"] = req.ip;
|
||||
}
|
||||
let originalResponse;
|
||||
try {
|
||||
// make request to original server
|
||||
originalResponse = await fetch(urlParam, {
|
||||
method: 'GET',
|
||||
headers: header
|
||||
method: "GET",
|
||||
headers: header,
|
||||
});
|
||||
// get response body as stream
|
||||
const originalBody = originalResponse.body;
|
||||
// get response headers
|
||||
const head = new Headers(originalResponse.headers);
|
||||
head.delete('content-security-policy');
|
||||
head.delete('content-security-policy-report-only');
|
||||
head.delete('clear-site-data');
|
||||
head.delete('Cache-Control');
|
||||
head.delete('Content-Encoding');
|
||||
head.delete("content-security-policy");
|
||||
head.delete("content-security-policy-report-only");
|
||||
head.delete("clear-site-data");
|
||||
head.delete("Cache-Control");
|
||||
head.delete("Content-Encoding");
|
||||
const headObj = {};
|
||||
for (let [k, v] of head) {
|
||||
headObj[k] = v;
|
||||
@@ -136,17 +155,15 @@ const reverseProxyFunc_get = async (req, res, next) => {
|
||||
res.status(originalResponse.status);
|
||||
// send response body to client
|
||||
await pipeline(originalResponse.body, res);
|
||||
}
|
||||
catch (err) {
|
||||
} catch (err) {
|
||||
next(err);
|
||||
return;
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
async function hubProxyFunc(req, res) {
|
||||
|
||||
try {
|
||||
const pathAndQuery = req.originalUrl.replace(/^\/hub-proxy/, '');
|
||||
const pathAndQuery = req.originalUrl.replace(/^\/hub-proxy/, "");
|
||||
const externalURL = hubURL + pathAndQuery;
|
||||
|
||||
const headersToSend = { ...req.headers };
|
||||
@@ -156,9 +173,9 @@ async function hubProxyFunc(req, res) {
|
||||
const response = await fetch(externalURL, {
|
||||
method: req.method,
|
||||
headers: headersToSend,
|
||||
body: req.method !== 'GET' && req.method !== 'HEAD' ? req : undefined,
|
||||
redirect: 'manual',
|
||||
duplex: 'half'
|
||||
body: req.method !== "GET" && req.method !== "HEAD" ? req : undefined,
|
||||
redirect: "manual",
|
||||
duplex: "half",
|
||||
});
|
||||
|
||||
for (const [key, value] of response.headers.entries()) {
|
||||
@@ -168,19 +185,15 @@ async function hubProxyFunc(req, res) {
|
||||
|
||||
if (response.status >= 300 && response.status < 400) {
|
||||
// Redirect handling (due to ‘/redirect/docs/lua’)
|
||||
const redirectUrl = response.headers.get('location');
|
||||
const redirectUrl = response.headers.get("location");
|
||||
if (redirectUrl) {
|
||||
|
||||
if (redirectUrl.startsWith('http')) {
|
||||
|
||||
if (redirectUrl.startsWith("http")) {
|
||||
if (redirectUrl.startsWith(hubURL)) {
|
||||
const newPath = redirectUrl.replace(hubURL, '/hub-proxy');
|
||||
res.setHeader('location', newPath);
|
||||
const newPath = redirectUrl.replace(hubURL, "/hub-proxy");
|
||||
res.setHeader("location", newPath);
|
||||
}
|
||||
|
||||
} else if (redirectUrl.startsWith('/')) {
|
||||
|
||||
res.setHeader('location', `/hub-proxy${redirectUrl}`);
|
||||
} else if (redirectUrl.startsWith("/")) {
|
||||
res.setHeader("location", `/hub-proxy${redirectUrl}`);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -188,116 +201,115 @@ async function hubProxyFunc(req, res) {
|
||||
}
|
||||
|
||||
await pipeline(response.body, res);
|
||||
|
||||
} catch (error) {
|
||||
console.error("[Hub Proxy] Error:", error);
|
||||
if (!res.headersSent) {
|
||||
res.status(502).send({ error: 'Proxy request failed: ' + error.message });
|
||||
res.status(502).send({ error: "Proxy request failed: " + error.message });
|
||||
} else {
|
||||
res.end();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
app.get('/proxy', reverseProxyFunc_get);
|
||||
app.get('/proxy2', reverseProxyFunc_get);
|
||||
app.get('/hub-proxy/*', hubProxyFunc);
|
||||
app.get("/proxy", reverseProxyFunc_get);
|
||||
app.get("/proxy2", reverseProxyFunc_get);
|
||||
app.get("/hub-proxy/*", hubProxyFunc);
|
||||
|
||||
app.post('/proxy', reverseProxyFunc);
|
||||
app.post('/proxy2', reverseProxyFunc);
|
||||
app.post('/hub-proxy/*', hubProxyFunc);
|
||||
app.post("/proxy", reverseProxyFunc);
|
||||
app.post("/proxy2", reverseProxyFunc);
|
||||
app.post("/hub-proxy/*", hubProxyFunc);
|
||||
|
||||
app.get('/api/password', async(req, res)=> {
|
||||
if(password === ''){
|
||||
res.send({status: 'unset'})
|
||||
app.get("/api/password", async (req, res) => {
|
||||
if (password === "") {
|
||||
res.send({ status: "unset" });
|
||||
} else if (req.headers["risu-auth"] === password) {
|
||||
res.send({ status: "correct" });
|
||||
} else {
|
||||
res.send({ status: "incorrect" });
|
||||
}
|
||||
else if(req.headers['risu-auth'] === password){
|
||||
res.send({status:'correct'})
|
||||
}
|
||||
else{
|
||||
res.send({status:'incorrect'})
|
||||
}
|
||||
})
|
||||
});
|
||||
|
||||
app.post('/api/crypto', async (req, res) => {
|
||||
app.post("/api/crypto", async (req, res) => {
|
||||
try {
|
||||
const hash = crypto.createHash('sha256')
|
||||
hash.update(Buffer.from(req.body.data, 'utf-8'))
|
||||
res.send(hash.digest('hex'))
|
||||
const hash = crypto.createHash("sha256");
|
||||
hash.update(Buffer.from(req.body.data, "utf-8"));
|
||||
res.send(hash.digest("hex"));
|
||||
} catch (error) {
|
||||
next(error)
|
||||
next(error);
|
||||
}
|
||||
})
|
||||
});
|
||||
|
||||
|
||||
app.post('/api/set_password', async (req, res) => {
|
||||
if(password === ''){
|
||||
password = req.body.password
|
||||
writeFileSync(passwordPath, password, 'utf-8')
|
||||
app.post("/api/set_password", async (req, res) => {
|
||||
if (password === "") {
|
||||
password = req.body.password;
|
||||
writeFileSync(passwordPath, password, "utf-8");
|
||||
}
|
||||
res.status(400).send("already set")
|
||||
})
|
||||
res.status(400).send("already set");
|
||||
});
|
||||
|
||||
app.get('/api/read', async (req, res, next) => {
|
||||
if(req.headers['risu-auth'].trim() !== password.trim()){
|
||||
console.log('incorrect')
|
||||
app.get("/api/read", async (req, res, next) => {
|
||||
if (req.headers["risu-auth"].trim() !== password.trim()) {
|
||||
console.log("incorrect");
|
||||
res.status(400).send({
|
||||
error:'Password Incorrect'
|
||||
error: "Password Incorrect",
|
||||
});
|
||||
return
|
||||
return;
|
||||
}
|
||||
const filePath = req.headers['file-path'];
|
||||
const filePath = req.headers["file-path"];
|
||||
if (!filePath) {
|
||||
console.log('no path')
|
||||
console.log("no path");
|
||||
res.status(400).send({
|
||||
error:'File path required'
|
||||
error: "File path required",
|
||||
});
|
||||
return;
|
||||
}
|
||||
|
||||
if(!isHex(filePath)){
|
||||
if (!isHex(filePath)) {
|
||||
res.status(400).send({
|
||||
error:'Invaild Path'
|
||||
error: "Invaild Path",
|
||||
});
|
||||
return;
|
||||
}
|
||||
try {
|
||||
if(!existsSync(path.join(savePath, filePath))){
|
||||
try {
|
||||
const fileData = await cosmosStore.getData(filePath);
|
||||
res.setHeader("Content-Type", "application/octet-stream");
|
||||
res.send(fileData);
|
||||
} catch (e) {
|
||||
res.send();
|
||||
}
|
||||
else{
|
||||
res.setHeader('Content-Type','application/octet-stream');
|
||||
res.sendFile(path.join(savePath, filePath));
|
||||
if (!existsSync(path.join(savePath, filePath))) {
|
||||
} else {
|
||||
}
|
||||
} catch (error) {
|
||||
next(error);
|
||||
}
|
||||
});
|
||||
|
||||
app.get('/api/remove', async (req, res, next) => {
|
||||
if(req.headers['risu-auth'].trim() !== password.trim()){
|
||||
console.log('incorrect')
|
||||
app.get("/api/remove", async (req, res, next) => {
|
||||
if (req.headers["risu-auth"].trim() !== password.trim()) {
|
||||
console.log("incorrect");
|
||||
res.status(400).send({
|
||||
error:'Password Incorrect'
|
||||
error: "Password Incorrect",
|
||||
});
|
||||
return
|
||||
return;
|
||||
}
|
||||
const filePath = req.headers['file-path'];
|
||||
const filePath = req.headers["file-path"];
|
||||
if (!filePath) {
|
||||
res.status(400).send({
|
||||
error:'File path required'
|
||||
error: "File path required",
|
||||
});
|
||||
return;
|
||||
}
|
||||
if(!isHex(filePath)){
|
||||
if (!isHex(filePath)) {
|
||||
res.status(400).send({
|
||||
error:'Invaild Path'
|
||||
error: "Invaild Path",
|
||||
});
|
||||
return;
|
||||
}
|
||||
|
||||
try {
|
||||
await fs.rm(path.join(savePath, filePath));
|
||||
await cosmosStore.removeData(filePath);
|
||||
res.send({
|
||||
success: true,
|
||||
});
|
||||
@@ -306,54 +318,54 @@ app.get('/api/remove', async (req, res, next) => {
|
||||
}
|
||||
});
|
||||
|
||||
app.get('/api/list', async (req, res, next) => {
|
||||
if(req.headers['risu-auth'].trim() !== password.trim()){
|
||||
console.log('incorrect')
|
||||
app.get("/api/list", async (req, res, next) => {
|
||||
if (req.headers["risu-auth"].trim() !== password.trim()) {
|
||||
console.log("incorrect");
|
||||
res.status(400).send({
|
||||
error:'Password Incorrect'
|
||||
error: "Password Incorrect",
|
||||
});
|
||||
return
|
||||
return;
|
||||
}
|
||||
try {
|
||||
const data = (await fs.readdir(path.join(savePath))).map((v) => {
|
||||
return Buffer.from(v, 'hex').toString('utf-8')
|
||||
})
|
||||
const data = (await cosmosStore.listData()).map((v) => {
|
||||
return Buffer.from(v, "hex").toString("utf-8");
|
||||
});
|
||||
res.send({
|
||||
success: true,
|
||||
content: data
|
||||
content: data,
|
||||
});
|
||||
} catch (error) {
|
||||
next(error);
|
||||
}
|
||||
});
|
||||
|
||||
app.post('/api/write', async (req, res, next) => {
|
||||
if(req.headers['risu-auth'].trim() !== password.trim()){
|
||||
console.log('incorrect')
|
||||
app.post("/api/write", async (req, res, next) => {
|
||||
if (req.headers["risu-auth"].trim() !== password.trim()) {
|
||||
console.log("incorrect");
|
||||
res.status(400).send({
|
||||
error:'Password Incorrect'
|
||||
error: "Password Incorrect",
|
||||
});
|
||||
return
|
||||
return;
|
||||
}
|
||||
const filePath = req.headers['file-path'];
|
||||
const fileContent = req.body
|
||||
const filePath = req.headers["file-path"];
|
||||
const fileContent = req.body;
|
||||
if (!filePath || !fileContent) {
|
||||
res.status(400).send({
|
||||
error:'File path required'
|
||||
error: "File path required",
|
||||
});
|
||||
return;
|
||||
}
|
||||
if(!isHex(filePath)){
|
||||
if (!isHex(filePath)) {
|
||||
res.status(400).send({
|
||||
error:'Invaild Path'
|
||||
error: "Invaild Path",
|
||||
});
|
||||
return;
|
||||
}
|
||||
|
||||
try {
|
||||
await fs.writeFile(path.join(savePath, filePath), fileContent);
|
||||
await cosmosStore.createData(filePath, fileContent);
|
||||
res.send({
|
||||
success: true
|
||||
success: true,
|
||||
});
|
||||
} catch (error) {
|
||||
next(error);
|
||||
@@ -361,32 +373,28 @@ app.post('/api/write', async (req, res, next) => {
|
||||
});
|
||||
|
||||
async function getHttpsOptions() {
|
||||
|
||||
const keyPath = path.join(sslPath, 'server.key');
|
||||
const certPath = path.join(sslPath, 'server.crt');
|
||||
const keyPath = path.join(sslPath, "server.key");
|
||||
const certPath = path.join(sslPath, "server.crt");
|
||||
|
||||
try {
|
||||
|
||||
await fs.access(keyPath);
|
||||
await fs.access(certPath);
|
||||
|
||||
const [key, cert] = await Promise.all([
|
||||
fs.readFile(keyPath),
|
||||
fs.readFile(certPath)
|
||||
fs.readFile(certPath),
|
||||
]);
|
||||
|
||||
return { key, cert };
|
||||
|
||||
} catch (error) {
|
||||
console.error('[Server] SSL setup errors:', error.message);
|
||||
console.log('[Server] Start the server with HTTP instead of HTTPS...');
|
||||
console.error("[Server] SSL setup errors:", error.message);
|
||||
console.log("[Server] Start the server with HTTP instead of HTTPS...");
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
async function startServer() {
|
||||
try {
|
||||
|
||||
const port = process.env.PORT || 6001;
|
||||
const httpsOptions = await getHttpsOptions();
|
||||
|
||||
@@ -404,7 +412,7 @@ async function startServer() {
|
||||
});
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('[Server] Failed to start server :', error);
|
||||
console.error("[Server] Failed to start server :", error);
|
||||
process.exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user