@@ -11,15 +11,18 @@ public class EnvironmentVariables {
1111 public static final String CODEQL_EXTRACTOR_JAVASCRIPT_SCRATCH_DIR_ENV_VAR =
1212 "CODEQL_EXTRACTOR_JAVASCRIPT_SCRATCH_DIR" ;
1313
14+ public static final String LGTM_WORKSPACE_ENV_VAR =
15+ "LGTM_WORKSPACE" ;
16+
1417 /**
1518 * Gets the extractor root based on the <code>CODEQL_EXTRACTOR_JAVASCRIPT_ROOT</code> or <code>
1619 * SEMMLE_DIST</code> or environment variable, or <code>null</code> if neither is set.
1720 */
1821 public static String tryGetExtractorRoot () {
19- String env = Env .systemEnv ().get (CODEQL_EXTRACTOR_JAVASCRIPT_ROOT_ENV_VAR );
20- if (env != null && ! env . isEmpty () ) return env ;
21- env = Env .systemEnv ().get (Var .SEMMLE_DIST );
22- if (env != null && ! env . isEmpty () ) return env ;
22+ String env = Env .systemEnv ().getNonEmpty (CODEQL_EXTRACTOR_JAVASCRIPT_ROOT_ENV_VAR );
23+ if (env != null ) return env ;
24+ env = Env .systemEnv ().getNonEmpty (Var .SEMMLE_DIST );
25+ if (env != null ) return env ;
2326 return null ;
2427 }
2528
@@ -35,11 +38,15 @@ public static String getExtractorRoot() {
3538 return env ;
3639 }
3740
41+ /**
42+ * Gets the scratch directory from the appropriate environment variable.
43+ */
3844 public static String getScratchDir () {
39- String env = Env .systemEnv ().get (CODEQL_EXTRACTOR_JAVASCRIPT_SCRATCH_DIR_ENV_VAR );
40- if (env == null ) {
41- throw new UserError (CODEQL_EXTRACTOR_JAVASCRIPT_SCRATCH_DIR_ENV_VAR + " must be set" );
42- }
43- return env ;
45+ String env = Env .systemEnv ().getNonEmpty (CODEQL_EXTRACTOR_JAVASCRIPT_SCRATCH_DIR_ENV_VAR );
46+ if (env != null ) return env ;
47+ env = Env .systemEnv ().getNonEmpty (LGTM_WORKSPACE_ENV_VAR );
48+ if (env != null ) return env ;
49+
50+ throw new UserError (CODEQL_EXTRACTOR_JAVASCRIPT_SCRATCH_DIR_ENV_VAR + " or " + LGTM_WORKSPACE_ENV_VAR + " must be set" );
4451 }
4552}
0 commit comments