@prosopo/datasets-fs 0.2.7 → 0.2.9

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 (230) hide show
  1. package/README.md +21 -0
  2. package/dist/cjs/cli/cli.cjs +55 -0
  3. package/dist/cjs/index.cjs +3 -1
  4. package/dist/cli/cli.d.ts +9 -0
  5. package/dist/cli/cli.d.ts.map +1 -0
  6. package/dist/cli/cli.js +54 -0
  7. package/dist/cli/cli.js.map +1 -0
  8. package/dist/cli/cliCommand.d.ts +17 -0
  9. package/dist/cli/cliCommand.d.ts.map +1 -0
  10. package/dist/cli/cliCommand.js +36 -0
  11. package/dist/cli/cliCommand.js.map +1 -0
  12. package/dist/cli/cliCommandComposite.d.ts +14 -0
  13. package/dist/cli/cliCommandComposite.d.ts.map +1 -0
  14. package/dist/cli/cliCommandComposite.js +34 -0
  15. package/dist/cli/cliCommandComposite.js.map +1 -0
  16. package/dist/cli.js +21 -35
  17. package/dist/cli.js.map +1 -1
  18. package/dist/commands/flatten.d.ts +56 -0
  19. package/dist/commands/flatten.d.ts.map +1 -0
  20. package/dist/commands/flatten.js +103 -0
  21. package/dist/commands/flatten.js.map +1 -0
  22. package/dist/commands/generate.d.ts +114 -0
  23. package/dist/commands/generate.d.ts.map +1 -0
  24. package/dist/commands/generate.js +151 -0
  25. package/dist/commands/generate.js.map +1 -0
  26. package/dist/commands/generateV1.d.ts +174 -0
  27. package/dist/commands/generateV1.d.ts.map +1 -0
  28. package/dist/commands/generateV1.js +178 -0
  29. package/dist/commands/generateV1.js.map +1 -0
  30. package/dist/commands/generateV2.d.ts +184 -0
  31. package/dist/commands/generateV2.d.ts.map +1 -0
  32. package/dist/commands/generateV2.js +204 -0
  33. package/dist/commands/generateV2.js.map +1 -0
  34. package/dist/commands/get.d.ts +35 -0
  35. package/dist/commands/get.d.ts.map +1 -0
  36. package/dist/commands/get.js +81 -0
  37. package/dist/commands/get.js.map +1 -0
  38. package/dist/commands/labels.d.ts +46 -0
  39. package/dist/commands/labels.d.ts.map +1 -0
  40. package/dist/commands/labels.js +43 -0
  41. package/dist/commands/labels.js.map +1 -0
  42. package/dist/commands/relocate.d.ts +68 -0
  43. package/dist/commands/relocate.d.ts.map +1 -0
  44. package/dist/commands/relocate.js +75 -0
  45. package/dist/commands/relocate.js.map +1 -0
  46. package/dist/commands/resize.d.ts +66 -0
  47. package/dist/commands/resize.d.ts.map +1 -0
  48. package/dist/commands/resize.js +118 -0
  49. package/dist/commands/resize.js.map +1 -0
  50. package/dist/{generate/distinct/dummy.d.ts.map → dummy.d.ts.map} +1 -1
  51. package/dist/dummy.js.map +1 -0
  52. package/dist/index.d.ts +1 -1
  53. package/dist/index.d.ts.map +1 -1
  54. package/dist/index.js +1 -1
  55. package/dist/index.js.map +1 -1
  56. package/dist/tests/data/flat_resized/labels.json +3 -0
  57. package/dist/tests/data/flat_resized/relocated_data.json +364 -0
  58. package/dist/tests/mocked.test.d.ts +2 -0
  59. package/dist/tests/mocked.test.d.ts.map +1 -0
  60. package/dist/tests/mocked.test.js +271 -0
  61. package/dist/tests/mocked.test.js.map +1 -0
  62. package/dist/tests/utils.d.ts +35 -0
  63. package/dist/tests/utils.d.ts.map +1 -0
  64. package/dist/tests/utils.js +120 -0
  65. package/dist/tests/utils.js.map +1 -0
  66. package/dist/utils/input.d.ts +24 -0
  67. package/dist/utils/input.d.ts.map +1 -0
  68. package/dist/utils/input.js +34 -0
  69. package/dist/utils/input.js.map +1 -0
  70. package/dist/utils/inputOutput.d.ts +20 -0
  71. package/dist/utils/inputOutput.d.ts.map +1 -0
  72. package/dist/utils/inputOutput.js +10 -0
  73. package/dist/utils/inputOutput.js.map +1 -0
  74. package/dist/utils/output.d.ts +37 -0
  75. package/dist/utils/output.d.ts.map +1 -0
  76. package/dist/utils/output.js +63 -0
  77. package/dist/utils/output.js.map +1 -0
  78. package/package.json +5 -6
  79. package/dist/cjs/cli.cjs +0 -34
  80. package/dist/cjs/flatten/args.cjs +0 -9
  81. package/dist/cjs/flatten/cli.cjs +0 -28
  82. package/dist/cjs/flatten/flatten.cjs +0 -55
  83. package/dist/cjs/generate/args.cjs +0 -16
  84. package/dist/cjs/generate/cli.cjs +0 -50
  85. package/dist/cjs/generate/distinct/args.cjs +0 -11
  86. package/dist/cjs/generate/distinct/cli.cjs +0 -28
  87. package/dist/cjs/generate/distinct/generate.cjs +0 -174
  88. package/dist/cjs/generate/union/args.cjs +0 -12
  89. package/dist/cjs/generate/union/cli.cjs +0 -31
  90. package/dist/cjs/generate/union/generate.cjs +0 -151
  91. package/dist/cjs/generate/util.cjs +0 -21
  92. package/dist/cjs/get/args.cjs +0 -7
  93. package/dist/cjs/get/cli.cjs +0 -20
  94. package/dist/cjs/get/get.cjs +0 -53
  95. package/dist/cjs/labels/args.cjs +0 -7
  96. package/dist/cjs/labels/cli.cjs +0 -20
  97. package/dist/cjs/labels/labels.cjs +0 -21
  98. package/dist/cjs/relocate/args.cjs +0 -9
  99. package/dist/cjs/relocate/cli.cjs +0 -27
  100. package/dist/cjs/relocate/relocate.cjs +0 -36
  101. package/dist/cjs/scale/args.cjs +0 -11
  102. package/dist/cjs/scale/cli.cjs +0 -34
  103. package/dist/cjs/scale/scale.cjs +0 -55
  104. package/dist/flatten/args.d.ts +0 -16
  105. package/dist/flatten/args.d.ts.map +0 -1
  106. package/dist/flatten/args.js +0 -8
  107. package/dist/flatten/args.js.map +0 -1
  108. package/dist/flatten/cli.d.ts +0 -18
  109. package/dist/flatten/cli.d.ts.map +0 -1
  110. package/dist/flatten/cli.js +0 -30
  111. package/dist/flatten/cli.js.map +0 -1
  112. package/dist/flatten/flatten.d.ts +0 -5
  113. package/dist/flatten/flatten.d.ts.map +0 -1
  114. package/dist/flatten/flatten.js +0 -65
  115. package/dist/flatten/flatten.js.map +0 -1
  116. package/dist/flatten/index.d.ts +0 -4
  117. package/dist/flatten/index.d.ts.map +0 -1
  118. package/dist/flatten/index.js +0 -4
  119. package/dist/flatten/index.js.map +0 -1
  120. package/dist/generate/args.d.ts +0 -37
  121. package/dist/generate/args.d.ts.map +0 -1
  122. package/dist/generate/args.js +0 -15
  123. package/dist/generate/args.js.map +0 -1
  124. package/dist/generate/cli.d.ts +0 -32
  125. package/dist/generate/cli.d.ts.map +0 -1
  126. package/dist/generate/cli.js +0 -62
  127. package/dist/generate/cli.js.map +0 -1
  128. package/dist/generate/distinct/args.d.ts +0 -49
  129. package/dist/generate/distinct/args.d.ts.map +0 -1
  130. package/dist/generate/distinct/args.js +0 -10
  131. package/dist/generate/distinct/args.js.map +0 -1
  132. package/dist/generate/distinct/cli.d.ts +0 -20
  133. package/dist/generate/distinct/cli.d.ts.map +0 -1
  134. package/dist/generate/distinct/cli.js +0 -31
  135. package/dist/generate/distinct/cli.js.map +0 -1
  136. package/dist/generate/distinct/dummy.js.map +0 -1
  137. package/dist/generate/distinct/generate.d.ts +0 -5
  138. package/dist/generate/distinct/generate.d.ts.map +0 -1
  139. package/dist/generate/distinct/generate.js +0 -186
  140. package/dist/generate/distinct/generate.js.map +0 -1
  141. package/dist/generate/distinct/index.d.ts +0 -4
  142. package/dist/generate/distinct/index.d.ts.map +0 -1
  143. package/dist/generate/distinct/index.js +0 -4
  144. package/dist/generate/distinct/index.js.map +0 -1
  145. package/dist/generate/index.d.ts +0 -5
  146. package/dist/generate/index.d.ts.map +0 -1
  147. package/dist/generate/index.js +0 -5
  148. package/dist/generate/index.js.map +0 -1
  149. package/dist/generate/union/args.d.ts +0 -52
  150. package/dist/generate/union/args.d.ts.map +0 -1
  151. package/dist/generate/union/args.js +0 -11
  152. package/dist/generate/union/args.js.map +0 -1
  153. package/dist/generate/union/cli.d.ts +0 -22
  154. package/dist/generate/union/cli.d.ts.map +0 -1
  155. package/dist/generate/union/cli.js +0 -35
  156. package/dist/generate/union/cli.js.map +0 -1
  157. package/dist/generate/union/generate.d.ts +0 -5
  158. package/dist/generate/union/generate.d.ts.map +0 -1
  159. package/dist/generate/union/generate.js +0 -168
  160. package/dist/generate/union/generate.js.map +0 -1
  161. package/dist/generate/union/index.d.ts +0 -4
  162. package/dist/generate/union/index.d.ts.map +0 -1
  163. package/dist/generate/union/index.js +0 -4
  164. package/dist/generate/union/index.js.map +0 -1
  165. package/dist/generate/util.d.ts +0 -6
  166. package/dist/generate/util.d.ts.map +0 -1
  167. package/dist/generate/util.js +0 -20
  168. package/dist/generate/util.js.map +0 -1
  169. package/dist/get/args.d.ts +0 -10
  170. package/dist/get/args.d.ts.map +0 -1
  171. package/dist/get/args.js +0 -6
  172. package/dist/get/args.js.map +0 -1
  173. package/dist/get/cli.d.ts +0 -14
  174. package/dist/get/cli.d.ts.map +0 -1
  175. package/dist/get/cli.js +0 -19
  176. package/dist/get/cli.js.map +0 -1
  177. package/dist/get/get.d.ts +0 -5
  178. package/dist/get/get.d.ts.map +0 -1
  179. package/dist/get/get.js +0 -60
  180. package/dist/get/get.js.map +0 -1
  181. package/dist/get/index.d.ts +0 -4
  182. package/dist/get/index.d.ts.map +0 -1
  183. package/dist/get/index.js +0 -4
  184. package/dist/get/index.js.map +0 -1
  185. package/dist/labels/args.d.ts +0 -10
  186. package/dist/labels/args.d.ts.map +0 -1
  187. package/dist/labels/args.js +0 -6
  188. package/dist/labels/args.js.map +0 -1
  189. package/dist/labels/cli.d.ts +0 -14
  190. package/dist/labels/cli.d.ts.map +0 -1
  191. package/dist/labels/cli.js +0 -19
  192. package/dist/labels/cli.js.map +0 -1
  193. package/dist/labels/index.d.ts +0 -4
  194. package/dist/labels/index.d.ts.map +0 -1
  195. package/dist/labels/index.js +0 -4
  196. package/dist/labels/index.js.map +0 -1
  197. package/dist/labels/labels.d.ts +0 -5
  198. package/dist/labels/labels.d.ts.map +0 -1
  199. package/dist/labels/labels.js +0 -22
  200. package/dist/labels/labels.js.map +0 -1
  201. package/dist/relocate/args.d.ts +0 -16
  202. package/dist/relocate/args.d.ts.map +0 -1
  203. package/dist/relocate/args.js +0 -8
  204. package/dist/relocate/args.js.map +0 -1
  205. package/dist/relocate/cli.d.ts +0 -18
  206. package/dist/relocate/cli.d.ts.map +0 -1
  207. package/dist/relocate/cli.js +0 -29
  208. package/dist/relocate/cli.js.map +0 -1
  209. package/dist/relocate/relocate.d.ts +0 -5
  210. package/dist/relocate/relocate.d.ts.map +0 -1
  211. package/dist/relocate/relocate.js +0 -40
  212. package/dist/relocate/relocate.js.map +0 -1
  213. package/dist/scale/args.d.ts +0 -22
  214. package/dist/scale/args.d.ts.map +0 -1
  215. package/dist/scale/args.js +0 -10
  216. package/dist/scale/args.js.map +0 -1
  217. package/dist/scale/cli.d.ts +0 -22
  218. package/dist/scale/cli.d.ts.map +0 -1
  219. package/dist/scale/cli.js +0 -38
  220. package/dist/scale/cli.js.map +0 -1
  221. package/dist/scale/index.d.ts +0 -4
  222. package/dist/scale/index.d.ts.map +0 -1
  223. package/dist/scale/index.js +0 -4
  224. package/dist/scale/index.js.map +0 -1
  225. package/dist/scale/scale.d.ts +0 -5
  226. package/dist/scale/scale.d.ts.map +0 -1
  227. package/dist/scale/scale.js +0 -67
  228. package/dist/scale/scale.js.map +0 -1
  229. /package/dist/{generate/distinct/dummy.d.ts → dummy.d.ts} +0 -0
  230. /package/dist/{generate/distinct/dummy.js → dummy.js} +0 -0
@@ -0,0 +1,184 @@
1
+ /// <reference types="yargs" />
2
+ import { Generate } from './generate.js';
3
+ import { z } from 'zod';
4
+ export declare const ArgsSchema: z.ZodObject<{
5
+ output: z.ZodString;
6
+ overwrite: z.ZodOptional<z.ZodBoolean>;
7
+ allowDuplicates: z.ZodOptional<z.ZodBoolean>;
8
+ labels: z.ZodOptional<z.ZodString>;
9
+ labelled: z.ZodOptional<z.ZodString>;
10
+ unlabelled: z.ZodOptional<z.ZodString>;
11
+ seed: z.ZodNumber;
12
+ size: z.ZodOptional<z.ZodNumber>;
13
+ allowDuplicatesLabelled: z.ZodOptional<z.ZodBoolean>;
14
+ allowDuplicatesUnlabelled: z.ZodOptional<z.ZodBoolean>;
15
+ minCorrect: z.ZodOptional<z.ZodNumber>;
16
+ minIncorrect: z.ZodOptional<z.ZodNumber>;
17
+ minLabelled: z.ZodOptional<z.ZodNumber>;
18
+ maxLabelled: z.ZodOptional<z.ZodNumber>;
19
+ count: z.ZodOptional<z.ZodNumber>;
20
+ }, "strip", z.ZodTypeAny, {
21
+ output: string;
22
+ seed: number;
23
+ overwrite?: boolean | undefined;
24
+ allowDuplicates?: boolean | undefined;
25
+ labels?: string | undefined;
26
+ labelled?: string | undefined;
27
+ unlabelled?: string | undefined;
28
+ size?: number | undefined;
29
+ allowDuplicatesLabelled?: boolean | undefined;
30
+ allowDuplicatesUnlabelled?: boolean | undefined;
31
+ minCorrect?: number | undefined;
32
+ minIncorrect?: number | undefined;
33
+ minLabelled?: number | undefined;
34
+ maxLabelled?: number | undefined;
35
+ count?: number | undefined;
36
+ }, {
37
+ output: string;
38
+ seed: number;
39
+ overwrite?: boolean | undefined;
40
+ allowDuplicates?: boolean | undefined;
41
+ labels?: string | undefined;
42
+ labelled?: string | undefined;
43
+ unlabelled?: string | undefined;
44
+ size?: number | undefined;
45
+ allowDuplicatesLabelled?: boolean | undefined;
46
+ allowDuplicatesUnlabelled?: boolean | undefined;
47
+ minCorrect?: number | undefined;
48
+ minIncorrect?: number | undefined;
49
+ minLabelled?: number | undefined;
50
+ maxLabelled?: number | undefined;
51
+ count?: number | undefined;
52
+ }>;
53
+ export type ArgsSchemaType = typeof ArgsSchema;
54
+ export type Args = z.infer<ArgsSchemaType>;
55
+ export declare class GenerateV2 extends Generate<ArgsSchemaType> {
56
+ #private;
57
+ getArgSchema(): z.ZodObject<{
58
+ output: z.ZodString;
59
+ overwrite: z.ZodOptional<z.ZodBoolean>;
60
+ allowDuplicates: z.ZodOptional<z.ZodBoolean>;
61
+ labels: z.ZodOptional<z.ZodString>;
62
+ labelled: z.ZodOptional<z.ZodString>;
63
+ unlabelled: z.ZodOptional<z.ZodString>;
64
+ seed: z.ZodNumber;
65
+ size: z.ZodOptional<z.ZodNumber>;
66
+ allowDuplicatesLabelled: z.ZodOptional<z.ZodBoolean>;
67
+ allowDuplicatesUnlabelled: z.ZodOptional<z.ZodBoolean>;
68
+ minCorrect: z.ZodOptional<z.ZodNumber>;
69
+ minIncorrect: z.ZodOptional<z.ZodNumber>;
70
+ minLabelled: z.ZodOptional<z.ZodNumber>;
71
+ maxLabelled: z.ZodOptional<z.ZodNumber>;
72
+ count: z.ZodOptional<z.ZodNumber>;
73
+ }, "strip", z.ZodTypeAny, {
74
+ output: string;
75
+ seed: number;
76
+ overwrite?: boolean | undefined;
77
+ allowDuplicates?: boolean | undefined;
78
+ labels?: string | undefined;
79
+ labelled?: string | undefined;
80
+ unlabelled?: string | undefined;
81
+ size?: number | undefined;
82
+ allowDuplicatesLabelled?: boolean | undefined;
83
+ allowDuplicatesUnlabelled?: boolean | undefined;
84
+ minCorrect?: number | undefined;
85
+ minIncorrect?: number | undefined;
86
+ minLabelled?: number | undefined;
87
+ maxLabelled?: number | undefined;
88
+ count?: number | undefined;
89
+ }, {
90
+ output: string;
91
+ seed: number;
92
+ overwrite?: boolean | undefined;
93
+ allowDuplicates?: boolean | undefined;
94
+ labels?: string | undefined;
95
+ labelled?: string | undefined;
96
+ unlabelled?: string | undefined;
97
+ size?: number | undefined;
98
+ allowDuplicatesLabelled?: boolean | undefined;
99
+ allowDuplicatesUnlabelled?: boolean | undefined;
100
+ minCorrect?: number | undefined;
101
+ minIncorrect?: number | undefined;
102
+ minLabelled?: number | undefined;
103
+ maxLabelled?: number | undefined;
104
+ count?: number | undefined;
105
+ }>;
106
+ getDescription(): string;
107
+ getOptions(): {
108
+ [key: string]: import("yargs").Options;
109
+ } & {
110
+ output: {
111
+ alias: string;
112
+ string: boolean;
113
+ demand: boolean;
114
+ description: string;
115
+ };
116
+ overwrite: {
117
+ boolean: boolean;
118
+ description: string;
119
+ };
120
+ } & {
121
+ output: {
122
+ description: string;
123
+ };
124
+ labelled: {
125
+ string: boolean;
126
+ demand: boolean;
127
+ description: string;
128
+ };
129
+ unlabelled: {
130
+ string: boolean;
131
+ demand: boolean;
132
+ description: string;
133
+ };
134
+ seed: {
135
+ number: boolean;
136
+ demand: boolean;
137
+ description: string;
138
+ };
139
+ size: {
140
+ number: boolean;
141
+ description: string;
142
+ };
143
+ labels: {
144
+ string: boolean;
145
+ description: string;
146
+ };
147
+ allowDuplicates: {
148
+ boolean: boolean;
149
+ description: string;
150
+ };
151
+ allowDuplicatesLabelled: {
152
+ boolean: boolean;
153
+ description: string;
154
+ };
155
+ allowDuplicatesUnlabelled: {
156
+ boolean: boolean;
157
+ description: string;
158
+ };
159
+ } & {
160
+ count: {
161
+ number: boolean;
162
+ description: string;
163
+ };
164
+ minCorrect: {
165
+ number: boolean;
166
+ description: string;
167
+ };
168
+ minIncorrect: {
169
+ number: boolean;
170
+ description: string;
171
+ };
172
+ minLabelled: {
173
+ number: boolean;
174
+ description: string;
175
+ };
176
+ maxLabelled: {
177
+ number: boolean;
178
+ description: string;
179
+ };
180
+ };
181
+ private setupTarget;
182
+ _run(args: Args): Promise<void>;
183
+ }
184
+ //# sourceMappingURL=generateV2.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"generateV2.d.ts","sourceRoot":"","sources":["../../src/commands/generateV2.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,QAAQ,EAAoC,MAAM,eAAe,CAAA;AAI1E,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAKvB,eAAO,MAAM,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAMrB,CAAA;AACF,MAAM,MAAM,cAAc,GAAG,OAAO,UAAU,CAAA;AAC9C,MAAM,MAAM,IAAI,GAAG,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAA;AAE1C,qBAAa,UAAW,SAAQ,QAAQ,CAAC,cAAc,CAAC;;IAmBpC,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAIZ,cAAc,IAAI,MAAM;IAIxB,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAyB1B,OAAO,CAAC,WAAW;IAiDG,IAAI,CAAC,IAAI,EAAE,IAAI;CAiHxC"}
@@ -0,0 +1,204 @@
1
+ var _GenerateV2_size, _GenerateV2_minCorrect, _GenerateV2_saltRounds, _GenerateV2_allowDuplicatesLabelled, _GenerateV2_allowDuplicatesUnlabelled, _GenerateV2_minIncorrect, _GenerateV2_minLabelled, _GenerateV2_maxLabelled, _GenerateV2_count, _GenerateV2_nCorrect, _GenerateV2_nIncorrect, _GenerateV2_nLabelled, _GenerateV2_nUnlabelled, _GenerateV2_target, _GenerateV2_targets, _GenerateV2_targetItems, _GenerateV2_notTargetItems;
2
+ import { __classPrivateFieldGet, __classPrivateFieldSet } from "tslib";
3
+ import { CaptchaTypes, CaptchasContainerSchema } from '@prosopo/types';
4
+ import { Generate, ArgsSchema as GenerateArgsSchema } from './generate.js';
5
+ import { ProsopoEnvError } from '@prosopo/common';
6
+ import { at, get, lodash } from '@prosopo/util';
7
+ import { blake2AsHex } from '@polkadot/util-crypto';
8
+ import { z } from 'zod';
9
+ import bcrypt from 'bcrypt';
10
+ import cliProgress from 'cli-progress';
11
+ import fs from 'fs';
12
+ export const ArgsSchema = GenerateArgsSchema.extend({
13
+ minCorrect: z.number().optional(),
14
+ minIncorrect: z.number().optional(),
15
+ minLabelled: z.number().optional(),
16
+ maxLabelled: z.number().optional(),
17
+ count: z.number().optional(),
18
+ });
19
+ export class GenerateV2 extends Generate {
20
+ constructor() {
21
+ super(...arguments);
22
+ _GenerateV2_size.set(this, 0);
23
+ _GenerateV2_minCorrect.set(this, 0);
24
+ _GenerateV2_saltRounds.set(this, 10);
25
+ _GenerateV2_allowDuplicatesLabelled.set(this, false);
26
+ _GenerateV2_allowDuplicatesUnlabelled.set(this, false);
27
+ _GenerateV2_minIncorrect.set(this, 0);
28
+ _GenerateV2_minLabelled.set(this, 0);
29
+ _GenerateV2_maxLabelled.set(this, 0);
30
+ _GenerateV2_count.set(this, 0);
31
+ _GenerateV2_nCorrect.set(this, 0);
32
+ _GenerateV2_nIncorrect.set(this, 0);
33
+ _GenerateV2_nLabelled.set(this, 0);
34
+ _GenerateV2_nUnlabelled.set(this, 0);
35
+ _GenerateV2_target.set(this, '');
36
+ _GenerateV2_targets.set(this, []);
37
+ _GenerateV2_targetItems.set(this, []);
38
+ _GenerateV2_notTargetItems.set(this, []);
39
+ }
40
+ getArgSchema() {
41
+ return ArgsSchema;
42
+ }
43
+ getDescription() {
44
+ return 'Generate distinct captchas producing captcha challenges comprising one or more rounds, mixing labelled and unlabelled data into a single round';
45
+ }
46
+ getOptions() {
47
+ return lodash().merge(super.getOptions(), {
48
+ count: {
49
+ number: true,
50
+ description: 'Number of captchas to generate',
51
+ },
52
+ minCorrect: {
53
+ number: true,
54
+ description: 'Minimum number of correct images in each captcha',
55
+ },
56
+ minIncorrect: {
57
+ number: true,
58
+ description: 'Minimum number of incorrect images in each captcha',
59
+ },
60
+ minLabelled: {
61
+ number: true,
62
+ description: 'Minimum number of labelled images in each captcha',
63
+ },
64
+ maxLabelled: {
65
+ number: true,
66
+ description: 'Maximum number of labelled images in each captcha',
67
+ },
68
+ });
69
+ }
70
+ setupTarget(i) {
71
+ const _ = lodash();
72
+ if (this.targets.length <= 1) {
73
+ throw new ProsopoEnvError(new Error(`not enough different labels in labelled data`), 'DATASET.NOT_ENOUGH_LABELS');
74
+ }
75
+ // uniformly sample targets
76
+ const target = at(this.targets, i % this.targets.length);
77
+ const notTargets = this.targets.filter((t) => t !== target);
78
+ // how many labelled images should be in the captcha?
79
+ const nLabelled = _.random(__classPrivateFieldGet(this, _GenerateV2_minLabelled, "f"), __classPrivateFieldGet(this, _GenerateV2_maxLabelled, "f"));
80
+ // how many correct labelled images should be in the captcha?
81
+ const maxCorrect = nLabelled - __classPrivateFieldGet(this, _GenerateV2_minCorrect, "f");
82
+ const nCorrect = _.random(__classPrivateFieldGet(this, _GenerateV2_minCorrect, "f"), maxCorrect);
83
+ const nIncorrect = nLabelled - nCorrect;
84
+ const nUnlabelled = __classPrivateFieldGet(this, _GenerateV2_size, "f") - nLabelled;
85
+ const targetItems = get(this.labelToImages, target);
86
+ const notTargetItems = notTargets.map((notTarget) => get(this.labelToImages, notTarget)).flat();
87
+ if (nUnlabelled > this.unlabelled.length) {
88
+ throw new ProsopoEnvError(new Error(`not enough unlabelled data`), 'DATASET.NOT_ENOUGH_IMAGES');
89
+ }
90
+ if (nCorrect > targetItems.length) {
91
+ throw new ProsopoEnvError(new Error(`not enough images for target (${target})`), 'DATASET.NOT_ENOUGH_IMAGES');
92
+ }
93
+ if (nIncorrect > notTargetItems.length) {
94
+ throw new ProsopoEnvError(new Error(`not enough non-matching images for target (${target})`), 'DATASET.NOT_ENOUGH_IMAGES');
95
+ }
96
+ __classPrivateFieldSet(this, _GenerateV2_nCorrect, nCorrect, "f");
97
+ __classPrivateFieldSet(this, _GenerateV2_nIncorrect, nIncorrect, "f");
98
+ __classPrivateFieldSet(this, _GenerateV2_nLabelled, nLabelled, "f");
99
+ __classPrivateFieldSet(this, _GenerateV2_nUnlabelled, nUnlabelled, "f");
100
+ __classPrivateFieldSet(this, _GenerateV2_target, target, "f");
101
+ __classPrivateFieldSet(this, _GenerateV2_targets, notTargets, "f");
102
+ __classPrivateFieldSet(this, _GenerateV2_targetItems, targetItems, "f");
103
+ __classPrivateFieldSet(this, _GenerateV2_notTargetItems, notTargetItems, "f");
104
+ }
105
+ async _run(args) {
106
+ await super._run(args);
107
+ const outFile = args.output;
108
+ // get lodash (with seeded rng)
109
+ const _ = lodash();
110
+ __classPrivateFieldSet(this, _GenerateV2_size, args.size || 9, "f");
111
+ __classPrivateFieldSet(this, _GenerateV2_minCorrect, args.minCorrect || 1, "f");
112
+ __classPrivateFieldSet(this, _GenerateV2_saltRounds, 10, "f");
113
+ __classPrivateFieldSet(this, _GenerateV2_allowDuplicatesLabelled, args.allowDuplicatesLabelled || args.allowDuplicates || false, "f");
114
+ __classPrivateFieldSet(this, _GenerateV2_allowDuplicatesUnlabelled, args.allowDuplicatesUnlabelled || args.allowDuplicates || false, "f");
115
+ __classPrivateFieldSet(this, _GenerateV2_minIncorrect, Math.max(args.minIncorrect || 1, 1), "f"); // at least 1 incorrect image
116
+ __classPrivateFieldSet(this, _GenerateV2_minLabelled, __classPrivateFieldGet(this, _GenerateV2_minCorrect, "f") + __classPrivateFieldGet(this, _GenerateV2_minIncorrect, "f"), "f"); // min incorrect + correct
117
+ __classPrivateFieldSet(this, _GenerateV2_maxLabelled, Math.min(args.maxLabelled || __classPrivateFieldGet(this, _GenerateV2_size, "f"), __classPrivateFieldGet(this, _GenerateV2_size, "f")), "f"); // at least 1 labelled image
118
+ __classPrivateFieldSet(this, _GenerateV2_count, args.count || 0, "f");
119
+ // the captcha contains n images. Each of these images are either labelled, being correct or incorrect against the target, or unlabelled. To construct one of these captchas, we need to decide how many of the images should be labelled vs unlabelled, and then how many of the labelled images should be correct vs incorrect
120
+ // in the traditional captcha, two rounds are produced, one with labelled images and the other with unlabelled images. This gives 18 images overall, 9 labels produced.
121
+ // the parameters for generation can regulate how many labels are collected vs how much of a test the captcha posses. E.g. 18 images could have 16 unlabelled and 2 labelled, or 2 unlabelled and 16 labelled. The former is a better test of the user being human, but the latter is a better for maximising label collection.
122
+ // if we focus on a single captcha round of 9 images, we must have at least 1 labelled correct image in the captcha for it to work, otherwise it's just a labelling phase, which normally isn't a problem but if we're treating these as tests for humanity too then we need some kind of test in there. (e.g. we abolish the labelled then unlabelled pattern of the challenge rounds in favour of mixing labelled and unlabelled data, but we then run a small chance of serving two completely unlabelled rounds if we don't set the min number of labelled images to 1 per captcha round)
123
+ const bar = new cliProgress.SingleBar({}, cliProgress.Presets.shades_classic);
124
+ bar.start(__classPrivateFieldGet(this, _GenerateV2_count, "f"), 0);
125
+ // generate n captchas
126
+ const captchas = [];
127
+ for (let i = 0; i < __classPrivateFieldGet(this, _GenerateV2_count, "f"); i++) {
128
+ bar.increment();
129
+ // this.logger.info(`generating captcha ${i + 1} of ${count}`)
130
+ this.setupTarget(i);
131
+ // get the correct items
132
+ const correctItems = _.sampleSize(__classPrivateFieldGet(this, _GenerateV2_targetItems, "f"), __classPrivateFieldGet(this, _GenerateV2_nCorrect, "f"));
133
+ // get the incorrect items
134
+ const incorrectItems = _.sampleSize(__classPrivateFieldGet(this, _GenerateV2_notTargetItems, "f"), __classPrivateFieldGet(this, _GenerateV2_nIncorrect, "f"));
135
+ // get the unlabelled items
136
+ const unlabelledItems = new Set();
137
+ while (unlabelledItems.size < __classPrivateFieldGet(this, _GenerateV2_size, "f") - __classPrivateFieldGet(this, _GenerateV2_nLabelled, "f")) {
138
+ // get a random image from the unlabelled data
139
+ const image = at(this.unlabelled, _.random(0, this.unlabelled.length - 1));
140
+ unlabelledItems.add(image);
141
+ }
142
+ const itemsConcat = [...correctItems, ...incorrectItems, ...unlabelledItems];
143
+ let indices = [...Array(itemsConcat.length).keys()];
144
+ indices = _.shuffle(indices);
145
+ const items = indices
146
+ .map((i) => at(itemsConcat, i))
147
+ .map((item) => {
148
+ return {
149
+ data: item.data,
150
+ hash: item.hash,
151
+ type: item.type,
152
+ };
153
+ });
154
+ // the first n indices are the correct items
155
+ const solution = indices
156
+ .map((index, i) => {
157
+ return {
158
+ pre: index,
159
+ post: i, // the index of the item in the shuffled array
160
+ };
161
+ })
162
+ .filter((item) => item.pre < correctItems.length) // keep all items that were in the first n slots of the original item array - these were the correct items
163
+ .map((item) => {
164
+ return item.post; // return the index in the shuffled array
165
+ });
166
+ // the unlabelled indices were after the correct and incorrect
167
+ const unlabelledIndices = indices
168
+ .map((index, i) => {
169
+ return {
170
+ pre: index,
171
+ post: i, // the index of the item in the shuffled array
172
+ };
173
+ })
174
+ .filter((item) => item.pre >= correctItems.length + incorrectItems.length) // keep all items that were in the first n slots of the original item array - these were the correct items
175
+ .map((item) => {
176
+ return item.post; // return the index in the shuffled array
177
+ });
178
+ const salt = blake2AsHex(bcrypt.genSaltSync(__classPrivateFieldGet(this, _GenerateV2_saltRounds, "f")));
179
+ // create the captcha
180
+ const captcha = {
181
+ salt,
182
+ target: __classPrivateFieldGet(this, _GenerateV2_target, "f"),
183
+ items,
184
+ solution,
185
+ unlabelled: unlabelledIndices,
186
+ };
187
+ captchas.push(captcha);
188
+ }
189
+ bar.stop();
190
+ // write to file
191
+ const output = {
192
+ captchas,
193
+ format: CaptchaTypes.SelectAll,
194
+ };
195
+ // verify the output
196
+ this.logger.info('verifying data');
197
+ CaptchasContainerSchema.parse(output);
198
+ this.logger.info(`writing data`);
199
+ fs.mkdirSync(args.output.split('/').slice(0, -1).join('/'), { recursive: true });
200
+ fs.writeFileSync(outFile, JSON.stringify(output, null, 4));
201
+ }
202
+ }
203
+ _GenerateV2_size = new WeakMap(), _GenerateV2_minCorrect = new WeakMap(), _GenerateV2_saltRounds = new WeakMap(), _GenerateV2_allowDuplicatesLabelled = new WeakMap(), _GenerateV2_allowDuplicatesUnlabelled = new WeakMap(), _GenerateV2_minIncorrect = new WeakMap(), _GenerateV2_minLabelled = new WeakMap(), _GenerateV2_maxLabelled = new WeakMap(), _GenerateV2_count = new WeakMap(), _GenerateV2_nCorrect = new WeakMap(), _GenerateV2_nIncorrect = new WeakMap(), _GenerateV2_nLabelled = new WeakMap(), _GenerateV2_nUnlabelled = new WeakMap(), _GenerateV2_target = new WeakMap(), _GenerateV2_targets = new WeakMap(), _GenerateV2_targetItems = new WeakMap(), _GenerateV2_notTargetItems = new WeakMap();
204
+ //# sourceMappingURL=generateV2.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"generateV2.js","sourceRoot":"","sources":["../../src/commands/generateV2.ts"],"names":[],"mappings":";;AAAA,OAAO,EAAE,YAAY,EAA8B,uBAAuB,EAAqB,MAAM,gBAAgB,CAAA;AACrH,OAAO,EAAE,QAAQ,EAAE,UAAU,IAAI,kBAAkB,EAAE,MAAM,eAAe,CAAA;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAA;AACjD,OAAO,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,eAAe,CAAA;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAA;AACnD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,MAAM,MAAM,QAAQ,CAAA;AAC3B,OAAO,WAAW,MAAM,cAAc,CAAA;AACtC,OAAO,EAAE,MAAM,IAAI,CAAA;AAEnB,MAAM,CAAC,MAAM,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC;IAChD,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACjC,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACnC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC/B,CAAC,CAAA;AAIF,MAAM,OAAO,UAAW,SAAQ,QAAwB;IAAxD;;QACI,2BAAQ,CAAC,EAAA;QACT,iCAAc,CAAC,EAAA;QACf,iCAAc,EAAE,EAAA;QAChB,8CAA2B,KAAK,EAAA;QAChC,gDAA6B,KAAK,EAAA;QAClC,mCAAgB,CAAC,EAAA;QACjB,kCAAe,CAAC,EAAA;QAChB,kCAAe,CAAC,EAAA;QAChB,4BAAS,CAAC,EAAA;QACV,+BAAY,CAAC,EAAA;QACb,iCAAc,CAAC,EAAA;QACf,gCAAa,CAAC,EAAA;QACd,kCAAe,CAAC,EAAA;QAChB,6BAAU,EAAE,EAAA;QACZ,8BAAqB,EAAE,EAAA;QACvB,kCAAuB,EAAE,EAAA;QACzB,qCAA0B,EAAE,EAAA;IAqMhC,CAAC;IAnMmB,YAAY;QACxB,OAAO,UAAU,CAAA;IACrB,CAAC;IAEe,cAAc;QAC1B,OAAO,gJAAgJ,CAAA;IAC3J,CAAC;IAEe,UAAU;QACtB,OAAO,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE;YACtC,KAAK,EAAE;gBACH,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,gCAAgC;aAChD;YACD,UAAU,EAAE;gBACR,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,kDAAkD;aAClE;YACD,YAAY,EAAE;gBACV,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,oDAAoD;aACpE;YACD,WAAW,EAAE;gBACT,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,mDAAmD;aACnE;YACD,WAAW,EAAE;gBACT,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,mDAAmD;aACnE;SACJ,CAAC,CAAA;IACN,CAAC;IAEO,WAAW,CAAC,CAAS;QACzB,MAAM,CAAC,GAAG,MAAM,EAAE,CAAA;QAClB,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,EAAE;YAC1B,MAAM,IAAI,eAAe,CACrB,IAAI,KAAK,CAAC,8CAA8C,CAAC,EACzD,2BAA2B,CAC9B,CAAA;SACJ;QAED,2BAA2B;QAC3B,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;QACxD,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,MAAM,CAAC,CAAA;QAC3D,qDAAqD;QACrD,MAAM,SAAS,GAAG,CAAC,CAAC,MAAM,CAAC,uBAAA,IAAI,+BAAa,EAAE,uBAAA,IAAI,+BAAa,CAAC,CAAA;QAChE,6DAA6D;QAC7D,MAAM,UAAU,GAAG,SAAS,GAAG,uBAAA,IAAI,8BAAY,CAAA;QAC/C,MAAM,QAAQ,GAAG,CAAC,CAAC,MAAM,CAAC,uBAAA,IAAI,8BAAY,EAAE,UAAU,CAAC,CAAA;QACvD,MAAM,UAAU,GAAG,SAAS,GAAG,QAAQ,CAAA;QACvC,MAAM,WAAW,GAAG,uBAAA,IAAI,wBAAM,GAAG,SAAS,CAAA;QAE1C,MAAM,WAAW,GAAG,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,MAAM,CAAC,CAAA;QACnD,MAAM,cAAc,GAAW,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;QAEvG,IAAI,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE;YACtC,MAAM,IAAI,eAAe,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,EAAE,2BAA2B,CAAC,CAAA;SAClG;QACD,IAAI,QAAQ,GAAG,WAAW,CAAC,MAAM,EAAE;YAC/B,MAAM,IAAI,eAAe,CACrB,IAAI,KAAK,CAAC,iCAAiC,MAAM,GAAG,CAAC,EACrD,2BAA2B,CAC9B,CAAA;SACJ;QACD,IAAI,UAAU,GAAG,cAAc,CAAC,MAAM,EAAE;YACpC,MAAM,IAAI,eAAe,CACrB,IAAI,KAAK,CAAC,8CAA8C,MAAM,GAAG,CAAC,EAClE,2BAA2B,CAC9B,CAAA;SACJ;QAED,uBAAA,IAAI,wBAAa,QAAQ,MAAA,CAAA;QACzB,uBAAA,IAAI,0BAAe,UAAU,MAAA,CAAA;QAC7B,uBAAA,IAAI,yBAAc,SAAS,MAAA,CAAA;QAC3B,uBAAA,IAAI,2BAAgB,WAAW,MAAA,CAAA;QAC/B,uBAAA,IAAI,sBAAW,MAAM,MAAA,CAAA;QACrB,uBAAA,IAAI,uBAAY,UAAU,MAAA,CAAA;QAC1B,uBAAA,IAAI,2BAAgB,WAAW,MAAA,CAAA;QAC/B,uBAAA,IAAI,8BAAmB,cAAc,MAAA,CAAA;IACzC,CAAC;IAEe,KAAK,CAAC,IAAI,CAAC,IAAU;QACjC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAEtB,MAAM,OAAO,GAAW,IAAI,CAAC,MAAM,CAAA;QAEnC,+BAA+B;QAC/B,MAAM,CAAC,GAAG,MAAM,EAAE,CAAA;QAElB,uBAAA,IAAI,oBAAS,IAAI,CAAC,IAAI,IAAI,CAAC,MAAA,CAAA;QAC3B,uBAAA,IAAI,0BAAe,IAAI,CAAC,UAAU,IAAI,CAAC,MAAA,CAAA;QACvC,uBAAA,IAAI,0BAAe,EAAE,MAAA,CAAA;QACrB,uBAAA,IAAI,uCAA4B,IAAI,CAAC,uBAAuB,IAAI,IAAI,CAAC,eAAe,IAAI,KAAK,MAAA,CAAA;QAC7F,uBAAA,IAAI,yCAA8B,IAAI,CAAC,yBAAyB,IAAI,IAAI,CAAC,eAAe,IAAI,KAAK,MAAA,CAAA;QACjG,uBAAA,IAAI,4BAAiB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,EAAE,CAAC,CAAC,MAAA,CAAA,CAAC,6BAA6B;QACtF,uBAAA,IAAI,2BAAgB,uBAAA,IAAI,8BAAY,GAAG,uBAAA,IAAI,gCAAc,MAAA,CAAA,CAAC,0BAA0B;QACpF,uBAAA,IAAI,2BAAgB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,IAAI,uBAAA,IAAI,wBAAM,EAAE,uBAAA,IAAI,wBAAM,CAAC,MAAA,CAAA,CAAC,4BAA4B;QACrG,uBAAA,IAAI,qBAAU,IAAI,CAAC,KAAK,IAAI,CAAC,MAAA,CAAA;QAE7B,gUAAgU;QAChU,uKAAuK;QACvK,+TAA+T;QAC/T,6jBAA6jB;QAE7jB,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,EAAE,EAAE,WAAW,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;QAC7E,GAAG,CAAC,KAAK,CAAC,uBAAA,IAAI,yBAAO,EAAE,CAAC,CAAC,CAAA;QAEzB,sBAAsB;QACtB,MAAM,QAAQ,GAAuB,EAAE,CAAA;QACvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,uBAAA,IAAI,yBAAO,EAAE,CAAC,EAAE,EAAE;YAClC,GAAG,CAAC,SAAS,EAAE,CAAA;YACf,8DAA8D;YAC9D,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAA;YAEnB,wBAAwB;YACxB,MAAM,YAAY,GAAW,CAAC,CAAC,UAAU,CAAC,uBAAA,IAAI,+BAAa,EAAE,uBAAA,IAAI,4BAAU,CAAC,CAAA;YAE5E,0BAA0B;YAC1B,MAAM,cAAc,GAAW,CAAC,CAAC,UAAU,CAAC,uBAAA,IAAI,kCAAgB,EAAE,uBAAA,IAAI,8BAAY,CAAC,CAAA;YAEnF,2BAA2B;YAC3B,MAAM,eAAe,GAAG,IAAI,GAAG,EAAQ,CAAA;YACvC,OAAO,eAAe,CAAC,IAAI,GAAG,uBAAA,IAAI,wBAAM,GAAG,uBAAA,IAAI,6BAAW,EAAE;gBACxD,8CAA8C;gBAC9C,MAAM,KAAK,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAA;gBAC1E,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;aAC7B;YAED,MAAM,WAAW,GAAW,CAAC,GAAG,YAAY,EAAE,GAAG,cAAc,EAAE,GAAG,eAAe,CAAC,CAAA;YACpF,IAAI,OAAO,GAAa,CAAC,GAAG,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC,CAAA;YAC7D,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;YAC5B,MAAM,KAAK,GAAG,OAAO;iBAChB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;iBAC9B,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBACV,OAAO;oBACH,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI,EAAE,IAAI,CAAC,IAAI;iBAClB,CAAA;YACL,CAAC,CAAC,CAAA;YAEN,4CAA4C;YAC5C,MAAM,QAAQ,GAAkB,OAAO;iBAClC,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;gBACd,OAAO;oBACH,GAAG,EAAE,KAAK;oBACV,IAAI,EAAE,CAAC,EAAE,8CAA8C;iBAC1D,CAAA;YACL,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,0GAA0G;iBAC3J,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBACV,OAAO,IAAI,CAAC,IAAI,CAAA,CAAC,yCAAyC;YAC9D,CAAC,CAAC,CAAA;YAEN,8DAA8D;YAC9D,MAAM,iBAAiB,GAAkB,OAAO;iBAC3C,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;gBACd,OAAO;oBACH,GAAG,EAAE,KAAK;oBACV,IAAI,EAAE,CAAC,EAAE,8CAA8C;iBAC1D,CAAA;YACL,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,IAAI,YAAY,CAAC,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,0GAA0G;iBACpL,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBACV,OAAO,IAAI,CAAC,IAAI,CAAA,CAAC,yCAAyC;YAC9D,CAAC,CAAC,CAAA;YAEN,MAAM,IAAI,GAAG,WAAW,CAAC,MAAM,CAAC,WAAW,CAAC,uBAAA,IAAI,8BAAY,CAAC,CAAC,CAAA;YAC9D,qBAAqB;YACrB,MAAM,OAAO,GAAqB;gBAC9B,IAAI;gBACJ,MAAM,EAAE,uBAAA,IAAI,0BAAQ;gBACpB,KAAK;gBACL,QAAQ;gBACR,UAAU,EAAE,iBAAiB;aAChC,CAAA;YACD,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;SACzB;QACD,GAAG,CAAC,IAAI,EAAE,CAAA;QAEV,gBAAgB;QAChB,MAAM,MAAM,GAAa;YACrB,QAAQ;YACR,MAAM,EAAE,YAAY,CAAC,SAAS;SACjC,CAAA;QAED,oBAAoB;QACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAClC,uBAAuB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;QAErC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;QAChC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;QAChF,EAAE,CAAC,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;IAC9D,CAAC;CACJ"}
@@ -0,0 +1,35 @@
1
+ import { InputCliCommand } from '../utils/input.js';
2
+ import { z } from 'zod';
3
+ export declare const ArgsSchema: z.ZodObject<{
4
+ input: z.ZodString;
5
+ }, "strip", z.ZodTypeAny, {
6
+ input: string;
7
+ }, {
8
+ input: string;
9
+ }>;
10
+ export type ArgsSchemaType = typeof ArgsSchema;
11
+ export type Args = z.infer<ArgsSchemaType>;
12
+ export declare class Get extends InputCliCommand<ArgsSchemaType> {
13
+ getArgSchema(): z.ZodObject<{
14
+ input: z.ZodString;
15
+ }, "strip", z.ZodTypeAny, {
16
+ input: string;
17
+ }, {
18
+ input: string;
19
+ }>;
20
+ getOptions(): {
21
+ input: {
22
+ string: boolean;
23
+ alias: string;
24
+ demand: boolean;
25
+ description: string;
26
+ };
27
+ } & {
28
+ input: {
29
+ description: string;
30
+ };
31
+ };
32
+ _run(args: Args): Promise<void>;
33
+ getDescription(): string;
34
+ }
35
+ //# sourceMappingURL=get.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"get.d.ts","sourceRoot":"","sources":["../../src/commands/get.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmB,eAAe,EAAE,MAAM,mBAAmB,CAAA;AAGpE,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAIvB,eAAO,MAAM,UAAU;;;;;;EAA6B,CAAA;AACpD,MAAM,MAAM,cAAc,GAAG,OAAO,UAAU,CAAA;AAC9C,MAAM,MAAM,IAAI,GAAG,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAA;AAC1C,qBAAa,GAAI,SAAQ,eAAe,CAAC,cAAc,CAAC;IACpC,YAAY;;;;;;;IAIZ,UAAU;;;;;;;;;;;;IAQJ,IAAI,CAAC,IAAI,EAAE,IAAI;IAyDrB,cAAc,IAAI,MAAM;CAG3C"}
@@ -0,0 +1,81 @@
1
+ import { InputArgsSchema, InputCliCommand } from '../utils/input.js';
2
+ import { get } from '@prosopo/util';
3
+ import { lodash } from '@prosopo/util';
4
+ import { z } from 'zod';
5
+ import cliProgress from 'cli-progress';
6
+ import fs from 'fs';
7
+ export const ArgsSchema = InputArgsSchema.extend({});
8
+ export class Get extends InputCliCommand {
9
+ getArgSchema() {
10
+ return ArgsSchema;
11
+ }
12
+ getOptions() {
13
+ return lodash().merge(super.getOptions(), {
14
+ input: {
15
+ description: 'JSON file containing urls under a "data" key',
16
+ },
17
+ });
18
+ }
19
+ async _run(args) {
20
+ await super._run(args);
21
+ const bar = new cliProgress.SingleBar({}, cliProgress.Presets.shades_classic);
22
+ const list = [];
23
+ const traverse = async (data) => {
24
+ if (data instanceof Array) {
25
+ for (let i = 0; i < data.length; i++) {
26
+ data[i] = await traverse(data[i]);
27
+ }
28
+ }
29
+ else if (data instanceof Object) {
30
+ for (const key of Object.keys(data)) {
31
+ if (key == 'data') {
32
+ const value = get(data, key);
33
+ const url = z.string().parse(value);
34
+ list.push(url);
35
+ }
36
+ else {
37
+ await traverse(get(data, key));
38
+ }
39
+ }
40
+ }
41
+ return data;
42
+ };
43
+ const file = args.input;
44
+ // read the map file
45
+ const data = JSON.parse(fs.readFileSync(file, 'utf8'));
46
+ await traverse(data);
47
+ bar.start(list.length, 0);
48
+ for (const url of list) {
49
+ bar.increment();
50
+ if (url.startsWith('http')) {
51
+ try {
52
+ const response = await fetch(url);
53
+ if (!response.ok) {
54
+ this.logger.error(`GET ${url} ${response.status} ${response.statusText}`);
55
+ }
56
+ else {
57
+ this.logger.log(`GET ${url} OK`);
58
+ }
59
+ }
60
+ catch (err) {
61
+ this.logger.error(err);
62
+ }
63
+ }
64
+ else {
65
+ // resolve locally
66
+ try {
67
+ fs.readFileSync(url);
68
+ this.logger.log(`GET ${url} OK`);
69
+ }
70
+ catch (err) {
71
+ this.logger.error(`GET ${url} ${err}`);
72
+ }
73
+ }
74
+ }
75
+ bar.stop();
76
+ }
77
+ getDescription() {
78
+ return 'Test a GET request at image URLs';
79
+ }
80
+ }
81
+ //# sourceMappingURL=get.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"get.js","sourceRoot":"","sources":["../../src/commands/get.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAA;AACpE,OAAO,EAAE,GAAG,EAAE,MAAM,eAAe,CAAA;AACnC,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAA;AACtC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,WAAW,MAAM,cAAc,CAAA;AACtC,OAAO,EAAE,MAAM,IAAI,CAAA;AAEnB,MAAM,CAAC,MAAM,UAAU,GAAG,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;AAGpD,MAAM,OAAO,GAAI,SAAQ,eAA+B;IACpC,YAAY;QACxB,OAAO,UAAU,CAAA;IACrB,CAAC;IAEe,UAAU;QACtB,OAAO,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE;YACtC,KAAK,EAAE;gBACH,WAAW,EAAE,8CAA8C;aAC9D;SACJ,CAAC,CAAA;IACN,CAAC;IAEe,KAAK,CAAC,IAAI,CAAC,IAAU;QACjC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACtB,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,EAAE,EAAE,WAAW,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;QAE7E,MAAM,IAAI,GAAa,EAAE,CAAA;QACzB,MAAM,QAAQ,GAAG,KAAK,EAAE,IAAS,EAAE,EAAE;YACjC,IAAI,IAAI,YAAY,KAAK,EAAE;gBACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;oBAClC,IAAI,CAAC,CAAC,CAAC,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;iBACpC;aACJ;iBAAM,IAAI,IAAI,YAAY,MAAM,EAAE;gBAC/B,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;oBACjC,IAAI,GAAG,IAAI,MAAM,EAAE;wBACf,MAAM,KAAK,GAAG,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;wBAC5B,MAAM,GAAG,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;wBACnC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;qBACjB;yBAAM;wBACH,MAAM,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAA;qBACjC;iBACJ;aACJ;YACD,OAAO,IAAI,CAAA;QACf,CAAC,CAAA;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAA;QAEvB,oBAAoB;QACpB,MAAM,IAAI,GAAQ,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAA;QAC3D,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAA;QAEpB,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACzB,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE;YACpB,GAAG,CAAC,SAAS,EAAE,CAAA;YACf,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE;gBACxB,IAAI;oBACA,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,CAAA;oBACjC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE;wBACd,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAA;qBAC5E;yBAAM;wBACH,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,GAAG,KAAK,CAAC,CAAA;qBACnC;iBACJ;gBAAC,OAAO,GAAG,EAAE;oBACV,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;iBACzB;aACJ;iBAAM;gBACH,kBAAkB;gBAClB,IAAI;oBACA,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;oBACpB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,GAAG,KAAK,CAAC,CAAA;iBACnC;gBAAC,OAAO,GAAG,EAAE;oBACV,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,GAAG,EAAE,CAAC,CAAA;iBACzC;aACJ;SACJ;QACD,GAAG,CAAC,IAAI,EAAE,CAAA;IACd,CAAC;IAEe,cAAc;QAC1B,OAAO,kCAAkC,CAAA;IAC7C,CAAC;CACJ"}
@@ -0,0 +1,46 @@
1
+ /// <reference types="yargs" />
2
+ import { InputOutputCliCommand } from '../utils/inputOutput.js';
3
+ import { z } from 'zod';
4
+ export declare const ArgsSchema: z.ZodObject<{
5
+ input: z.ZodString;
6
+ output: z.ZodString;
7
+ overwrite: z.ZodOptional<z.ZodBoolean>;
8
+ }, "strip", z.ZodTypeAny, {
9
+ input: string;
10
+ output: string;
11
+ overwrite?: boolean | undefined;
12
+ }, {
13
+ input: string;
14
+ output: string;
15
+ overwrite?: boolean | undefined;
16
+ }>;
17
+ export type ArgsSchemaType = typeof ArgsSchema;
18
+ export type Args = z.infer<ArgsSchemaType>;
19
+ export declare class Labels extends InputOutputCliCommand<ArgsSchemaType> {
20
+ getArgSchema(): z.ZodObject<{
21
+ input: z.ZodString;
22
+ output: z.ZodString;
23
+ overwrite: z.ZodOptional<z.ZodBoolean>;
24
+ }, "strip", z.ZodTypeAny, {
25
+ input: string;
26
+ output: string;
27
+ overwrite?: boolean | undefined;
28
+ }, {
29
+ input: string;
30
+ output: string;
31
+ overwrite?: boolean | undefined;
32
+ }>;
33
+ getOptions(): {
34
+ [key: string]: import("yargs").Options;
35
+ } & {
36
+ input: {
37
+ description: string;
38
+ };
39
+ output: {
40
+ description: string;
41
+ };
42
+ };
43
+ _run(args: Args): Promise<void>;
44
+ getDescription(): string;
45
+ }
46
+ //# sourceMappingURL=labels.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"labels.d.ts","sourceRoot":"","sources":["../../src/commands/labels.ts"],"names":[],"mappings":";AAAA,OAAO,EAAkD,qBAAqB,EAAE,MAAM,yBAAyB,CAAA;AAI/G,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,eAAO,MAAM,UAAU;;;;;;;;;;;;EAAmC,CAAA;AAC1D,MAAM,MAAM,cAAc,GAAG,OAAO,UAAU,CAAA;AAC9C,MAAM,MAAM,IAAI,GAAG,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAA;AAE1C,qBAAa,MAAO,SAAQ,qBAAqB,CAAC,cAAc,CAAC;IAC7C,YAAY;;;;;;;;;;;;;IAIZ,UAAU;;;;;;;;;;IAWJ,IAAI,CAAC,IAAI,EAAE,IAAI;IAuBrB,cAAc,IAAI,MAAM;CAG3C"}
@@ -0,0 +1,43 @@
1
+ import { InputOutputArgsSchema as InputOutputArgsSchema, InputOutputCliCommand } from '../utils/inputOutput.js';
2
+ import { LabelledDataSchema } from '@prosopo/types';
3
+ import { ProsopoEnvError } from '@prosopo/common';
4
+ import { lodash } from '@prosopo/util';
5
+ import fs from 'fs';
6
+ export const ArgsSchema = InputOutputArgsSchema.extend({});
7
+ export class Labels extends InputOutputCliCommand {
8
+ getArgSchema() {
9
+ return ArgsSchema;
10
+ }
11
+ getOptions() {
12
+ return lodash().merge(super.getOptions(), {
13
+ input: {
14
+ description: 'JSON file containing labelled data',
15
+ },
16
+ output: {
17
+ description: 'Where to put the JSON file containing labels',
18
+ },
19
+ });
20
+ }
21
+ async _run(args) {
22
+ await super._run(args);
23
+ const file = args.input;
24
+ if (!fs.existsSync(file)) {
25
+ throw new ProsopoEnvError(new Error(`file does not exist: ${file}`), 'FS.FILE_NOT_FOUND');
26
+ }
27
+ const labelled = file
28
+ ? LabelledDataSchema.parse(JSON.parse(fs.readFileSync(file, 'utf8'))).items
29
+ : [];
30
+ const labels = new Set();
31
+ for (const item of labelled) {
32
+ labels.add(item.label);
33
+ }
34
+ const labelArray = Array.from(labels);
35
+ labelArray.sort();
36
+ fs.mkdirSync(args.output.split('/').slice(0, -1).join('/'), { recursive: true });
37
+ fs.writeFileSync(args.output, JSON.stringify({ labels: labelArray }, null, 4));
38
+ }
39
+ getDescription() {
40
+ return 'get all labels from some data';
41
+ }
42
+ }
43
+ //# sourceMappingURL=labels.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"labels.js","sourceRoot":"","sources":["../../src/commands/labels.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,IAAI,qBAAqB,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAA;AAC/G,OAAO,EAAE,kBAAkB,EAAgB,MAAM,gBAAgB,CAAA;AACjE,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAA;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAA;AAEtC,OAAO,EAAE,MAAM,IAAI,CAAA;AAEnB,MAAM,CAAC,MAAM,UAAU,GAAG,qBAAqB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;AAI1D,MAAM,OAAO,MAAO,SAAQ,qBAAqC;IAC7C,YAAY;QACxB,OAAO,UAAU,CAAA;IACrB,CAAC;IAEe,UAAU;QACtB,OAAO,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE;YACtC,KAAK,EAAE;gBACH,WAAW,EAAE,oCAAoC;aACpD;YACD,MAAM,EAAE;gBACJ,WAAW,EAAE,8CAA8C;aAC9D;SACJ,CAAC,CAAA;IACN,CAAC;IAEe,KAAK,CAAC,IAAI,CAAC,IAAU;QACjC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAEtB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE;YACtB,MAAM,IAAI,eAAe,CAAC,IAAI,KAAK,CAAC,wBAAwB,IAAI,EAAE,CAAC,EAAE,mBAAmB,CAAC,CAAA;SAC5F;QAED,MAAM,QAAQ,GAAmB,IAAI;YACjC,CAAC,CAAC,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK;YAC3E,CAAC,CAAC,EAAE,CAAA;QAER,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAA;QAChC,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE;YACzB,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;SACzB;QACD,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACrC,UAAU,CAAC,IAAI,EAAE,CAAA;QAEjB,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;QAChF,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;IAClF,CAAC;IAEe,cAAc;QAC1B,OAAO,+BAA+B,CAAA;IAC1C,CAAC;CACJ"}