api_key.go

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,
}