@snowtop/ent 0.0.23 → 0.0.28-alpha.2

package/schema/field.d.ts

@@ -1,4 +1,4 @@
-import { Type, Field, FieldOptions, PolymorphicOptions, ForeignKey } from "./schema";
+import { Field, FieldOptions, ForeignKey, PolymorphicOptions, Type } from "./schema";
 export declare abstract class BaseField {
     name: string;
     nullable?: boolean;
@@ -21,10 +21,23 @@ export declare class UUIDField extends BaseField implements Field {
     constructor(options: FieldOptions);
 }
 export declare function UUIDType(options: FieldOptions): UUIDField;
+export interface IntegerOptions extends FieldOptions {
+    min?: number;
+    max?: number;
+}
 export declare class IntegerField extends BaseField implements Field {
     type: Type;
+    private validators;
+    private options;
+    constructor(options?: IntegerOptions);
+    getOptions(): IntegerOptions;
+    private handleOptions;
+    min(l: number): this;
+    max(l: number): this;
+    valid(val: any): boolean;
+    validate(validator: (str: number) => boolean): this;
 }
-export declare function IntegerType(options: FieldOptions): IntegerField;
+export declare function IntegerType(options: IntegerOptions): IntegerField;
 export declare class BigIntegerField extends BaseField implements Field {
     type: Type;
 }
@@ -53,23 +66,24 @@ export declare class StringField extends BaseField implements Field {
     type: Type;
     private validators;
     private formatters;
+    private options;
     constructor(options?: StringOptions);
     getOptions(): StringOptions;
     private handleOptions;
-    minLen(l: number): StringField;
-    maxLen(l: number): StringField;
-    length(l: number): StringField;
+    minLen(l: number): this;
+    maxLen(l: number): this;
+    length(l: number): this;
     valid(val: any): boolean;
     format(val: any): any;
-    validate(validator: (str: string) => boolean): StringField;
-    formatter(formatter: (str: string) => string): StringField;
-    match(pattern: string | RegExp): StringField;
-    doesNotMatch(pattern: string | RegExp): StringField;
-    toLowerCase(): StringField;
-    toUpperCase(): StringField;
-    trim(): StringField;
-    trimLeft(): StringField;
-    trimRight(): StringField;
+    validate(validator: (str: string) => boolean): this;
+    formatter(formatter: (str: string) => string): this;
+    match(pattern: string | RegExp): this;
+    doesNotMatch(pattern: string | RegExp): this;
+    toLowerCase(): this;
+    toUpperCase(): this;
+    trim(): this;
+    trimLeft(): this;
+    trimRight(): this;
 }
 export declare function StringType(options: StringOptions): StringField;
 export interface TimestampOptions extends FieldOptions {

package/schema/field.js

@@ -20,10 +20,10 @@ var __importStar = (this && this.__importStar) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.UUIDListType = exports.EnumListType = exports.DateListType = exports.TimetzListType = exports.TimeListType = exports.TimestamptzListType = exports.TimestampListType = exports.BooleanListType = exports.BigIntegerListType = exports.FloatListType = exports.IntegerListType = exports.IntListType = exports.StringListType = exports.ListField = exports.EnumType = exports.EnumField = exports.DateType = exports.DateField = exports.TimetzType = exports.TimeType = exports.TimeField = exports.leftPad = exports.TimestamptzType = exports.TimestampType = exports.TimestampField = exports.StringType = exports.StringField = exports.BooleanType = exports.BooleanField = exports.FloatType = exports.FloatField = exports.BigIntegerType = exports.BigIntegerField = exports.IntegerType = exports.IntegerField = exports.UUIDType = exports.UUIDField = exports.BaseField = void 0;
-const schema_1 = require("./schema");
-const snake_case_1 = require("snake-case");
 const luxon_1 = require("luxon");
+const snake_case_1 = require("snake-case");
 const db_1 = __importStar(require("../core/db"));
+const schema_1 = require("./schema");
 class BaseField {
     logValue(val) {
         if (this.sensitive) {
@@ -89,14 +89,52 @@ function UUIDType(options) {
 }
 exports.UUIDType = UUIDType;
 class IntegerField extends BaseField {
-    constructor() {
-        super(...arguments);
+    constructor(options) {
+        super();
         this.type = { dbType: schema_1.DBType.Int };
+        this.validators = [];
+        this.options = { name: "field" };
+        // for legacy callers
+        this.handleOptions(options || this.options);
+    }
+    getOptions() {
+        return this.options;
+    }
+    handleOptions(options) {
+        const params = {
+            min: this.min,
+            max: this.max,
+        };
+        for (const k in params) {
+            const v = options[k];
+            if (v !== undefined) {
+                params[k].apply(this, [v]);
+            }
+        }
+        this.options = options;
+    }
+    min(l) {
+        return this.validate((val) => val >= l);
+    }
+    max(l) {
+        return this.validate((val) => val <= l);
+    }
+    valid(val) {
+        for (const validator of this.validators) {
+            if (!validator(val)) {
+                return false;
+            }
+        }
+        return true;
+    }
+    validate(validator) {
+        this.validators.push(validator);
+        return this;
     }
 }
 exports.IntegerField = IntegerField;
 function IntegerType(options) {
-    let result = new IntegerField();
+    let result = new IntegerField(options);
     return Object.assign(result, options);
 }
 exports.IntegerType = IntegerType;
@@ -142,11 +180,12 @@ class StringField extends BaseField {
         this.type = { dbType: schema_1.DBType.String };
         this.validators = [];
         this.formatters = [];
+        this.options = { name: "field" };
         // for legacy callers
         this.handleOptions(options || { name: "field" });
     }
     getOptions() {
-        return this.getOptions;
+        return this.options;
     }
     handleOptions(options) {
         const noParams = {
@@ -174,6 +213,7 @@ class StringField extends BaseField {
                 noParams[k].apply(this);
             }
         }
+        this.options = options;
     }
     minLen(l) {
         return this.validate((val) => {

package/scripts/custom_graphql.js

@@ -25,6 +25,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const glob_1 = __importDefault(require("glob"));
 const minimist_1 = __importDefault(require("minimist"));
+const graphql_1 = require("../graphql/graphql");
 const readline = __importStar(require("readline"));
 const path = __importStar(require("path"));
 const fs = __importStar(require("fs"));
@@ -52,7 +53,17 @@ async function readInputs() {
         });
     });
 }
-async function captureCustom(filePath) {
+async function captureCustom(filePath, filesCsv) {
+    if (filesCsv !== undefined) {
+        let files = filesCsv.split(",");
+        for (let i = 0; i < files.length; i++) {
+            // TODO fix. we have "src" in the path we get here
+            files[i] = path.join(filePath, "..", files[i]);
+        }
+        await requireFiles(files);
+        return;
+    }
+    // TODO delete all of this eventually
     // TODO configurable paths eventually
     // for now only files that are in the include path of the roots are allowed
     const rootFiles = [
@@ -81,18 +92,28 @@ async function captureCustom(filePath) {
         ignore: ignore,
     });
     const files = rootFiles.concat(customGQLResolvers, customGQLMutations);
-    //console.log(files);
-    let promises = [];
-    files.forEach((file) => {
+    await requireFiles(files);
+}
+async function requireFiles(files) {
+    await Promise.all(files.map(async (file) => {
         if (fs.existsSync(file)) {
-            promises.push(require(file));
+            try {
+                await require(file);
+            }
+            catch (e) {
+                throw new Error(`${e.message} loading ${file}`);
+            }
+        }
+        else {
+            throw new Error(`file ${file} doesn't exist`);
         }
+    })).catch((err) => {
+        throw new Error(err);
     });
-    await Promise.all(promises);
 }
 async function parseImports(filePath) {
     // only do graphql files...
-    return (0, imports_1.parseCustomInput)(path.join(filePath, "graphql"), {
+    return (0, imports_1.parseCustomImports)(path.join(filePath, "graphql"), {
         ignore: ["**/generated/**", "**/tests/**"],
     });
 }
@@ -110,6 +131,24 @@ function findGraphQLPath(filePath) {
     return undefined;
 }
 async function main() {
+    // known custom types that are not required
+    // scalar Time will successfully parse
+    // TODO add a mechanism for a type we should check for to
+    // add this to schema.gql
+    // e.g. scalar Time will be added to schemas which don't use it (yet)
+    // TODO foreign non-scalars??
+    (0, graphql_1.addCustomType)({
+        importPath: "../graphql/scalars/time",
+        type: "GraphQLTime",
+    });
+    // if not in the schema, will be ignored
+    // something like GraphQLUpload gotten via gqlArg({type: gqlFileUpload})
+    // these 2 need this because they're added by the schema
+    // if this list grows too long, need to build this into golang types and passed here
+    (0, graphql_1.addCustomType)({
+        importPath: "graphql-type-json",
+        type: "GraphQLJSON",
+    });
     const options = (0, minimist_1.default)(process.argv.slice(2));
     if (!options.path) {
         throw new Error("path required");
@@ -126,7 +165,7 @@ async function main() {
     GQLCapture.enable(true);
     const [inputsRead, _, imports] = await Promise.all([
         readInputs(),
-        captureCustom(options.path),
+        captureCustom(options.path, options.files),
         parseImports(options.path),
     ]);
     const { nodes, nodesMap } = inputsRead;

package/testutils/builder.d.ts

@@ -9,6 +9,7 @@ export declare class User implements Ent {
     accountID: string;
     nodeType: string;
     privacyPolicy: import("../core/base").PrivacyPolicy;
+    firstName: string;
     constructor(viewer: Viewer, data: Data);
 }
 export declare class Event implements Ent {

package/testutils/builder.js

@@ -23,6 +23,7 @@ class User {
         this.data.created_at = (0, convert_1.convertDate)(data.created_at);
         this.data.updated_at = (0, convert_1.convertDate)(data.updated_at);
         this.id = data.id;
+        this.firstName = data.first_name;
     }
 }
 exports.User = User;

package/testutils/fake_data/events_query.d.ts

@@ -1,25 +1,30 @@
-import { Viewer } from "../../core/base";
+import { ID, Viewer } from "../../core/base";
 import { AssocEdge } from "../../core/ent";
 import { AssocEdgeQueryBase, EdgeQuerySource } from "../../core/query/assoc_query";
 import { FakeUser } from "./fake_user";
 import { FakeEvent } from "./internal";
 export declare class EventToAttendeesQuery extends AssocEdgeQueryBase<FakeEvent, FakeUser, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeEvent>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeEvent>): EventToAttendeesQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeEvent, FakeUser>);
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeEvent, FakeUser>): EventToAttendeesQuery;
+    sourceEnt(id: ID): Promise<FakeEvent | null>;
 }
 export declare class EventToInvitedQuery extends AssocEdgeQueryBase<FakeEvent, FakeUser, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeEvent>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeEvent>): EventToInvitedQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeEvent, FakeUser>);
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeEvent, FakeUser>): EventToInvitedQuery;
+    sourceEnt(id: ID): Promise<FakeEvent | null>;
 }
 export declare class EventToDeclinedQuery extends AssocEdgeQueryBase<FakeEvent, FakeUser, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeEvent>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeEvent>): EventToDeclinedQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeEvent, FakeUser>);
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeEvent, FakeUser>): EventToDeclinedQuery;
+    sourceEnt(id: ID): Promise<FakeEvent | null>;
 }
 export declare class EventToMaybeQuery extends AssocEdgeQueryBase<FakeEvent, FakeUser, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeEvent>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeEvent>): EventToMaybeQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeEvent, FakeUser>);
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeEvent, FakeUser>): EventToMaybeQuery;
+    sourceEnt(id: ID): Promise<FakeEvent | null>;
 }
 export declare class EventToHostsQuery extends AssocEdgeQueryBase<FakeEvent, FakeUser, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeEvent>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeEvent>): EventToHostsQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeEvent, FakeUser>);
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeEvent, FakeUser>): EventToHostsQuery;
+    sourceEnt(id: ID): Promise<FakeEvent | null>;
 }

package/testutils/fake_data/events_query.js

@@ -14,6 +14,9 @@ class EventToAttendeesQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new EventToAttendeesQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeEvent.load(this.viewer, id);
+    }
 }
 exports.EventToAttendeesQuery = EventToAttendeesQuery;
 class EventToInvitedQuery extends assoc_query_1.AssocEdgeQueryBase {
@@ -23,6 +26,9 @@ class EventToInvitedQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new EventToInvitedQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeEvent.load(this.viewer, id);
+    }
 }
 exports.EventToInvitedQuery = EventToInvitedQuery;
 class EventToDeclinedQuery extends assoc_query_1.AssocEdgeQueryBase {
@@ -32,6 +38,9 @@ class EventToDeclinedQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new EventToDeclinedQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeEvent.load(this.viewer, id);
+    }
 }
 exports.EventToDeclinedQuery = EventToDeclinedQuery;
 class EventToMaybeQuery extends assoc_query_1.AssocEdgeQueryBase {
@@ -41,6 +50,9 @@ class EventToMaybeQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new EventToMaybeQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeEvent.load(this.viewer, id);
+    }
 }
 exports.EventToMaybeQuery = EventToMaybeQuery;
 class EventToHostsQuery extends assoc_query_1.AssocEdgeQueryBase {
@@ -50,5 +62,8 @@ class EventToHostsQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new EventToHostsQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeEvent.load(this.viewer, id);
+    }
 }
 exports.EventToHostsQuery = EventToHostsQuery;

package/testutils/fake_data/test_helpers.d.ts

@@ -1,14 +1,18 @@
-import { Data } from "../../core/base";
+import { Data, Ent } from "../../core/base";
 import { AssocEdge } from "../../core/ent";
 import { TempDB } from "../db/test_db";
 import { FakeUser, UserCreateInput, ContactCreateInput, FakeContact } from ".";
 import { EventCreateInput, FakeEvent } from "./fake_event";
+import { BuilderSchema } from "../builder";
 export declare function getContactInput(user: FakeUser, input?: Partial<ContactCreateInput>): ContactCreateInput;
 export declare function getUserInput(input?: Partial<UserCreateInput>): UserCreateInput;
 export declare function getEventInput(user: FakeUser, input?: Partial<EventCreateInput>): EventCreateInput;
 export declare function createTestUser(input?: Partial<UserCreateInput>): Promise<FakeUser>;
 export declare const inputs: Partial<ContactCreateInput>[];
 export declare function createAllContacts(input?: Partial<UserCreateInput>, slice?: number): Promise<[FakeUser, FakeContact[]]>;
+export declare function createUserPlusFriendRequests(input?: Partial<UserCreateInput>, slice?: number): Promise<[FakeUser, FakeUser[]]>;
+export declare function addEdge<T extends Ent>(source: T, schema: BuilderSchema<T>, edgeType: string, inbound: boolean, // inbound or outbound
+...dest: Ent[]): Promise<void>;
 export declare function verifyUserToContactEdges(user: FakeUser, edges: AssocEdge[], contacts: FakeContact[]): void;
 export declare function verifyUserToContactRawData(user: FakeUser, edges: Data[], contacts: FakeContact[]): void;
 export declare function verifyUserToContacts(user: FakeUser, ents: FakeContact[], contacts: FakeContact[]): void;

package/testutils/fake_data/test_helpers.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createAllEvents = exports.tempDBTables = exports.setupTempDB = exports.createTestEvent = exports.edgeTableNames = exports.createEdges = exports.verifyUserToContacts = exports.verifyUserToContactRawData = exports.verifyUserToContactEdges = exports.createAllContacts = exports.inputs = exports.createTestUser = exports.getEventInput = exports.getUserInput = exports.getContactInput = void 0;
+exports.createAllEvents = exports.tempDBTables = exports.setupTempDB = exports.createTestEvent = exports.edgeTableNames = exports.createEdges = exports.verifyUserToContacts = exports.verifyUserToContactRawData = exports.verifyUserToContactEdges = exports.addEdge = exports.createUserPlusFriendRequests = exports.createAllContacts = exports.inputs = exports.createTestUser = exports.getEventInput = exports.getUserInput = exports.getContactInput = void 0;
 const assert_1 = require("assert");
 const jest_date_mock_1 = require("jest-date-mock");
 const viewer_1 = require("../../core/viewer");
@@ -12,6 +12,8 @@ const _1 = require(".");
 const fake_event_1 = require("./fake_event");
 const const_1 = require("./const");
 const mock_date_1 = require("./../mock_date");
+const builder_1 = require("../builder");
+const action_1 = require("../../action");
 function getContactInput(user, input) {
     return {
         firstName: "Jon",
@@ -99,6 +101,38 @@ async function createAllContacts(input, slice) {
     return [user, contacts];
 }
 exports.createAllContacts = createAllContacts;
+async function createUserPlusFriendRequests(input, slice) {
+    const user = await createTestUser(input);
+    let userInputs = exports.inputs.slice(0, slice || exports.inputs.length);
+    const friendRequests = await Promise.all(userInputs.map(async (input) => {
+        return createTestUser(input);
+    }));
+    expect(friendRequests.length).toBe(userInputs.length);
+    await addEdge(user, new _1.FakeUserSchema(), _1.EdgeType.UserToFriendRequests, true, ...friendRequests);
+    return [user, friendRequests];
+}
+exports.createUserPlusFriendRequests = createUserPlusFriendRequests;
+async function addEdge(source, schema, edgeType, inbound, // inbound or outbound
+...dest) {
+    const action = new builder_1.SimpleAction(source.viewer, schema, new Map(), action_1.WriteOperation.Edit, source);
+    dest.forEach(async (friendRequest) => {
+        // just to make times deterministic so that tests can consistently work
+        (0, jest_date_mock_1.advanceBy)(100);
+        // add edge
+        if (inbound) {
+            action.builder.orchestrator.addInboundEdge(friendRequest.id, edgeType, dest[0].nodeType, {
+                time: new Date(), // set time to advanceBy time
+            });
+        }
+        else {
+            action.builder.orchestrator.addOutboundEdge(friendRequest.id, edgeType, dest[0].nodeType, {
+                time: new Date(), // set time to advanceBy time
+            });
+        }
+    });
+    await action.saveX();
+}
+exports.addEdge = addEdge;
 function verifyUserToContactEdges(user, edges, contacts) {
     expect(edges.length).toBe(contacts.length);
     for (let i = 0; i < contacts.length; i++) {

package/testutils/fake_data/user_query.d.ts

@@ -10,16 +10,19 @@ import { QueryLoaderFactory } from "../../core/loaders/query_loader";
 export declare class UserToContactsQuery extends AssocEdgeQueryBase<FakeUser, FakeContact, AssocEdge> {
     constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
     static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToContactsQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
 }
 export declare const userToContactsCountLoaderFactory: RawCountLoaderFactory;
 export declare const userToContactsDataLoaderFactory: IndexLoaderFactory;
-export declare class UserToContactsFkeyQuery extends CustomEdgeQueryBase<FakeContact> {
+export declare class UserToContactsFkeyQuery extends CustomEdgeQueryBase<FakeUser, FakeContact> {
     constructor(viewer: Viewer, src: ID | FakeUser);
     static query(viewer: Viewer, src: FakeUser | ID): UserToContactsFkeyQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
 }
 export declare class UserToFriendsQuery extends AssocEdgeQueryBase<FakeUser, FakeUser, AssocEdge> {
     constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
     static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToFriendsQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
     queryContacts(): UserToContactsQuery;
     queryFriends(): UserToFriendsQuery;
     queryHostedEvents(): UserToHostedEventsQuery;
@@ -32,6 +35,7 @@ export declare class CustomEdge extends AssocEdge {
 export declare class UserToCustomEdgeQuery extends AssocEdgeQueryBase<FakeUser, FakeUser, CustomEdge> {
     constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
     static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToCustomEdgeQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
     queryContacts(): UserToContactsQuery;
     queryFriends(): UserToFriendsQuery;
     queryHostedEvents(): UserToHostedEventsQuery;
@@ -40,6 +44,7 @@ export declare class UserToCustomEdgeQuery extends AssocEdgeQueryBase<FakeUser,
 export declare class UserToFriendRequestsQuery extends AssocEdgeQueryBase<FakeUser, FakeUser, AssocEdge> {
     constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
     static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToFriendRequestsQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
     queryContacts(): UserToContactsQuery;
     queryFriends(): UserToFriendsQuery;
     queryHostedEvents(): UserToHostedEventsQuery;
@@ -47,8 +52,10 @@ export declare class UserToFriendRequestsQuery extends AssocEdgeQueryBase<FakeUs
     queryCustomEdge(): UserToCustomEdgeQuery;
 }
 export declare class UserToIncomingFriendRequestsQuery extends AssocEdgeQueryBase<FakeUser, FakeUser, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToIncomingFriendRequestsQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeUser>);
+    getPrivacyPolicy(): import("../../core/base").PrivacyPolicy;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeUser>): UserToIncomingFriendRequestsQuery;
     queryContacts(): UserToContactsQuery;
     queryFriends(): UserToFriendsQuery;
     queryHostedEvents(): UserToHostedEventsQuery;
@@ -56,8 +63,9 @@ export declare class UserToIncomingFriendRequestsQuery extends AssocEdgeQueryBas
     queryCustomEdge(): UserToCustomEdgeQuery;
 }
 export declare class UserToEventsAttendingQuery extends AssocEdgeQueryBase<FakeUser, FakeEvent, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToEventsAttendingQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeEvent>);
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeEvent>): UserToEventsAttendingQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
     queryHosts(): EventToHostsQuery;
     queryAttendees(): EventToAttendeesQuery;
     queryInvited(): EventToInvitedQuery;
@@ -65,8 +73,9 @@ export declare class UserToEventsAttendingQuery extends AssocEdgeQueryBase<FakeU
     queryMaybe(): EventToMaybeQuery;
 }
 export declare class UserToHostedEventsQuery extends AssocEdgeQueryBase<FakeUser, FakeEvent, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToHostedEventsQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeEvent>);
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeEvent>): UserToHostedEventsQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
     queryHosts(): EventToHostsQuery;
     queryAttendees(): EventToAttendeesQuery;
     queryInvited(): EventToInvitedQuery;
@@ -77,11 +86,14 @@ export declare const getNextWeekClause: () => clause.Clause;
 export declare function getCompleteClause(id: ID): clause.Clause;
 export declare const userToEventsInNextWeekCountLoaderFactory: RawCountLoaderFactory;
 export declare const userToEventsInNextWeekDataLoaderFactory: QueryLoaderFactory<unknown>;
-export declare class UserToEventsInNextWeekQuery extends CustomEdgeQueryBase<FakeEvent> {
+export declare class UserToEventsInNextWeekQuery extends CustomEdgeQueryBase<FakeUser, FakeEvent> {
     constructor(viewer: Viewer, src: ID | FakeUser);
     static query(viewer: Viewer, src: FakeUser | ID): UserToEventsInNextWeekQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
+    getPrivacyPolicy(): import("../../core/base").PrivacyPolicy;
 }
 export declare class UserToFollowingQuery extends AssocEdgeQueryBase<FakeUser, Ent, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToFollowingQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeUser>);
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeUser>): UserToFollowingQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
 }

package/testutils/fake_data/user_query.js

@@ -35,6 +35,7 @@ const luxon_1 = require("luxon");
 const query_loader_1 = require("../../core/loaders/query_loader");
 const mock_date_1 = require("./../mock_date");
 const _1 = require(".");
+const privacy_1 = require("../../core/privacy");
 class UserToContactsQuery extends assoc_query_1.AssocEdgeQueryBase {
     constructor(viewer, src) {
         super(viewer, src, new assoc_count_loader_1.AssocEdgeCountLoaderFactory(internal_1.EdgeType.UserToContacts), new assoc_edge_loader_1.AssocEdgeLoaderFactory(internal_1.EdgeType.UserToContacts, ent_1.AssocEdge), internal_1.FakeContact.loaderOptions());
@@ -42,6 +43,9 @@ class UserToContactsQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToContactsQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
 }
 exports.UserToContactsQuery = UserToContactsQuery;
 exports.userToContactsCountLoaderFactory = new raw_count_loader_1.RawCountLoaderFactory({
@@ -64,6 +68,9 @@ class UserToContactsFkeyQuery extends custom_query_1.CustomEdgeQueryBase {
     static query(viewer, src) {
         return new UserToContactsFkeyQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
 }
 exports.UserToContactsFkeyQuery = UserToContactsFkeyQuery;
 class UserToFriendsQuery extends assoc_query_1.AssocEdgeQueryBase {
@@ -73,6 +80,9 @@ class UserToFriendsQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToFriendsQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     queryContacts() {
         return UserToContactsQuery.query(this.viewer, this);
     }
@@ -104,6 +114,9 @@ class UserToCustomEdgeQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToCustomEdgeQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     queryContacts() {
         return UserToContactsQuery.query(this.viewer, this);
     }
@@ -125,6 +138,9 @@ class UserToFriendRequestsQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToFriendRequestsQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     queryContacts() {
         return UserToContactsQuery.query(this.viewer, this);
     }
@@ -146,6 +162,12 @@ class UserToIncomingFriendRequestsQuery extends assoc_query_1.AssocEdgeQueryBase
     constructor(viewer, src) {
         super(viewer, src, new assoc_count_loader_1.AssocEdgeCountLoaderFactory(internal_1.EdgeType.UserToIncomingFriendRequests), new assoc_edge_loader_1.AssocEdgeLoaderFactory(internal_1.EdgeType.UserToIncomingFriendRequests, ent_1.AssocEdge), internal_1.FakeUser.loaderOptions());
     }
+    getPrivacyPolicy() {
+        return privacy_1.AllowIfViewerPrivacyPolicy;
+    }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     static query(viewer, src) {
         return new UserToIncomingFriendRequestsQuery(viewer, src);
     }
@@ -173,6 +195,9 @@ class UserToEventsAttendingQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToEventsAttendingQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     queryHosts() {
         return internal_1.EventToHostsQuery.query(this.viewer, this);
     }
@@ -197,6 +222,9 @@ class UserToHostedEventsQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToHostedEventsQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     queryHosts() {
         return internal_1.EventToHostsQuery.query(this.viewer, this);
     }
@@ -256,6 +284,12 @@ class UserToEventsInNextWeekQuery extends custom_query_1.CustomEdgeQueryBase {
     static query(viewer, src) {
         return new UserToEventsInNextWeekQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
+    getPrivacyPolicy() {
+        return privacy_1.AllowIfViewerPrivacyPolicy;
+    }
 }
 exports.UserToEventsInNextWeekQuery = UserToEventsInNextWeekQuery;
 class UserToFollowingQuery extends assoc_query_1.AssocEdgeQueryBase {
@@ -265,5 +299,8 @@ class UserToFollowingQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToFollowingQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
 }
 exports.UserToFollowingQuery = UserToFollowingQuery;