@snowtop/ent 0.1.0-alpha → 0.1.0-alpha12

package/testutils/fake_data/fake_user.js

@@ -26,7 +26,18 @@ class FakeUser {
     constructor(viewer, data) {
         this.viewer = viewer;
         this.nodeType = const_1.NodeType.FakeUser;
-        this.privacyPolicy = {
+        this.data = data;
+        this.id = data.id;
+        this.createdAt = (0, convert_1.convertDate)(data.created_at);
+        this.updatedAt = (0, convert_1.convertDate)(data.updated_at);
+        this.firstName = data.first_name;
+        this.lastName = data.last_name;
+        this.emailAddress = data.email_address;
+        this.phoneNumber = data.phone_number;
+        this.password = data.password;
+    }
+    getPrivacyPolicy() {
+        return {
             rules: [
                 privacy_1.AllowIfViewerRule,
                 //can view user if friends
@@ -48,15 +59,6 @@ class FakeUser {
                 privacy_1.AlwaysDenyRule,
             ],
         };
-        this.data = data;
-        this.id = data.id;
-        this.createdAt = (0, convert_1.convertDate)(data.created_at);
-        this.updatedAt = (0, convert_1.convertDate)(data.updated_at);
-        this.firstName = data.first_name;
-        this.lastName = data.last_name;
-        this.emailAddress = data.email_address;
-        this.phoneNumber = data.phone_number;
-        this.password = data.password;
     }
     static getFields() {
         return [
@@ -89,31 +91,15 @@ class FakeUser {
     }
 }
 exports.FakeUser = FakeUser;
-class FakeUserSchema extends schema_1.BaseEntSchema {
-    constructor() {
-        super(...arguments);
-        this.ent = FakeUser;
-        this.fields = [
-            (0, schema_1.StringType)({
-                name: "firstName",
-            }),
-            (0, schema_1.StringType)({
-                name: "lastName",
-            }),
-            (0, schema_1.StringType)({
-                name: "emailAddress",
-            }),
-            (0, schema_1.StringType)({
-                name: "phoneNumber",
-            }),
-            (0, schema_1.StringType)({
-                name: "password",
-                nullable: true,
-            }),
-        ];
-    }
-}
-exports.FakeUserSchema = FakeUserSchema;
+exports.FakeUserSchema = (0, builder_1.getBuilderSchemaFromFields)({
+    firstName: (0, schema_1.StringType)(),
+    lastName: (0, schema_1.StringType)(),
+    emailAddress: (0, schema_1.StringType)(),
+    phoneNumber: (0, schema_1.StringType)(),
+    password: (0, schema_1.StringType)({
+        nullable: true,
+    }),
+}, FakeUser);
 function getUserBuilder(viewer, input) {
     const action = getUserAction(viewer, input);
     return action.builder;
@@ -124,7 +110,7 @@ function getUserAction(viewer, input) {
     for (const key in input) {
         m.set(key, input[key]);
     }
-    const action = new builder_1.SimpleAction(viewer, new FakeUserSchema(), m);
+    const action = new builder_1.SimpleAction(viewer, exports.FakeUserSchema, m);
     action.viewerForEntLoad = (data) => {
         // load the created ent using a VC of the newly created user.
         return new viewer_1.IDViewer(data.id);

package/testutils/fake_data/test_helpers.js

@@ -107,7 +107,7 @@ async function createUserPlusFriendRequests(input, slice) {
         return createTestUser(input);
     }));
     expect(friendRequests.length).toBe(userInputs.length);
-    await addEdge(user, new _1.FakeUserSchema(), _1.EdgeType.UserToFriendRequests, true, ...friendRequests);
+    await addEdge(user, _1.FakeUserSchema, _1.EdgeType.UserToFriendRequests, true, ...friendRequests);
     return [user, friendRequests];
 }
 exports.createUserPlusFriendRequests = createUserPlusFriendRequests;

package/tsc/ast.d.ts

@@ -0,0 +1,21 @@
+import ts from "typescript";
+export declare function getPreText(fileContents: string, node: ts.Node, sourceFile: ts.SourceFile): string;
+export interface ClassInfo {
+    extends?: string;
+    comment: string;
+    name: string;
+    export?: boolean;
+    default?: boolean;
+    implements?: string[];
+    wrapClassContents(inner: string): string;
+}
+export declare function getClassInfo(fileContents: string, sourceFile: ts.SourceFile, node: ts.ClassDeclaration): ClassInfo | undefined;
+declare type transformImportFn = (imp: string) => string;
+interface transformOpts {
+    removeImports?: string[];
+    newImports?: string[];
+    transform?: transformImportFn;
+}
+export declare function transformImport(fileContents: string, importNode: ts.ImportDeclaration, sourceFile: ts.SourceFile, opts?: transformOpts): string | undefined;
+export declare function updateImportPath(fileContents: string, importNode: ts.ImportDeclaration, sourceFile: ts.SourceFile, newPath: string): string | undefined;
+export {};

package/tsc/ast.js

@@ -0,0 +1,154 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.updateImportPath = exports.transformImport = exports.getClassInfo = exports.getPreText = void 0;
+const typescript_1 = __importDefault(require("typescript"));
+function getPreText(fileContents, node, sourceFile) {
+    return fileContents.substring(node.getFullStart(), node.getStart(sourceFile));
+}
+exports.getPreText = getPreText;
+function getClassInfo(fileContents, sourceFile, node) {
+    let className = node.name?.text;
+    let classExtends;
+    let impl = [];
+    if (node.heritageClauses) {
+        for (const hc of node.heritageClauses) {
+            switch (hc.token) {
+                case typescript_1.default.SyntaxKind.ImplementsKeyword:
+                    for (const type of hc.types) {
+                        impl.push(type.expression.getText(sourceFile));
+                    }
+                    break;
+                case typescript_1.default.SyntaxKind.ExtendsKeyword:
+                    // can only extend one class
+                    for (const type of hc.types) {
+                        const text = type.expression.getText(sourceFile);
+                        classExtends = text;
+                    }
+                    break;
+            }
+        }
+    }
+    // we probably still don't need all of this...
+    if (!className || !node.heritageClauses || !classExtends) {
+        return undefined;
+    }
+    let hasExport = false;
+    let hasDefault = false;
+    let comment = getPreText(fileContents, node, sourceFile);
+    const wrapClassContents = (inner) => {
+        let ret = `${comment}`;
+        if (hasExport) {
+            ret += "export ";
+        }
+        if (hasDefault) {
+            ret += "default ";
+        }
+        ret += `class ${className} `;
+        if (classExtends) {
+            ret += `extends ${classExtends} `;
+        }
+        if (impl.length) {
+            ret += `implements ${impl.join(", ")}`;
+        }
+        return `${ret}{
+      ${inner}
+    }`;
+    };
+    if (node.modifiers) {
+        for (const mod of node.modifiers) {
+            const text = mod.getText(sourceFile);
+            if (text === "export") {
+                hasExport = true;
+            }
+            else if (text === "default") {
+                hasDefault = true;
+            }
+        }
+    }
+    return {
+        name: className,
+        extends: classExtends,
+        comment,
+        implements: impl,
+        wrapClassContents,
+        export: hasExport,
+        default: hasDefault,
+    };
+}
+exports.getClassInfo = getClassInfo;
+function transformImport(fileContents, importNode, sourceFile, opts) {
+    // remove quotes too
+    const text = importNode.moduleSpecifier.getText(sourceFile).slice(1, -1);
+    if (text !== "@snowtop/ent" &&
+        text !== "@snowtop/ent/schema" &&
+        text !== "@snowtop/ent/schema/") {
+        return;
+    }
+    const importText = importNode.importClause?.getText(sourceFile) || "";
+    const start = importText.indexOf("{");
+    const end = importText.lastIndexOf("}");
+    if (start === -1 || end === -1) {
+        return;
+    }
+    const imports = importText
+        .substring(start + 1, end)
+        //    .trim()
+        .split(",");
+    let removeImportsMap = {};
+    if (opts?.removeImports) {
+        opts.removeImports.forEach((imp) => (removeImportsMap[imp] = true));
+    }
+    let finalImports = new Set();
+    for (let i = 0; i < imports.length; i++) {
+        let imp = imports[i].trim();
+        if (imp === "") {
+            continue;
+        }
+        if (opts?.transform) {
+            imp = opts.transform(imp);
+        }
+        if (removeImportsMap[imp]) {
+            continue;
+        }
+        finalImports.add(imp);
+    }
+    if (opts?.newImports) {
+        opts.newImports.forEach((imp) => finalImports.add(imp));
+    }
+    const comment = getPreText(fileContents, importNode, sourceFile);
+    return (comment +
+        "import " +
+        importText.substring(0, start + 1) +
+        Array.from(finalImports).join(", ") +
+        importText.substring(end) +
+        ' from "' +
+        text +
+        '";');
+}
+exports.transformImport = transformImport;
+function updateImportPath(fileContents, importNode, sourceFile, newPath) {
+    const comment = getPreText(fileContents, importNode, sourceFile);
+    // all this copied from above...
+    const importText = importNode.importClause?.getText(sourceFile) || "";
+    const start = importText.indexOf("{");
+    const end = importText.lastIndexOf("}");
+    if (start === -1 || end === -1) {
+        return;
+    }
+    const imports = importText
+        .substring(start + 1, end)
+        //    .trim()
+        .split(",");
+    return (comment +
+        "import " +
+        importText.substring(0, start + 1) +
+        Array.from(imports).join(", ") +
+        importText.substring(end) +
+        ' from "' +
+        newPath +
+        '";');
+}
+exports.updateImportPath = updateImportPath;

package/tsc/compilerOptions.d.ts

@@ -0,0 +1,8 @@
+import ts from "typescript";
+export declare function readCompilerOptions(filePath: string): ts.CompilerOptions;
+export declare function getTarget(target?: string): ts.ScriptTarget;
+export declare function getTargetFromCurrentDir(): ts.ScriptTarget;
+export declare function createSourceFile(target: ts.ScriptTarget, file: string): {
+    contents: string;
+    sourceFile: ts.SourceFile;
+};

package/tsc/compilerOptions.js

@@ -0,0 +1,100 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createSourceFile = exports.getTargetFromCurrentDir = exports.getTarget = exports.readCompilerOptions = void 0;
+const fs = __importStar(require("fs"));
+const json5_1 = __importDefault(require("json5"));
+const typescript_1 = __importDefault(require("typescript"));
+const path = __importStar(require("path"));
+function findTSConfigFile(filePath) {
+    while (filePath != "/") {
+        let configPath = `${filePath}/tsconfig.json`;
+        if (fs.existsSync(configPath)) {
+            return configPath;
+        }
+        filePath = path.join(filePath, "..");
+    }
+    return null;
+}
+function readCompilerOptions(filePath) {
+    let configPath = findTSConfigFile(filePath);
+    if (!configPath) {
+        return {};
+    }
+    let json = {};
+    try {
+        json = json5_1.default.parse(fs.readFileSync(configPath, {
+            encoding: "utf8",
+        }));
+    }
+    catch (e) {
+        console.error("couldn't read tsconfig.json file");
+    }
+    let options = json["compilerOptions"] || {};
+    // @ts-ignore
+    if (options.moduleResolution === "node") {
+        options.moduleResolution = typescript_1.default.ModuleResolutionKind.NodeJs;
+    }
+    return options;
+}
+exports.readCompilerOptions = readCompilerOptions;
+function getTarget(target) {
+    switch (target?.toLowerCase()) {
+        case "es2015":
+            return typescript_1.default.ScriptTarget.ES2015;
+        case "es2016":
+            return typescript_1.default.ScriptTarget.ES2016;
+        case "es2017":
+            return typescript_1.default.ScriptTarget.ES2017;
+        case "es2018":
+            return typescript_1.default.ScriptTarget.ES2018;
+        case "es2019":
+            return typescript_1.default.ScriptTarget.ES2019;
+        case "es2020":
+            return typescript_1.default.ScriptTarget.ES2020;
+        case "es2021":
+            return typescript_1.default.ScriptTarget.ES2021;
+        case "es3":
+            return typescript_1.default.ScriptTarget.ES3;
+        case "es5":
+            return typescript_1.default.ScriptTarget.ES5;
+        case "esnext":
+            return typescript_1.default.ScriptTarget.ESNext;
+        default:
+            return typescript_1.default.ScriptTarget.ESNext;
+    }
+}
+exports.getTarget = getTarget;
+function getTargetFromCurrentDir() {
+    const options = readCompilerOptions(".");
+    return getTarget(options.target?.toString());
+}
+exports.getTargetFromCurrentDir = getTargetFromCurrentDir;
+function createSourceFile(target, file) {
+    let contents = fs.readFileSync(file).toString();
+    // go through the file and print everything back if not starting immediately after other position
+    const sourceFile = typescript_1.default.createSourceFile(file, contents, target, false, typescript_1.default.ScriptKind.TS);
+    return { contents, sourceFile };
+}
+exports.createSourceFile = createSourceFile;