/** ~ Copyright © 2019 Bell Canada. ~ ~ Licensed under the Apache License, Version 2.0 (the "License"); ~ you may not use this file except in compliance with the License. ~ You may obtain a copy of the License at ~ ~ http://www.apache.org/licenses/LICENSE-2.0 ~ ~ Unless required by applicable law or agreed to in writing, software ~ distributed under the License is distributed on an "AS IS" BASIS, ~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. ~ See the License for the specific language governing permissions and ~ limitations under the License. */ import * as fs from 'fs'; import * as uuidv1 from 'uuid/v1'; const grpc = require('@grpc/grpc-js'); import * as protoLoader from '@grpc/proto-loader'; import { processorApiConfig } from '../config/app-config'; const PROTO_PATH = processorApiConfig.grpc.bluePrintManagement.protoPath; // Suggested options for similarity to existing grpc.load behavior const packageDefinition: protoLoader.PackageDefinition = protoLoader.loadSync( PROTO_PATH, { keepCase: true, longs: String, enums: String, defaults: true, oneofs: true }); const protoDescriptor = grpc.loadPackageDefinition(packageDefinition); // The protoDescriptor object has the full package hierarchy const stub = new protoDescriptor.org.onap.ccsdk.cds.controllerblueprints.management.api.BluePrintManagementService( "" + processorApiConfig.grpc.host + ":" + processorApiConfig.grpc.port + "", grpc.credentials.createInsecure()); const metadata = new grpc.Metadata(); metadata.add('Authorization', processorApiConfig.grpc.authToken); class BluePrintManagementServiceGrpcClient { async uploadBlueprint(filePath: string, actionName: string): Promise { let input = { commonHeader: { timestamp: new Date(), originatorId: "cds-ui", requestId: uuidv1(), subRequestId: "1234-56", }, fileChunk: { chunk: fs.readFileSync(filePath) }, actionIdentifiers: { mode: "sync", blueprintName: "cds.zip", actionName: actionName } } let removeTempFile = () => { fs.unlink(filePath, (err: any) => { if (err) { console.error(err); } }); } return new Promise((resolve, reject) => { stub.uploadBlueprint(input, metadata, (err: any, output: any) => { if (err) { removeTempFile(); reject(err); return; } removeTempFile(); resolve(output); }); }); } async downloadBlueprint(blueprintName: string,blueprintVersion: string): Promise { let input = { commonHeader: { timestamp: new Date(), originatorId: "cds-ui", requestId: uuidv1(), subRequestId: "1234-56", }, actionIdentifiers: { mode: "sync", blueprintName: blueprintName, blueprintVersion: blueprintVersion } } return new Promise((resolve, reject) => { stub.downloadBlueprint(input, metadata, (err: any, output: any) => { if (err) { reject(err); return; } resolve(output); }); }); } } export const bluePrintManagementServiceGrpcClient = new BluePrintManagementServiceGrpcClient();