Merge pull request #687 from LIT-Protocol/LIT-3959-raw-wrapped-keys-litaction-functions

LIT-3959 - Export raw wrapped-keys LIT action functions

Author: Ansonhkg

packages/wrapped-keys-lit-actions/esbuild.config.js

@@ -50,14 +50,14 @@ module.exports = {
 (async () => {
   await esbuild.build({
     entryPoints: [
-      './src/lib/solana/signTransactionWithEncryptedSolanaKey.js',
-      './src/lib/solana/signMessageWithEncryptedSolanaKey.js',
-      './src/lib/solana/generateEncryptedSolanaPrivateKey.js',
-      './src/lib/ethereum/signTransactionWithEncryptedEthereumKey.js',
-      './src/lib/ethereum/signMessageWithEncryptedEthereumKey.js',
-      './src/lib/ethereum/generateEncryptedEthereumPrivateKey.js',
-      './src/lib/common/exportPrivateKey.js',
-      './src/lib/common/batchGenerateEncryptedKeys.js',
+      './src/lib/self-executing-actions/solana/signTransactionWithEncryptedSolanaKey.js',
+      './src/lib/self-executing-actions/solana/signMessageWithEncryptedSolanaKey.js',
+      './src/lib/self-executing-actions/solana/generateEncryptedSolanaPrivateKey.js',
+      './src/lib/self-executing-actions/ethereum/signTransactionWithEncryptedEthereumKey.js',
+      './src/lib/self-executing-actions/ethereum/signMessageWithEncryptedEthereumKey.js',
+      './src/lib/self-executing-actions/ethereum/generateEncryptedEthereumPrivateKey.js',
+      './src/lib/self-executing-actions/common/exportPrivateKey.js',
+      './src/lib/self-executing-actions/common/batchGenerateEncryptedKeys.js',
     ],
     bundle: true,
     minify: true,

packages/wrapped-keys-lit-actions/src/index.ts

@@ -6,6 +6,7 @@ import * as signTransactionWithEthereumEncryptedKey from './generated/ethereum/s
 import * as generateEncryptedSolanaPrivateKey from './generated/solana/generateEncryptedSolanaPrivateKey';
 import * as signMessageWithSolanaEncryptedKey from './generated/solana/signMessageWithEncryptedSolanaKey';
 import * as signTransactionWithSolanaEncryptedKey from './generated/solana/signTransactionWithEncryptedSolanaKey';
+import { rawActionFunctions } from './lib/raw-action-functions';
 
 import type {
   LitActionCodeRepository,
@@ -36,6 +37,10 @@ const litActionRepositoryCommon: LitActionCodeRepositoryCommon = {
 };
 
 export {
+  // Raw functions, <not wrapped in IIFEs>, for consumers to be able to compose these into their own LIT actions
+  // Facilitates running e.g. `batchGenerateEncryptedKeys` using `Lit.Actions.runOnce` inside of another action
+  rawActionFunctions,
+
   // Individual exports to allow tree-shaking and only importing the lit actions you need
   batchGenerateEncryptedKeys,
   exportPrivateKey,

packages/wrapped-keys-lit-actions/src/lib/abortError.js

@@ -0,0 +1,9 @@
+export class AbortError extends Error {
+  name = 'AbortError';
+}
+
+export const rethrowIfAbortError = (err) => {
+  if (err instanceof AbortError) {
+    throw err;
+  }
+};

packages/wrapped-keys-lit-actions/src/lib/common/exportPrivateKey.js

@@ -0,0 +1,42 @@
+/* global Lit */
+
+import { AbortError } from '../../abortError';
+import { removeSaltFromDecryptedKey } from '../../utils';
+
+async function tryDecryptToSingleNode({
+  accessControlConditions,
+  ciphertext,
+  dataToEncryptHash,
+}) {
+  try {
+    // May be undefined, since we're using `decryptToSingleNode`
+    return await Lit.Actions.decryptToSingleNode({
+      accessControlConditions,
+      ciphertext,
+      dataToEncryptHash,
+      chain: 'ethereum',
+      authSig: null,
+    });
+  } catch (err) {
+    throw new Error(`When decrypting key to a single node - ${err.message}`);
+  }
+}
+
+export async function getDecryptedKeyToSingleNode({
+  accessControlConditions,
+  ciphertext,
+  dataToEncryptHash,
+}) {
+  const decryptedPrivateKey = await tryDecryptToSingleNode({
+    accessControlConditions,
+    ciphertext,
+    dataToEncryptHash,
+  });
+
+  if (!decryptedPrivateKey) {
+    // Silently exit on nodes which didn't run the `decryptToSingleNode` code
+    throw new AbortError();
+  }
+
+  return removeSaltFromDecryptedKey(decryptedPrivateKey);
+}