@avalix/chroma 0.0.8 → 0.0.10

package/src/wallets/talisman.ts

@@ -5,22 +5,32 @@ import path from 'node:path'
 import process from 'node:process'
 
 // Talisman specific configuration
+// https://github.com/avalix-labs/polkadot-wallets/tree/main/talisman
+const VERSION = '3.1.13'
 export const TALISMAN_CONFIG = {
-  downloadUrl: 'https://github.com/avalix-labs/polkadot-wallets/raw/refs/heads/main/talisman/talisman-3.0.5.zip',
-  extensionName: 'talisman-extension-3.0.5',
+  downloadUrl: `https://github.com/avalix-labs/polkadot-wallets/raw/refs/heads/main/talisman/talisman-${VERSION}.zip`,
+  extensionName: `talisman-extension-${VERSION}`,
 } as const
 
 // Helper function to find extension popup
 async function findExtensionPopup(context: BrowserContext, extensionId: string): Promise<Page> {
-  // delay for 1 second
-  await new Promise(resolve => setTimeout(resolve, 1000))
-
-  const pages = context.pages()
-  for (const p of pages) {
-    if (p.url().includes(`chrome-extension://${extensionId}/`)) {
-      p.setViewportSize({ width: 400, height: 600 })
-      p.waitForLoadState('domcontentloaded')
-      return p
+  // Wait for extension popup to appear with retry logic
+  const maxAttempts = 10
+  const retryDelay = 500
+
+  for (let attempt = 0; attempt < maxAttempts; attempt++) {
+    const pages = context.pages()
+    for (const p of pages) {
+      if (p.url().includes(`chrome-extension://${extensionId}/`)) {
+        await p.setViewportSize({ width: 400, height: 600 })
+        await p.waitForLoadState('domcontentloaded')
+        return p
+      }
+    }
+
+    // If not found, wait a bit before retrying
+    if (attempt < maxAttempts - 1) {
+      await new Promise(resolve => setTimeout(resolve, retryDelay))
     }
   }
 
@@ -37,13 +47,10 @@ export async function getTalismanExtensionPath(): Promise<string> {
     throw new Error(
       `Talisman extension not found at: ${extensionDir}\n\n`
       + `Please download the extension first by running:\n`
-      + `  npx @avalix/chroma download-extensions\n\n`
-      + `Or if you're using this as a dependency:\n`
-      + `  npm run chroma:download\n`,
+      + `  npx @avalix/chroma download-extensions\n`,
     )
   }
 
-  console.log(`✅ Found Talisman extension at: ${extensionDir}`)
   return extensionDir
 }
 
@@ -55,38 +62,42 @@ export async function importEthPrivateKey(
   const context = page.__extensionContext
   const extensionId = page.__extensionId
 
-  // Wait for Talisman to open its onboarding tab
-  await new Promise(resolve => setTimeout(resolve, 2000))
-
-  // Find the onboarding tab
+  // Wait for Talisman to open its onboarding tab with retry logic
+  const maxAttempts = 20
+  const retryDelay = 500
   let extensionPage: Page | null = null
-  const pages = context.pages()
-
-  for (const page of pages) {
-    const url = page.url()
-    console.log(`📄 Found page: ${url}`)
-    if (url.includes('onboarding.html') || url.includes(`chrome-extension://${extensionId}/`)) {
-      extensionPage = page
-      console.log(`✅ Found Talisman onboarding page: ${url}`)
+
+  for (let attempt = 0; attempt < maxAttempts; attempt++) {
+    const pages = context.pages()
+
+    for (const p of pages) {
+      const url = p.url()
+      if (url.includes('onboarding.html') || url.includes(`chrome-extension://${extensionId}/`)) {
+        extensionPage = p
+        break
+      }
+    }
+
+    if (extensionPage) {
       break
     }
+
+    // If not found, wait before retrying
+    if (attempt < maxAttempts - 1) {
+      await new Promise(resolve => setTimeout(resolve, retryDelay))
+    }
   }
 
   if (!extensionPage) {
-    throw new Error(`Talisman onboarding page not found`)
+    throw new Error(`Talisman onboarding page not found after ${maxAttempts} attempts`)
   }
 
   try {
     // Bring the onboarding page to front
     await extensionPage.bringToFront()
 
-    // Reload the onboarding page to ensure fresh state
-    console.log('🔄 Reloading onboarding page for fresh state...')
-    await extensionPage.reload()
-
     // Wait for the page to load and become interactive
     await extensionPage.waitForLoadState('domcontentloaded')
-    // await extensionPage.waitForTimeout(20000) // Give Talisman more time to initialize
 
     // Click the get started button
     await extensionPage.getByTestId('onboarding-get-started-button').click()
@@ -111,8 +122,6 @@ export async function importEthPrivateKey(
     await extensionPage.getByRole('button', { name: 'Save' }).click()
 
     await extensionPage.close()
-
-    console.log('✅ Talisman account import completed')
   }
   catch (error) {
     console.error('❌ Error during Talisman account import:', error)
@@ -128,7 +137,6 @@ export async function authorizeTalisman(
   const { accountName = 'Test Account' } = options
   const context = page.__extensionContext
   const extensionId = page.__extensionId
-  await new Promise(resolve => setTimeout(resolve, 1000))
 
   const extensionPopup = await findExtensionPopup(context, extensionId)
 
@@ -141,7 +149,6 @@ export async function authorizeTalisman(
     await anotherPopup.getByRole('button', { name: 'Approve' }).click()
   }
   catch {
-    console.log('No another popup found, skipping')
   }
 }
 
@@ -152,8 +159,25 @@ export async function approveTalismanTx(
   const context = page.__extensionContext
   const extensionId = page.__extensionId
 
-  await new Promise(resolve => setTimeout(resolve, 1000))
   const extensionPopup = await findExtensionPopup(context, extensionId)
 
+  try {
+    await extensionPopup.getByRole('button', { name: 'Yes' }).click()
+  }
+  catch {
+    console.log('No another popup found, skipping')
+  }
+
   await extensionPopup.getByRole('button', { name: 'Approve' }).click()
 }
+
+// Talisman specific transaction rejection implementation
+export async function rejectTalismanTx(
+  page: Page & { __extensionContext: BrowserContext, __extensionId: string },
+): Promise<void> {
+  const context = page.__extensionContext
+  const extensionId = page.__extensionId
+
+  const extensionPopup = await findExtensionPopup(context, extensionId)
+  await extensionPopup.getByTestId('connection-reject-button').click()
+}

package/dist/index.d.ts

@@ -1,64 +0,0 @@
-import { BrowserContext, Page, expect } from "@playwright/test";
-import * as playwright_test0 from "playwright/test";
-
-//#region src/context-playwright/types.d.ts
-type WalletType = 'polkadot-js' | 'talisman';
-interface WalletAccount {
-  seed: string;
-  name?: string;
-  password?: string;
-}
-interface WalletConfig {
-  type: WalletType;
-  downloadUrl?: string;
-}
-interface BaseWalletInstance {
-  extensionId: string;
-  importMnemonic: (options: WalletAccount) => Promise<void>;
-  authorize: (options?: {
-    accountName?: string;
-  }) => Promise<void>;
-  approveTx: (options?: {
-    password?: string;
-  }) => Promise<void>;
-}
-interface PolkadotJsWalletInstance extends BaseWalletInstance {
-  type: 'polkadot-js';
-}
-interface TalismanWalletInstance extends BaseWalletInstance {
-  type: 'talisman';
-  importEthPrivateKey: (options: {
-    privateKey: string;
-    name?: string;
-    password?: string;
-  }) => Promise<void>;
-}
-interface WalletTypeMap {
-  'polkadot-js': PolkadotJsWalletInstance;
-  'talisman': TalismanWalletInstance;
-}
-type Wallets = WalletTypeMap;
-type ConfiguredWallets<T extends readonly WalletConfig[]> = { [K in T[number]['type']]: WalletTypeMap[K] };
-type ExtendedPage = Page & {
-  __extensionContext: BrowserContext;
-  __walletExtensionIds: Map<string, string>;
-};
-interface ChromaTestOptions<T extends readonly WalletConfig[] = WalletConfig[]> {
-  wallets?: T;
-  headless?: boolean;
-  slowMo?: number;
-}
-interface WalletFixtures<W = Wallets> {
-  page: ExtendedPage;
-  wallets: W;
-}
-interface WalletWorkerFixtures {
-  walletContext: BrowserContext;
-  walletExtensionIds: Map<string, string>;
-}
-//#endregion
-//#region src/context-playwright/index.d.ts
-declare function createWalletTest<const T extends readonly WalletConfig[]>(options?: ChromaTestOptions<T>): playwright_test0.TestType<playwright_test0.PlaywrightTestArgs & playwright_test0.PlaywrightTestOptions & WalletFixtures<T extends readonly WalletConfig[] ? ConfiguredWallets<T> : WalletTypeMap>, playwright_test0.PlaywrightWorkerArgs & playwright_test0.PlaywrightWorkerOptions & WalletWorkerFixtures>;
-declare const test: ReturnType<typeof createWalletTest>;
-//#endregion
-export { createWalletTest, expect, test };