2015-03-24 23:57:23 -04:00
package client
import (
"bufio"
"fmt"
"io"
"os"
2015-11-05 17:19:48 -05:00
"runtime"
2015-03-24 23:57:23 -04:00
"strings"
2015-05-05 00:18:28 -04:00
Cli "github.com/docker/docker/cli"
2015-03-24 23:57:23 -04:00
flag "github.com/docker/docker/pkg/mflag"
"github.com/docker/docker/pkg/term"
2016-01-04 19:05:26 -05:00
"github.com/docker/engine-api/client"
"github.com/docker/engine-api/types"
2015-03-24 23:57:23 -04:00
)
2015-03-25 13:34:41 -04:00
// CmdLogin logs in or registers a user to a Docker registry service.
//
// If no server is specified, the user will be logged into or registered to the registry's index server.
//
// Usage: docker login SERVER
2015-03-24 23:57:23 -04:00
func ( cli * DockerCli ) CmdLogin ( args ... string ) error {
2016-02-01 14:13:11 -05:00
cmd := Cli . Subcmd ( "login" , [ ] string { "[SERVER]" } , Cli . DockerCommands [ "login" ] . Description + ".\nIf no server is specified, the default is defined by the daemon." , true )
2015-03-24 23:57:23 -04:00
cmd . Require ( flag . Max , 1 )
2016-01-27 18:26:48 -05:00
flUser := cmd . String ( [ ] string { "u" , "-username" } , "" , "Username" )
flPassword := cmd . String ( [ ] string { "p" , "-password" } , "" , "Password" )
flEmail := cmd . String ( [ ] string { "e" , "-email" } , "" , "Email" )
2015-03-24 23:57:23 -04:00
2015-03-28 21:22:46 -04:00
cmd . ParseFlags ( args , true )
2015-03-24 23:57:23 -04:00
2015-11-05 17:19:48 -05:00
// On Windows, force the use of the regular OS stdin stream. Fixes #14336/#14210
if runtime . GOOS == "windows" {
cli . in = os . Stdin
}
2016-02-03 13:30:17 -05:00
var serverAddress string
2015-03-24 23:57:23 -04:00
if len ( cmd . Args ( ) ) > 0 {
serverAddress = cmd . Arg ( 0 )
2016-02-03 13:30:17 -05:00
} else {
serverAddress = cli . electAuthServer ( )
2015-03-24 23:57:23 -04:00
}
2016-01-27 18:26:48 -05:00
authConfig , err := cli . configureAuth ( * flUser , * flPassword , * flEmail , serverAddress )
if err != nil {
return err
2015-03-24 23:57:23 -04:00
}
2016-01-27 18:26:48 -05:00
response , err := cli . client . RegistryLogin ( authConfig )
if err != nil {
if client . IsErrUnauthorized ( err ) {
delete ( cli . configFile . AuthConfigs , serverAddress )
if err2 := cli . configFile . Save ( ) ; err2 != nil {
fmt . Fprintf ( cli . out , "WARNING: could not save config file: %v\n" , err2 )
}
2015-03-24 23:57:23 -04:00
}
2016-01-27 18:26:48 -05:00
return err
2015-03-24 23:57:23 -04:00
}
2016-01-27 18:26:48 -05:00
if err := cli . configFile . Save ( ) ; err != nil {
return fmt . Errorf ( "Error saving config file: %v" , err )
}
fmt . Fprintf ( cli . out , "WARNING: login credentials saved in %s\n" , cli . configFile . Filename ( ) )
if response . Status != "" {
fmt . Fprintf ( cli . out , "%s\n" , response . Status )
}
return nil
}
func ( cli * DockerCli ) promptWithDefault ( prompt string , configDefault string ) {
if configDefault == "" {
fmt . Fprintf ( cli . out , "%s: " , prompt )
} else {
fmt . Fprintf ( cli . out , "%s (%s): " , prompt , configDefault )
}
}
func ( cli * DockerCli ) configureAuth ( flUser , flPassword , flEmail , serverAddress string ) ( types . AuthConfig , error ) {
2015-04-01 18:39:37 -04:00
authconfig , ok := cli . configFile . AuthConfigs [ serverAddress ]
2015-03-24 23:57:23 -04:00
if ! ok {
2015-12-11 23:11:42 -05:00
authconfig = types . AuthConfig { }
2015-03-24 23:57:23 -04:00
}
2016-01-28 09:17:28 -05:00
authconfig . Username = strings . TrimSpace ( authconfig . Username )
2015-03-24 23:57:23 -04:00
2016-01-28 09:17:28 -05:00
if flUser = strings . TrimSpace ( flUser ) ; flUser == "" {
2016-01-27 18:26:48 -05:00
cli . promptWithDefault ( "Username" , authconfig . Username )
flUser = readInput ( cli . in , cli . out )
flUser = strings . TrimSpace ( flUser )
if flUser == "" {
flUser = authconfig . Username
2015-03-24 23:57:23 -04:00
}
}
2016-01-28 09:17:28 -05:00
if flUser == "" {
return authconfig , fmt . Errorf ( "Error: Non-null Username Required" )
}
2016-01-27 18:26:48 -05:00
if flPassword == "" {
oldState , err := term . SaveState ( cli . inFd )
if err != nil {
return authconfig , err
}
fmt . Fprintf ( cli . out , "Password: " )
term . DisableEcho ( cli . inFd , oldState )
2015-03-24 23:57:23 -04:00
2016-01-27 18:26:48 -05:00
flPassword = readInput ( cli . in , cli . out )
fmt . Fprint ( cli . out , "\n" )
term . RestoreTerminal ( cli . inFd , oldState )
if flPassword == "" {
2016-01-28 09:17:28 -05:00
return authconfig , fmt . Errorf ( "Error: Password Required" )
2015-03-24 23:57:23 -04:00
}
2016-01-27 18:26:48 -05:00
}
2015-03-24 23:57:23 -04:00
2016-01-27 18:26:48 -05:00
// Assume that a different username means they may not want to use
// the email from the config file, so prompt it
if flUser != authconfig . Username {
if flEmail == "" {
cli . promptWithDefault ( "Email" , authconfig . Email )
flEmail = readInput ( cli . in , cli . out )
if flEmail == "" {
flEmail = authconfig . Email
2015-03-24 23:57:23 -04:00
}
}
} else {
// However, if they don't override the username use the
2016-01-27 18:26:48 -05:00
// email from the cmd line if specified. IOW, allow
2015-03-24 23:57:23 -04:00
// then to change/override them. And if not specified, just
// use what's in the config file
2016-01-27 18:26:48 -05:00
if flEmail == "" {
flEmail = authconfig . Email
2015-03-24 23:57:23 -04:00
}
}
2016-01-27 18:26:48 -05:00
authconfig . Username = flUser
authconfig . Password = flPassword
authconfig . Email = flEmail
2015-03-24 23:57:23 -04:00
authconfig . ServerAddress = serverAddress
2015-04-01 18:39:37 -04:00
cli . configFile . AuthConfigs [ serverAddress ] = authconfig
2016-01-27 18:26:48 -05:00
return authconfig , nil
}
2015-03-24 23:57:23 -04:00
2016-01-27 18:26:48 -05:00
func readInput ( in io . Reader , out io . Writer ) string {
reader := bufio . NewReader ( in )
line , _ , err := reader . ReadLine ( )
2015-03-24 23:57:23 -04:00
if err != nil {
2016-01-27 18:26:48 -05:00
fmt . Fprintln ( out , err . Error ( ) )
os . Exit ( 1 )
2015-04-01 18:39:37 -04:00
}
2016-01-27 18:26:48 -05:00
return string ( line )
2015-03-24 23:57:23 -04:00
}