Skip to content

Commit

Permalink
fix: missing env LINKLORE_PREFIX/ LINKLORE_BASE_URL
Browse files Browse the repository at this point in the history
  • Loading branch information
pluveto committed Nov 26, 2023
1 parent 939c140 commit ae4c846
Showing 1 changed file with 18 additions and 7 deletions.
25 changes: 18 additions & 7 deletions main.go
Original file line number Diff line number Diff line change
Expand Up @@ -87,10 +87,19 @@ func loadConfig() Config {

loadEnvVariables(&inputFile, &outputFile, &baseDir, &prefix, &force)

prefixDefault := "/"
prefixEnv := os.Getenv("LINKLORE_PREFIX")
prefixEnvAlias := os.Getenv("LINKLORE_BASE_URL")
if prefixEnv != "" {
prefixDefault = prefixEnv
} else if prefixEnvAlias != "" {
prefixDefault = prefixEnvAlias
}

flag.StringVar(&inputFile, "i", os.Getenv("LINKLORE_INPUT_FILE"), "input file")
flag.StringVar(&outputFile, "o", os.Getenv("LINKLORE_OUTPUT_FILE"), "output file")
flag.StringVar(&baseDir, "d", os.Getenv("LINKLORE_BASE_DIR"), "base directory")
flag.StringVar(&prefix, "p", os.Getenv("LINKLORE_PREFIX"), "prefix")
flag.StringVar(&prefix, "p", prefixDefault, "prefix")
flag.BoolVar(&force, "f", false, "force overwrite output file")

flag.Usage = func() {
Expand Down Expand Up @@ -230,16 +239,18 @@ func loadEnvVariables(inputFile, outputFile, baseDir, prefix *string, force *boo
continue
}
key, value := parts[0], parts[1]
switch strings.ToLower(key) {
case "input_file":
switch strings.ToUpper(key) {
case "LINKLORE_INPUT_FILE":
*inputFile = value
case "output_file":
case "LINKLORE_OUTPUT_FILE":
*outputFile = value
case "base_dir":
case "LINKLORE_BASE_DIR":
*baseDir = value
case "prefix":
case "LINKLORE_PREFIX":
*prefix = value
case "LINKLORE_BASE_URL":
*prefix = value
case "force":
case "LINKLORE_FORCE":
*force = value == "true" || value == "1"
}
}
Expand Down

0 comments on commit ae4c846

Please sign in to comment.