feat: implement getCID methods on store and upload tables

this enables the new capability invocation handlers in storacha/w3up#918
storacha · Sep 13, 2023 · b81c071 · b81c071
1 parent ff92360
commit b81c071
Show file tree

Hide file tree

Showing 4 changed files with 80 additions and 58 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/upload-api/package.json b/upload-api/package.json
@@ -23,8 +23,8 @@
     "@ucanto/validator": "^8.0.0",
     "@web-std/fetch": "^4.1.0",
     "@web3-storage/access": "^14.0.0",
-    "@web3-storage/capabilities": "^9.0.0",
-    "@web3-storage/upload-api": "^5.3.0",
+    "@web3-storage/capabilities": "^9.3.0",
+    "@web3-storage/upload-api": "^5.4.0",
     "@web3-storage/w3infra-ucan-invocation": "*",
     "multiformats": "^11.0.1",
     "nanoid": "^4.0.2",

diff --git a/upload-api/tables/store.js b/upload-api/tables/store.js
@@ -186,6 +186,33 @@ export function useStoreTable(dynamoDb, tableName) {
         }
       }
     },
+
+    /**
+     * Get information about a CID.
+     * 
+     * @param {import('@web3-storage/upload-api').UnknownLink} link 
+     */
+    async getCID (link) {
+      const response = await dynamoDb.send(new QueryCommand({
+        TableName: tableName,
+        IndexName: 'cid',
+        KeyConditionExpression: "link = :link",
+        ExpressionAttributeValues: {
+          ':link': { S: link.toString() }
+        }
+      }))
+      return {
+        spaces: response.Items ? response.Items.map(
+          i => {
+            const item = unmarshall(i)
+            return ({
+              did: item.space,
+              insertedAt: item.insertedAt
+            })
+          }
+        ) : []
+      }
+    }
   }
 }
 

diff --git a/upload-api/tables/upload.js b/upload-api/tables/upload.js
@@ -182,6 +182,33 @@ export function useUploadTable(dynamoDb, tableName) {
         results: options.pre ? results.reverse() : results,
       }
     },
+
+    /**
+     * Get information about a CID.
+     * 
+     * @param {import('@web3-storage/upload-api').UnknownLink} link 
+     */
+    async getCID (link) {
+      const response = await dynamoDb.send(new QueryCommand({
+        TableName: tableName,
+        IndexName: 'cid',
+        KeyConditionExpression: "root = :root",
+        ExpressionAttributeValues: {
+          ':root': { S: link.toString() }
+        }
+      }))
+      return {
+        spaces: response.Items ? response.Items.map(
+          i => {
+            const item = unmarshall(i)
+            return ({
+              did: item.space,
+              insertedAt: item.insertedAt
+            })
+          }
+        ) : []
+      }
+    }
   }
 }