Skip to content

Commit 92dbfb2

Browse files
committed
JS: Handle LGTM_WORKSPACE and fix emptiness check
1 parent 141d4bf commit 92dbfb2

1 file changed

Lines changed: 16 additions & 9 deletions

File tree

javascript/extractor/src/com/semmle/js/extractor/EnvironmentVariables.java

Lines changed: 16 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)