add tx based v2 auth

Author: prsvic

packages/blockchain-sdk-solana/src/auth/sign-message-v2/solana-sign-message-v2-authentication-facade-factory.ts renamed to packages/blockchain-sdk-solana/src/auth/v2/sign-message/solana-sign-message-v2-authentication-facade-factory.ts

@@ -4,34 +4,32 @@ import {
   Authenticator,
   Ed25519TokenBodyParser,
   TokenParser,
+  TokenSigner,
 } from '@dialectlabs/sdk';
 
-import { SolanaSignMessageV2TokenValidator } from './solana-sign-message-v2-token-validator';
 import { SolanaSignMessageV2TokenGenerator } from './solana-sign-message-v2-token-generator';
-import type { SolanaEd25519TokenSigner } from '../ed25519/solana-ed25519-token-signer';
+import { SolanaV2TokenValidator } from '../solana-v2-token-validator';
 
 export class SolanaSignMessageV2AuthenticationFacadeFactory extends AuthenticationFacadeFactory {
   constructor(
-    private readonly tokenSigner: SolanaEd25519TokenSigner,
+    private readonly tokenSigner: TokenSigner,
     private readonly baseUrl: string,
   ) {
     super();
   }
 
-  static createAuthenticator(baseUrl: string): Authenticator {
+  static createAuthenticator(): Authenticator {
     return new Authenticator(
       new TokenParser(new Ed25519TokenBodyParser()),
-      new SolanaSignMessageV2TokenValidator(baseUrl),
+      new SolanaV2TokenValidator(),
     );
   }
 
   get(): AuthenticationFacade {
     return new AuthenticationFacade(
       this.tokenSigner,
       new SolanaSignMessageV2TokenGenerator(this.tokenSigner, this.baseUrl),
-      SolanaSignMessageV2AuthenticationFacadeFactory.createAuthenticator(
-        this.baseUrl,
-      ),
+      SolanaSignMessageV2AuthenticationFacadeFactory.createAuthenticator(),
     );
   }
 }

packages/blockchain-sdk-solana/src/auth/sign-message-v2/solana-sign-message-v2-token-generator.ts renamed to packages/blockchain-sdk-solana/src/auth/v2/sign-message/solana-sign-message-v2-token-generator.ts

@@ -0,0 +1,35 @@
+import {
+  AuthenticationFacade,
+  AuthenticationFacadeFactory,
+  Authenticator,
+  Ed25519TokenBodyParser,
+  TokenParser,
+} from '@dialectlabs/sdk';
+
+import { SolanaSignTxV2TokenGenerator } from './solana-sign-tx-v2-token-generator';
+import { SolanaV2TokenValidator } from '../solana-v2-token-validator';
+import type { SolanaEd25519TokenSigner } from '../../ed25519/solana-ed25519-token-signer';
+
+export class SolanaSignTxV2AuthenticationFacadeFactory extends AuthenticationFacadeFactory {
+  constructor(
+    private readonly tokenSigner: SolanaEd25519TokenSigner,
+    private readonly baseUrl: string,
+  ) {
+    super();
+  }
+
+  static createAuthenticator(): Authenticator {
+    return new Authenticator(
+      new TokenParser(new Ed25519TokenBodyParser()),
+      new SolanaV2TokenValidator(),
+    );
+  }
+
+  get(): AuthenticationFacade {
+    return new AuthenticationFacade(
+      this.tokenSigner,
+      new SolanaSignTxV2TokenGenerator(this.tokenSigner, this.baseUrl),
+      SolanaSignTxV2AuthenticationFacadeFactory.createAuthenticator(),
+    );
+  }
+}

packages/blockchain-sdk-solana/src/auth/v2/sign-tx/solana-sign-tx-v2-authentication-facade-factory.ts

@@ -0,0 +1,51 @@
+import {
+  bytesFromBase64,
+  jsonParseFromBase64,
+  type Token,
+  TokenGenerator,
+  type TokenSigner,
+} from '@dialectlabs/sdk';
+import axios from 'axios';
+
+export class SolanaSignTxV2TokenGenerator extends TokenGenerator {
+  constructor(signer: TokenSigner, private readonly v2BaseUrl: string) {
+    super(signer);
+  }
+
+  override async generate(): Promise<Token> {
+    const prepareResponse = await axios.post<{ transaction: string }>(
+      `${this.v2BaseUrl}/v2/auth/solana-tx/prepare`,
+      { walletAddress: this.signer.subject },
+    );
+
+    const transactionBase64 = prepareResponse.data.transaction;
+
+    const { signature } = await this.signer.sign(
+      Buffer.from(transactionBase64, 'base64'),
+    );
+
+    const tokenResponse = await axios.post<{ token: string }>(
+      `${this.v2BaseUrl}/v2/auth/solana-tx/verify`,
+      {
+        transaction: Buffer.from(signature).toString('base64'),
+      },
+    );
+
+    const tokenStr = tokenResponse.data.token;
+
+    const [base64Header, base64Body, base64Signature] = tokenStr.split('.');
+    if (!base64Header || !base64Body || !base64Signature) {
+      throw new Error('Invalid token format from backend');
+    }
+
+    return {
+      rawValue: tokenStr,
+      header: jsonParseFromBase64(base64Header),
+      body: jsonParseFromBase64(base64Body),
+      signature: bytesFromBase64(base64Signature),
+      base64Header,
+      base64Body,
+      base64Signature,
+    };
+  }
+}

packages/blockchain-sdk-solana/src/auth/v2/sign-tx/solana-sign-tx-v2-token-generator.ts

@@ -0,0 +1,46 @@
+import type { DialectSolanaWalletAdapterWrapper } from '../../../wallet-adapter/dialect-solana-wallet-adapter-wrapper';
+import type {
+  AccountAddress,
+  PublicKey,
+  TokenSigner,
+  TokenSignerResult,
+} from '@dialectlabs/sdk';
+import { DIALECT_BLOCKCHAIN_SDK_TYPE_SOLANA } from '../../../sdk/constants';
+import { VersionedTransaction } from '@solana/web3.js';
+
+export const SOLANA_TX_TOKEN_SIGNER_ALG = `${DIALECT_BLOCKCHAIN_SDK_TYPE_SOLANA}-v2-tx`;
+
+export abstract class SolanaTxSigner implements TokenSigner {
+  readonly alg = SOLANA_TX_TOKEN_SIGNER_ALG;
+
+  abstract subject: AccountAddress;
+  abstract subjectPublicKey: PublicKey;
+
+  abstract sign(payload: Uint8Array): Promise<TokenSignerResult>;
+}
+
+export class DialectWalletAdapterSolanaTxSigner extends SolanaTxSigner {
+  constructor(
+    readonly dialectWalletAdapter: DialectSolanaWalletAdapterWrapper,
+  ) {
+    super();
+  }
+
+  get subject(): AccountAddress {
+    return this.dialectWalletAdapter.publicKey.toBase58();
+  }
+
+  get subjectPublicKey(): PublicKey {
+    return this.dialectWalletAdapter.publicKey;
+  }
+
+  async sign(payload: Uint8Array): Promise<TokenSignerResult> {
+    const tx = VersionedTransaction.deserialize(payload);
+    const signedTx = await this.dialectWalletAdapter.signTransaction(tx);
+
+    return {
+      payload,
+      signature: signedTx.serialize(),
+    };
+  }
+}

packages/blockchain-sdk-solana/src/auth/v2/sign-tx/solana-tx-signer.ts

@@ -1,7 +1,7 @@
 import type { TokenHeader } from '@dialectlabs/sdk';
 import { TokenValidator } from '@dialectlabs/sdk';
 
-export class SolanaSignMessageV2TokenValidator extends TokenValidator {
+export class SolanaV2TokenValidator extends TokenValidator {
   constructor() {
     super();
   }

packages/blockchain-sdk-solana/src/auth/sign-message-v2/solana-sign-message-v2-token-validator.ts renamed to packages/blockchain-sdk-solana/src/auth/v2/solana-v2-token-validator.ts

@@ -12,8 +12,9 @@ import { SolanaEd25519AuthenticationFacadeFactory } from '../auth/ed25519/solana
 import { DialectWalletAdapterSolanaEd25519TokenSigner } from '../auth/ed25519/solana-ed25519-token-signer';
 import { SolanaTxAuthenticationFacadeFactory } from '../auth/tx/solana-tx-authentication-facade-factory';
 import { DialectWalletAdapterSolanaTxTokenSigner } from '../auth/tx/solana-tx-token-signer';
-import { SolanaSignMessageV2AuthenticationFacadeFactory } from '../auth/sign-message-v2/solana-sign-message-v2-authentication-facade-factory';
+import { SolanaSignMessageV2AuthenticationFacadeFactory } from '../auth/v2/sign-message/solana-sign-message-v2-authentication-facade-factory';
 import { DIALECT_BLOCKCHAIN_SDK_TYPE_SOLANA } from './constants';
+import { SolanaSignTxV2AuthenticationFacadeFactory } from '../auth/v2/sign-tx/solana-sign-tx-v2-authentication-facade-factory';
 
 export interface SolanaConfigProps {
   wallet: DialectSolanaWalletAdapter;
@@ -108,6 +109,12 @@ Solana settings:
           config.dialectCloud.v2Url,
         );
       }
+      if (solanaConfig.wallet.canSignTransaction()) {
+        return new SolanaSignTxV2AuthenticationFacadeFactory(
+          new DialectWalletAdapterSolanaTxTokenSigner(solanaConfig.wallet),
+          config.dialectCloud.v2Url,
+        );
+      }
     }
     throw new IllegalArgumentError('Unsupported authentication');
   }