@avaprotocol/sdk-js 0.6.4 → 0.6.6

package/.env.example

@@ -0,0 +1,7 @@
+# Testing environment variables to be used in the .env.test
+
+# JWT token for authentication
+TEST_JWT_TOKEN=
+
+# Private key for authentication
+TEST_PRIVATE_KEY=

package/dist/index.cjs

@@ -32,7 +32,16 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 
 // grpc_codegen/avs_pb.js
 var avs_pb_exports = {};
-var jspb, goog, global, google_protobuf_timestamp_pb, google_protobuf_wrappers_pb;
+__export(avs_pb_exports, {
+  CreateTaskReq: () => CreateTaskReq,
+  CreateTaskResp: () => CreateTaskResp,
+  GetKeyReq: () => GetKeyReq,
+  KeyResp: () => KeyResp,
+  Task: () => Task,
+  UpdateChecksReq: () => UpdateChecksReq,
+  UpdateChecksResp: () => UpdateChecksResp
+});
+var jspb, goog, global, google_protobuf_timestamp_pb, google_protobuf_wrappers_pb, Task, CreateTaskReq, CreateTaskResp, GetKeyReq, KeyResp, UpdateChecksReq, UpdateChecksResp;
 var init_avs_pb = __esm({
   "grpc_codegen/avs_pb.js"() {
     "use strict";
@@ -4093,19 +4102,35 @@ var init_avs_pb = __esm({
       JAVASCRIPT: 0
     };
     goog.object.extend(exports, proto.aggregator);
+    ({ Task, CreateTaskReq, CreateTaskResp, GetKeyReq, KeyResp, UpdateChecksReq, UpdateChecksResp } = proto.aggregator);
   }
 });
 
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
-  BaseClient: () => BaseClient,
-  Client: () => Client,
+  default: () => Client,
   getKeyRequestMessage: () => getKeyRequestMessage
 });
 module.exports = __toCommonJS(src_exports);
 var grpc2 = __toESM(require("@grpc/grpc-js"), 1);
-var import_grpc_js = require("@grpc/grpc-js");
+
+// src/config.ts
+var DEFAULT_JWT_EXPIRATION = 24 * 60 * 60;
+var configs = {
+  development: {
+    AVS_RPC_URL: process.env.AVS_RPC_URL || "localhost:2206"
+  },
+  staging: {
+    AVS_RPC_URL: "aggregator-holesky.avaprotocol.org:2206"
+  },
+  production: {
+    AVS_RPC_URL: "aggregator.avaprotocol.org:2206"
+  }
+};
+function getRpcEndpoint(env) {
+  return configs[env].AVS_RPC_URL;
+}
 
 // src/auth.ts
 var getKeyRequestMessage = (address, expiredAt) => {
@@ -4288,7 +4313,7 @@ function serialize_google_protobuf_BoolValue(arg) {
 function deserialize_google_protobuf_BoolValue(buffer_arg) {
   return google_protobuf_wrappers_pb2.BoolValue.deserializeBinary(new Uint8Array(buffer_arg));
 }
-var AggregatorService = exports.AggregatorService = {
+var AggregatorService = {
   // Auth
   getKey: {
     path: "/aggregator.Aggregator/GetKey",
@@ -4415,20 +4440,20 @@ var AggregatorService = exports.AggregatorService = {
     responseDeserialize: deserialize_aggregator_UpdateChecksResp
   }
 };
-exports.AggregatorClient = grpc.makeGenericClientConstructor(AggregatorService);
+var AggregatorClient = grpc.makeGenericClientConstructor(AggregatorService);
 
 // src/index.ts
 init_avs_pb();
-var grpcMetadata = new import_grpc_js.Metadata();
+var metadata = new grpc2.Metadata();
 var BaseClient = class {
-  // A JWT token for admin operations
-  // protected wallet?: any;
-  constructor(config) {
-    this.endpoint = config.endpoint;
-    this.rpcClient = new (void 0)(
-      config.endpoint,
+  constructor(opts) {
+    this.env = opts.env || "production";
+    this.rpcClient = new AggregatorClient(
+      getRpcEndpoint(this.env),
+      // TODO: switch to the TLS after we're able to update all the operator
       grpc2.credentials.createInsecure()
     );
+    console.log("this.rpcClient:", this.rpcClient);
   }
   // async authWithJwtToken(
   //   address: string,
@@ -4436,21 +4461,21 @@ var BaseClient = class {
   //   expiredAt?: number
   // ): Promise<KeyExchangeResp> {
   //   console.log("Authenticating with JWT token: ", jwtToken);
+  //   // Use the provided expiredAt or set it to 24 hours from now if not provided
   //   const expirationTime =
   //     expiredAt || Math.floor(Date.now() / 1000) + DEFAULT_JWT_EXPIRATION;
-  //   const method = this.rpcClient.getKey.bind(this.rpcClient) as (
-  //     request: GetKeyReq,
-  //     metadata: Metadata | undefined,
-  //     callback: (error: grpc.ServiceError | null, response: KeyResp) => void
-  //   ) => grpc.ClientUnaryCall;
-  //   const request = new GetKeyReq()
-  //     .setOwner(address)
-  //     .setExpiredAt(expirationTime)
-  //     .setSignature(jwtToken);
-  //   const response = await this._callRPC<KeyResp, GetKeyReq>(method, request);
-  //   console.log("response:", response.getKey());
-  //   return { key: response.getKey() };
+  //   const result: avsPb.KeyResp = await this._callRPC<
+  //     avsPb.KeyResp,
+  //     avsPb.GetKeyReq
+  //   >("getKey", {
+  //     owner: address,
+  //     expired_at: expirationTime,
+  //     signature: jwtToken,
+  //   });
+  //   this.authkey = result.getKey();
+  //   return { key: result.getKey() };
   // }
+  // This flow can be used where the signature is generate from outside, such as in front-end and pass in
   async authWithSignature(address, signature, expiredAtEpoch) {
     console.log(
       "Authenticating with signature:",
@@ -4458,52 +4483,29 @@ var BaseClient = class {
       "Expired at epoch:",
       expiredAtEpoch
     );
-    const request = new (void 0)().setOwner(address).setExpiredAt(expiredAtEpoch).setSignature(signature);
-    console.log("request:", request);
-    const response = await this._callRPC("getKey", request);
-    console.log("response:", response.getKey());
-    return { key: response.getKey() };
+    const request = new GetKeyReq();
+    request.setOwner(address);
+    request.setExpiredAt(expiredAtEpoch);
+    request.setSignature(signature);
+    let result = await this._callRPC(
+      "getKey",
+      request
+    );
+    console.log("result:", result);
+    this.authkey = result.getKey();
+    return { key: result.getKey() };
   }
-  // protected async sendRequest<TResponse, TRequest extends object = {}>(
-  //   method: string,
-  //   request: TRequest = {} as TRequest,
-  //   metadata?: Metadata
-  // ): Promise<TResponse> {
-  //   if (metadata === undefined) {
-  //     metadata = new grpc.Metadata();
-  //   }
-  //   if (!this.adminToken) {
-  //     throw new Error(
-  //       "Authentication required. Please call authWithJwtToken() or authWithSignature() before making requests."
-  //     );
-  //   }
-  //   metadata.add("adminToken", this.adminToken);
-  //   return this._callRPC<TResponse, TRequest>(
-  //     method as (
-  //       request: TRequest,
-  //       metadata: grpc.Metadata | undefined,
-  //       callback: (error: grpc.ServiceError | null, response: TResponse) => void
-  //     ),
-  //     request,
-  //     metadata
-  //   );
-  // }
   isAuthenticated() {
-    return !!this.adminToken;
+    return !!this.authkey;
   }
-  _callRPC(methodName, request) {
+  _callRPC(method, request) {
     return new Promise((resolve, reject) => {
-      const method = this.rpcClient[methodName];
-      method.call(
-        this.rpcClient,
+      this.rpcClient[method].bind(this.rpcClient)(
         request,
-        grpcMetadata,
+        metadata,
         (error, response) => {
-          if (error) {
-            reject(error);
-          } else {
-            resolve(response);
-          }
+          if (error) reject(error);
+          else resolve(response);
         }
       );
     });
@@ -4513,33 +4515,8 @@ var Client = class extends BaseClient {
   constructor(config) {
     super(config);
   }
-  async listTask() {
-    console.log("Listing tasks");
-    const request = new (void 0)();
-    const response = await this._callRPC(
-      "listTasks",
-      request
-    );
-    console.log("response:", response);
-    return response;
-  }
-  async getSmartWalletAddress(address) {
-    const request = new (void 0)().setOwner(address);
-    const response = await this._callRPC(
-      "getSmartAccountAddress",
-      request
-    );
-    return response;
-  }
-  async getTask(taskId) {
-    const request = new (void 0)().setBytes(taskId);
-    const response = await this._callRPC("getTask", request);
-    return response;
-  }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  BaseClient,
-  Client,
   getKeyRequestMessage
 });