allow to provide an existing id when creating an entity, image, type or property

Author: nikgraf

.changeset/shaggy-windows-camp.md

@@ -0,0 +1,5 @@
+---
+"@graphprotocol/grc-20": minor
+---
+
+allow to provide an existing id when creating an entity, image, type or property

src/core/image.ts

@@ -21,23 +21,28 @@ type MakeImageParams = {
     width: number;
     height: number;
   };
+  id?: Id;
 };
 
 /**
  * Creates an entity representing an Image.
  *
  * @example
  * ```ts
- * const { id, ops } = Image.make({ cid: 'https://example.com/image.png', dimensions: { width: 100, height: 100 } });
+ * const { id, ops } = Image.make({
+ *   cid: 'https://example.com/image.png',
+ *   dimensions: { width: 100, height: 100 },
+ *   id: imageId, // optional and will be generated if not provided
+ * });
  * console.log(id); // 'gw9uTVTnJdhtczyuzBkL3X'
  * console.log(ops); // [...]
  * ```
  *
  * @returns id – base58 encoded v4 uuid representing the image entity: {@link MakeImageReturnType}
  * @returns ops – The ops for the Image entity: {@link MakeImageReturnType}
  */
-export function make({ cid, dimensions }: MakeImageParams): MakeImageReturnType {
-  const entityId = generate();
+export function make({ cid, dimensions, id }: MakeImageParams): MakeImageReturnType {
+  const entityId = id ?? generate();
   const ops: Array<Op> = [
     Relation.make({
       fromId: entityId,

src/graph/create-entity.test.ts

@@ -310,4 +310,23 @@ describe('createEntity', () => {
     expect(typeof entity.id).toBe('string');
     expect(entity.ops).toHaveLength(9);
   });
+
+  it('creates an entity with a provided id', async () => {
+    const entity = createEntity({
+      id: Id('WeUPYRkhnQLmHPH4S1ioc4'),
+      name: 'Yummy Coffee',
+    });
+
+    expect(entity).toBeDefined();
+    expect(entity.id).toBe('WeUPYRkhnQLmHPH4S1ioc4');
+    expect(entity.ops).toHaveLength(1);
+    expect(entity.ops[0]).toMatchObject({
+      type: 'SET_TRIPLE',
+      triple: {
+        attribute: NAME_PROPERTY,
+        entity: entity.id,
+        value: { type: 'TEXT', value: 'Yummy Coffee' },
+      },
+    });
+  });
 });

src/graph/create-entity.ts

@@ -21,6 +21,7 @@ type CreateEntityParams = DefaultProperties & {
  *   description: 'description of the entity',
  *   cover: imageEntityId,
  *   types: [typeEntityId1, typeEntityId2],
+ *   id: entityId, // optional and will be generated if not provided
  *   properties: {
  *     // value property like text, number, url, time, point, checkbox
  *     [propertyId]: {
@@ -49,8 +50,8 @@ type CreateEntityParams = DefaultProperties & {
  * @returns – {@link CreateResult}
  * @throws Will throw an error if any provided ID is invalid
  */
-export const createEntity = ({ name, description, cover, properties, types }: CreateEntityParams): CreateResult => {
-  const id = generate();
+export const createEntity = ({ id: providedId, name, description, cover, properties, types }: CreateEntityParams): CreateResult => {
+  const id = providedId ?? generate();
   const ops: Array<Op> = [];
 
   ops.push(...createDefaultProperties({ entityId: id, name, description, cover }));

src/graph/create-image.test.ts

@@ -1,4 +1,5 @@
 import { beforeEach, describe, expect, it, vi } from 'vitest';
+import { Id } from '../id.js';
 import { SystemIds } from '../system-ids.js';
 import { createImage } from './create-image.js';
 
@@ -129,6 +130,16 @@ describe('createImage', () => {
     }
   });
 
+  it('creates an image with a provided id', async () => {
+    const image = await createImage({
+      url: 'http://localhost:3000/image',
+      id: Id('WeUPYRkhnQLmHPH4S1ioc4'),
+    });
+
+    expect(image).toBeDefined();
+    expect(image.id).toBe('WeUPYRkhnQLmHPH4S1ioc4');  
+  });
+
   it('throws an error if the image cannot be uploaded to IPFS', async () => {
     vi.spyOn(global, 'fetch').mockImplementation(url => {
       if (url.toString() === 'http://localhost:3000/image') {

src/graph/create-image.ts

@@ -1,3 +1,4 @@
+import type { Id } from '../id.js';
 import { Image } from '../image.js';
 import { uploadImage } from '../ipfs.js';
 import type { CreateResult, Op } from '../types.js';
@@ -7,11 +8,13 @@ type CreateImageParams =
       blob: Blob;
       name?: string;
       description?: string;
+      id?: Id;
     }
   | {
       url: string;
       name?: string;
       description?: string;
+      id?: Id;
     };
 
 /**
@@ -25,22 +28,24 @@ type CreateImageParams =
  *   url: 'https://example.com/image.png',
  *   name: 'name of the image', // optional
  *   description: 'description of the image', // optional
+ *   id: imageId, // optional and will be generated if not provided
  * });
  *
  * const { id, ops } = createImage({
  *   blob: new Blob(…),
  *   name: 'name of the image', // optional
  *   description: 'description of the image', // optional
+ *   id: imageId, // optional and will be generated if not provided
  * });
  * ```
  * @param params – {@link CreateImageParams}
  * @returns – {@link CreateResult}
  * @throws Will throw an IpfsUploadError if the image cannot be uploaded to IPFS
  */
-export const createImage = async ({ name, description, ...params }: CreateImageParams): Promise<CreateResult> => {
+export const createImage = async ({ name, description, id: providedId, ...params }: CreateImageParams): Promise<CreateResult> => {
   const ops: Array<Op> = [];
   const { cid, dimensions } = await uploadImage(params);
-  const { id, ops: imageOps } = Image.make({ cid, dimensions });
+  const { id, ops: imageOps } = Image.make({ cid, dimensions, id: providedId });
   ops.push(...imageOps);
   ops.push(...createDefaultProperties({ entityId: id, name, description }));
 

src/graph/create-property.test.ts

@@ -1,5 +1,6 @@
 import { describe, expect, it } from 'vitest';
 import { JOB_TYPE, ROLES_PROPERTY } from '../core/ids/content.js';
+import { Id } from '../id.js';
 import { createProperty } from './create-property.js';
 
 describe('createProperty', () => {
@@ -73,4 +74,15 @@ describe('createProperty', () => {
     expect(property.ops[4]?.type).toBe('CREATE_RELATION');
     expect(property.ops[5]?.type).toBe('CREATE_RELATION');
   });
+
+  it('creates a property with a provided id', async () => {
+    const property = createProperty({
+      id: Id('WeUPYRkhnQLmHPH4S1ioc4'),
+      name: 'Price',
+      type: 'NUMBER',
+    });
+
+    expect(property).toBeDefined();
+    expect(property.id).toBe('WeUPYRkhnQLmHPH4S1ioc4');
+  });
 });

src/graph/create-property.ts

@@ -32,15 +32,16 @@ type CreatePropertyParams = DefaultProperties &
  *   type: 'TEXT'
  *   description: 'description of the property',
  *   cover: imageEntityId,
+ *   id: propertyId, // optional and will be generated if not provided
  * });
  * ```
  * @param params – {@link CreatePropertyParams}
  * @returns – {@link CreateResult}
  * @throws Will throw an error if any provided ID is invalid
  */
 export const createProperty = (params: CreatePropertyParams): CreateResult => {
-  const { name, description, cover } = params;
-  const entityId = generate();
+  const { id, name, description, cover } = params;
+  const entityId = id ?? generate();
   const ops: Op[] = [];
 
   ops.push(...createDefaultProperties({ entityId, name, description, cover }));

src/graph/create-type.test.ts

@@ -1,6 +1,7 @@
 import { describe, expect, it } from 'vitest';
 import { AUTHORS_PROPERTY, WEBSITE_PROPERTY } from '../core/ids/content.js';
 import { NAME_PROPERTY, PROPERTY, SCHEMA_TYPE, TYPES_PROPERTY } from '../core/ids/system.js';
+import { Id } from '../id.js';
 import { createType } from './create-type.js';
 
 describe('createType', () => {
@@ -100,4 +101,14 @@ describe('createType', () => {
       type: 'CREATE_RELATION',
     });
   });
+
+  it('creates a type with a provided id', async () => {
+    const type = createType({
+      id: Id('WeUPYRkhnQLmHPH4S1ioc4'),
+      name: 'Article',
+    });
+
+    expect(type).toBeDefined();
+    expect(type.id).toBe('WeUPYRkhnQLmHPH4S1ioc4');
+  });
 });

src/graph/create-type.ts

@@ -18,15 +18,16 @@ type CreateTypeParams = DefaultProperties & {
  *   name: 'name of the type',
  *   description: 'description of the type',
  *   cover: imageEntityId,
+ *   id: typeId, // optional and will be generated if not provided
  *   properties: [propertyEntityId1, propertyEntityId2],
  * });
  * ```
  * @param params – {@link CreateTypeParams}
  * @returns – {@link CreateResult}
  * @throws Will throw an error if any provided ID is invalid
  */
-export const createType = ({ name, description, cover, properties }: CreateTypeParams): CreateResult => {
-  const id = generate();
+export const createType = ({ id: providedId, name, description, cover, properties }: CreateTypeParams): CreateResult => {
+  const id = providedId ?? generate();
   const ops: Op[] = [];
 
   ops.push(...createDefaultProperties({ entityId: id, name, description, cover }));