feat: add pre run scripts

Author: benfdking

js/packages/proto/src/generated/quary/service/v1/connection_config.ts

@@ -267,6 +267,7 @@ const DatabaseSelection: React.FC<{
                     sqliteInMemory: {},
                   },
                   vars: [],
+                  preRunScripts: [],
                 })
                 break
               }
@@ -329,6 +330,7 @@ const ConfigurationDetails: React.FC<ConfigurationDetailsProps> = ({
             },
           },
           vars: [],
+          preRunScripts: [],
         }
       case DatabaseOnboardingOptions.SQLite:
         return {
@@ -339,6 +341,7 @@ const ConfigurationDetails: React.FC<ConfigurationDetailsProps> = ({
             },
           },
           vars: [],
+          preRunScripts: [],
         }
       case DatabaseOnboardingOptions.Snowflake:
         return {
@@ -351,6 +354,7 @@ const ConfigurationDetails: React.FC<ConfigurationDetailsProps> = ({
             },
           },
           vars: [],
+          preRunScripts: [],
         }
       default: {
         throw new Error(`invalid database type ${states.sourceDetails.type}`)

js/packages/quary-extension-ui/src/views/OnboardingView.tsx

@@ -119,6 +119,16 @@ export class ConnectionConfig extends Message<ConnectionConfig> {
    */
   vars: Var[] = [];
 
+  /**
+   * pre_run_scripts is a list of scripts that should be run before a run job against the database. This is useful for
+   * setting up the database for the run job. For example creating UDFs or installing plugins ...
+   * The scripts are run in order and should be idempotent.
+   * The list points to the location of the script in the project.
+   *
+   * @generated from field: repeated string pre_run_scripts = 11;
+   */
+  preRunScripts: string[] = [];
+
   constructor(data?: PartialMessage<ConnectionConfig>) {
     super();
     proto3.util.initPartial(data, this);
@@ -137,6 +147,7 @@ export class ConnectionConfig extends Message<ConnectionConfig> {
     { no: 9, name: "redshift", kind: "message", T: ConnectionConfig_ConnectionConfigRedshift, oneof: "config" },
     { no: 10, name: "clickhouse", kind: "message", T: ConnectionConfig_ConnectionConfigClickHouse, oneof: "config" },
     { no: 8, name: "vars", kind: "message", T: Var, repeated: true },
+    { no: 11, name: "pre_run_scripts", kind: "scalar", T: 9 /* ScalarType.STRING */, repeated: true },
   ]);
 
   static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): ConnectionConfig {

proto/gen/es/quary/service/v1/connection_config_pb.ts

@@ -46,6 +46,13 @@
                     },
                     "additionalProperties": false,
                     "type": "array"
+                },
+                "pre_run_scripts": {
+                    "items": {
+                        "type": "string"
+                    },
+                    "type": "array",
+                    "description": "pre_run_scripts is a list of scripts that should be run before a run job against the database. This is useful for setting up the database for the run job. For example creating UDFs or installing plugins ... The scripts are run in order and should be idempotent. The list points to the location of the script in the project."
                 }
             },
             "additionalProperties": false,

proto/gen/json_schema/ConnectionConfig.json

@@ -57,6 +57,13 @@
                     },
                     "additionalProperties": false,
                     "type": "array"
+                },
+                "pre_run_scripts": {
+                    "items": {
+                        "type": "string"
+                    },
+                    "type": "array",
+                    "description": "pre_run_scripts is a list of scripts that should be run before a run job against the database. This is useful for setting up the database for the run job. For example creating UDFs or installing plugins ... The scripts are run in order and should be idempotent. The list points to the location of the script in the project."
                 }
             },
             "additionalProperties": false,

proto/gen/json_schema/GenerateProjectFilesRequest.json

@@ -57,6 +57,13 @@
                     },
                     "additionalProperties": false,
                     "type": "array"
+                },
+                "pre_run_scripts": {
+                    "items": {
+                        "type": "string"
+                    },
+                    "type": "array",
+                    "description": "pre_run_scripts is a list of scripts that should be run before a run job against the database. This is useful for setting up the database for the run job. For example creating UDFs or installing plugins ... The scripts are run in order and should be idempotent. The list points to the location of the script in the project."
                 }
             },
             "additionalProperties": false,

proto/gen/json_schema/GetProjectConfigResponse.json

@@ -135,6 +135,13 @@
                     },
                     "additionalProperties": false,
                     "type": "array"
+                },
+                "pre_run_scripts": {
+                    "items": {
+                        "type": "string"
+                    },
+                    "type": "array",
+                    "description": "pre_run_scripts is a list of scripts that should be run before a run job against the database. This is useful for setting up the database for the run job. For example creating UDFs or installing plugins ... The scripts are run in order and should be idempotent. The list points to the location of the script in the project."
                 }
             },
             "additionalProperties": false,

proto/gen/json_schema/ParseProjectResponse.json

@@ -191,6 +191,13 @@
                     },
                     "additionalProperties": false,
                     "type": "array"
+                },
+                "pre_run_scripts": {
+                    "items": {
+                        "type": "string"
+                    },
+                    "type": "array",
+                    "description": "pre_run_scripts is a list of scripts that should be run before a run job against the database. This is useful for setting up the database for the run job. For example creating UDFs or installing plugins ... The scripts are run in order and should be idempotent. The list points to the location of the script in the project."
                 }
             },
             "additionalProperties": false,

proto/gen/json_schema/Project.json

@@ -141,6 +141,13 @@
                     },
                     "additionalProperties": false,
                     "type": "array"
+                },
+                "pre_run_scripts": {
+                    "items": {
+                        "type": "string"
+                    },
+                    "type": "array",
+                    "description": "pre_run_scripts is a list of scripts that should be run before a run job against the database. This is useful for setting up the database for the run job. For example creating UDFs or installing plugins ... The scripts are run in order and should be idempotent. The list points to the location of the script in the project."
                 }
             },
             "additionalProperties": false,