@snowtop/ent 0.1.0-alpha15 → 0.1.0-alpha150

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (170) hide show
  1. package/action/action.d.ts +27 -16
  2. package/action/action.js +22 -7
  3. package/action/executor.d.ts +16 -3
  4. package/action/executor.js +90 -23
  5. package/action/experimental_action.d.ts +25 -16
  6. package/action/experimental_action.js +35 -9
  7. package/action/index.d.ts +3 -1
  8. package/action/index.js +7 -1
  9. package/action/operations.d.ts +125 -0
  10. package/action/operations.js +684 -0
  11. package/action/orchestrator.d.ts +38 -12
  12. package/action/orchestrator.js +427 -102
  13. package/action/relative_value.d.ts +47 -0
  14. package/action/relative_value.js +125 -0
  15. package/action/transaction.d.ts +10 -0
  16. package/action/transaction.js +23 -0
  17. package/auth/auth.d.ts +1 -1
  18. package/core/base.d.ts +51 -21
  19. package/core/base.js +7 -1
  20. package/core/clause.d.ts +85 -40
  21. package/core/clause.js +375 -64
  22. package/core/config.d.ts +15 -1
  23. package/core/config.js +10 -1
  24. package/core/const.d.ts +3 -0
  25. package/core/const.js +6 -0
  26. package/core/context.d.ts +4 -2
  27. package/core/context.js +20 -2
  28. package/core/convert.d.ts +1 -1
  29. package/core/date.js +1 -5
  30. package/core/db.d.ts +12 -8
  31. package/core/db.js +18 -8
  32. package/core/ent.d.ts +68 -94
  33. package/core/ent.js +535 -586
  34. package/core/global_schema.d.ts +7 -0
  35. package/core/global_schema.js +51 -0
  36. package/core/loaders/assoc_count_loader.d.ts +1 -0
  37. package/core/loaders/assoc_count_loader.js +10 -2
  38. package/core/loaders/assoc_edge_loader.d.ts +1 -1
  39. package/core/loaders/assoc_edge_loader.js +10 -13
  40. package/core/loaders/index.d.ts +1 -1
  41. package/core/loaders/index.js +1 -3
  42. package/core/loaders/index_loader.d.ts +3 -3
  43. package/core/loaders/loader.d.ts +2 -2
  44. package/core/loaders/loader.js +5 -5
  45. package/core/loaders/object_loader.d.ts +30 -9
  46. package/core/loaders/object_loader.js +225 -78
  47. package/core/loaders/query_loader.d.ts +6 -12
  48. package/core/loaders/query_loader.js +54 -13
  49. package/core/loaders/raw_count_loader.js +5 -1
  50. package/core/logger.d.ts +1 -1
  51. package/core/logger.js +1 -0
  52. package/core/privacy.d.ts +7 -6
  53. package/core/privacy.js +21 -25
  54. package/core/query/assoc_query.d.ts +3 -2
  55. package/core/query/assoc_query.js +9 -1
  56. package/core/query/custom_clause_query.d.ts +27 -0
  57. package/core/query/custom_clause_query.js +88 -0
  58. package/core/query/custom_query.d.ts +17 -2
  59. package/core/query/custom_query.js +88 -13
  60. package/core/query/index.d.ts +1 -0
  61. package/core/query/index.js +3 -1
  62. package/core/query/query.d.ts +15 -3
  63. package/core/query/query.js +128 -53
  64. package/core/query/shared_assoc_test.d.ts +2 -1
  65. package/core/query/shared_assoc_test.js +44 -54
  66. package/core/query/shared_test.d.ts +8 -1
  67. package/core/query/shared_test.js +532 -236
  68. package/core/viewer.d.ts +2 -0
  69. package/core/viewer.js +3 -1
  70. package/graphql/graphql.d.ts +52 -19
  71. package/graphql/graphql.js +174 -136
  72. package/graphql/graphql_field_helpers.d.ts +7 -1
  73. package/graphql/graphql_field_helpers.js +21 -1
  74. package/graphql/index.d.ts +2 -2
  75. package/graphql/index.js +3 -5
  76. package/graphql/query/connection_type.d.ts +9 -9
  77. package/graphql/query/shared_assoc_test.js +1 -1
  78. package/graphql/query/shared_edge_connection.js +1 -19
  79. package/graphql/scalars/orderby_direction.d.ts +2 -0
  80. package/graphql/scalars/orderby_direction.js +15 -0
  81. package/imports/dataz/example1/_auth.js +128 -47
  82. package/imports/dataz/example1/_viewer.js +87 -39
  83. package/imports/index.d.ts +7 -2
  84. package/imports/index.js +20 -5
  85. package/index.d.ts +14 -5
  86. package/index.js +26 -10
  87. package/package.json +18 -17
  88. package/parse_schema/parse.d.ts +31 -9
  89. package/parse_schema/parse.js +179 -32
  90. package/schema/base_schema.d.ts +13 -3
  91. package/schema/base_schema.js +13 -0
  92. package/schema/field.d.ts +78 -21
  93. package/schema/field.js +231 -71
  94. package/schema/index.d.ts +2 -2
  95. package/schema/index.js +7 -2
  96. package/schema/json_field.d.ts +16 -4
  97. package/schema/json_field.js +32 -2
  98. package/schema/schema.d.ts +109 -20
  99. package/schema/schema.js +42 -53
  100. package/schema/struct_field.d.ts +15 -3
  101. package/schema/struct_field.js +117 -22
  102. package/schema/union_field.d.ts +1 -1
  103. package/scripts/custom_compiler.js +12 -8
  104. package/scripts/custom_graphql.js +145 -34
  105. package/scripts/migrate_v0.1.js +36 -0
  106. package/scripts/move_types.js +120 -0
  107. package/scripts/read_schema.js +22 -7
  108. package/testutils/action/complex_schemas.d.ts +69 -0
  109. package/testutils/action/complex_schemas.js +405 -0
  110. package/testutils/builder.d.ts +39 -43
  111. package/testutils/builder.js +75 -49
  112. package/testutils/db/fixture.d.ts +10 -0
  113. package/testutils/db/fixture.js +26 -0
  114. package/testutils/db/{test_db.d.ts → temp_db.d.ts} +32 -8
  115. package/testutils/db/{test_db.js → temp_db.js} +244 -48
  116. package/testutils/db/value.d.ts +7 -0
  117. package/testutils/db/value.js +251 -0
  118. package/testutils/db_mock.d.ts +16 -4
  119. package/testutils/db_mock.js +52 -9
  120. package/testutils/db_time_zone.d.ts +4 -0
  121. package/testutils/db_time_zone.js +41 -0
  122. package/testutils/ent-graphql-tests/index.d.ts +7 -1
  123. package/testutils/ent-graphql-tests/index.js +56 -26
  124. package/testutils/fake_comms.js +1 -1
  125. package/testutils/fake_data/const.d.ts +2 -1
  126. package/testutils/fake_data/const.js +3 -0
  127. package/testutils/fake_data/fake_contact.d.ts +7 -3
  128. package/testutils/fake_data/fake_contact.js +13 -7
  129. package/testutils/fake_data/fake_event.d.ts +4 -1
  130. package/testutils/fake_data/fake_event.js +7 -6
  131. package/testutils/fake_data/fake_tag.d.ts +36 -0
  132. package/testutils/fake_data/fake_tag.js +89 -0
  133. package/testutils/fake_data/fake_user.d.ts +8 -5
  134. package/testutils/fake_data/fake_user.js +16 -15
  135. package/testutils/fake_data/index.js +5 -1
  136. package/testutils/fake_data/internal.d.ts +2 -0
  137. package/testutils/fake_data/internal.js +7 -1
  138. package/testutils/fake_data/tag_query.d.ts +13 -0
  139. package/testutils/fake_data/tag_query.js +43 -0
  140. package/testutils/fake_data/test_helpers.d.ts +11 -4
  141. package/testutils/fake_data/test_helpers.js +29 -13
  142. package/testutils/fake_data/user_query.d.ts +11 -4
  143. package/testutils/fake_data/user_query.js +54 -22
  144. package/testutils/fake_log.js +1 -1
  145. package/testutils/parse_sql.d.ts +6 -0
  146. package/testutils/parse_sql.js +16 -2
  147. package/testutils/test_edge_global_schema.d.ts +15 -0
  148. package/testutils/test_edge_global_schema.js +62 -0
  149. package/testutils/write.d.ts +2 -2
  150. package/testutils/write.js +33 -7
  151. package/tsc/ast.d.ts +25 -2
  152. package/tsc/ast.js +141 -17
  153. package/tsc/compilerOptions.js +5 -1
  154. package/tsc/move_generated.d.ts +1 -0
  155. package/tsc/move_generated.js +164 -0
  156. package/tsc/transform.d.ts +22 -0
  157. package/tsc/transform.js +182 -0
  158. package/tsc/transform_action.d.ts +22 -0
  159. package/tsc/transform_action.js +183 -0
  160. package/tsc/transform_ent.d.ts +17 -0
  161. package/tsc/transform_ent.js +60 -0
  162. package/tsc/transform_schema.d.ts +27 -0
  163. package/{scripts → tsc}/transform_schema.js +146 -117
  164. package/graphql/enums.d.ts +0 -3
  165. package/graphql/enums.js +0 -25
  166. package/scripts/move_generated.js +0 -142
  167. package/scripts/transform_code.js +0 -113
  168. package/scripts/transform_schema.d.ts +0 -1
  169. /package/scripts/{move_generated.d.ts → migrate_v0.1.d.ts} +0 -0
  170. /package/scripts/{transform_code.d.ts → move_types.d.ts} +0 -0
@@ -0,0 +1,120 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
14
+ Object.defineProperty(o, "default", { enumerable: true, value: v });
15
+ }) : function(o, v) {
16
+ o["default"] = v;
17
+ });
18
+ var __importStar = (this && this.__importStar) || function (mod) {
19
+ if (mod && mod.__esModule) return mod;
20
+ var result = {};
21
+ if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
22
+ __setModuleDefault(result, mod);
23
+ return result;
24
+ };
25
+ var __importDefault = (this && this.__importDefault) || function (mod) {
26
+ return (mod && mod.__esModule) ? mod : { "default": mod };
27
+ };
28
+ Object.defineProperty(exports, "__esModule", { value: true });
29
+ const transform_1 = require("../tsc/transform");
30
+ const ast_1 = require("../tsc/ast");
31
+ const typescript_1 = __importStar(require("typescript"));
32
+ const path_1 = __importDefault(require("path"));
33
+ class GatherExportsInGeneratedTypes {
34
+ constructor() {
35
+ this.glob = "src/ent/generated/types.ts";
36
+ this.names = [];
37
+ }
38
+ traverseChild(sourceFile, contents, file, node) {
39
+ const modifiers = typescript_1.default.canHaveModifiers(node)
40
+ ? typescript_1.default.getModifiers(node)
41
+ : undefined;
42
+ const exported = modifiers?.filter((mod) => mod.getText(sourceFile) === "export");
43
+ if (exported?.length) {
44
+ if (typescript_1.default.isEnumDeclaration(node) ||
45
+ typescript_1.default.isInterfaceDeclaration(node) ||
46
+ typescript_1.default.isTypeAliasDeclaration(node)) {
47
+ this.names.push(node.name.text);
48
+ }
49
+ if (typescript_1.default.isFunctionDeclaration(node) && node.name?.text) {
50
+ this.names.push(node.name.text);
51
+ }
52
+ }
53
+ return { node };
54
+ }
55
+ }
56
+ class TransformImports {
57
+ constructor() {
58
+ this.glob = "src/**/*.ts";
59
+ this.prettierGlob = "src/**/*.ts";
60
+ this.impsToMove = new Map();
61
+ this.cwd = "";
62
+ this.relative = false;
63
+ this.cwd = process.cwd();
64
+ const gt = new GatherExportsInGeneratedTypes();
65
+ (0, transform_1.transform)(gt);
66
+ gt.names.forEach((v) => this.impsToMove.set(v, true));
67
+ this.relative = (0, ast_1.getCustomInfo)().relativeImports ?? this.relative;
68
+ }
69
+ getSrcPath(file, text) {
70
+ if (text.startsWith("src")) {
71
+ return text;
72
+ }
73
+ const dir = path_1.default.dirname(path_1.default.join(this.cwd, file));
74
+ const fullPath = path_1.default.join(dir, text);
75
+ return path_1.default.relative(this.cwd, fullPath);
76
+ }
77
+ traverseChild(sourceFile, contents, file, node) {
78
+ if (!(0, typescript_1.isImportDeclaration)(node)) {
79
+ return { node };
80
+ }
81
+ const impInfo = (0, ast_1.getImportInfo)(node, sourceFile);
82
+ if (!impInfo) {
83
+ return { node };
84
+ }
85
+ const text = node.moduleSpecifier.getText(sourceFile).slice(1, -1);
86
+ const pathToWrite = (0, ast_1.transformRelative)(file, "src/ent/generated/types", this.relative);
87
+ const current = (0, ast_1.transformRelative)(file, text, this.relative);
88
+ const srcPath = this.getSrcPath(file, text);
89
+ // nothing to do here
90
+ if (!srcPath.startsWith("src/ent") || pathToWrite === current) {
91
+ return { node };
92
+ }
93
+ // let's see if we can simplify
94
+ let seenImports = [];
95
+ for (let imp of impInfo.imports) {
96
+ imp = imp.trim();
97
+ if (this.impsToMove.has(imp)) {
98
+ seenImports.push(imp);
99
+ }
100
+ }
101
+ if (!seenImports.length) {
102
+ return { node };
103
+ }
104
+ let imports = new Map([[pathToWrite, seenImports]]);
105
+ return {
106
+ rawString: (0, ast_1.transformImport)(contents, node, sourceFile, {
107
+ removeImports: seenImports,
108
+ transformPath: text,
109
+ }),
110
+ traversed: true,
111
+ imports,
112
+ allowSeenImportsAdded: true,
113
+ };
114
+ }
115
+ }
116
+ // ts-node-script --swc --project ./tsconfig.json -r tsconfig-paths/register ../../ts/src/scripts/move_types.ts
117
+ function main() {
118
+ (0, transform_1.transform)(new TransformImports());
119
+ }
120
+ main();
@@ -1,7 +1,11 @@
1
1
  "use strict";
2
2
  var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
3
  if (k2 === undefined) k2 = k;
4
- Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
5
9
  }) : (function(o, m, k, k2) {
6
10
  if (k2 === undefined) k2 = k;
7
11
  o[k2] = m[k];
@@ -22,24 +26,32 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
22
26
  return (mod && mod.__esModule) ? mod : { "default": mod };
23
27
  };
24
28
  Object.defineProperty(exports, "__esModule", { value: true });
25
- const glob_1 = __importDefault(require("glob"));
29
+ const glob = __importStar(require("glob"));
26
30
  const path = __importStar(require("path"));
27
31
  const pascal_case_1 = require("pascal-case");
28
32
  const minimist_1 = __importDefault(require("minimist"));
29
- const process_1 = require("process");
30
33
  const parse_1 = require("../parse_schema/parse");
34
+ const ast_1 = require("../tsc/ast");
31
35
  function main() {
32
36
  const options = (0, minimist_1.default)(process.argv.slice(2));
33
37
  if (!options.path) {
34
38
  throw new Error("path required");
35
39
  }
40
+ const customInfo = (0, ast_1.getCustomInfo)();
41
+ const globalSchemaPath = customInfo.globalSchemaPath || "__global__schema.ts";
42
+ let globalSchema;
36
43
  const r = /(\w+).ts/;
37
- const paths = glob_1.default.sync(path.join(options.path, "*.ts"), {
44
+ // do we still even need this...
45
+ const paths = glob.sync(path.join(options.path, "*.ts"), {
38
46
  ignore: [`\d+_read_schema.ts`],
39
47
  });
40
48
  let potentialSchemas = {};
41
49
  for (const p of paths) {
42
50
  const basename = path.basename(p);
51
+ if (basename === globalSchemaPath) {
52
+ globalSchema = require(p).default;
53
+ continue;
54
+ }
43
55
  const match = r.exec(basename);
44
56
  if (!match) {
45
57
  throw new Error(`non-typescript file ${p} returned by glob`);
@@ -61,13 +73,16 @@ function main() {
61
73
  potentialSchemas[(0, pascal_case_1.pascalCase)(schema)] = s;
62
74
  }
63
75
  // console.log(potentialSchemas);
64
- const result = (0, parse_1.parseSchema)(potentialSchemas);
65
- console.log(JSON.stringify(result));
76
+ // NB: do not change this to async/await
77
+ // doing so runs it buffer limit on linux (65536 bytes) and we lose data reading in go
78
+ (0, parse_1.parseSchema)(potentialSchemas, globalSchema).then((result) => {
79
+ console.log(JSON.stringify(result));
80
+ });
66
81
  }
67
82
  try {
68
83
  main();
69
84
  }
70
85
  catch (err) {
71
86
  console.error(err);
72
- (0, process_1.exit)(1);
87
+ process.exit(1);
73
88
  }
@@ -0,0 +1,69 @@
1
+ import { Ent, ID, Viewer, Data } from "../../core/base";
2
+ import { DataOperation } from "../../action/operations";
3
+ import { Action, Builder, Executor, WriteOperation, Trigger, Observer, TriggerReturn } from "../../action/action";
4
+ import { EdgeInputData } from "../../action/orchestrator";
5
+ import { User, Group, Message, Contact, SimpleBuilder, BuilderSchema, SimpleAction, BaseEnt } from "../../testutils/builder";
6
+ import { MockLogs } from "../../testutils/mock_log";
7
+ import { Table } from "../db/temp_db";
8
+ export declare function setupTest(): void;
9
+ export declare function getML(): MockLogs;
10
+ export declare function getOperations(): DataOperation<Ent<Viewer<Ent<any> | null, ID | null>>>[];
11
+ export declare const UserSchema: BuilderSchema<User>;
12
+ export declare class UserWithBalance extends User {
13
+ }
14
+ export declare const UserBalanceSchema: BuilderSchema<UserWithBalance>;
15
+ export declare class UserWithBalanceWithCheck extends UserWithBalance {
16
+ }
17
+ export declare const UserBalanceWithCheckSchema: BuilderSchema<UserWithBalanceWithCheck>;
18
+ export declare class Account extends BaseEnt {
19
+ accountID: string;
20
+ nodeType: string;
21
+ }
22
+ export declare const AccountSchema: BuilderSchema<Account>;
23
+ export declare const ContactSchema: BuilderSchema<Contact>;
24
+ export declare const GroupSchema: BuilderSchema<Group>;
25
+ export declare class GroupMembership extends BaseEnt {
26
+ nodeType: string;
27
+ }
28
+ export declare const GroupMembershipSchema: BuilderSchema<GroupMembership>;
29
+ export declare class Changelog extends BaseEnt {
30
+ nodeType: string;
31
+ }
32
+ export declare const ChangelogSchema: BuilderSchema<Changelog>;
33
+ export declare const MessageSchema: BuilderSchema<Message>;
34
+ export declare function executeAction<T extends Ent, E = any>(action: Action<T, Builder<T>>, name?: E): Promise<Executor>;
35
+ export declare function createGroup(): Promise<Group>;
36
+ export declare function createUser(): Promise<User>;
37
+ export declare class MessageAction extends SimpleAction<Message> {
38
+ constructor(viewer: Viewer, fields: Map<string, any>, operation: WriteOperation, existingEnt: Message | null);
39
+ getTriggers(): Trigger<Message, SimpleBuilder<Message, Message | null>>[];
40
+ getObservers(): Observer<Message, SimpleBuilder<Message>>[];
41
+ }
42
+ export declare class UserAction extends SimpleAction<User> {
43
+ contactAction: SimpleAction<Contact> | undefined;
44
+ constructor(viewer: Viewer, fields: Map<string, any>, operation: WriteOperation, existingEnt: User | null);
45
+ getTriggers(): Trigger<User, SimpleBuilder<User>>[];
46
+ getObservers(): Observer<User, SimpleBuilder<User>>[];
47
+ }
48
+ type getMembershipFunction = (viewer: Viewer, edge: EdgeInputData) => SimpleAction<Ent>;
49
+ export declare class GroupMembershipTrigger implements Trigger<Group, SimpleBuilder<Group>> {
50
+ private getter;
51
+ constructor(getter: getMembershipFunction);
52
+ changeset(builder: SimpleBuilder<Group>, input: Data): TriggerReturn;
53
+ }
54
+ export declare class EditGroupAction extends SimpleAction<Group> {
55
+ viewer: Viewer;
56
+ private getter;
57
+ constructor(viewer: Viewer, schema: BuilderSchema<Group>, fields: Map<string, any>, existingEnt: Group, getter: getMembershipFunction);
58
+ getTriggers: () => GroupMembershipTrigger[];
59
+ }
60
+ export declare function verifyGroupMembers(group: Group, members: User[]): Promise<ID[]>;
61
+ export declare function loadMemberships(viewer: Viewer, membershipids: ID[]): Promise<GroupMembership[]>;
62
+ export declare function loadChangelogs(viewer: Viewer, clids: ID[]): Promise<Changelog[]>;
63
+ export declare function verifyChangelogFromMeberships(user: User, memberships: GroupMembership[]): Promise<void>;
64
+ export declare class GroupMemberOf extends BaseEnt {
65
+ nodeType: string;
66
+ }
67
+ export declare const GroupMemberOfSchema: BuilderSchema<GroupMemberOf>;
68
+ export declare const getTables: () => Table[];
69
+ export {};
@@ -0,0 +1,405 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
14
+ Object.defineProperty(o, "default", { enumerable: true, value: v });
15
+ }) : function(o, v) {
16
+ o["default"] = v;
17
+ });
18
+ var __importStar = (this && this.__importStar) || function (mod) {
19
+ if (mod && mod.__esModule) return mod;
20
+ var result = {};
21
+ if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
22
+ __setModuleDefault(result, mod);
23
+ return result;
24
+ };
25
+ Object.defineProperty(exports, "__esModule", { value: true });
26
+ exports.getTables = exports.GroupMemberOfSchema = exports.GroupMemberOf = exports.verifyChangelogFromMeberships = exports.loadChangelogs = exports.loadMemberships = exports.verifyGroupMembers = exports.EditGroupAction = exports.GroupMembershipTrigger = exports.UserAction = exports.MessageAction = exports.createUser = exports.createGroup = exports.executeAction = exports.MessageSchema = exports.ChangelogSchema = exports.Changelog = exports.GroupMembershipSchema = exports.GroupMembership = exports.GroupSchema = exports.ContactSchema = exports.AccountSchema = exports.Account = exports.UserBalanceWithCheckSchema = exports.UserWithBalanceWithCheck = exports.UserBalanceSchema = exports.UserWithBalance = exports.UserSchema = exports.getOperations = exports.getML = exports.setupTest = void 0;
27
+ const ent_1 = require("../../core/ent");
28
+ const object_loader_1 = require("../../core/loaders/object_loader");
29
+ const action_1 = require("../../action/action");
30
+ const executor_1 = require("../../action/executor");
31
+ const builder_1 = require("../../testutils/builder");
32
+ const viewer_1 = require("../../core/viewer");
33
+ const field_1 = require("../../schema/field");
34
+ const json_field_1 = require("../../schema/json_field");
35
+ const fake_log_1 = require("../../testutils/fake_log");
36
+ const write_1 = require("../../testutils/write");
37
+ const mock_log_1 = require("../../testutils/mock_log");
38
+ const logger_1 = require("../../core/logger");
39
+ const action = __importStar(require("../../action"));
40
+ const convert_1 = require("../../core/convert");
41
+ const uuid_1 = require("uuid");
42
+ const temp_db_1 = require("../db/temp_db");
43
+ const db_1 = require("../../core/db");
44
+ const schema_1 = require("../../schema");
45
+ const ml = new mock_log_1.MockLogs();
46
+ let operations = [];
47
+ const edges = [
48
+ "fake_edge",
49
+ "selfContact",
50
+ "channelMember",
51
+ "senderToMessage",
52
+ "workspaceMember",
53
+ "recipientToMessage",
54
+ "objectToChangelog",
55
+ "changelogToParent",
56
+ ];
57
+ function setupTest() {
58
+ beforeEach(async () => {
59
+ // does assoc_edge_config loader need to be cleared?
60
+ for (const edge of edges) {
61
+ await (0, write_1.createRowForTest)({
62
+ tableName: "assoc_edge_config",
63
+ fields: {
64
+ edge_table: `${edge}_table`,
65
+ symmetric_edge: false,
66
+ inverse_edge_type: null,
67
+ edge_type: edge,
68
+ edge_name: "name",
69
+ created_at: new Date(),
70
+ updated_at: new Date(),
71
+ },
72
+ });
73
+ }
74
+ ml.clear();
75
+ });
76
+ beforeAll(() => {
77
+ (0, logger_1.setLogLevels)(["query", "error", "cache"]);
78
+ ml.mock();
79
+ });
80
+ afterEach(() => {
81
+ ml.clear();
82
+ });
83
+ afterAll(() => {
84
+ ml.restore();
85
+ });
86
+ afterEach(() => {
87
+ fake_log_1.FakeLogger.clear();
88
+ operations = [];
89
+ });
90
+ }
91
+ exports.setupTest = setupTest;
92
+ function getML() {
93
+ return ml;
94
+ }
95
+ exports.getML = getML;
96
+ function getOperations() {
97
+ return operations;
98
+ }
99
+ exports.getOperations = getOperations;
100
+ exports.UserSchema = (0, builder_1.getBuilderSchemaFromFields)({
101
+ FirstName: (0, field_1.StringType)(),
102
+ LastName: (0, field_1.StringType)(),
103
+ EmailAddress: (0, field_1.StringType)({ nullable: true }),
104
+ AccountID: (0, field_1.UUIDType)({ nullable: true }),
105
+ }, builder_1.User);
106
+ class UserWithBalance extends builder_1.User {
107
+ }
108
+ exports.UserWithBalance = UserWithBalance;
109
+ exports.UserBalanceSchema = (0, builder_1.getBuilderSchemaFromFields)({
110
+ first_name: (0, field_1.StringType)(),
111
+ last_name: (0, field_1.StringType)(),
112
+ email_address: (0, field_1.StringType)({ nullable: true }),
113
+ balance: (0, field_1.FloatType)(),
114
+ }, UserWithBalance);
115
+ class UserWithBalanceWithCheck extends UserWithBalance {
116
+ }
117
+ exports.UserWithBalanceWithCheck = UserWithBalanceWithCheck;
118
+ exports.UserBalanceWithCheckSchema = (0, builder_1.getBuilderSchemaFromFields)({
119
+ first_name: (0, field_1.StringType)(),
120
+ last_name: (0, field_1.StringType)(),
121
+ email_address: (0, field_1.StringType)({ nullable: true }),
122
+ balance: (0, field_1.FloatType)(),
123
+ }, UserWithBalanceWithCheck, {
124
+ constraints: [
125
+ {
126
+ type: schema_1.ConstraintType.Check,
127
+ name: "positive_balance",
128
+ columns: ["balance"],
129
+ condition: "balance >= 0",
130
+ },
131
+ ],
132
+ });
133
+ class Account extends builder_1.BaseEnt {
134
+ constructor() {
135
+ super(...arguments);
136
+ this.accountID = "";
137
+ this.nodeType = "Account";
138
+ }
139
+ }
140
+ exports.Account = Account;
141
+ exports.AccountSchema = (0, builder_1.getBuilderSchemaFromFields)({}, Account);
142
+ exports.ContactSchema = (0, builder_1.getBuilderSchemaFromFields)({
143
+ FirstName: (0, field_1.StringType)(),
144
+ LastName: (0, field_1.StringType)(),
145
+ UserID: (0, field_1.StringType)(),
146
+ }, builder_1.Contact);
147
+ exports.GroupSchema = (0, builder_1.getBuilderSchemaFromFields)({
148
+ name: (0, field_1.StringType)(),
149
+ funField: (0, field_1.StringType)({ nullable: true }),
150
+ }, builder_1.Group);
151
+ class GroupMembership extends builder_1.BaseEnt {
152
+ constructor() {
153
+ super(...arguments);
154
+ this.nodeType = "GroupMembership";
155
+ }
156
+ }
157
+ exports.GroupMembership = GroupMembership;
158
+ exports.GroupMembershipSchema = (0, builder_1.getBuilderSchemaFromFields)({
159
+ ownerID: (0, field_1.UUIDType)(),
160
+ addedBy: (0, field_1.UUIDType)(),
161
+ notificationsEnabled: (0, field_1.BooleanType)(),
162
+ }, GroupMembership);
163
+ class Changelog extends builder_1.BaseEnt {
164
+ constructor() {
165
+ super(...arguments);
166
+ this.nodeType = "Changelog";
167
+ }
168
+ }
169
+ exports.Changelog = Changelog;
170
+ exports.ChangelogSchema = (0, builder_1.getBuilderSchemaFromFields)({
171
+ parentID: (0, field_1.UUIDType)({ polymorphic: true }),
172
+ log: (0, json_field_1.JSONBType)(),
173
+ }, Changelog);
174
+ exports.MessageSchema = (0, builder_1.getBuilderSchemaFromFields)({
175
+ // TODO both id fields
176
+ sender: (0, field_1.StringType)({
177
+ index: true,
178
+ }),
179
+ recipient: (0, field_1.StringType)(),
180
+ message: (0, field_1.StringType)(),
181
+ transient: (0, field_1.BooleanType)({ nullable: true }),
182
+ expiresAt: (0, field_1.TimestampType)({ nullable: true }),
183
+ delivered: (0, field_1.BooleanType)({ defaultValueOnCreate: () => false }),
184
+ }, builder_1.Message);
185
+ jest.spyOn(action, "saveBuilder").mockImplementation(saveBuilder);
186
+ jest.spyOn(action, "saveBuilderX").mockImplementation(saveBuilderX);
187
+ async function saveBuilder(builder) {
188
+ const changeset = await builder.build();
189
+ const executor = changeset.executor();
190
+ operations = await (0, executor_1.executeOperations)(executor, builder.viewer.context, true);
191
+ }
192
+ async function saveBuilderX(builder) {
193
+ return saveBuilder(builder);
194
+ }
195
+ async function executeAction(action, name) {
196
+ const exec = await executor(action.builder);
197
+ if (name !== undefined) {
198
+ expect(exec).toBeInstanceOf(name);
199
+ }
200
+ operations = await (0, executor_1.executeOperations)(exec, action.builder.viewer.context, true);
201
+ return exec;
202
+ }
203
+ exports.executeAction = executeAction;
204
+ async function executor(builder) {
205
+ const changeset = await builder.build();
206
+ return changeset.executor();
207
+ }
208
+ async function createGroup() {
209
+ let groupID = (0, uuid_1.v4)();
210
+ let groupFields = {
211
+ id: groupID,
212
+ name: "group",
213
+ created_at: new Date(),
214
+ updated_at: new Date(),
215
+ };
216
+ // need to create the group first
217
+ await (0, write_1.createRowForTest)({
218
+ tableName: "groups",
219
+ fields: groupFields,
220
+ });
221
+ return new builder_1.Group(new viewer_1.LoggedOutViewer(), groupFields);
222
+ }
223
+ exports.createGroup = createGroup;
224
+ async function createUser() {
225
+ const id = (0, uuid_1.v4)();
226
+ return new builder_1.User(new viewer_1.IDViewer(id), { id });
227
+ }
228
+ exports.createUser = createUser;
229
+ class MessageAction extends builder_1.SimpleAction {
230
+ constructor(viewer, fields, operation, existingEnt) {
231
+ super(viewer, exports.MessageSchema, fields, operation, existingEnt);
232
+ }
233
+ getTriggers() {
234
+ return [
235
+ {
236
+ changeset: (builder, _input) => {
237
+ let sender = builder.fields.get("sender");
238
+ let recipient = builder.fields.get("recipient");
239
+ if (sender) {
240
+ builder.orchestrator.addInboundEdge(sender, "senderToMessage", "user");
241
+ }
242
+ if (recipient) {
243
+ builder.orchestrator.addInboundEdge(recipient, "recipientToMessage", "user");
244
+ }
245
+ },
246
+ },
247
+ ];
248
+ }
249
+ getObservers() {
250
+ return [new fake_log_1.EntCreationObserver()];
251
+ }
252
+ }
253
+ exports.MessageAction = MessageAction;
254
+ class UserAction extends builder_1.SimpleAction {
255
+ constructor(viewer, fields, operation, existingEnt) {
256
+ super(viewer, exports.UserSchema, fields, operation, existingEnt);
257
+ }
258
+ getTriggers() {
259
+ return [
260
+ {
261
+ changeset: (builder) => {
262
+ let firstName = builder.fields.get("FirstName");
263
+ let lastName = builder.fields.get("LastName");
264
+ this.contactAction = new builder_1.SimpleAction(builder.viewer, exports.ContactSchema, new Map([
265
+ ["FirstName", firstName],
266
+ ["LastName", lastName],
267
+ ["UserID", builder],
268
+ ]), action_1.WriteOperation.Insert, null);
269
+ this.contactAction.getObservers = () => [
270
+ new fake_log_1.EntCreationObserver(),
271
+ ];
272
+ builder.orchestrator.addOutboundEdge(this.contactAction.builder, "selfContact", "contact");
273
+ return this.contactAction.changeset();
274
+ },
275
+ },
276
+ ];
277
+ }
278
+ getObservers() {
279
+ return [new fake_log_1.EntCreationObserver()];
280
+ }
281
+ }
282
+ exports.UserAction = UserAction;
283
+ class GroupMembershipTrigger {
284
+ constructor(getter) {
285
+ this.getter = getter;
286
+ }
287
+ changeset(builder, input) {
288
+ const inputEdges = builder.orchestrator.getInputEdges("workspaceMember", action_1.WriteOperation.Insert);
289
+ const changesets = [];
290
+ for (const edge of inputEdges) {
291
+ // we're going to simplify and assume it doesn't currently exist
292
+ const memberAction = this.getter(builder.viewer, edge);
293
+ // store the membership edge in data field of member edge
294
+ builder.orchestrator.addOutboundEdge(edge.id, "workspaceMember", "User", {
295
+ data: memberAction.builder,
296
+ });
297
+ changesets.push(memberAction.changeset());
298
+ }
299
+ return Promise.all(changesets);
300
+ }
301
+ }
302
+ exports.GroupMembershipTrigger = GroupMembershipTrigger;
303
+ class EditGroupAction extends builder_1.SimpleAction {
304
+ constructor(viewer, schema, fields, existingEnt, getter) {
305
+ super(viewer, schema, fields, action_1.WriteOperation.Edit, existingEnt);
306
+ this.viewer = viewer;
307
+ this.getter = getter;
308
+ this.getTriggers = () => [new GroupMembershipTrigger(this.getter)];
309
+ }
310
+ }
311
+ exports.EditGroupAction = EditGroupAction;
312
+ async function verifyGroupMembers(group, members) {
313
+ const memberEdges = await (0, ent_1.loadEdges)({
314
+ edgeType: "workspaceMember",
315
+ id1: group.id,
316
+ });
317
+ const memberIDs = members.map((ent) => ent.id);
318
+ expect(memberIDs.sort()).toEqual(memberEdges.map((edge) => edge.id2).sort());
319
+ // @ts-ignore
320
+ const membershipids = memberEdges
321
+ .map((edge) => edge.data)
322
+ .filter((str) => str !== null && str !== undefined);
323
+ return membershipids;
324
+ }
325
+ exports.verifyGroupMembers = verifyGroupMembers;
326
+ async function loadMemberships(viewer, membershipids) {
327
+ const tableName = (0, builder_1.getTableName)(exports.GroupMembershipSchema);
328
+ const ents = await (0, ent_1.loadEnts)(viewer, {
329
+ tableName,
330
+ ent: GroupMembership,
331
+ fields: ["id", "owner_id", "added_by", "notifications_enabled"],
332
+ loaderFactory: new object_loader_1.ObjectLoaderFactory({
333
+ tableName,
334
+ fields: ["id", "owner_id", "added_by", "notifications_enabled"],
335
+ key: "id",
336
+ }),
337
+ }, ...membershipids);
338
+ return Array.from(ents.values());
339
+ }
340
+ exports.loadMemberships = loadMemberships;
341
+ async function loadChangelogs(viewer, clids) {
342
+ const ents = await (0, ent_1.loadEnts)(viewer, {
343
+ tableName: "changelogs",
344
+ ent: Changelog,
345
+ fields: ["id", "parent_id", "log"],
346
+ loaderFactory: new object_loader_1.ObjectLoaderFactory({
347
+ tableName: "changelogs",
348
+ fields: ["id", "parent_id", "log"],
349
+ key: "id",
350
+ }),
351
+ }, ...clids);
352
+ return Array.from(ents.values());
353
+ }
354
+ exports.loadChangelogs = loadChangelogs;
355
+ async function verifyChangelogFromMeberships(user, memberships) {
356
+ await Promise.all(memberships.map(async (membership) => {
357
+ const edges = await (0, ent_1.loadEdges)({
358
+ edgeType: "objectToChangelog",
359
+ id1: membership.id,
360
+ });
361
+ expect(edges.length).toBe(1);
362
+ const clIDs = edges.map((edge) => edge.id2);
363
+ const cls = await loadChangelogs(user.viewer, clIDs);
364
+ expect(cls.length).toBe(1);
365
+ const cl = cls[0];
366
+ expect(edges[0].id2).toBe(cl.id);
367
+ expect((0, convert_1.convertJSON)(cl.data.log)).toMatchObject({
368
+ // also has ownerID...
369
+ addedBy: user.id,
370
+ notificationsEnabled: true,
371
+ });
372
+ }));
373
+ }
374
+ exports.verifyChangelogFromMeberships = verifyChangelogFromMeberships;
375
+ class GroupMemberOf extends builder_1.BaseEnt {
376
+ constructor() {
377
+ super(...arguments);
378
+ this.nodeType = "GroupMemberOf";
379
+ }
380
+ }
381
+ exports.GroupMemberOf = GroupMemberOf;
382
+ exports.GroupMemberOfSchema = (0, builder_1.getBuilderSchemaFromFields)({
383
+ userID: (0, field_1.UUIDType)({ index: true }),
384
+ addedBy: (0, field_1.UUIDType)(),
385
+ groupID: (0, field_1.UUIDType)(),
386
+ notificationsEnabled: (0, field_1.BooleanType)(),
387
+ }, GroupMemberOf);
388
+ const getTables = () => {
389
+ const tables = [(0, temp_db_1.assoc_edge_config_table)()];
390
+ edges.map((edge) => tables.push((0, temp_db_1.assoc_edge_table)(`${edge}_table`)));
391
+ [
392
+ exports.AccountSchema,
393
+ exports.ContactSchema,
394
+ exports.GroupSchema,
395
+ exports.UserSchema,
396
+ exports.MessageSchema,
397
+ exports.GroupMembershipSchema,
398
+ exports.ChangelogSchema,
399
+ exports.GroupMemberOfSchema,
400
+ exports.UserBalanceSchema,
401
+ exports.UserBalanceWithCheckSchema,
402
+ ].map((s) => tables.push((0, temp_db_1.getSchemaTable)(s, db_1.Dialect.SQLite)));
403
+ return tables;
404
+ };
405
+ exports.getTables = getTables;