Move snowflake specific configuration into SnowflakeConfig

Author: sfc-gh-hyu

src/main/groovy/nextflow/snowflake/SnowflakeExecutor.groovy

@@ -7,6 +7,7 @@ import nextflow.processor.TaskHandler
 import nextflow.processor.TaskMonitor
 import nextflow.processor.TaskPollingMonitor
 import nextflow.processor.TaskRun
+import nextflow.snowflake.config.SnowflakeConfig
 import nextflow.util.ServiceName
 import nextflow.util.Duration
 import org.pf4j.ExtensionPoint
@@ -23,7 +24,7 @@ import java.sql.ResultSet
 @ServiceName('snowflake')
 @CompileStatic
 class SnowflakeExecutor extends Executor implements ExtensionPoint {
-    Map snowflakeConfig
+    SnowflakeConfig snowflakeConfig
 
     /**
      * A path where executable scripts from bin directory are copied
@@ -47,7 +48,7 @@ class SnowflakeExecutor extends Executor implements ExtensionPoint {
         Statement stmt = conn.createStatement();
         final Map<String, String> registryMappings = new HashMap<>()
 
-        String mappings = snowflakeConfig.get("registryMappings", "")
+        String mappings = snowflakeConfig.registryMappings ?: ""
 
         // Skip processing if mappings is empty
         if (mappings == null || mappings.trim().isEmpty()) {
@@ -119,7 +120,8 @@ class SnowflakeExecutor extends Executor implements ExtensionPoint {
     @Override
     protected void register() {
         super.register()
-        snowflakeConfig = session.config.navigate("snowflake") as Map
+        final Map configMap = session.config.navigate("snowflake") as Map
+        snowflakeConfig = new SnowflakeConfig(configMap ?: [:])
 
         // Validate that workDir uses snowflake:// scheme
         validateWorkDir()

src/main/groovy/nextflow/snowflake/SnowflakeTaskHandler.groovy

@@ -174,7 +174,7 @@ class SnowflakeTaskHandler extends TaskHandler {
 
         final String spec = buildJobServiceSpec()
         final String computePoolEnv = System.getenv("computePool")
-        final String defaultComputePool = computePoolEnv!=null ? computePoolEnv : executor.snowflakeConfig.get("computePool")
+        final String defaultComputePool = computePoolEnv!=null ? computePoolEnv : executor.snowflakeConfig.computePool
         final String jobComment = String.format("nextflow task name: %s", task.getName())
 
         String executeSql = String.format("""

src/main/groovy/nextflow/snowflake/config/SnowflakeConfig.groovy

@@ -0,0 +1,73 @@
+package nextflow.snowflake.config
+
+import groovy.transform.CompileStatic
+import groovy.transform.ToString
+import nextflow.config.spec.ConfigOption
+import nextflow.config.spec.ConfigScope
+import nextflow.config.spec.ScopeName
+import nextflow.script.dsl.Description
+
+/**
+ * Configuration scope for Snowflake executor plugin.
+ *
+ * This class defines all configuration options available under the 'snowflake' scope
+ * in Nextflow configuration files.
+ *
+ * @author Hongxin Yu <hongxin.yu@snowflake.com>
+ */
+@ScopeName('snowflake')
+@Description('''
+    The `snowflake` scope allows you to configure the Snowflake executor plugin.
+
+    Example configuration:
+    ```
+    snowflake {
+        computePool = 'MY_COMPUTE_POOL'
+        registryMappings = 'docker.io:my_registry,ghcr.io:github_registry'
+    }
+    ```
+''')
+@CompileStatic
+@ToString(includeNames = true, includePackage = false)
+class SnowflakeConfig implements ConfigScope {
+
+    @ConfigOption
+    @Description('''
+        The name of the Snowflake compute pool to use for executing jobs.
+        This can also be configured via the `computePool` environment variable.
+
+        Example: `computePool = 'MY_COMPUTE_POOL'`
+    ''')
+    String computePool
+
+    @ConfigOption
+    @Description('''
+        Docker registry mappings for container images. This allows mapping external
+        Docker registries to Snowflake image repositories.
+
+        Format: Comma-separated list of mappings in the form `original:replacement`
+
+        Example: `registryMappings = 'docker.io:my_registry,ghcr.io:github_registry'`
+    ''')
+    String registryMappings
+
+    /**
+     * No-argument constructor required for configuration validation support.
+     */
+    SnowflakeConfig() {
+    }
+
+    /**
+     * Constructor accepting a Map for initialization from Nextflow configuration.
+     *
+     * @param opts Configuration options map
+     */
+    SnowflakeConfig(Map opts) {
+        if (opts == null) {
+            return
+        }
+
+        this.computePool = opts.computePool as String
+        this.registryMappings = opts.registryMappings as String
+    }
+}