Update tests and types for package privacy and enums

commonalityco · Jul 14, 2024 · ec3e4d6 · ec3e4d6
1 parent b2b1e2a
commit ec3e4d6
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/packages/data-packages/src/get-package.test.ts b/packages/data-packages/src/get-package.test.ts
@@ -56,7 +56,7 @@ describe('getPackage', () => {
       churn: 0.3,
       complexity: 0.3,
       license: 'MIT',
-      private: true,
+      privacy: 'PRIVATE',
     });
   });
 });
diff --git a/packages/types/src/index.ts b/packages/types/src/index.ts
@@ -7,6 +7,10 @@ import { z } from 'zod';
 
 export const privacyEnum = z.enum(['PUBLIC', 'PRIVATE']);
 
+export const dependencyTypeEnum = z.enum(['PRODUCTION', 'DEVELOPMENT', 'PEER']);
+
+export const blockTypeEnum = z.enum(['REACT', 'NODE', 'NEXT']);
+
 export type Constraint =
   | {
       allow: string[] | typeof AllPackagesWildcard;
@@ -59,6 +63,20 @@ export type Tag = string;
 
 export type TagsData = { packageName: string; tags: Tag[] };
 
+const blockSchema = z.object({
+  path: z.string(),
+  type: blockTypeEnum,
+  name: z.string(),
+  description: z.string().optional(),
+  version: z.string(),
+  churn: z.number(),
+  complexity: z.number(),
+  license: z.string().optional(),
+  privacy: privacyEnum,
+});
+
+export type Block = z.infer<typeof blockSchema>;
+
 export type Package = {
   path: string;
   type: BlockType;