fix: configurable extractor address

satisfactorymodding · Oct 20, 2023 · 11cb63b · 11cb63b
1 parent 2c09e07
commit 11cb63b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/config/config.go b/config/config.go
@@ -102,4 +102,6 @@ func initializeDefaults() {
 	viper.SetDefault("virustotal.key", "")
 
 	viper.SetDefault("feature_flags.allow_multi_target_upload", false)
+
+	viper.SetDefault("extractor_host", "localhost:50051")
 }
diff --git a/validation/validation.go b/validation/validation.go
@@ -8,6 +8,7 @@ import (
 	"encoding/hex"
 	"encoding/json"
 	"fmt"
+	"github.com/spf13/viper"
 	"io"
 	"path"
 	"path/filepath"
@@ -134,7 +135,7 @@ func ExtractModInfo(ctx context.Context, body []byte, withMetadata bool, withVal
 
 	if withMetadata {
 		// Extract all possible metadata
-		conn, err := grpc.Dial("localhost:50051", grpc.WithTransportCredentials(insecure.NewCredentials()))
+		conn, err := grpc.Dial(viper.GetString("extractor_host"), grpc.WithTransportCredentials(insecure.NewCredentials()))
 		if err != nil {
 			return nil, errors.Wrap(err, "failed to connect to metadata server")
 		}