コード例 #1
0
ファイル: main.go プロジェクト: dgoodlad/aws-vault
func main() {
	var (
		prompts          = prompt.Available()
		debug            = kingpin.Flag("debug", "Show debugging output").Bool()
		promptDriver     = kingpin.Flag("prompt", fmt.Sprintf("Prompt driver to use  %v", prompts)).Default("terminal").OverrideDefaultFromEnvar("AWS_VAULT_PROMPT").Enum(prompts...)
		add              = kingpin.Command("add", "Adds credentials, prompts if none provided")
		addProfile       = add.Arg("profile", "Name of the profile").Required().String()
		addFromEnv       = add.Flag("env", "Read the credentials from the environment").Bool()
		ls               = kingpin.Command("ls", "List profiles")
		exec             = kingpin.Command("exec", "Executes a command with AWS credentials in the environment")
		execNoSession    = exec.Flag("no-session", "Use root credentials, no session created").Short('n').Bool()
		execSessDuration = exec.Flag("session-ttl", "Expiration time for aws session").Default("4h").OverrideDefaultFromEnvar("AWS_SESSION_TTL").Short('t').Duration()
		execRoleDuration = exec.Flag("assume-role-ttl", "Expiration time for aws assumed role").Default("15m").Duration()
		execMfaToken     = exec.Flag("mfa-token", "The mfa token to use").Short('m').String()
		execServer       = exec.Flag("server", "Run the server in the background for credentials").Short('s').Bool()
		execProfile      = exec.Arg("profile", "Name of the profile").Required().String()
		execCmd          = exec.Arg("cmd", "Command to execute").Default(os.Getenv("SHELL")).String()
		execCmdArgs      = exec.Arg("args", "Command arguments").Strings()
		rotate           = kingpin.Command("rotate", "Rotates credentials")
		rotateProfile    = rotate.Arg("profile", "Name of the profile").Required().String()
		rm               = kingpin.Command("rm", "Removes credentials, including sessions")
		rmProfile        = rm.Arg("profile", "Name of the profile").Required().String()
		rmSessionsOnly   = rm.Flag("sessions-only", "Only remove sessions, leave credentials intact").Short('s').Bool()
		login            = kingpin.Command("login", "Generate a login link for the AWS Console")
		loginProfile     = login.Arg("profile", "Name of the profile").Required().String()
		loginMfaToken    = login.Flag("mfa-token", "The mfa token to use").Short('t').String()
		server           = kingpin.Command("server", "Run an ec2 instance role server locally")
	)

	kingpin.Version(Version)
	kingpin.CommandLine.Help =
		`A vault for securely storing and accessing AWS credentials in development environments.`

	ui := Ui{
		Logger: log.New(os.Stdout, "", 0),
		Error:  log.New(os.Stderr, "", 0),
		Debug:  log.New(ioutil.Discard, "", 0),
		Exit:   os.Exit,
	}

	keyring, err := keyring.Open("aws-vault")
	if err != nil {
		ui.Error.Fatal(err)
	}

	cmd := kingpin.Parse()

	if *debug {
		ui.Debug = log.New(os.Stderr, "DEBUG ", log.LstdFlags)
		log.SetFlags(0)
		log.SetOutput(&logWriter{ui.Debug})
	} else {
		log.SetOutput(ioutil.Discard)
	}

	switch cmd {
	case ls.FullCommand():
		LsCommand(ui, LsCommandInput{
			Keyring: keyring,
		})

	case rm.FullCommand():
		RemoveCommand(ui, RemoveCommandInput{
			Profile:      *rmProfile,
			Keyring:      keyring,
			SessionsOnly: *rmSessionsOnly,
		})

	case add.FullCommand():
		AddCommand(ui, AddCommandInput{
			Profile: *addProfile,
			Keyring: keyring,
			FromEnv: *addFromEnv,
		})

	case exec.FullCommand():
		signals := make(chan os.Signal)
		signal.Notify(signals, os.Interrupt, os.Kill)

		ExecCommand(ui, ExecCommandInput{
			Profile:      *execProfile,
			Command:      *execCmd,
			Args:         *execCmdArgs,
			Keyring:      keyring,
			Duration:     *execSessDuration,
			RoleDuration: *execRoleDuration,
			Signals:      signals,
			MfaToken:     *execMfaToken,
			MfaPrompt:    prompt.Method(*promptDriver),
			StartServer:  *execServer,
			NoSession:    *execNoSession,
		})

	case login.FullCommand():
		LoginCommand(ui, LoginCommandInput{
			Profile:   *loginProfile,
			Keyring:   keyring,
			MfaToken:  *loginMfaToken,
			MfaPrompt: prompt.Method(*promptDriver),
		})

	case server.FullCommand():
		ServerCommand(ui, ServerCommandInput{})

	case rotate.FullCommand():
		RotateCommand(ui, RotateCommandInput{
			Profile:   *rotateProfile,
			Keyring:   keyring,
			MfaToken:  *loginMfaToken,
			MfaPrompt: prompt.Method(*promptDriver),
		})
	}
}
コード例 #2
0
ファイル: main.go プロジェクト: simpsora/aws-vault
func main() {
	var (
		debug            = kingpin.Flag("debug", "Show debugging output").Bool()
		add              = kingpin.Command("add", "Adds credentials, prompts if none provided")
		addProfile       = add.Arg("profile", "Name of the profile").Required().String()
		addFromEnv       = add.Flag("env", "Read the credentials from the environment").Bool()
		ls               = kingpin.Command("ls", "List profiles")
		exec             = kingpin.Command("exec", "Executes a command with AWS credentials in the environment")
		execProfile      = exec.Arg("profile", "Name of the profile").Required().String()
		execSessDuration = exec.Flag("session-ttl", "Expiration time for aws session").Default("4h").OverrideDefaultFromEnvar("AWS_SESSION_TTL").Short('t').Duration()
		execWriteEnv     = exec.Flag("write-env", "Write AWS env vars").Short('e').Bool()
		execCmd          = exec.Arg("cmd", "Command to execute").Default(os.Getenv("SHELL")).String()
		execCmdArgs      = exec.Arg("args", "Command arguments").Strings()
		rm               = kingpin.Command("rm", "Removes credentials")
		rmProfile        = rm.Arg("profile", "Name of the profile").Required().String()
		login            = kingpin.Command("login", "Generate a login link for the AWS Console")
		loginProfile     = login.Arg("profile", "Name of the profile").Required().String()
	)

	kingpin.Version(Version)
	kingpin.CommandLine.Help =
		`A vault for securely storing and accessing AWS credentials in development environments.`

	ui := Ui{
		Logger: log.New(os.Stdout, "", 0),
		Error:  log.New(os.Stderr, "", 0),
		Debug:  log.New(ioutil.Discard, "", 0),
		Exit:   os.Exit,
	}

	keyring, err := keyring.Open("aws-vault")
	if err != nil {
		ui.Error.Fatal(err)
	}

	cmd := kingpin.Parse()

	if *debug {
		ui.Debug = log.New(os.Stderr, "DEBUG ", log.LstdFlags)
		log.SetFlags(0)
		log.SetOutput(&logWriter{ui.Debug})
	} else {
		log.SetOutput(ioutil.Discard)
	}

	switch cmd {
	case ls.FullCommand():
		LsCommand(ui, LsCommandInput{
			Keyring: keyring,
		})

	case rm.FullCommand():
		RemoveCommand(ui, RemoveCommandInput{
			Profile: *rmProfile,
			Keyring: keyring,
		})

	case add.FullCommand():
		AddCommand(ui, AddCommandInput{
			Profile: *addProfile,
			Keyring: keyring,
			FromEnv: *addFromEnv,
		})

	case exec.FullCommand():
		signals := make(chan os.Signal)
		signal.Notify(signals, os.Interrupt, os.Kill)

		ExecCommand(ui, ExecCommandInput{
			Profile:  *execProfile,
			Command:  *execCmd,
			Args:     *execCmdArgs,
			Keyring:  keyring,
			Duration: *execSessDuration,
			WriteEnv: *execWriteEnv,
			Signals:  signals,
		})

	case login.FullCommand():
		LoginCommand(ui, LoginCommandInput{
			Profile: *loginProfile,
			Keyring: keyring,
		})
	}
}