Implementing wallet.createWallet

src/coinbase/address.ts

@@ -62,15 +62,6 @@ export class Address {
     return this.model.network_id;
   }
 
-  /**
-   * Returns the public key.
-   *
-   * @returns {string} The public key.
-   */
-  public getPublicKey(): string {
-    return this.model.public_key;
-  }
-
   /**
    * Returns the wallet ID.
    *

src/coinbase/coinbase.ts

@@ -1,6 +1,12 @@
 import globalAxios from "axios";
 import fs from "fs";
-import { User as UserModel, UsersApiFactory, TransfersApiFactory } from "../client";
+import {
+  AddressesApiFactory,
+  User as UserModel,
+  UsersApiFactory,
+  TransfersApiFactory,
+  WalletsApiFactory,
+} from "../client";
 import { ethers } from "ethers";
 import { BASE_PATH } from "./../client/base";
 import { Configuration } from "./../client/configuration";
@@ -80,6 +86,8 @@ export class Coinbase {
     );
 
     this.apiClients.user = UsersApiFactory(config, BASE_PATH, axiosInstance);
+    this.apiClients.wallet = WalletsApiFactory(config, BASE_PATH, axiosInstance);
+    this.apiClients.address = AddressesApiFactory(config, BASE_PATH, axiosInstance);
     this.apiClients.transfer = TransfersApiFactory(config, BASE_PATH, axiosInstance);
     this.apiClients.baseSepoliaProvider = new ethers.JsonRpcProvider("https://sepolia.base.org");
   }

src/coinbase/index.ts

@@ -0,0 +1 @@
+export { Coinbase } from "./coinbase";

src/coinbase/tests/address_test.ts

@@ -49,10 +49,6 @@ describe("Address", () => {
     expect(address.getNetworkId()).toBe(VALID_ADDRESS_MODEL.network_id);
   });
 
-  it("should return the public key", () => {
-    expect(address.getPublicKey()).toBe(newEthAddress.publicKey);
-  });
-
   it("should return the wallet ID", () => {
     expect(address.getWalletId()).toBe(VALID_ADDRESS_MODEL.wallet_id);
   });

src/coinbase/tests/coinbase_test.ts

@@ -2,6 +2,7 @@ import { Coinbase } from "../coinbase";
 import MockAdapter from "axios-mock-adapter";
 import axios from "axios";
 import { APIError } from "../api_error";
+import { VALID_WALLET_MODEL } from "./wallet_test";
 
 const axiosMock = new MockAdapter(axios);
 const PATH_PREFIX = "./src/coinbase/tests/config";
@@ -39,17 +40,43 @@ describe("Coinbase tests", () => {
     );
   });
 
-  it("should be able to get the default user", async () => {
-    axiosMock.onGet().reply(200, {
-      id: 123,
-    });
+  describe("should able to interact with the API", () => {
     const cbInstance = Coinbase.configureFromJson(
       `${PATH_PREFIX}/coinbase_cloud_api_key.json`,
       true,
     );
-    const user = await cbInstance.getDefaultUser();
-    expect(user.getId()).toBe(123);
-    expect(user.toString()).toBe("Coinbase:User{userId: 123}");
+    let user;
+    beforeEach(async () => {
+      axiosMock.reset();
+      axiosMock
+        .onPost(/\/v1\/wallets\/.*\/addresses\/.*\/faucet/)
+        .reply(200, { transaction_hash: "0xdeadbeef" })
+        .onGet(/\/me/)
+        .reply(200, {
+          id: 123,
+        })
+        .onPost(/\/v1\/wallets/)
+        .reply(200, VALID_WALLET_MODEL)
+        .onGet(/\/v1\/wallets\/.*/)
+        .reply(200, VALID_WALLET_MODEL);
+      user = await cbInstance.getDefaultUser();
+    });
+
+    it("should return the correct user ID", () => {
+      expect(user.getId()).toBe(123);
+      expect(user.toString()).toBe("User{ userId: 123 }");
+    });
+
+    it("should be able to get faucet funds", async () => {
+      const wallet = await user.createWallet();
+      expect(wallet.getId()).toBe(VALID_WALLET_MODEL.id);
+
+      const defaultAddress = wallet.defaultAddress();
+      expect(defaultAddress?.getId()).toBe(VALID_WALLET_MODEL.default_address.address_id);
+
+      const faucetTransaction = await wallet?.faucet();
+      expect(faucetTransaction.getTransactionHash()).toBe("0xdeadbeef");
+    });
   });
 
   it("should raise an error if the user is not found", async () => {

src/coinbase/tests/user_test.ts

@@ -22,6 +22,6 @@ describe("User Class", () => {
 
   it("should return a correctly formatted string representation of the User instance", () => {
     const user = new User(mockUserModel, mockApiClients);
-    expect(user.toString()).toBe(`Coinbase:User{userId: ${mockUserModel.id}}`);
+    expect(user.toString()).toBe(`User{ userId: ${mockUserModel.id} }`);
   });
 });

src/coinbase/tests/wallet_test.ts

@@ -8,7 +8,7 @@ import { Wallet } from "../wallet";
 import { createAxiosMock } from "./utils";
 
 const walletId = randomUUID();
-const VALID_WALLET_MODEL = {
+export const VALID_WALLET_MODEL = {
   id: randomUUID(),
   network_id: Coinbase.networkList.BaseSepolia,
   default_address: {
@@ -56,13 +56,12 @@ describe("Wallet Class", () => {
     it("should derive the correct number of addresses", async () => {
       expect(wallet.addresses.length).toBe(2);
     });
-  });
 
-  it("should return the correct string representation", async () => {
-    const wallet = await Wallet.init(VALID_WALLET_MODEL, client);
-    expect(wallet.toString()).toBe(
-      `Wallet{id: '${VALID_WALLET_MODEL.id}', networkId: 'base-sepolia'}`,
-    );
+    it("should return the correct string representation", async () => {
+      expect(wallet.toString()).toBe(
+        `Wallet{id: '${VALID_WALLET_MODEL.id}', networkId: '${Coinbase.networkList.BaseSepolia}'}`,
+      );
+    });
   });
 
   it("should throw an ArgumentError when the API client is not provided", async () => {

src/coinbase/types.ts

@@ -1,6 +1,8 @@
 import { AxiosPromise, AxiosRequestConfig, RawAxiosRequestConfig } from "axios";
 import { ethers } from "ethers";
 import {
+  Address,
+  CreateAddressRequest,
   CreateWalletRequest,
   BroadcastTransferRequest,
   CreateTransferRequest,
@@ -27,6 +29,15 @@ export type WalletAPIClient = {
     createWalletRequest?: CreateWalletRequest,
     options?: RawAxiosRequestConfig,
   ) => AxiosPromise<WalletModel>;
+
+  /**
+   * Returns the wallet model with the given ID.
+   *
+   * @param walletId - The ID of the wallet to fetch.
+   * @param options - Override http request option.
+   * @throws {APIError} If the request fails.
+   */
+  getWallet: (walletId: string, options?: RawAxiosRequestConfig) => AxiosPromise<WalletModel>;
 };
 
 /**
@@ -38,7 +49,7 @@ export type AddressAPIClient = {
    *
    * @param walletId - The wallet ID.
    * @param addressId - The address ID.
-   * @returns The transaction hash
+   * @returns The transaction hash.
    * @throws {APIError} If the request fails.
    */
   requestFaucetFunds(
@@ -47,7 +58,7 @@ export type AddressAPIClient = {
   ): Promise<{ data: { transaction_hash: string } }>;
 
   /**
-   * Get address by onchain address
+   * Get address by onchain address.
    *
    * @param walletId - The ID of the wallet the address belongs to.
    * @param addressId - The onchain address of the address that is being fetched.
@@ -58,6 +69,20 @@ export type AddressAPIClient = {
     walletId: string,
     addressId: string,
     options?: AxiosRequestConfig,
+  ): AxiosPromise<Address>;
+
+  /**
+   * Create a new address scoped to the wallet.
+   *
+   * @param walletId - The ID of the wallet to create the address in.
+   * @param createAddressRequest - The address creation request.
+   * @param options - Axios request options.
+   * @throws {APIError} If the request fails.
+   */
+  createAddress(
+    walletId: string,
+    createAddressRequest?: CreateAddressRequest,
+    options?: AxiosRequestConfig,
   ): AxiosPromise<AddressModel>;
 };
 

src/coinbase/user.ts

@@ -1,5 +1,7 @@
 import { ApiClients } from "./types";
 import { User as UserModel } from "./../client/api";
+import { Coinbase } from "./coinbase";
+import { Wallet } from "./wallet";
 
 /**
  * A representation of a User.
@@ -13,18 +15,39 @@ export class User {
   /**
    * Initializes a new User instance.
    *
-   * @param {UserModel} user - The user model.
-   * @param {ApiClients} client - The API clients.
+   * @param user - The user model.
+   * @param client - The API clients.
    */
   constructor(user: UserModel, client: ApiClients) {
     this.client = client;
     this.model = user;
   }
 
+  /**
+   * Creates a new Wallet belonging to the User.
+   *
+   * @throws {APIError} - If the request fails.
+   * @throws {ArgumentError} - If the model or client is not provided.
+   * @throws {InternalError} - If address derivation or caching fails.
+   * @returns the new Wallet
+   */
+  async createWallet(): Promise<Wallet> {
+    const payload = {
+      wallet: {
+        network_id: Coinbase.networkList.BaseSepolia,
+      },
+    };
+    const walletData = await this.client.wallet!.createWallet(payload);
+    return Wallet.init(walletData.data!, {
+      wallet: this.client.wallet!,
+      address: this.client.address!,
+    });
+  }
+
   /**
    * Returns the user's ID.
    *
-   * @returns {string} The user's ID.
+   * @returns The user's ID.
    */
   public getId(): string {
     return this.model.id;
@@ -33,9 +56,9 @@ export class User {
   /**
    * Returns a string representation of the User.
    *
-   * @returns {string} The string representation of the User.
+   * @returns The string representation of the User.
    */
   toString(): string {
-    return `Coinbase:User{userId: ${this.model.id}}`;
+    return `User{ userId: ${this.model.id} }`;
   }
 }