Merge pull request #327 from parthiv11/vsql

Added Vertica Shell Plugin (vsql)

Author: hculea

plugins/vertica/database_credentials.go

@@ -0,0 +1,51 @@
+package vertica
+
+import (
+	"github.com/1Password/shell-plugins/sdk"
+	"github.com/1Password/shell-plugins/sdk/importer"
+	"github.com/1Password/shell-plugins/sdk/provision"
+	"github.com/1Password/shell-plugins/sdk/schema"
+	"github.com/1Password/shell-plugins/sdk/schema/credname"
+	"github.com/1Password/shell-plugins/sdk/schema/fieldname"
+)
+
+func DatabaseCredentials() schema.CredentialType {
+	return schema.CredentialType{
+		Name:    credname.DatabaseCredentials,
+		DocsURL: sdk.URL("https://www.vertica.com/docs/9.2.x/HTML/Content/Authoring/AdministratorsGuide/DBUsersAndPrivileges/Users/CreatingADatabaseUser.htm"),
+		Fields: []schema.CredentialField{
+			{
+				Name:                fieldname.Host,
+				MarkdownDescription: "Vertica host to connect to.",
+				Optional:            true,
+			},
+			{
+				Name:                fieldname.Port,
+				MarkdownDescription: "Port used to connect to Vertica.",
+				Optional:            true,
+			},
+			{
+				Name:                fieldname.Username,
+				MarkdownDescription: "Vertica user to authenticate as.",
+			},
+			{
+				Name:                fieldname.Password,
+				MarkdownDescription: "Password used to authenticate to Vertica.",
+				Secret:              true,
+			},
+			{
+				Name:                fieldname.Database,
+				MarkdownDescription: "Database name to connect to.",
+			},
+		},
+		DefaultProvisioner: provision.EnvVars(defaultEnvVarMapping),
+		Importer:           importer.TryEnvVarPair(defaultEnvVarMapping)}
+}
+
+var defaultEnvVarMapping = map[string]sdk.FieldName{
+	"VSQL_USER":     fieldname.Username,
+	"VSQL_PASSWORD": fieldname.Password,
+	"VSQL_HOST":     fieldname.Host,
+	"VSQL_PORT":     fieldname.Port,
+	"VSQL_DATABASE": fieldname.Database,
+}

plugins/vertica/database_credentials_test.go

@@ -0,0 +1,57 @@
+package vertica
+
+import (
+	"testing"
+
+	"github.com/1Password/shell-plugins/sdk"
+	"github.com/1Password/shell-plugins/sdk/plugintest"
+	"github.com/1Password/shell-plugins/sdk/schema/fieldname"
+)
+
+func TestDatabaseCredentialsProvisioner(t *testing.T) {
+	plugintest.TestProvisioner(t, DatabaseCredentials().DefaultProvisioner, map[string]plugintest.ProvisionCase{
+		"default": {
+			ItemFields: map[sdk.FieldName]string{
+				fieldname.Username: "vertica",
+				fieldname.Password: "Il0v3y04",
+				fieldname.Host:     "localhost",
+				fieldname.Port:     "5433",
+				fieldname.Database: "VMart",
+			},
+			ExpectedOutput: sdk.ProvisionOutput{
+				Environment: map[string]string{
+					"VSQL_USER":     "vertica",
+					"VSQL_PASSWORD": "Il0v3y04",
+					"VSQL_HOST":     "localhost",
+					"VSQL_PORT":     "5433",
+					"VSQL_DATABASE": "VMart",
+				},
+			},
+		},
+	})
+}
+
+func TestAPIKeyImporter(t *testing.T) {
+	plugintest.TestImporter(t, DatabaseCredentials().Importer, map[string]plugintest.ImportCase{
+		"environment": {
+			Environment: map[string]string{
+				"VSQL_USER":     "vertica",
+				"VSQL_PASSWORD": "Il0v3y04",
+				"VSQL_HOST":     "localhost",
+				"VSQL_PORT":     "5433",
+				"VSQL_DATABASE": "VMart",
+			},
+			ExpectedCandidates: []sdk.ImportCandidate{
+				{
+					Fields: map[sdk.FieldName]string{
+						fieldname.Username: "vertica",
+						fieldname.Password: "Il0v3y04",
+						fieldname.Host:     "localhost",
+						fieldname.Port:     "5433",
+						fieldname.Database: "VMart",
+					},
+				},
+			},
+		},
+	})
+}

plugins/vertica/plugin.go

@@ -0,0 +1,22 @@
+package vertica
+
+import (
+	"github.com/1Password/shell-plugins/sdk"
+	"github.com/1Password/shell-plugins/sdk/schema"
+)
+
+func New() schema.Plugin {
+	return schema.Plugin{
+		Name: "vertica",
+		Platform: schema.PlatformInfo{
+			Name:     "Vertica",
+			Homepage: sdk.URL("https://www.vertica.com/"),
+		},
+		Credentials: []schema.CredentialType{
+			DatabaseCredentials(),
+		},
+		Executables: []schema.Executable{
+			VerticaCLI(),
+		},
+	}
+}

plugins/vertica/vsql.go

@@ -0,0 +1,24 @@
+package vertica
+
+import (
+	"github.com/1Password/shell-plugins/sdk"
+	"github.com/1Password/shell-plugins/sdk/needsauth"
+	"github.com/1Password/shell-plugins/sdk/schema"
+	"github.com/1Password/shell-plugins/sdk/schema/credname"
+)
+
+func VerticaCLI() schema.Executable {
+	return schema.Executable{
+		Name:    "Vertica CLI",
+		Runs:    []string{"vsql"},
+		DocsURL: sdk.URL("https://www.vertica.com/docs/9.2.x/HTML/Content/Authoring/ConnectingToVertica/vsql/UsingVsql.htm"),
+		NeedsAuth: needsauth.IfAll(
+			needsauth.NotForHelpOrVersion(),
+		),
+		Uses: []schema.CredentialUsage{
+			{
+				Name: credname.DatabaseCredentials,
+			},
+		},
+	}
+}