add support for snowflake integrations (#53)

* add support for snowflake integrations

* bumping sdk version and removing replace in go mod

---------

Co-authored-by: bradlo <[email protected]>

Author: torkins

CHANGELOG.md

@@ -1,17 +1,22 @@
 # Changelog
 
-## v0.1.3
+## v0.1.4-alpha
+
+* Add Snowflake integration commands
+
+## v0.1.3-alpha
 
 * Fix create-user role argument
 
-## v0.1.2
+## v0.1.2-alpha
 
 * Add list-edb-names command
 * Add --prefix option to load-models
 * Add get-model-source command
 * Add prebuilt binaries on release
-## V0.1.1
 
-## v0.1.0
+## v0.1.1-alpha
+
+## v0.1.0-alpha
 
 * Initial preview release

go.mod

@@ -4,7 +4,7 @@ go 1.18
 
 require (
 	github.com/pkg/errors v0.9.1
-	github.com/relationalai/rai-sdk-go v0.3.1-alpha
+	github.com/relationalai/rai-sdk-go v0.4.3-alpha
 	github.com/spf13/cobra v1.5.0
 )
 
@@ -15,6 +15,7 @@ require (
 	github.com/inconshreveable/mousetrap v1.0.0 // indirect
 	github.com/klauspost/compress v1.15.8 // indirect
 	github.com/pierrec/lz4/v4 v4.1.15 // indirect
+	github.com/shopspring/decimal v1.3.1 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect
 	golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4 // indirect
 	golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8 // indirect

go.sum

@@ -139,8 +139,8 @@ github.com/google/go-cmp v0.5.6/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/
 github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
 github.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=
 github.com/google/uuid v1.0.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/google/uuid v1.1.2 h1:EVhdT+1Kseyi1/pUmXKaFxYsDNy9RQYkMWRH68J/W7Y=
 github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
-github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
 github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY=
 github.com/gorilla/context v1.1.1/go.mod h1:kBGZzfjB9CEq2AlWe17Uuf7NDRt0dE0s8S51q0aT7Yg=
 github.com/gorilla/mux v1.6.2/go.mod h1:1lud6UwP+6orDFRuTfBEV8e9/aOM/c4fVVCaMa2zaAs=
@@ -277,8 +277,8 @@ github.com/prometheus/procfs v0.0.0-20190117184657-bf6a532e95b1/go.mod h1:c3At6R
 github.com/prometheus/procfs v0.0.2/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA=
 github.com/prometheus/procfs v0.0.8/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A=
 github.com/rcrowley/go-metrics v0.0.0-20181016184325-3113b8401b8a/go.mod h1:bCqnVzQkZxMG4s8nGwiZ5l3QUCyqpo9Y+/ZMZ9VjZe4=
-github.com/relationalai/rai-sdk-go v0.3.1-alpha h1:pXU6bdVdvRoeS1nHZBH7G0UU5OQxNgqERGfKUCBQOlQ=
-github.com/relationalai/rai-sdk-go v0.3.1-alpha/go.mod h1:SPIBXtTY6gHuS343dHRuv/PfUImOw0HkL2sANljk2Hg=
+github.com/relationalai/rai-sdk-go v0.4.3-alpha h1:uz3Dkn8Qg45yM9NACYdizUGJhrU6DRDoLONNwTMy94o=
+github.com/relationalai/rai-sdk-go v0.4.3-alpha/go.mod h1:qfIZ7OiUpM+ZLPB8m358DFQPRPFzMjM4ghX4iKR8UHo=
 github.com/rogpeppe/fastuuid v0.0.0-20150106093220-6724a57986af/go.mod h1:XWv6SoW27p1b0cqNHllgS5HIMJraePCO15w5zCzIWYg=
 github.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ=
 github.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=
@@ -290,6 +290,8 @@ github.com/ruudk/golang-pdf417 v0.0.0-20181029194003-1af4ab5afa58/go.mod h1:6lfF
 github.com/ryanuber/columnize v0.0.0-20160712163229-9b3edd62028f/go.mod h1:sm1tb6uqfes/u+d4ooFouqFdy9/2g9QGwK3SQygK0Ts=
 github.com/samuel/go-zookeeper v0.0.0-20190923202752-2cc03de413da/go.mod h1:gi+0XIa01GRL2eRQVjQkKGqKF3SF9vZR/HnPullcV2E=
 github.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529/go.mod h1:DxrIzT+xaE7yg65j358z/aeFdxmN0P9QXhEzd20vsDc=
+github.com/shopspring/decimal v1.3.1 h1:2Usl1nmF/WZucqkFZhnfFYxxxu8LG21F6nPQBE5gKV8=
+github.com/shopspring/decimal v1.3.1/go.mod h1:DKyhrW/HYNuLGql+MJL6WCR6knT2jwCFRcu2hWCYk4o=
 github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=
 github.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=
 github.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE=

rai/cmds.go

@@ -1,4 +1,4 @@
-// Copyright 2022 RelationalAI, Inc.
+// Copyright 2022-2023 RelationalAI, Inc.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -783,6 +783,55 @@ func updateUser(cmd *cobra.Command, args []string) {
 	action.Exit(rsp, err)
 }
 
+//
+// Integrations
+//
+
+func createSnowflakeIntegration(cmd *cobra.Command, args []string) {
+	action := newAction(cmd)
+	name := args[0]
+	account := action.getString("account")
+	adminUsername := action.getString("admin-username")
+	adminPassword := action.getString("admin-password")
+	proxyUsername := action.getString("proxy-username")
+	proxyPassword := action.getString("proxy-password")
+	adminCreds := rai.SnowflakeCredentials{
+		Username: adminUsername, Password: adminPassword}
+	proxyCreds := rai.SnowflakeCredentials{
+		Username: proxyUsername, Password: proxyPassword}
+	action.Start("Create Snowflake integration '%s' account='%s'", name, account)
+	rsp, err := action.Client().CreateSnowflakeIntegration(
+		name, account, &adminCreds, &proxyCreds)
+	action.Exit(rsp, err)
+}
+
+func deleteSnowflakeIntegration(cmd *cobra.Command, args []string) {
+	action := newAction(cmd)
+	name := args[0]
+	adminUsername := action.getString("admin-username")
+	adminPassword := action.getString("admin-password")
+	adminCreds := rai.SnowflakeCredentials{
+		Username: adminUsername, Password: adminPassword}
+	action.Start("Delete Snowflake integration '%s'", name)
+	err := action.Client().DeleteSnowflakeIntegration(name, &adminCreds)
+	action.Exit(nil, err)
+}
+
+func getSnowflakeIntegration(cmd *cobra.Command, args []string) {
+	action := newAction(cmd)
+	name := args[0]
+	action.Start("Get Snowflake integration '%s'", name)
+	rsp, err := action.Client().GetSnowflakeIntegration(name)
+	action.Exit(rsp, err)
+}
+
+func listSnowflakeIntegrations(cmd *cobra.Command, _ []string) {
+	action := newAction(cmd)
+	action.Start("List Snowflake integrations")
+	rsp, err := action.Client().ListSnowflakeIntegrations()
+	action.Exit(rsp, err)
+}
+
 //
 // Misc
 //

rai/main.go

@@ -1,4 +1,4 @@
-// Copyright 2022 RelationalAI, Inc.
+// Copyright 2022-2023 RelationalAI, Inc.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -36,7 +36,7 @@ func addCommands(root *cobra.Command) {
 
 	cmd = &cobra.Command{
 		Use:   "delete-database database",
-		Short: "Delete a database databaase",
+		Short: "Delete a database",
 		Args:  cobra.ExactArgs(1),
 		Run:   deleteDatabase}
 	root.AddCommand(cmd)
@@ -290,8 +290,43 @@ func addCommands(root *cobra.Command) {
 	cmd.Flags().StringArray("role", nil, "user roles")
 	root.AddCommand(cmd)
 
-	// Misc
+	// Integrations
+	cmd = &cobra.Command{
+		Use:   "create-snowflake-integration name",
+		Short: "Create a Snowflake integration",
+		Args:  cobra.ExactArgs(1),
+		Run:   createSnowflakeIntegration}
+	cmd.Flags().String("account", "", "Snowflake account")
+	cmd.Flags().String("admin-username", "", "admin username")
+	cmd.Flags().String("admin-password", "", "admin password")
+	cmd.Flags().String("proxy-username", "", "proxy username")
+	cmd.Flags().String("proxy-password", "", "proxy password")
+	root.AddCommand(cmd)
+
+	cmd = &cobra.Command{
+		Use:   "delete-snowflake-integration name",
+		Short: "Delete a Snowflake integration",
+		Args:  cobra.ExactArgs(1),
+		Run:   deleteSnowflakeIntegration}
+	cmd.Flags().String("admin-username", "", "admin username")
+	cmd.Flags().String("admin-password", "", "admin password")
+	root.AddCommand(cmd)
 
+	cmd = &cobra.Command{
+		Use:   "get-snowflake-integration name",
+		Short: "Get information about the given Snowflake integration",
+		Args:  cobra.ExactArgs(1),
+		Run:   getSnowflakeIntegration}
+	root.AddCommand(cmd)
+
+	cmd = &cobra.Command{
+		Use:   "list-snowflake-integrations",
+		Short: "List all Snowflake integrations",
+		Args:  cobra.ExactArgs(0),
+		Run:   listSnowflakeIntegrations}
+	root.AddCommand(cmd)
+
+	// Misc
 	cmd = &cobra.Command{
 		Use:   "get-access-token",
 		Short: "Get OAuth access token",