diff --git a/plugins/gemini/api_key.go b/plugins/gemini/api_key.go new file mode 100644 index 00000000..abec5da8 --- /dev/null +++ b/plugins/gemini/api_key.go @@ -0,0 +1,40 @@ +package gemini + +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 APIKey() schema.CredentialType { + return schema.CredentialType{ + Name: credname.APIKey, + DocsURL: sdk.URL("https://github.com/google-gemini/gemini-cli/blob/main/docs/cli/authentication.md"), + ManagementURL: sdk.URL("https://aistudio.google.com/app/apikey"), + Fields: []schema.CredentialField{ + { + Name: fieldname.APIKey, + MarkdownDescription: "API Key used to authenticate to Google Gemini.", + Secret: true, + Composition: &schema.ValueComposition{ + Length: 100, + Charset: schema.Charset{ + Uppercase: true, + }, + }, + }, + }, + DefaultProvisioner: provision.EnvVars(defaultEnvVarMapping), + Importer: importer.TryAll( + importer.TryEnvVarPair(defaultEnvVarMapping), + ), + } +} + +var defaultEnvVarMapping = map[string]sdk.FieldName{ + "GEMINI_API_KEY": fieldname.APIKey, + "GOOGLE_API_KEY": fieldname.APIKey, +} diff --git a/plugins/gemini/api_key_test.go b/plugins/gemini/api_key_test.go new file mode 100644 index 00000000..2f4d17e4 --- /dev/null +++ b/plugins/gemini/api_key_test.go @@ -0,0 +1,55 @@ +package gemini + +import ( + "testing" + + "github.com/1Password/shell-plugins/sdk" + "github.com/1Password/shell-plugins/sdk/plugintest" + "github.com/1Password/shell-plugins/sdk/schema/fieldname" +) + +func TestAPIKeyProvisioner(t *testing.T) { + plugintest.TestProvisioner(t, APIKey().DefaultProvisioner, map[string]plugintest.ProvisionCase{ + "default": { + ItemFields: map[sdk.FieldName]string{ // TODO: Check if this is correct + fieldname.APIKey: "UIYISUJNXTDEEXAMPLE", + }, + ExpectedOutput: sdk.ProvisionOutput{ + Environment: map[string]string{ + "GEMINI_API_KEY": "UIYISUJNXTDEEXAMPLE", + }, + }, + }, + }) +} + +func TestAPIKeyImporter(t *testing.T) { + plugintest.TestImporter(t, APIKey().Importer, map[string]plugintest.ImportCase{ + "environment": { + Environment: map[string]string{ // TODO: Check if this is correct + "GEMINI_API_KEY": "UIYISUJNXTDEEXAMPLE", + }, + ExpectedCandidates: []sdk.ImportCandidate{ + { + Fields: map[sdk.FieldName]string{ + fieldname.APIKey: "UIYISUJNXTDEEXAMPLE", + }, + }, + }, + }, + // TODO: If you implemented a config file importer, add a test file example in gemini/test-fixtures + // and fill the necessary details in the test template below. + "config file": { + Files: map[string]string{ + // "~/path/to/config.yml": plugintest.LoadFixture(t, "config.yml"), + }, + ExpectedCandidates: []sdk.ImportCandidate{ + // { + // Fields: map[sdk.FieldName]string{ + // fieldname.Token: "UIYISUJNXTDEEXAMPLE", + // }, + // }, + }, + }, + }) +} diff --git a/plugins/gemini/gemini.go b/plugins/gemini/gemini.go new file mode 100644 index 00000000..264ccdb3 --- /dev/null +++ b/plugins/gemini/gemini.go @@ -0,0 +1,25 @@ +package gemini + +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 GoogleGeminiCLI() schema.Executable { + return schema.Executable{ + Name: "Google Gemini CLI", + Runs: []string{"gemini"}, + DocsURL: sdk.URL("https://github.com/google-gemini/gemini-cli/blob/main/docs/cli/index.md"), + NeedsAuth: needsauth.IfAll( + needsauth.NotForHelpOrVersion(), + needsauth.NotWithoutArgs(), + ), + Uses: []schema.CredentialUsage{ + { + Name: credname.APIKey, + }, + }, + } +} diff --git a/plugins/gemini/plugin.go b/plugins/gemini/plugin.go new file mode 100644 index 00000000..e3a78265 --- /dev/null +++ b/plugins/gemini/plugin.go @@ -0,0 +1,22 @@ +package gemini + +import ( + "github.com/1Password/shell-plugins/sdk" + "github.com/1Password/shell-plugins/sdk/schema" +) + +func New() schema.Plugin { + return schema.Plugin{ + Name: "gemini", + Platform: schema.PlatformInfo{ + Name: "Google Gemini", + Homepage: sdk.URL("https://gemini.google.com"), + }, + Credentials: []schema.CredentialType{ + APIKey(), + }, + Executables: []schema.Executable{ + GoogleGeminiCLI(), + }, + } +}