Compare commits
18 Commits
latest
...
v2.0.0-bet
Author | SHA1 | Date | |
---|---|---|---|
2a62c3d3ac | |||
1434036b42 | |||
88b0cb68d8 | |||
904b986e22 | |||
2ac9def153 | |||
d2621fdd3c | |||
4cee0048f5 | |||
3432ea5e23 | |||
0175822699 | |||
72e5c0dedd | |||
d48bf46231 | |||
e287890ca1 | |||
3e0dc06521 | |||
50268d05c5 | |||
10f3b4fa50 | |||
e42f0d3991 | |||
87985def82 | |||
1f193fd5a1 |
21
.drone.yml
Normal file
21
.drone.yml
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
kind: pipeline
|
||||||
|
type: docker
|
||||||
|
name: default
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: Build with node
|
||||||
|
image: node:12
|
||||||
|
commands:
|
||||||
|
- npm install
|
||||||
|
- npm run build
|
||||||
|
- name: Publish to docker
|
||||||
|
image: plugins/docker
|
||||||
|
settings:
|
||||||
|
username:
|
||||||
|
from_secret: docker_username
|
||||||
|
password:
|
||||||
|
from_secret: docker_password
|
||||||
|
auto_tag: true
|
||||||
|
repo: hibas123.azurecr.io/realtimedb
|
||||||
|
registry: hibas123.azurecr.io
|
||||||
|
debug: true
|
@ -1,3 +1,5 @@
|
|||||||
|
[*]
|
||||||
charset = utf-8
|
charset = utf-8
|
||||||
indent_size = 3
|
indent_size = 3
|
||||||
indent_style = space
|
indent_style = space
|
||||||
|
insert_final_newline = true
|
2466
package-lock.json
generated
2466
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
39
package.json
39
package.json
@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "@hibas123/realtimedb",
|
"name": "@hibas123/realtimedb",
|
||||||
"version": "2.0.0-beta.3",
|
"version": "2.0.0-beta.9",
|
||||||
"description": "",
|
"description": "",
|
||||||
"main": "lib/index.js",
|
"main": "lib/index.js",
|
||||||
"private": true,
|
"private": true,
|
||||||
@ -17,32 +17,31 @@
|
|||||||
"license": "ISC",
|
"license": "ISC",
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@types/dotenv": "^8.2.0",
|
"@types/dotenv": "^8.2.0",
|
||||||
"@types/jsonwebtoken": "^8.3.5",
|
"@types/jsonwebtoken": "^8.3.8",
|
||||||
"@types/koa": "^2.0.51",
|
"@types/koa": "^2.11.2",
|
||||||
"@types/koa-router": "^7.0.42",
|
"@types/koa-router": "^7.4.0",
|
||||||
"@types/leveldown": "^4.0.1",
|
"@types/leveldown": "^4.0.2",
|
||||||
"@types/levelup": "^3.1.1",
|
"@types/levelup": "^4.3.0",
|
||||||
"@types/nanoid": "^2.1.0",
|
"@types/nanoid": "^2.1.0",
|
||||||
"@types/node": "^12.12.5",
|
"@types/node": "^13.9.3",
|
||||||
"@types/ws": "^6.0.3",
|
"@types/ws": "^7.2.3",
|
||||||
"concurrently": "^5.0.0",
|
"concurrently": "^5.1.0",
|
||||||
"nodemon": "^1.19.4",
|
"nodemon": "^2.0.2",
|
||||||
"typescript": "^3.6.4"
|
"typescript": "^3.8.3"
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@hibas123/logging": "^2.1.1",
|
"@hibas123/nodelogging": "^2.1.5",
|
||||||
"@hibas123/nodelogging": "^2.1.1",
|
"@hibas123/utils": "^2.2.3",
|
||||||
"@hibas123/utils": "^2.1.1",
|
|
||||||
"dotenv": "^8.2.0",
|
"dotenv": "^8.2.0",
|
||||||
"handlebars": "^4.5.1",
|
"handlebars": "^4.7.3",
|
||||||
"jsonwebtoken": "^8.5.1",
|
"jsonwebtoken": "^8.5.1",
|
||||||
"koa": "^2.11.0",
|
"koa": "^2.11.0",
|
||||||
"koa-body": "^4.1.1",
|
"koa-body": "^4.1.1",
|
||||||
"koa-router": "^7.4.0",
|
"koa-router": "^8.0.8",
|
||||||
"leveldown": "^5.4.1",
|
"leveldown": "^5.5.1",
|
||||||
"levelup": "^4.3.2",
|
"levelup": "^4.3.2",
|
||||||
"nanoid": "^2.1.6",
|
"nanoid": "^2.1.11",
|
||||||
"what-the-pack": "^2.0.3",
|
"what-the-pack": "^2.0.3",
|
||||||
"ws": "^7.2.0"
|
"ws": "^7.2.3"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,5 +1,6 @@
|
|||||||
import Logging from "@hibas123/nodelogging";
|
import Logging from "@hibas123/nodelogging";
|
||||||
import * as dotenv from "dotenv";
|
import * as dotenv from "dotenv";
|
||||||
|
import { LoggingTypes } from "@hibas123/logging";
|
||||||
dotenv.config()
|
dotenv.config()
|
||||||
|
|
||||||
|
|
||||||
@ -17,4 +18,8 @@ const config: IConfig = {
|
|||||||
dev: (process.env.DEV || "").toLowerCase() === "true"
|
dev: (process.env.DEV || "").toLowerCase() === "true"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (config.dev) {
|
||||||
|
Logging.logLevel = LoggingTypes.Log;
|
||||||
|
}
|
||||||
|
|
||||||
export default config;
|
export default config;
|
@ -1,43 +1,12 @@
|
|||||||
|
import Logging from "@hibas123/nodelogging";
|
||||||
|
import { IncomingMessage, Server } from "http";
|
||||||
import * as WebSocket from "ws";
|
import * as WebSocket from "ws";
|
||||||
import { Server, IncomingMessage } from "http";
|
|
||||||
import { DatabaseManager } from "./database/database";
|
import { DatabaseManager } from "./database/database";
|
||||||
import Logging from "@hibas123/logging";
|
import { CollectionQuery, DocumentQuery, IQuery, ITypedQuery } from "./database/query";
|
||||||
import { Query, CollectionQuery, DocumentQuery } from "./database/query";
|
|
||||||
import Session from "./database/session";
|
import Session from "./database/session";
|
||||||
|
import { verifyJWT } from "./helper/jwt";
|
||||||
import nanoid = require("nanoid");
|
import nanoid = require("nanoid");
|
||||||
|
|
||||||
import * as JWT from "jsonwebtoken";
|
|
||||||
|
|
||||||
async function verifyJWT(token: string, publicKey: string) {
|
|
||||||
return new Promise<any | undefined>((yes) => {
|
|
||||||
JWT.verify(token, publicKey, (err, decoded) => {
|
|
||||||
if (err)
|
|
||||||
yes(undefined);
|
|
||||||
else
|
|
||||||
yes(decoded);
|
|
||||||
})
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
const StoreSym = Symbol("store");
|
|
||||||
function StoreQuery(result?: any) {
|
|
||||||
return {
|
|
||||||
[StoreSym]: true,
|
|
||||||
result
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function DeleteQuery(result?: any) {
|
|
||||||
return {
|
|
||||||
[StoreSym]: false,
|
|
||||||
result
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
import { URLSearchParams } from "url";
|
|
||||||
|
|
||||||
type QueryTypes = "keys" | "get" | "set" | "update" | "delete" | "push" | "subscribe" | "unsubscribe";
|
|
||||||
|
|
||||||
export class ConnectionManager {
|
export class ConnectionManager {
|
||||||
static server: WebSocket.Server;
|
static server: WebSocket.Server;
|
||||||
|
|
||||||
@ -50,10 +19,9 @@ export class ConnectionManager {
|
|||||||
Logging.log("New Connection:");
|
Logging.log("New Connection:");
|
||||||
const sendError = (error: string) => socket.send(JSON.stringify({ ns: "error_msg", data: error }));
|
const sendError = (error: string) => socket.send(JSON.stringify({ ns: "error_msg", data: error }));
|
||||||
|
|
||||||
|
|
||||||
const session = new Session(nanoid());
|
const session = new Session(nanoid());
|
||||||
|
|
||||||
let query = new URLSearchParams(req.url.split("?").pop());
|
const query = new URL(req.url, "http://localhost").searchParams;
|
||||||
|
|
||||||
const database = query.get("database");
|
const database = query.get("database");
|
||||||
const db = DatabaseManager.getDatabase(database);
|
const db = DatabaseManager.getDatabase(database);
|
||||||
@ -92,53 +60,42 @@ export class ConnectionManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const stored = new Map<string, Query>();
|
|
||||||
const answer = (id: string, data: any, error: boolean = false) => {
|
const answer = (id: string, data: any, error: boolean = false) => {
|
||||||
|
if (error)
|
||||||
|
Logging.error(error as any);
|
||||||
socket.send(JSON.stringify({ ns: "message", data: { id, error, data } }));
|
socket.send(JSON.stringify({ ns: "message", data: { id, error, data } }));
|
||||||
}
|
}
|
||||||
|
|
||||||
const handler = new Map<string, ((data: any) => void)>();
|
const handler = new Map<string, ((data: any) => void)>();
|
||||||
type QueryData = { id: string, type: QueryTypes, path: string[], data: any, options: any };
|
|
||||||
|
|
||||||
handler.set("query", async ({ id, type, path, data }: QueryData) => {
|
handler.set("v2", async ({ id, query }) => db.run(Array.isArray(query) ? query : [query], session)
|
||||||
//TODO: Handle case with no id, type, path
|
.then(res => answer(id, res))
|
||||||
Logging.debug(`Request with id '${id}' from type '${type}' and path '${path.join("/")}' with data`, data)
|
.catch(err => answer(id, undefined, err))
|
||||||
|
);
|
||||||
|
|
||||||
try {
|
// handler.set("bulk", async ({ id, query }) => db.run(query, session)
|
||||||
if (!db)
|
// .then(res => answer(id, res))
|
||||||
throw new Error("Database not found!");
|
// .catch(err => answer(id, undefined, err))
|
||||||
else {
|
// );
|
||||||
let isDoc = path.length % 2 === 0;
|
|
||||||
let handler = isDoc ? documentHandler.get(type) : collectionHandler.get(type);
|
|
||||||
|
|
||||||
if (!handler && session.root)
|
|
||||||
handler = rootHandler.get(type);
|
|
||||||
|
|
||||||
if (!handler)
|
const SnapshotMap = new Map<string, string>();
|
||||||
throw new Error("Invalid Request!");
|
handler.set("snapshot", async ({ id, query }: { id: string, query: ITypedQuery<"snapshot"> }) => {
|
||||||
|
db.snapshot(query, session, (data => {
|
||||||
|
socket.send(JSON.stringify({
|
||||||
|
ns: "snapshot", data: { id, data }
|
||||||
|
}));
|
||||||
|
})).then(s => {
|
||||||
|
answer(id, s.snaphot);
|
||||||
|
SnapshotMap.set(id, s.id);
|
||||||
|
}).catch(err => answer(id, undefined, err));
|
||||||
|
})
|
||||||
|
|
||||||
let query = db.getQuery(path || [], session.sessionid, isDoc ? "document" : "collection");
|
handler.set("unsubscribe", async ({ id }) => {
|
||||||
let res = await handler({
|
let i = SnapshotMap.get(id);
|
||||||
id,
|
if (i) {
|
||||||
data,
|
db.unsubscribe(i, session);
|
||||||
socket,
|
SnapshotMap.delete(i);
|
||||||
query: query as any // We know it is the right one
|
|
||||||
})
|
|
||||||
|
|
||||||
if (res && typeof res === "object" && res[StoreSym] !== undefined) {
|
|
||||||
if (res[StoreSym])
|
|
||||||
stored.set(id, query);
|
|
||||||
else
|
|
||||||
stored.delete(id);
|
|
||||||
|
|
||||||
res = res.result;
|
|
||||||
}
|
|
||||||
answer(id, res);
|
|
||||||
}
|
|
||||||
} catch (err) {
|
|
||||||
// Logging.error(err);
|
|
||||||
Logging.debug("Sending error:", err);
|
|
||||||
answer(id, err.message, true);
|
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
@ -156,102 +113,10 @@ export class ConnectionManager {
|
|||||||
})
|
})
|
||||||
|
|
||||||
socket.on("close", () => {
|
socket.on("close", () => {
|
||||||
Logging.log(`${session.sessionid} has disconnected!`);
|
Logging.log(`${session.id} has disconnected!`);
|
||||||
Logging.debug("Clearing stored:", stored);
|
session.subscriptions.forEach(unsubscribe => unsubscribe());
|
||||||
stored.forEach(query => (query as DocumentQuery | CollectionQuery).unsubscribe());
|
session.subscriptions.clear();
|
||||||
stored.clear();
|
|
||||||
socket.removeAllListeners();
|
socket.removeAllListeners();
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
type QueryHandler<T extends Query> = (api: {
|
|
||||||
id: string;
|
|
||||||
query: T;
|
|
||||||
// storedQuery(id: string): T | undefined;
|
|
||||||
socket: WebSocket;
|
|
||||||
data: any;
|
|
||||||
}) => any;
|
|
||||||
|
|
||||||
const NoPermissionError = new Error("No permisison!");
|
|
||||||
|
|
||||||
const rootHandler = new Map<string, QueryHandler<Query>>();
|
|
||||||
rootHandler.set("collections", ({ query }) => {
|
|
||||||
return CollectionQuery.fromQuery(query).collections();
|
|
||||||
})
|
|
||||||
|
|
||||||
rootHandler.set("delete-collection", ({ query }) => {
|
|
||||||
return CollectionQuery.fromQuery(query).deleteCollection();
|
|
||||||
})
|
|
||||||
|
|
||||||
|
|
||||||
const documentHandler = new Map<string, QueryHandler<DocumentQuery>>();
|
|
||||||
documentHandler.set("get", ({ query }) => {
|
|
||||||
return query.get();
|
|
||||||
})
|
|
||||||
|
|
||||||
documentHandler.set("set", ({ query, data }) => {
|
|
||||||
return query.set(data, {});
|
|
||||||
})
|
|
||||||
|
|
||||||
documentHandler.set("update", ({ query, data }) => {
|
|
||||||
return query.update(data);
|
|
||||||
})
|
|
||||||
|
|
||||||
documentHandler.set("delete", ({ query }) => {
|
|
||||||
return query.delete();
|
|
||||||
})
|
|
||||||
|
|
||||||
documentHandler.set("snapshot", async ({ query, data, id, socket }) => {
|
|
||||||
let res = await query.snapshot((data) => {
|
|
||||||
socket.send(JSON.stringify({
|
|
||||||
ns: "snapshot", data: { id, data }
|
|
||||||
}));
|
|
||||||
});
|
|
||||||
|
|
||||||
return StoreQuery(res);
|
|
||||||
})
|
|
||||||
|
|
||||||
documentHandler.set("unsubscribe", async ({ query }) => {
|
|
||||||
query.unsubscribe();
|
|
||||||
return DeleteQuery(true);
|
|
||||||
})
|
|
||||||
|
|
||||||
|
|
||||||
const collectionHandler = new Map<string, QueryHandler<CollectionQuery>>();
|
|
||||||
collectionHandler.set("keys", ({ query }) => {
|
|
||||||
return query.keys();
|
|
||||||
})
|
|
||||||
|
|
||||||
|
|
||||||
collectionHandler.set("add", ({ query, data }) => {
|
|
||||||
return query.add(data);
|
|
||||||
})
|
|
||||||
|
|
||||||
collectionHandler.set("get", ({ query, data }) => {
|
|
||||||
if (data.where)
|
|
||||||
query.where = data.where;
|
|
||||||
if (data.limit)
|
|
||||||
query.limit = data.limit;
|
|
||||||
return query.get();
|
|
||||||
})
|
|
||||||
|
|
||||||
collectionHandler.set("snapshot", async ({ query, id, socket, data }) => {
|
|
||||||
if (data.where)
|
|
||||||
query.where = data.where;
|
|
||||||
if (data.limit)
|
|
||||||
query.limit = data.limit;
|
|
||||||
|
|
||||||
let res = await query.snapshot((data) => {
|
|
||||||
socket.send(JSON.stringify({
|
|
||||||
ns: "snapshot", data: { id, data }
|
|
||||||
}));
|
|
||||||
});
|
|
||||||
|
|
||||||
return StoreQuery(res);
|
|
||||||
})
|
|
||||||
|
|
||||||
collectionHandler.set("unsubscribe", async ({ query }) => {
|
|
||||||
query.unsubscribe();
|
|
||||||
return DeleteQuery(true);
|
|
||||||
})
|
|
@ -1,8 +1,18 @@
|
|||||||
import { Rules } from "./rules";
|
import { Rules } from "./rules";
|
||||||
import Settings from "../settings";
|
import Settings from "../settings";
|
||||||
import getLevelDB, { LevelDB, deleteLevelDB } from "../storage";
|
import getLevelDB, { LevelDB, deleteLevelDB, resNull } from "../storage";
|
||||||
import DocumentLock from "./lock";
|
import DocumentLock from "./lock";
|
||||||
import { DocumentQuery, CollectionQuery, Query } from "./query";
|
import { DocumentQuery, CollectionQuery, Query, QueryError, ITypedQuery, IQuery } from "./query";
|
||||||
|
import Logging from "@hibas123/nodelogging";
|
||||||
|
import Session from "./session";
|
||||||
|
import nanoid = require("nanoid/generate");
|
||||||
|
import { Observable } from "@hibas123/utils";
|
||||||
|
|
||||||
|
const ALPHABET = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
|
||||||
|
|
||||||
|
// interface ITransaction {
|
||||||
|
// queries: ITypedQuery<IWriteQueries>[];
|
||||||
|
// }
|
||||||
|
|
||||||
export class DatabaseManager {
|
export class DatabaseManager {
|
||||||
static databases = new Map<string, Database>();
|
static databases = new Map<string, Database>();
|
||||||
@ -45,12 +55,17 @@ export type ChangeTypes = "added" | "modified" | "deleted";
|
|||||||
export type Change = {
|
export type Change = {
|
||||||
data: any;
|
data: any;
|
||||||
document: string;
|
document: string;
|
||||||
|
collection: string;
|
||||||
type: ChangeTypes;
|
type: ChangeTypes;
|
||||||
sender: string;
|
sender: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
export class Database {
|
export class Database {
|
||||||
|
public static getKey(collectionid: string, documentid?: string) {
|
||||||
|
return `${collectionid || ""}/${documentid || ""}`;
|
||||||
|
}
|
||||||
|
|
||||||
private level = getLevelDB(this.name);
|
private level = getLevelDB(this.name);
|
||||||
|
|
||||||
get data() {
|
get data() {
|
||||||
@ -63,10 +78,15 @@ export class Database {
|
|||||||
|
|
||||||
|
|
||||||
public rules: Rules;
|
public rules: Rules;
|
||||||
public locks = new DocumentLock()
|
private locks = new DocumentLock()
|
||||||
public collectionLocks = new DocumentLock()
|
public collectionLocks = new DocumentLock()
|
||||||
|
|
||||||
public changes = new Map<string, Set<(change: Change) => void>>();
|
public changeListener = new Map<string, Set<(change: Change[]) => void>>();
|
||||||
|
public collectionChangeListener = new Observable<{
|
||||||
|
key: string;
|
||||||
|
id: string;
|
||||||
|
type: "create" | "delete"
|
||||||
|
}>();
|
||||||
|
|
||||||
toJSON() {
|
toJSON() {
|
||||||
return {
|
return {
|
||||||
@ -103,17 +123,305 @@ export class Database {
|
|||||||
this.publickey = key;
|
this.publickey = key;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public async resolve(path: string[], create = false): Promise<{ collection: string, document: string, collectionKey: string }> {
|
||||||
|
path = [...path]; // Create modifiable copy
|
||||||
|
let collectionID: string = undefined;
|
||||||
|
let documentKey = path.length % 2 === 0 ? path.pop() : undefined;
|
||||||
|
let key = path.join("/");
|
||||||
|
|
||||||
getQuery(path: string[], sender: string, type: "document" | "collection" | "any") {
|
const lock = await this.collectionLocks.lock(key);
|
||||||
if (type === "document")
|
|
||||||
return new DocumentQuery(this, path, sender);
|
try {
|
||||||
else if (type === "collection")
|
collectionID = await this.collections.get(key).then(r => r.toString()).catch(resNull);
|
||||||
return new CollectionQuery(this, path, sender);
|
if (!collectionID && create) {
|
||||||
|
collectionID = nanoid(ALPHABET, 32);
|
||||||
|
await this.collections.put(key, collectionID);
|
||||||
|
setImmediate(() => {
|
||||||
|
this.collectionChangeListener.send({
|
||||||
|
id: collectionID,
|
||||||
|
key,
|
||||||
|
type: "create"
|
||||||
|
})
|
||||||
|
})
|
||||||
|
}
|
||||||
|
} finally {
|
||||||
|
lock();
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
collection: collectionID,
|
||||||
|
document: documentKey,
|
||||||
|
collectionKey: key
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
private sendChanges(changes: Change[]) {
|
||||||
|
let col = new Map<string, Map<string, Change[]>>();
|
||||||
|
changes.forEach(change => {
|
||||||
|
let e = col.get(change.collection);
|
||||||
|
if (!e) {
|
||||||
|
e = new Map()
|
||||||
|
col.set(change.collection, e);
|
||||||
|
}
|
||||||
|
|
||||||
|
let d = e.get(change.document);
|
||||||
|
if (!d) {
|
||||||
|
d = [];
|
||||||
|
e.set(change.document, d);
|
||||||
|
}
|
||||||
|
|
||||||
|
d.push(change);
|
||||||
|
})
|
||||||
|
|
||||||
|
setImmediate(() => {
|
||||||
|
for (let [collection, documents] of col.entries()) {
|
||||||
|
let collectionChanges = [];
|
||||||
|
for (let [document, documentChanges] of documents.entries()) {
|
||||||
|
let s = this.changeListener.get(Database.getKey(collection, document));
|
||||||
|
if (s)
|
||||||
|
s.forEach(e => setImmediate(() => e(documentChanges)));
|
||||||
|
|
||||||
|
collectionChanges.push(...documentChanges);
|
||||||
|
}
|
||||||
|
let s = this.changeListener.get(Database.getKey(collection))
|
||||||
|
if (s)
|
||||||
|
s.forEach(e => setImmediate(() => e(collectionChanges)))
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
private validate(query: ITypedQuery<any>) {
|
||||||
|
const inv = new QueryError("Malformed query!");
|
||||||
|
if (!query || typeof query !== "object")
|
||||||
|
throw inv;
|
||||||
|
|
||||||
|
if (!query.type)
|
||||||
|
throw inv;
|
||||||
|
|
||||||
|
if (!query.path)
|
||||||
|
throw inv;
|
||||||
|
}
|
||||||
|
|
||||||
|
async run(queries: IQuery[], session: Session) {
|
||||||
|
let resolve: { path: string[], create: boolean, resolved?: [string, string, string] }[] = [];
|
||||||
|
|
||||||
|
const addToResolve = (path: string[], create?: boolean) => {
|
||||||
|
let entry = resolve.find(e => { //TODO: Find may be slow...
|
||||||
|
if (e.path.length !== path.length)
|
||||||
|
return false;
|
||||||
|
for (let i = 0; i < e.path.length; i++) {
|
||||||
|
if (e.path[i] !== path[i])
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
})
|
||||||
|
|
||||||
|
if (!entry) {
|
||||||
|
entry = {
|
||||||
|
path,
|
||||||
|
create
|
||||||
|
}
|
||||||
|
resolve.push(entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
entry.create = entry.create || create;
|
||||||
|
|
||||||
|
return entry;
|
||||||
|
}
|
||||||
|
|
||||||
|
const isBatch = queries.length > 1;
|
||||||
|
let parsed = queries.map(rawQuery => {
|
||||||
|
this.validate(rawQuery);
|
||||||
|
const isCollection = rawQuery.path.length % 2 === 1;
|
||||||
|
|
||||||
|
let query = isCollection
|
||||||
|
? new CollectionQuery(this, session, rawQuery)
|
||||||
|
: new DocumentQuery(this, session, rawQuery);
|
||||||
|
|
||||||
|
if (isBatch && !query.batchCompatible)
|
||||||
|
throw new Error("There are queries that are not batch compatible!");
|
||||||
|
|
||||||
|
let path = addToResolve(rawQuery.path, query.createCollection);
|
||||||
|
if (query.additionalLock)
|
||||||
|
addToResolve(query.additionalLock);
|
||||||
|
|
||||||
|
return {
|
||||||
|
path,
|
||||||
|
query
|
||||||
|
};
|
||||||
|
});
|
||||||
|
|
||||||
|
resolve = resolve.sort((a, b) => a.path.length - b.path.length);
|
||||||
|
|
||||||
|
let locks: (() => void)[] = [];
|
||||||
|
for (let e of resolve) {
|
||||||
|
let { collection, document, collectionKey } = await this.resolve(e.path, e.create);
|
||||||
|
e.resolved = [collection, document, collectionKey];
|
||||||
|
|
||||||
|
locks.push(
|
||||||
|
await this.locks.lock(collection, document)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let result = [];
|
||||||
|
try {
|
||||||
|
let batch = this.data.batch();
|
||||||
|
let changes: Change[] = [];
|
||||||
|
for (let e of parsed) {
|
||||||
|
result.push(
|
||||||
|
await e.query.run(e.path.resolved[0], e.path.resolved[1], batch, e.path.resolved[2])
|
||||||
|
);
|
||||||
|
changes.push(...e.query.changes);
|
||||||
|
}
|
||||||
|
if (batch.length > 0)
|
||||||
|
await batch.write();
|
||||||
|
|
||||||
|
this.sendChanges(changes);
|
||||||
|
} finally {
|
||||||
|
locks.forEach(lock => lock());
|
||||||
|
}
|
||||||
|
|
||||||
|
if (isBatch)
|
||||||
|
return result;
|
||||||
else
|
else
|
||||||
return new Query(this, path, sender);
|
return result[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
async snapshot(rawQuery: ITypedQuery<"snapshot">, session: Session, onchange: (change: any) => void) {
|
||||||
|
Logging.debug("Snaphot request:", rawQuery.path);
|
||||||
|
this.validate(rawQuery);
|
||||||
|
|
||||||
|
if (rawQuery.type !== "snapshot")
|
||||||
|
throw new Error("Invalid query type!");
|
||||||
|
|
||||||
|
const isCollection = rawQuery.path.length % 2 === 1;
|
||||||
|
let query = isCollection
|
||||||
|
? new CollectionQuery(this, session, rawQuery, true)
|
||||||
|
: new DocumentQuery(this, session, rawQuery, true);
|
||||||
|
|
||||||
|
const {
|
||||||
|
unsubscribe,
|
||||||
|
value
|
||||||
|
} = await query.snapshot(onchange);
|
||||||
|
|
||||||
|
const id = nanoid(ALPHABET, 16);
|
||||||
|
session.subscriptions.set(id, unsubscribe);
|
||||||
|
return {
|
||||||
|
id,
|
||||||
|
snaphot: value
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
async unsubscribe(id: string, session: Session) {
|
||||||
|
let query = session.subscriptions.get(id);
|
||||||
|
if (query) {
|
||||||
|
query();
|
||||||
|
session.subscriptions.delete(id);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async stop() {
|
async stop() {
|
||||||
await this.data.close();
|
await this.data.close();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public async runCleanup() {
|
||||||
|
const should = await new Promise<Set<string>>((yes, no) => {
|
||||||
|
const stream = this.collections.iterator({
|
||||||
|
keyAsBuffer: false,
|
||||||
|
valueAsBuffer: false
|
||||||
|
})
|
||||||
|
|
||||||
|
const collections = new Set<string>();
|
||||||
|
const onValue = (err: Error, key: string, value: string) => {
|
||||||
|
if (err) {
|
||||||
|
Logging.error(err);
|
||||||
|
stream.end((err) => Logging.error(err))
|
||||||
|
no(err);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!key && !value) {
|
||||||
|
yes(collections);
|
||||||
|
} else {
|
||||||
|
collections.add(value)
|
||||||
|
stream.next(onValue);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
stream.next(onValue);
|
||||||
|
})
|
||||||
|
|
||||||
|
const existing = await new Promise<Set<string>>((yes, no) => {
|
||||||
|
const stream = this.data.iterator({
|
||||||
|
keyAsBuffer: false,
|
||||||
|
values: false
|
||||||
|
})
|
||||||
|
|
||||||
|
const collections = new Set<string>();
|
||||||
|
const onValue = (err: Error, key: string, value: Buffer) => {
|
||||||
|
if (err) {
|
||||||
|
Logging.error(err);
|
||||||
|
stream.end((err) => Logging.error(err))
|
||||||
|
no(err);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!key && !value) {
|
||||||
|
yes(collections);
|
||||||
|
} else {
|
||||||
|
let coll = key.split("/")[0];
|
||||||
|
collections.add(coll)
|
||||||
|
stream.next(onValue);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
stream.next(onValue);
|
||||||
|
})
|
||||||
|
|
||||||
|
const toDelete = new Set<string>();
|
||||||
|
existing.forEach(collection => {
|
||||||
|
if (!should.has(collection))
|
||||||
|
toDelete.add(collection);
|
||||||
|
})
|
||||||
|
|
||||||
|
for (let collection of toDelete) {
|
||||||
|
const batch = this.data.batch();
|
||||||
|
|
||||||
|
let gt = Buffer.from(collection + "/ ");
|
||||||
|
gt[gt.length - 1] = 0;
|
||||||
|
|
||||||
|
let lt = Buffer.alloc(gt.length);
|
||||||
|
lt.set(gt);
|
||||||
|
lt[gt.length - 1] = 0xFF;
|
||||||
|
|
||||||
|
await new Promise<void>((yes, no) => {
|
||||||
|
const stream = this.data.iterator({
|
||||||
|
keyAsBuffer: false,
|
||||||
|
values: false,
|
||||||
|
gt,
|
||||||
|
lt
|
||||||
|
})
|
||||||
|
|
||||||
|
const onValue = (err: Error, key: string, value: Buffer) => {
|
||||||
|
if (err) {
|
||||||
|
Logging.error(err);
|
||||||
|
stream.end((err) => Logging.error(err))
|
||||||
|
no(err);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!key && !value) {
|
||||||
|
yes();
|
||||||
|
} else {
|
||||||
|
batch.del(key);
|
||||||
|
stream.next(onValue);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
stream.next(onValue);
|
||||||
|
})
|
||||||
|
|
||||||
|
await batch.write();
|
||||||
|
}
|
||||||
|
|
||||||
|
return Array.from(toDelete.values());
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -8,6 +8,7 @@ export default class DocumentLock {
|
|||||||
}
|
}
|
||||||
|
|
||||||
async lock(collection: string = "", document: string = "") {
|
async lock(collection: string = "", document: string = "") {
|
||||||
|
//TODO: Check collection locks
|
||||||
let key = collection + "/" + document;
|
let key = collection + "/" + document;
|
||||||
let l = this.locks.get(key);
|
let l = this.locks.get(key);
|
||||||
if (l)
|
if (l)
|
||||||
|
File diff suppressed because it is too large
Load Diff
@ -2,13 +2,15 @@ import Session from "./session";
|
|||||||
import Logging from "@hibas123/nodelogging";
|
import Logging from "@hibas123/nodelogging";
|
||||||
|
|
||||||
interface IRule<T> {
|
interface IRule<T> {
|
||||||
".write"?: T
|
".write"?: T;
|
||||||
".read"?: T
|
".read"?: T;
|
||||||
}
|
}
|
||||||
|
|
||||||
type IRuleConfig<T> = {
|
type IRuleConfig<T> =
|
||||||
[segment: string]: IRuleConfig<T>;
|
| IRule<T>
|
||||||
} | IRule<T>;
|
| {
|
||||||
|
[segment: string]: IRuleConfig<T>;
|
||||||
|
};
|
||||||
|
|
||||||
type IRuleRaw = IRuleConfig<string>;
|
type IRuleRaw = IRuleConfig<string>;
|
||||||
type IRuleParsed = IRuleConfig<boolean>;
|
type IRuleParsed = IRuleConfig<boolean>;
|
||||||
@ -17,17 +19,16 @@ const resolve = (value: any) => {
|
|||||||
if (value === true) {
|
if (value === true) {
|
||||||
return true;
|
return true;
|
||||||
} else if (typeof value === "string") {
|
} else if (typeof value === "string") {
|
||||||
|
|
||||||
}
|
}
|
||||||
return undefined;
|
return undefined;
|
||||||
}
|
};
|
||||||
|
|
||||||
export class Rules {
|
export class Rules {
|
||||||
rules: IRuleParsed;
|
rules: IRuleParsed;
|
||||||
constructor(private config: string) {
|
constructor(private config: string) {
|
||||||
let parsed: IRuleRaw = JSON.parse(config);
|
let parsed: IRuleRaw = JSON.parse(config);
|
||||||
|
|
||||||
const analyze = (raw: IRuleRaw) => {
|
const analyse = (raw: IRuleRaw) => {
|
||||||
let r: IRuleParsed = {};
|
let r: IRuleParsed = {};
|
||||||
|
|
||||||
if (raw[".read"]) {
|
if (raw[".read"]) {
|
||||||
@ -47,18 +48,25 @@ export class Rules {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for (let segment in raw) {
|
for (let segment in raw) {
|
||||||
if (segment.startsWith("."))
|
if (segment.startsWith(".")) continue;
|
||||||
continue;
|
|
||||||
|
|
||||||
r[segment] = analyze(raw[segment]);
|
r[segment] = analyse(raw[segment]);
|
||||||
}
|
}
|
||||||
return r;
|
return r;
|
||||||
}
|
};
|
||||||
|
|
||||||
this.rules = analyze(parsed);
|
this.rules = analyse(parsed);
|
||||||
}
|
}
|
||||||
|
|
||||||
hasPermission(path: string[], session: Session): { read: boolean, write: boolean } {
|
hasPermission(
|
||||||
|
path: string[],
|
||||||
|
session: Session
|
||||||
|
): { read: boolean; write: boolean } {
|
||||||
|
if (session.root)
|
||||||
|
return {
|
||||||
|
read: true,
|
||||||
|
write: true
|
||||||
|
};
|
||||||
let read = this.rules[".read"] || false;
|
let read = this.rules[".read"] || false;
|
||||||
let write = this.rules[".write"] || false;
|
let write = this.rules[".write"] || false;
|
||||||
|
|
||||||
@ -77,22 +85,21 @@ export class Rules {
|
|||||||
.find(e => {
|
.find(e => {
|
||||||
switch (e) {
|
switch (e) {
|
||||||
case "$uid":
|
case "$uid":
|
||||||
if (segment === session.uid)
|
if (segment === session.uid) return true;
|
||||||
return true;
|
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
return false;
|
return false;
|
||||||
})
|
});
|
||||||
|
|
||||||
rules = (k ? rules[k] : undefined) || rules[segment] || rules["*"];
|
rules = (k ? rules[k] : undefined) || rules[segment] || rules["*"];
|
||||||
|
|
||||||
if (rules) {
|
if (rules) {
|
||||||
if (rules[".read"]) {
|
if (rules[".read"]) {
|
||||||
read = rules[".read"]
|
read = rules[".read"];
|
||||||
}
|
}
|
||||||
|
|
||||||
if (rules[".write"]) {
|
if (rules[".write"]) {
|
||||||
read = rules[".write"]
|
read = rules[".write"];
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
break;
|
break;
|
||||||
@ -102,10 +109,10 @@ export class Rules {
|
|||||||
return {
|
return {
|
||||||
read: read as boolean,
|
read: read as boolean,
|
||||||
write: write as boolean
|
write: write as boolean
|
||||||
}
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
toJSON() {
|
toJSON() {
|
||||||
return this.config;
|
return this.config;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,8 +1,11 @@
|
|||||||
|
|
||||||
export default class Session {
|
export default class Session {
|
||||||
constructor(private _sessionid: string) { }
|
constructor(private _sessionid: string) { }
|
||||||
get sessionid() {
|
get id() {
|
||||||
return this._sessionid;
|
return this._sessionid;
|
||||||
}
|
}
|
||||||
root: boolean = false;
|
root: boolean = false;
|
||||||
uid: string = undefined;
|
uid: string = undefined;
|
||||||
|
|
||||||
|
subscriptions = new Map<string, (() => void)>();
|
||||||
}
|
}
|
12
src/helper/jwt.ts
Normal file
12
src/helper/jwt.ts
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
import * as JWT from "jsonwebtoken";
|
||||||
|
|
||||||
|
export async function verifyJWT(token: string, publicKey: string) {
|
||||||
|
return new Promise<any | undefined>((yes) => {
|
||||||
|
JWT.verify(token, publicKey, (err, decoded) => {
|
||||||
|
if (err)
|
||||||
|
yes(undefined);
|
||||||
|
else
|
||||||
|
yes(decoded);
|
||||||
|
})
|
||||||
|
})
|
||||||
|
}
|
@ -5,9 +5,14 @@ import { DatabaseManager } from "./database/database";
|
|||||||
import { createServer } from "http";
|
import { createServer } from "http";
|
||||||
import { ConnectionManager } from "./connection";
|
import { ConnectionManager } from "./connection";
|
||||||
import { LoggingTypes } from "@hibas123/logging";
|
import { LoggingTypes } from "@hibas123/logging";
|
||||||
|
import { readFileSync } from "fs";
|
||||||
|
|
||||||
Logging.logLevel = config.dev ? LoggingTypes.Debug : LoggingTypes.Log;
|
Logging.logLevel = config.dev ? LoggingTypes.Debug : LoggingTypes.Log;
|
||||||
|
|
||||||
|
const version = JSON.parse(readFileSync("./package.json").toString()).version;
|
||||||
|
|
||||||
|
Logging.log("Starting Database version:", version);
|
||||||
|
|
||||||
DatabaseManager.init().then(() => {
|
DatabaseManager.init().then(() => {
|
||||||
const http = createServer(Web.callback());
|
const http = createServer(Web.callback());
|
||||||
ConnectionManager.bind(http);
|
ConnectionManager.bind(http);
|
||||||
|
@ -398,6 +398,12 @@ export class NoPermissionError extends HttpError {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export class UnauthorizedError extends HttpError {
|
||||||
|
constructor(message: string) {
|
||||||
|
super(message, HttpStatusCode.UNAUTHORIZED)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
export class BadRequestError extends HttpError {
|
export class BadRequestError extends HttpError {
|
||||||
constructor(message: string) {
|
constructor(message: string) {
|
||||||
super(message, HttpStatusCode.BAD_REQUEST)
|
super(message, HttpStatusCode.BAD_REQUEST)
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
import getTemplate from "./hb";
|
import { getTemplate } from "./hb";
|
||||||
import { Context } from "vm";
|
import { Context } from "vm";
|
||||||
|
|
||||||
interface IFormConfigField {
|
interface IFormConfigField {
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
import * as Handlebars from "handlebars";
|
import * as Handlebars from "handlebars";
|
||||||
import { readFileSync } from "fs";
|
import { readFileSync } from "fs";
|
||||||
import config from "../../config";
|
import config from "../../config";
|
||||||
import Logging from "@hibas123/logging";
|
import Logging from "@hibas123/nodelogging";
|
||||||
|
|
||||||
|
|
||||||
function checkCondition(v1, operator, v2) {
|
function checkCondition(v1, operator, v2) {
|
||||||
@ -37,8 +37,22 @@ Handlebars.registerHelper('ifCond', function (v1, operator, v2, options) {
|
|||||||
|
|
||||||
|
|
||||||
const cache = new Map<string, Handlebars.TemplateDelegate>();
|
const cache = new Map<string, Handlebars.TemplateDelegate>();
|
||||||
|
const htmlCache = new Map<string, string>();
|
||||||
|
|
||||||
export default function getTemplate(name: string) {
|
export function getView(name: string) {
|
||||||
|
let tl: string;
|
||||||
|
if (!config.dev)
|
||||||
|
tl = htmlCache.get(name);
|
||||||
|
|
||||||
|
if (!tl) {
|
||||||
|
tl = readFileSync(`./views/${name}.html`).toString();
|
||||||
|
htmlCache.set(name, tl);
|
||||||
|
}
|
||||||
|
|
||||||
|
return tl;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function getTemplate(name: string) {
|
||||||
let tl: Handlebars.TemplateDelegate;
|
let tl: Handlebars.TemplateDelegate;
|
||||||
if (!config.dev)
|
if (!config.dev)
|
||||||
tl = cache.get(name);
|
tl = cache.get(name);
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
import { Context } from "koa";
|
import { Context } from "koa";
|
||||||
import getTemplate from "./hb";
|
import { getTemplate } from "./hb";
|
||||||
|
|
||||||
export default function getTable(title: string, data: any[], ctx: Context) {
|
export default function getTable(title: string, data: any[], ctx: Context) {
|
||||||
let table: string[][] = [];
|
let table: string[][] = [];
|
||||||
|
@ -6,7 +6,8 @@ import { BadRequestError, NoPermissionError } from "../helper/errors";
|
|||||||
import { DatabaseManager } from "../../database/database";
|
import { DatabaseManager } from "../../database/database";
|
||||||
import { MP } from "../../database/query";
|
import { MP } from "../../database/query";
|
||||||
import config from "../../config";
|
import config from "../../config";
|
||||||
import Logging from "@hibas123/logging";
|
import Logging from "@hibas123/nodelogging";
|
||||||
|
import { getView } from "../helper/hb";
|
||||||
|
|
||||||
const AdminRoute = new Router();
|
const AdminRoute = new Router();
|
||||||
|
|
||||||
@ -17,6 +18,11 @@ AdminRoute.use(async (ctx, next) => {
|
|||||||
return next();
|
return next();
|
||||||
})
|
})
|
||||||
|
|
||||||
|
AdminRoute.get("/", async ctx => {
|
||||||
|
//TODO: Main Interface
|
||||||
|
ctx.body = getView("admin");
|
||||||
|
});
|
||||||
|
|
||||||
AdminRoute.get("/settings", async ctx => {
|
AdminRoute.get("/settings", async ctx => {
|
||||||
let res = await new Promise<string[][]>((yes, no) => {
|
let res = await new Promise<string[][]>((yes, no) => {
|
||||||
const stream = Settings.db.createReadStream({
|
const stream = Settings.db.createReadStream({
|
||||||
@ -51,11 +57,11 @@ AdminRoute.get("/data", async ctx => {
|
|||||||
keys: true,
|
keys: true,
|
||||||
values: true,
|
values: true,
|
||||||
valueAsBuffer: true,
|
valueAsBuffer: true,
|
||||||
keyAsBuffer: false
|
keyAsBuffer: false,
|
||||||
|
limit: 1000
|
||||||
});
|
});
|
||||||
let res = [["key", "value"]];
|
let res = [["key", "value"]];
|
||||||
stream.on("data", ({ key, value }: { key: string, value: Buffer }) => {
|
stream.on("data", ({ key, value }: { key: string, value: Buffer }) => {
|
||||||
Logging.debug("Admin Key:", key);
|
|
||||||
res.push([key, key.split("/").length > 2 ? value.toString() : JSON.stringify(MP.decode(value))]);
|
res.push([key, key.split("/").length > 2 ? value.toString() : JSON.stringify(MP.decode(value))]);
|
||||||
})
|
})
|
||||||
|
|
||||||
@ -114,6 +120,48 @@ AdminRoute
|
|||||||
ctx.body = "Success";
|
ctx.body = "Success";
|
||||||
})
|
})
|
||||||
|
|
||||||
|
AdminRoute.get("/collections", async ctx => {
|
||||||
|
const { database } = ctx.query;
|
||||||
|
let db = DatabaseManager.getDatabase(database);
|
||||||
|
if (!db)
|
||||||
|
throw new BadRequestError("Database not found");
|
||||||
|
|
||||||
|
let res = await new Promise<string[]>((yes, no) => {
|
||||||
|
|
||||||
|
const stream = db.collections.createKeyStream({
|
||||||
|
keyAsBuffer: false,
|
||||||
|
limit: 1000
|
||||||
|
});
|
||||||
|
let res = [];
|
||||||
|
stream.on("data", (key: string) => {
|
||||||
|
res.push(key);
|
||||||
|
})
|
||||||
|
|
||||||
|
stream.on("error", no);
|
||||||
|
stream.on("end", () => yes(res))
|
||||||
|
})
|
||||||
|
|
||||||
|
if (ctx.query.view) {
|
||||||
|
return getTable("Databases", res, ctx);
|
||||||
|
} else {
|
||||||
|
ctx.body = res;
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
AdminRoute.get("/collections/cleanup", async ctx => {
|
||||||
|
const { database } = ctx.query;
|
||||||
|
let db = DatabaseManager.getDatabase(database);
|
||||||
|
if (!db)
|
||||||
|
throw new BadRequestError("Database not found");
|
||||||
|
|
||||||
|
let deleted = await db.runCleanup();
|
||||||
|
if (ctx.query.view) {
|
||||||
|
return getTable("Databases", deleted, ctx);
|
||||||
|
} else {
|
||||||
|
ctx.body = deleted;
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
AdminRoute.get("/database/new", getForm("/v1/admin/database", "New/Change Database", {
|
AdminRoute.get("/database/new", getForm("/v1/admin/database", "New/Change Database", {
|
||||||
name: { label: "Name", type: "text", },
|
name: { label: "Name", type: "text", },
|
||||||
accesskey: { label: "Access Key", type: "text" },
|
accesskey: { label: "Access Key", type: "text" },
|
||||||
|
@ -1,5 +1,62 @@
|
|||||||
import * as Router from "koa-router";
|
import * as Router from "koa-router";
|
||||||
import AdminRoute from "./admin";
|
import AdminRoute from "./admin";
|
||||||
|
import { DatabaseManager } from "../../database/database";
|
||||||
|
import {
|
||||||
|
NotFoundError,
|
||||||
|
NoPermissionError,
|
||||||
|
BadRequestError
|
||||||
|
} from "../helper/errors";
|
||||||
|
import Logging from "@hibas123/nodelogging";
|
||||||
|
import Session from "../../database/session";
|
||||||
|
import nanoid = require("nanoid");
|
||||||
|
import { verifyJWT } from "../../helper/jwt";
|
||||||
|
import { QueryError } from "../../database/query";
|
||||||
const V1 = new Router({ prefix: "/v1" });
|
const V1 = new Router({ prefix: "/v1" });
|
||||||
|
|
||||||
V1.use("/admin", AdminRoute.routes(), AdminRoute.allowedMethods());
|
V1.use("/admin", AdminRoute.routes(), AdminRoute.allowedMethods());
|
||||||
export default V1;
|
|
||||||
|
V1.post("/db/:database/query", async ctx => {
|
||||||
|
const { database } = ctx.params;
|
||||||
|
const { accesskey, authkey, rootkey } = ctx.query;
|
||||||
|
|
||||||
|
const query = ctx.request.body;
|
||||||
|
if (!query) {
|
||||||
|
throw new BadRequestError("Query not defined!");
|
||||||
|
}
|
||||||
|
|
||||||
|
const session = new Session(nanoid());
|
||||||
|
const db = DatabaseManager.getDatabase(database);
|
||||||
|
if (!db) {
|
||||||
|
throw new NotFoundError("Database not found!");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (db.accesskey) {
|
||||||
|
if (!accesskey || accesskey !== db.accesskey) {
|
||||||
|
throw new NoPermissionError("Invalid Access Key");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (authkey && db.publickey) {
|
||||||
|
let res = await verifyJWT(authkey, db.publickey);
|
||||||
|
if (!res || !res.uid) {
|
||||||
|
throw new BadRequestError("Invalid JWT");
|
||||||
|
} else {
|
||||||
|
session.uid = res.uid;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (rootkey && db.rootkey) {
|
||||||
|
if (rootkey === db.rootkey) {
|
||||||
|
session.root = true;
|
||||||
|
Logging.warning(`Somebody logged into ${database} via rootkey`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.body = await db.run([query], session).catch(err => {
|
||||||
|
if (err instanceof QueryError) {
|
||||||
|
throw new BadRequestError(err.message);
|
||||||
|
}
|
||||||
|
throw err;
|
||||||
|
});
|
||||||
|
});
|
||||||
|
export default V1;
|
||||||
|
112
views/admin.html
Normal file
112
views/admin.html
Normal file
@ -0,0 +1,112 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en">
|
||||||
|
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<meta http-equiv="X-UA-Compatible" content="ie=edge">
|
||||||
|
<title>Admin Interface</title>
|
||||||
|
<link rel="stylesheet" href="https://unpkg.com/@hibas123/theme/out/base.css">
|
||||||
|
<link rel="stylesheet" href="https://unpkg.com/@hibas123/theme/out/light.css">
|
||||||
|
|
||||||
|
<script src="https://unpkg.com/handlebars/dist/handlebars.min.js"></script>
|
||||||
|
|
||||||
|
<style>
|
||||||
|
#message {
|
||||||
|
visibility: hidden;
|
||||||
|
background-color: lightgreen;
|
||||||
|
border: 1px solid lime;
|
||||||
|
border-radius: .5rem;
|
||||||
|
padding: 1rem;
|
||||||
|
font-size: 1.5rem;
|
||||||
|
margin-bottom: 1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.grid {
|
||||||
|
display: grid;
|
||||||
|
height: 100vh;
|
||||||
|
grid-template-columns: 360px auto;
|
||||||
|
}
|
||||||
|
|
||||||
|
#content {
|
||||||
|
position: absolute;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
border: 0;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
|
||||||
|
<body>
|
||||||
|
<div class="grid">
|
||||||
|
<div style="border-right: 1px solid darkgrey; padding: 1rem;">
|
||||||
|
<h2>Navigation: </h2>
|
||||||
|
<ul class="list list-clickable">
|
||||||
|
<li onclick="loadView('settings');">Settings</li>
|
||||||
|
<li onclick="loadView('database', {full:true});">Databases</li>
|
||||||
|
<li onclick="loadView('database/new');">New Database</li>
|
||||||
|
</ul>
|
||||||
|
Databases:
|
||||||
|
<div id="dbs" class="list list-clickable" style="margin: 1rem;"></div>
|
||||||
|
|
||||||
|
</div>
|
||||||
|
<div style="position:relative;">
|
||||||
|
<iframe id="content"></iframe>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<template>
|
||||||
|
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
const key = new URL(window.location.href).searchParams.get("key");
|
||||||
|
const content = document.getElementById("content");
|
||||||
|
const base = new URL(window.location.href).host;
|
||||||
|
|
||||||
|
function getUrl(name, params, view = true) {
|
||||||
|
const url = new URL(window.location.href);
|
||||||
|
url.pathname = "/v1/admin/" + name;
|
||||||
|
for (let key in params || {})
|
||||||
|
url.searchParams.set(key, params[key]);
|
||||||
|
|
||||||
|
url.searchParams.set("key", key);
|
||||||
|
if (view)
|
||||||
|
url.searchParams.set("view", "true");
|
||||||
|
|
||||||
|
return url.href;
|
||||||
|
}
|
||||||
|
|
||||||
|
function loadView(name, params) {
|
||||||
|
content.src = getUrl(name, params);
|
||||||
|
}
|
||||||
|
|
||||||
|
loadView("settings")
|
||||||
|
|
||||||
|
const dbsul = document.getElementById("dbs");
|
||||||
|
function reloadDBs() {
|
||||||
|
fetch(getUrl("database", {}, false))
|
||||||
|
.then(res => res.json())
|
||||||
|
.then(databases => databases.map(database => `
|
||||||
|
<div class="card margin elv-4">
|
||||||
|
<h3>${database}</h3>
|
||||||
|
<button class=btn onclick="loadView('data', {database:'${database}'})">Data</button>
|
||||||
|
<button class=btn onclick="loadView('collections', {database:'${database}'})">Collections</button>
|
||||||
|
<button class=btn onclick="loadView('collections/cleanup', {database:'${database}'})">Clean</button>
|
||||||
|
</div>`
|
||||||
|
))
|
||||||
|
.then(d => d.join("\n"))
|
||||||
|
.then(d => dbsul.innerHTML = d)
|
||||||
|
.catch(console.error)
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
reloadDBs();
|
||||||
|
setInterval(reloadDBs, 5000);
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
|
||||||
|
</html>
|
Reference in New Issue
Block a user