✨ Allow loading of local extensions in the CLI

Author: julien040

cmd/query.go

@@ -25,6 +25,7 @@ func init() {
 	queryCmd.Flags().Bool("read-only", false, "Start the server in read-only mode")
 	queryCmd.Flags().StringArray("init", []string{}, "Run SQL commands in a file before the query. You can specify multiple files.")
 	queryCmd.Flags().Bool("dev", false, "Run the program in developer mode")
+	queryCmd.Flags().StringSlice("extension", []string{}, "Load one or more extensions by specifying their path. Separate multiple extensions with a comma.")
 
 	// Query flags
 	queryCmd.Flags().StringP("query", "q", "", "Query to run")

cmd/root.go

@@ -38,6 +38,7 @@ func init() {
 	rootCmd.Flags().Bool("read-only", false, "Start the server in read-only mode")
 	rootCmd.Flags().StringArray("init", []string{}, "Run SQL commands in a file before the query. You can specify multiple files.")
 	rootCmd.Flags().Bool("dev", false, "Run the program in developer mode")
+	rootCmd.Flags().StringSlice("extension", []string{}, "Load one or more extensions by specifying their path. Separate multiple extensions with a comma.")
 
 	// Query flags
 	rootCmd.Flags().StringP("query", "q", "", "Query to run")

cmd/server.go

@@ -25,6 +25,7 @@ func init() {
 	serverCmd.Flags().String("log-file", "/dev/stdout", "Log file")
 	serverCmd.Flags().String("auth-file", "", "Path to the authentication file")
 	serverCmd.Flags().Bool("dev", false, "Run the program in developer mode")
+	serverCmd.Flags().StringSlice("extension", []string{}, "Load one or more extensions by specifying their path. Separate multiple extensions with a comma.")
 
 	addFlag_commandModifiesConfiguration(serverCmd)
 }

controller/profiles.go

@@ -123,6 +123,11 @@ func createOrUpdateProfile(queries *model.Queries, registryName string, pluginNa
 		return err
 	}
 
+	// If the plugin is a shared extension, we don't need to ask for the configuration
+	if pluginInfo.Issharedextension == 1 {
+		return nil
+	}
+
 	// Parse the asked configuration required by the plugin
 	var toAskConfig []registry.UserConfig
 	err = json.Unmarshal([]byte(pluginInfo.Config), &toAskConfig)

controller/query.go

@@ -106,6 +106,15 @@ func Query(cmd *cobra.Command, args []string) error {
 		return fmt.Errorf("failed to load config: %w", err)
 	}
 
+	// Get the extensions
+	extensions, _ := cmd.Flags().GetStringSlice("extension")
+	for _, extension := range extensions {
+		err = namespace.LoadSharedExtension(extension, "")
+		if err != nil {
+			return fmt.Errorf("failed to load extension: %w", err)
+		}
+	}
+
 	// Register the namespace
 	db, err := namespace.Register("main")
 	if err != nil {

controller/server.go

@@ -117,6 +117,15 @@ func Server(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	// Get the extensions
+	extensions, _ := cmd.Flags().GetStringSlice("extension")
+	for _, extension := range extensions {
+		err = instance.LoadSharedExtension(extension, "")
+		if err != nil {
+			return fmt.Errorf("failed to load extension: %w", err)
+		}
+	}
+
 	// We register the namespace
 	db, err := instance.Register("")
 	if err != nil {

go.mod

@@ -2,7 +2,7 @@ module github.com/julien040/anyquery
 
 go 1.22.3
 
-replace github.com/mattn/go-sqlite3 => github.com/julien040/go-sqlite3-anyquery v1.17.5
+replace github.com/mattn/go-sqlite3 => github.com/julien040/go-sqlite3-anyquery v1.17.6
 
 replace github.com/runreveal/pql => github.com/julien040/pql-anyquery v0.2.1
 

go.sum

@@ -442,8 +442,8 @@ github.com/jmoiron/sqlx v1.4.0 h1:1PLqN7S1UYp5t4SrVVnt4nUVNemrDAtxlulVe+Qgm3o=
 github.com/jmoiron/sqlx v1.4.0/go.mod h1:ZrZ7UsYB/weZdl2Bxg6jCRO9c3YHl8r3ahlKmRT4JLY=
 github.com/jstemmer/go-junit-report v0.0.0-20190106144839-af01ea7f8024/go.mod h1:6v2b51hI/fHJwM22ozAgKL4VKDeJcHhJFhtBdhmNjmU=
 github.com/jstemmer/go-junit-report v0.9.1/go.mod h1:Brl9GWCQeLvo8nXZwPNNblvFj/XSXhF0NWZEnDohbsk=
-github.com/julien040/go-sqlite3-anyquery v1.17.5 h1:rSP5owX6B9NTmJR4qMo6aDBzEntuMIq4/1rFvOgX2tw=
-github.com/julien040/go-sqlite3-anyquery v1.17.5/go.mod h1:9t7/JQ99yNR2b18cBR11VGyrJYg+Q7IQNHMWoCt6yGE=
+github.com/julien040/go-sqlite3-anyquery v1.17.6 h1:AJAPWCDhpZVVu10XXBOg5artrVatNaUyXR5txU9xdrY=
+github.com/julien040/go-sqlite3-anyquery v1.17.6/go.mod h1:9t7/JQ99yNR2b18cBR11VGyrJYg+Q7IQNHMWoCt6yGE=
 github.com/julien040/go-ternary v0.0.0-20230119180150-f0435f66948e h1:q8lhYSYDzN8slDRCVRt2TD2ShyjNcuQU+I9LZNPv4TM=
 github.com/julien040/go-ternary v0.0.0-20230119180150-f0435f66948e/go.mod h1:XXIcjDHL7vyuHA7V0UwaTKMscsqKzFkE9FTGbBeqJHM=
 github.com/julien040/pql-anyquery v0.2.1 h1:k4eF28S8diwFYWg1r6fKiA7ZmeiF65PTersyUY3GqVw=