first commit

yUnreal · May 20, 2024 · c8da569 · c8da569
1 parent adb0aa6
commit c8da569
Show file tree

Hide file tree

Showing 30 changed files with 1,099 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,4 @@
+# Unnecessary files
 node_modules
-# Keep environment variables out of version control
-.env
+.editorconfig
+index.test.ts
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
-    "name": "wuue",
+    "name": "wuuedb",
     "version": "0.0.1",
-    "description": "TypeScript schema validation",
+    "description": "Wuue db",
     "main": "src/index.ts",
     "scripts": {
         "test": "tsx index.test.ts",
@@ -18,12 +18,19 @@
     },
     "license": "ISC",
     "devDependencies": {
+        "@types/lodash.isequal": "^4.5.8",
         "@typescript-eslint/eslint-plugin": "^5.60.1",
         "@typescript-eslint/parser": "^5.60.1",
         "eslint": "^8.43.0",
         "eslint-config-prettier": "^8.8.0",
         "prettier": "^3.0.0",
         "tsx": "^4.7.0",
         "typescript": "^5.1.6"
+    },
+    "dependencies": {
+        "bson": "^6.7.0",
+        "is-plain-obj": "^4.1.0",
+        "lodash.isequal": "^4.5.0",
+        "uuid-random": "^1.3.2"
     }
 }
diff --git a/src/drivers/JSONBasedDriver.ts b/src/drivers/JSONBasedDriver.ts
@@ -0,0 +1,31 @@
+import { basename } from 'path';
+import { DriverError } from '../errors/DriverError';
+import { writeFileSync, readFileSync, existsSync } from 'fs';
+
+export class JSONBasedDriver<
+    T extends Record<string, unknown> = Record<string, unknown>,
+> {
+    public constructor(public readonly path: string) {
+        if (!path.endsWith('.json'))
+            throw new DriverError('Invalid path', this);
+        if (!existsSync(path)) writeFileSync(path, '{}');
+    }
+
+    protected getPathName() {
+        return basename(this.path, '.json');
+    }
+
+    public get data() {
+        const data = readFileSync(this.path, 'utf8');
+
+        return <T>JSON.parse(data);
+    }
+
+    public update(fn: (data: T) => T) {
+        const data = fn(this.data);
+
+        writeFileSync(this.path, JSON.stringify(data, null, '\t'));
+
+        return this;
+    }
+}
diff --git a/src/errors/DriverError.ts b/src/errors/DriverError.ts
@@ -0,0 +1,10 @@
+import { JSONBasedDriver } from '../drivers/JSONBasedDriver';
+
+export class DriverError extends Error {
+    public constructor(
+        message: string,
+        public driver: JSONBasedDriver
+    ) {
+        super(message);
+    }
+}
diff --git a/src/errors/QueryError.ts b/src/errors/QueryError.ts
@@ -0,0 +1,8 @@
+export class QueryError extends Error {
+    public constructor(
+        message: string,
+        public query: object
+    ) {
+        super(message);
+    }
+}
diff --git a/src/errors/SchemaError.ts b/src/errors/SchemaError.ts
@@ -0,0 +1,10 @@
+import { Schema } from '../structs/Schema';
+
+export class SchemaError extends Error {
+    public constructor(
+        message: string,
+        public schema: Schema<Record<string, unknown>>
+    ) {
+        super(message);
+    }
+}
diff --git a/src/managers/BaseManager.ts b/src/managers/BaseManager.ts
@@ -0,0 +1,3 @@
+export abstract class BaseManager<K, V> {
+    public cache = new Map<K, V>();
+}
diff --git a/src/managers/QueryManager.ts b/src/managers/QueryManager.ts
@@ -0,0 +1,61 @@
+import { QueryError } from '../errors/QueryError';
+import { Model } from '../structs/Model';
+import { Query } from '../structs/Query';
+import { Schema } from '../structs/Schema';
+import { QueryOptions } from '../typings/query';
+import { AnyObject } from '../typings/utils';
+import { BaseManager } from './BaseManager';
+
+export class QueryManager<T extends AnyObject = AnyObject> extends BaseManager<
+    string,
+    Query<T>
+> {
+    public constructor(public model: Model<Schema<T>>) {
+        super();
+    }
+
+    public resolve(query: QueryOptions<T> | string) {
+        if (typeof query === 'object') return query;
+
+        const fetchedQuery = this.get(query);
+
+        if (!fetchedQuery)
+            throw new QueryError(`Uknown query "${query}"`, this);
+
+        return fetchedQuery.query;
+    }
+
+    public delete(query: string) {
+        return this.cache.delete(query);
+    }
+
+    public create(name: string, options?: QueryOptions<T>) {
+        const query = new Query(this.model, name, options);
+
+        this.cache.set(name, query);
+
+        return query;
+    }
+
+    public get(name: string) {
+        return this.cache.get(name) ?? null;
+    }
+
+    public merge(first: Query<T>, second: Query<T>) {
+        first.query = { ...first.query, ...second.query };
+
+        return first;
+    }
+
+    public edit(name: string, options: QueryOptions<T>) {
+        const query = this.get(name);
+
+        if (!query) throw new QueryError(`Unknown query "${name}"`, options);
+
+        query.query = { ...query.query, ...options };
+
+        this.cache.set(name, query);
+
+        return query;
+    }
+}
diff --git a/src/structs/Doc.ts b/src/structs/Doc.ts
@@ -0,0 +1,32 @@
+import { Projection, UpdateOneOptions } from '../typings/query';
+import { Model } from './Model';
+import { Schema } from './Schema';
+
+export class Doc<T extends Record<string, unknown>> {
+    public constructor(
+        public _doc: T,
+        public model: Model<Schema<T>>
+    ) {}
+
+    public get id() {
+        return <string>this._doc.id;
+    }
+
+    public fetch(projection?: Projection<T>) {
+        return <Doc<T>>(
+            this.model.findUnique({ query: { id: this.id }, projection })
+        );
+    }
+
+    public delete() {
+        this.model.deleteOne({ query: { id: this.id } });
+    }
+
+    public update(options: UpdateOneOptions<T>) {
+        return this.model.updateOne({ query: { id: this.id } }, options);
+    }
+
+    public save() {
+        return this.update({ Set: this._doc });
+    }
+}