Add proxy options to location cmd

Author: emilioalvap

__tests__/locations/index.test.ts

@@ -33,6 +33,8 @@ import { LOCATIONS } from '../fixtures/locationinfo';
 import { Server } from '../utils/server';
 import { CLIMock } from '../utils/test-config';
 import { mkdir, rm, writeFile } from 'fs/promises';
+// import { createProxyServer } from 'http-proxy';
+// import http from "http";
 
 describe('Locations', () => {
   const apiKey = 'foo';
@@ -97,10 +99,11 @@ describe('Locations', () => {
     });
 
     const runLocations = async (args: Array<string> = []) => {
-      const cli = new CLIMock()
+      const cli = new CLIMock(true)
         .args(['locations', ...args])
         .run({ cwd: PROJECT_DIR, env: process.env });
-      expect(await cli.exitCode).toBe(0);
+      // expect(await cli.exitCode).toBe(0);
+      console.log(cli.stderr());
       return cli.stderr();
     };
 
@@ -125,5 +128,50 @@ describe('Locations', () => {
       expect(output).toContain(`custom location 1`);
       expect(output).toContain(`custom location 2`);
     });
+
+    describe('Proxy options', () => {
+      // let requests: Array<any> = [];
+      // let proxyServer;
+
+      beforeAll(async () => {
+        await fakeProjectSetup({ url: server.PREFIX });
+        // proxyServer = createProxyServer({ target: server.PREFIX }).listen(8019);
+        // proxyServer.on('proxyReq', function (proxyReq, req) {
+        //   requests.push(req);
+        // });
+        // const proxy = createProxyServer({});
+
+        //
+        // Create your custom server and just call `proxy.web()` to proxy
+        // a web request to the target passed in the options
+        // also you can use `proxy.ws()` to proxy a websockets request
+        //
+        // proxyServer = http.createServer(function (req, res) {
+        //   proxy.web(req, res, { target: server.PREFIX });
+        // }).listen(8019);
+      });
+
+      // afterAll(async () => {
+      //   proxyServer.close();
+      // });
+
+      // beforeEach(() => {
+      //   requests = []
+      // });
+
+      it('enables proxy based on HTTP_PROXY', async () => {
+        const output = await runLocations([
+          '--url',
+          server.PREFIX,
+          '--auth',
+          apiKey,
+          '--proxy-uri',
+          'http://localhost:9191',
+        ]);
+        console.log(output);
+        // expect(requests).toHaveLength(1);
+        expect(output).toContain(`custom location 1`);
+      });
+    });
   });
 });

src/cli.ts

@@ -275,10 +275,23 @@ program
   )
   .option('--url <url>', 'Kibana URL to fetch all public and private locations')
   .addOption(auth)
+  .addOption(proxyUri)
+  .addOption(proxyToken)
+  .addOption(proxyNoVerify)
+  .addOption(proxyCa)
+  .addOption(proxyCert)
   .action(async (cmdOpts: LocationCmdOptions) => {
     const revert = installTransform();
     const url = cmdOpts.url ?? (await loadSettings(null, true))?.url;
     try {
+      //Set up global proxy agent if any of the related options are set
+      setGlobalProxy(
+        cmdOpts.proxyUri,
+        cmdOpts.proxyToken,
+        cmdOpts.proxyNoVerify ? false : true,
+        cmdOpts.proxyCa,
+        cmdOpts.proxyCert
+      );
       if (url && cmdOpts.auth) {
         const allLocations = await getLocations({
           url,

src/locations/index.ts

@@ -30,6 +30,11 @@ import { generateURL } from '../push/utils';
 export type LocationCmdOptions = {
   auth: string;
   url: string;
+  proxyUri?: string;
+  proxyToken?: string;
+  proxyCa?: string;
+  proxyCert?: string;
+  proxyNoVerify?: string;
 };
 
 type LocationMetadata = {