From 2504b822131093d6aa7043c56aaf69e650a977f6 Mon Sep 17 00:00:00 2001
From: spalger <spalger@users.noreply.github.com>
Date: Fri, 2 Oct 2020 17:04:55 -0700
Subject: [PATCH] split up cache based on upstream branch

---
 packages/kbn-dev-utils/src/index.ts      |  2 +-
 packages/kbn-optimizer/src/node/cache.ts |  4 +-
 packages/kbn-utils/src/repo_root.ts      | 50 +++++++++++++++---------
 3 files changed, 34 insertions(+), 22 deletions(-)

diff --git a/packages/kbn-dev-utils/src/index.ts b/packages/kbn-dev-utils/src/index.ts
index 8217999b01128..6a845825f0fd4 100644
--- a/packages/kbn-dev-utils/src/index.ts
+++ b/packages/kbn-dev-utils/src/index.ts
@@ -17,7 +17,7 @@
  * under the License.
  */
 
-export { REPO_ROOT } from '@kbn/utils';
+export * from '@kbn/utils';
 export { withProcRunner, ProcRunner } from './proc_runner';
 export * from './tooling_log';
 export * from './serializers';
diff --git a/packages/kbn-optimizer/src/node/cache.ts b/packages/kbn-optimizer/src/node/cache.ts
index 5ebe4fb353129..7fbf009e38a7d 100644
--- a/packages/kbn-optimizer/src/node/cache.ts
+++ b/packages/kbn-optimizer/src/node/cache.ts
@@ -21,9 +21,9 @@ import Path from 'path';
 
 // @ts-expect-error no types available
 import * as LmdbStore from 'lmdb-store';
-import { REPO_ROOT } from '@kbn/dev-utils';
+import { REPO_ROOT, UPSTREAM_BRANCH } from '@kbn/dev-utils';
 
-const CACHE_DIR = Path.resolve(REPO_ROOT, 'data/node_auto_transpilation_cache');
+const CACHE_DIR = Path.resolve(REPO_ROOT, 'data/node_auto_transpilation_cache', UPSTREAM_BRANCH);
 const reportError = () => {
   // right now I'm not sure we need to worry about errors, the cache isn't actually
   // necessary, and if the cache is broken it should just rebuild on the next restart
diff --git a/packages/kbn-utils/src/repo_root.ts b/packages/kbn-utils/src/repo_root.ts
index b33b28d8d6e2f..ed3516a7304f9 100644
--- a/packages/kbn-utils/src/repo_root.ts
+++ b/packages/kbn-utils/src/repo_root.ts
@@ -22,38 +22,50 @@ import Fs from 'fs';
 
 import loadJsonFile from 'load-json-file';
 
-const isKibanaDir = (dir: string) => {
+const readKibanaPkgJson = (dir: string) => {
   try {
     const path = Path.resolve(dir, 'package.json');
     const json = loadJsonFile.sync(path);
     if (json && typeof json === 'object' && 'name' in json && json.name === 'kibana') {
-      return true;
+      return json;
     }
   } catch (error) {
     if (error && error.code === 'ENOENT') {
-      return false;
+      return;
     }
 
     throw error;
   }
 };
 
-// search for the kibana directory, since this file is moved around it might
-// not be where we think but should always be a relatively close parent
-// of this directory
-const startDir = Fs.realpathSync(__dirname);
-const { root: rootDir } = Path.parse(startDir);
-let cursor = startDir;
-while (true) {
-  if (isKibanaDir(cursor)) {
-    break;
-  }
+const findKibanaPackageJson = () => {
+  // search for the kibana directory, since this file is moved around it might
+  // not be where we think but should always be a relatively close parent
+  // of this directory
+  const startDir = Fs.realpathSync(__dirname);
+  const { root: rootDir } = Path.parse(startDir);
+  let cursor = startDir;
+  while (true) {
+    const kibanaPkgJson = readKibanaPkgJson(cursor);
+    if (kibanaPkgJson) {
+      return {
+        kibanaDir: cursor,
+        kibanaPkgJson: kibanaPkgJson as {
+          name: string;
+          branch: string;
+        },
+      };
+    }
 
-  const parent = Path.dirname(cursor);
-  if (parent === rootDir) {
-    throw new Error(`unable to find kibana directory from ${startDir}`);
+    const parent = Path.dirname(cursor);
+    if (parent === rootDir) {
+      throw new Error(`unable to find kibana directory from ${startDir}`);
+    }
+    cursor = parent;
   }
-  cursor = parent;
-}
+};
+
+const { kibanaDir, kibanaPkgJson } = findKibanaPackageJson();
 
-export const REPO_ROOT = cursor;
+export const REPO_ROOT = kibanaDir;
+export const UPSTREAM_BRANCH = kibanaPkgJson.branch;