[Feature] Greatly improved forms with dbt profile browsing

fetch_dbt_profiles.go

@@ -0,0 +1,50 @@
+package main
+
+import (
+	"log"
+	"os"
+	"path/filepath"
+
+	"gopkg.in/yaml.v2"
+)
+
+type DbtProfile struct {
+	Target  string `yaml:"target"`
+	Outputs map[string]struct {
+		ConnType      string `yaml:"type"`
+		Account       string `yaml:"account"`
+		User          string `yaml:"user"`
+		Role          string `yaml:"role"`
+		Authenticator string `yaml:"authenticator"`
+		Database      string `yaml:"database"`
+		Schema        string `yaml:"schema"`
+		Project       string `yaml:"project"`
+		Dataset       string `yaml:"dataset"`
+		Path          string `yaml:"path"`
+		Threads       int    `yaml:"threads"`
+	} `yaml:"outputs"`
+}
+
+type DbtProfiles map[string]DbtProfile
+
+func FetchDbtProfiles() (DbtProfiles, error) {
+	paths := []string{
+		filepath.Join(".", "profiles.yml"),
+		filepath.Join(os.Getenv("HOME"), ".dbt", "profiles.yml"),
+	}
+	ps := DbtProfiles{}
+	for _, path := range paths {
+		pf := DbtProfiles{}
+		yf, err := os.ReadFile(path)
+		if err != nil {
+			continue
+		}
+		if err = yaml.Unmarshal(yf, pf); err != nil {
+			log.Fatalf("Could not read dbt profile, \nlikely unsupported fields or formatting issues\n please open an issue: %v\n", err)
+		}
+		for k, v := range pf {
+			ps[k] = v
+		}
+	}
+	return ps, nil
+}

fetch_dbt_profiles_test.go

@@ -0,0 +1,28 @@
+package main
+
+import (
+	"os"
+	"testing"
+)
+
+func TestFetchDbtProfiles(t *testing.T) {
+	CreateTempDbtProfiles(t)
+	defer os.RemoveAll(os.Getenv("HOME"))
+	defer os.Unsetenv("HOME")
+	profiles, err := FetchDbtProfiles()
+	if err != nil {
+		t.Fatalf("Error fetching dbt profiles: %v\n", err)
+	}
+	if err != nil {
+		t.Fatalf("Error fetching dbt profiles: %v\n", err)
+	}
+	if profiles["elf"].Outputs["dev"].ConnType != "snowflake" {
+		t.Fatalf("Expected snowflake, got %s\n", profiles["elf"].Outputs["dev"].ConnType)
+	}
+	if profiles["human"].Outputs["dev"].ConnType != "bigquery" {
+		t.Fatalf("Expected bigquery, got %s\n", profiles["human"].Outputs["dev"].ConnType)
+	}
+	if profiles["dwarf"].Outputs["dev"].ConnType != "duckdb" {
+		t.Fatalf("Expected duckdb, got %s\n", profiles["dwarf"].Outputs["dev"].ConnType)
+	}
+}