@riocrypto/common-server 1.0.1230 → 1.0.1233

package/build/clients/cloud-vision-api-client.d.ts

@@ -0,0 +1,9 @@
+import { RioEnv } from "@riocrypto/common";
+declare class cloudVisionAPIClient {
+    private env;
+    googleAuth: any;
+    constructor(env: RioEnv);
+    getDataFromPDF(bucketName: string, fileName: string, outputPrefix: string): Promise<any>;
+}
+export declare const googleVisionAPIClient: cloudVisionAPIClient;
+export {};

package/build/clients/cloud-vision-api-client.js

@@ -0,0 +1,59 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.googleVisionAPIClient = void 0;
+const googleapis_1 = require("googleapis");
+const vision_1 = require("@google-cloud/vision");
+class cloudVisionAPIClient {
+    constructor(env) {
+        this.env = env;
+        this.googleAuth = new googleapis_1.google.auth.GoogleAuth({
+            keyFile: "/etc/secrets/cloud-vision/cloud-vision-service-account-key.json",
+        });
+    }
+    getDataFromPDF(bucketName, fileName, outputPrefix) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const client = new vision_1.v1.ImageAnnotatorClient({
+                auth: this.googleAuth,
+            });
+            const gcsSourceUri = `gs://${bucketName}/${fileName}`;
+            const gcsDestinationUri = `gs://${bucketName}/${outputPrefix}/`;
+            const inputConfig = {
+                // Supported mime_types are: 'application/pdf' and 'image/tiff'
+                mimeType: "application/pdf",
+                gcsSource: {
+                    uri: gcsSourceUri,
+                },
+            };
+            const outputConfig = {
+                gcsDestination: {
+                    uri: gcsDestinationUri,
+                },
+            };
+            const features = [{ type: "DOCUMENT_TEXT_DETECTION" }];
+            const request = {
+                requests: [
+                    {
+                        inputConfig: inputConfig,
+                        features: features,
+                        outputConfig: outputConfig,
+                    },
+                ],
+            };
+            //@ts-ignore
+            const [operation] = yield client.asyncBatchAnnotateFiles(request);
+            const [filesResponse] = yield operation.promise();
+            const destinationUri = filesResponse.responses[0].outputConfig.gcsDestination;
+            return destinationUri;
+        });
+    }
+}
+exports.googleVisionAPIClient = new cloudVisionAPIClient(process.env.RIO_ENV || process.env.NEXT_PUBLIC_RIO_ENV);

package/build/clients/gcloud-storage-client.d.ts

@@ -3,5 +3,6 @@ export declare class GCloudStorageClient {
     _bucket: Bucket;
     constructor(bucket: string);
     get bucket(): Bucket;
+    downloadJsonFile(filename: string): Promise<any>;
 }
 export declare const buildGCloudStorageClient: () => Promise<GCloudStorageClient>;

package/build/clients/gcloud-storage-client.js

@@ -22,6 +22,31 @@ class GCloudStorageClient {
     get bucket() {
         return this._bucket;
     }
+    downloadJsonFile(filename) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const file = this._bucket.file(filename);
+            return new Promise((resolve, reject) => {
+                let data = "";
+                file
+                    .createReadStream()
+                    .on("data", (chunk) => {
+                    data += chunk.toString();
+                })
+                    .on("end", () => {
+                    try {
+                        const jsonData = JSON.parse(data);
+                        resolve(jsonData);
+                    }
+                    catch (error) {
+                        reject(new Error(`Error parsing JSON file: ${error}`));
+                    }
+                })
+                    .on("error", (err) => {
+                    reject(new Error(`Error reading file: ${err.message}`));
+                });
+            });
+        });
+    }
 }
 exports.GCloudStorageClient = GCloudStorageClient;
 const buildGCloudStorageClient = () => __awaiter(void 0, void 0, void 0, function* () {

package/build/index.d.ts

@@ -50,6 +50,7 @@ export * from "./clients/kapstar-client";
 export * from "./clients/etherscan-client";
 export * from "./clients/secret-manager-client";
 export * from "./clients/gcloud-storage-client";
+export * from "./clients/cloud-vision-api-client";
 export * from "./clients/currency-api-client";
 export * from "./clients/aiprise-client";
 export * from "./clients/bridge-client";

package/build/index.js

@@ -66,6 +66,7 @@ __exportStar(require("./clients/kapstar-client"), exports);
 __exportStar(require("./clients/etherscan-client"), exports);
 __exportStar(require("./clients/secret-manager-client"), exports);
 __exportStar(require("./clients/gcloud-storage-client"), exports);
+__exportStar(require("./clients/cloud-vision-api-client"), exports);
 __exportStar(require("./clients/currency-api-client"), exports);
 __exportStar(require("./clients/aiprise-client"), exports);
 __exportStar(require("./clients/bridge-client"), exports);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@riocrypto/common-server",
-  "version": "1.0.1230",
+  "version": "1.0.1233",
   "description": "",
   "main": "./build/index.js",
   "types": "./build/index.d.ts",
@@ -26,6 +26,7 @@
     "@aws-sdk/client-s3": "^3.297.0",
     "@everapi/currencyapi-js": "^1.0.6",
     "@google-cloud/storage": "^6.9.5",
+    "@google-cloud/vision": "^4.0.2",
     "@riocrypto/common": "^1.0.1123",
     "@types/express": "^4.17.13",
     "axios": "^0.27.2",