pomerium / enterprise-client Goto Github PK

const gulp = require('gulp');
const through2 = require('through2');
const fs = require('fs');
const path = require('path');
const grpc = require('@grpc/grpc-js');
const protoLoader = require('@grpc/proto-loader');

gulp.task('generate-sdk', function () {
    const protoDir = 'proto';
    const includeDirs = [
        path.resolve(__dirname, 'lib/envoyproxy/data-plane-api'),
        path.resolve(__dirname, 'lib/googleapis/googleapis'),
        path.resolve(__dirname, 'lib/bufbuild/protoc-gen-validate'),
        path.resolve(__dirname, 'lib/cncf/udpa'),
        path.resolve(__dirname, 'proto/pomerium'),
    ]
    return gulp.src(`${protoDir}/**/*.proto`)
      .pipe(through2.obj(function (file, _, cb) {
        const packageDefinition = protoLoader.loadSync(file.path, {
          keepCase: true,
          longs: String,
          enums: String,
          defaults: true,
          oneofs: true,
          includeDirs
        });
        const protoDescriptor = grpc.loadPackageDefinition(packageDefinition);
        const serviceName = Object.keys(protoDescriptor)[0]; // Assume the service name is the first key
        const relativePath = path.relative(protoDir, path.dirname(file.path));
        const outputFilePath = path.join('build', relativePath, `${serviceName}Client.js`);
        const clientFactory = `
          const grpc = require('@grpc/grpc-js');
          const protoDescriptor = ${JSON.stringify(protoDescriptor)};
          class ${serviceName}Client {
            constructor(endpoint, channelCredentials) {
              this.service = new protoDescriptor.${serviceName}(endpoint, channelCredentials);
            }
          }
          module.exports = ${serviceName}Client;
        `;
        fs.mkdirSync(path.dirname(outputFilePath), { recursive: true });
        fs.writeFileSync(outputFilePath, clientFactory);
        cb(null, file);
      }));
  });
// Define a task that uses the generate file and includes the copy task
gulp.task('copy', () => {
    return gulp.src('src/**/*')
        .pipe(gulp.dest('build'))
});

// Define a task to clean the build directory
gulp.task('clean', async () => {
    const del = await import('del');
    return del(['build']);
});

// Default task

gulp.task('generate', gulp.series('copy','generate-sdk'));

gulp.task('default', gulp.series('clean', 'generate'));