Allow to download files from interncache

Summary: Added a query param to gate inerncache urls from the existing controller to the newer plugin veresions only

Reviewed By: mweststrate

Differential Revision: D64055855

fbshipit-source-id: 2ad02f688b0eeb1bb1f86673167ac6b0a7cb5e7c

desktop/flipper-common/src/server-types.tsx

@@ -506,6 +506,7 @@ export interface DownloadFileStartOptions {
   maxRedirects?: number;
   headers?: Record<string, string>;
   overwrite?: boolean;
+  proxy?: {host: string; port: number; protocol?: string};
 }
 
 export type DownloadFileUpdate = {

desktop/flipper-server/src/FlipperServerImpl.tsx

@@ -32,6 +32,7 @@ import {
   Settings,
   ClientQuery,
   RmOptions,
+  DownloadFileStartOptions,
 } from 'flipper-common';
 import {ServerDevice} from './devices/ServerDevice';
 import {Base64} from 'js-base64';
@@ -49,6 +50,7 @@ import {sendScribeLogs} from './fb-stubs/sendScribeLogs';
 import {
   internGraphGETAPIRequest,
   internGraphPOSTAPIRequest,
+  rewriteInternRequest,
 } from './fb-stubs/internRequests';
 import {commandNodeApiExec} from './commands/NodeApiExec';
 import {commandDownloadFileStartFactory} from './commands/DownloadFile';
@@ -501,9 +503,26 @@ export class FlipperServerImpl implements FlipperServer {
     'node-api-fs-writefile-binary': (path, base64contents) =>
       writeFile(path, Base64.toUint8Array(base64contents), 'binary'),
     // TODO: Do we need API to cancel an active download?
-    'download-file-start': commandDownloadFileStartFactory(
-      this.emit.bind(this),
-    ),
+    'download-file-start': async (
+      endpoint: string,
+      dest: string,
+      options?: DownloadFileStartOptions,
+    ) => {
+      const downloadFileStart = commandDownloadFileStartFactory(
+        this.emit.bind(this),
+      );
+
+      const {url, headers, proxy} = await rewriteInternRequest(
+        endpoint,
+        options?.headers ?? {},
+      );
+
+      return downloadFileStart(url, dest, {
+        ...options,
+        proxy: proxy ?? undefined,
+        headers: headers as Record<string, string>,
+      });
+    },
     'get-config': async () => this.config,
     'get-changelog': getChangelog,
     'device-find': async (deviceSerial) => {

desktop/flipper-server/src/commands/DownloadFile.tsx

@@ -31,7 +31,7 @@ export const commandDownloadFileStartFactory =
   async (
     url,
     dest,
-    {method = 'GET', timeout, maxRedirects, headers, overwrite} = {},
+    {method = 'GET', timeout, maxRedirects, headers, overwrite, proxy} = {},
   ) => {
     const destExists = await pathExists(dest);
 
@@ -70,6 +70,7 @@ export const commandDownloadFileStartFactory =
       timeout,
       maxRedirects,
       headers,
+      proxy,
     });
     let totalSize = parseInt(response.headers['content-length'], 10);
     if (Number.isNaN(totalSize)) {

desktop/flipper-server/src/fb-stubs/internRequests.tsx

@@ -7,6 +7,7 @@
  * @format
  */
 
+import type {AxiosProxyConfig} from 'axios';
 import {GraphFileUpload, GraphResponse} from 'flipper-common';
 
 /* eslint-disable @typescript-eslint/no-unused-vars */
@@ -43,3 +44,14 @@ export async function internGraphGETAPIRequest(
 ): Promise<GraphResponse> {
   throw new Error('Feature not implemented');
 }
+
+export async function rewriteInternRequest(
+  url: string,
+  headers: Record<string, string>,
+): Promise<{
+  url: string;
+  headers: Record<string, string>;
+  proxy: AxiosProxyConfig | null;
+}> {
+  return {url, headers, proxy: null};
+}