79acea5976
Some checks failed
goreleaser / goreleaser (push) Has been cancelled
- Moved login and service management logic to `internal/cmd` package (`login.go` and `run.go`). - Introduced `DoLogin` and `StartService` functions for modularity. - Enhanced error handling by using structured `ErrorMessage` in `Client`. - Improved token file saving process and added project-specific token identification. - Updated API handlers to handle more detailed error responses, including status codes.
63 lines
2.0 KiB
Go
63 lines
2.0 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"github.com/luispater/CLIProxyAPI/internal/auth"
|
|
"github.com/luispater/CLIProxyAPI/internal/client"
|
|
"github.com/luispater/CLIProxyAPI/internal/config"
|
|
log "github.com/sirupsen/logrus"
|
|
"os"
|
|
)
|
|
|
|
func DoLogin(cfg *config.Config, projectID string) {
|
|
var err error
|
|
var ts auth.TokenStorage
|
|
if projectID != "" {
|
|
ts.ProjectID = projectID
|
|
}
|
|
|
|
// 2. Initialize authenticated HTTP Client
|
|
clientCtx := context.Background()
|
|
|
|
log.Info("Initializing authentication...")
|
|
httpClient, errGetClient := auth.GetAuthenticatedClient(clientCtx, &ts, cfg)
|
|
if errGetClient != nil {
|
|
log.Fatalf("failed to get authenticated client: %v", errGetClient)
|
|
return
|
|
}
|
|
log.Info("Authentication successful.")
|
|
|
|
// 3. Initialize CLI Client
|
|
cliClient := client.NewClient(httpClient)
|
|
projectID, err = cliClient.SetupUser(clientCtx, ts.Email, projectID, ts.Auto)
|
|
if err != nil {
|
|
if err.Error() == "failed to start user onboarding, need define a project id" {
|
|
log.Error("failed to start user onboarding")
|
|
project, errGetProjectList := cliClient.GetProjectList(clientCtx)
|
|
if errGetProjectList != nil {
|
|
log.Fatalf("failed to complete user setup: %v", err)
|
|
} else {
|
|
log.Infof("Your account %s needs specify a project id.", ts.Email)
|
|
log.Info("========================================================================")
|
|
for i := 0; i < len(project.Projects); i++ {
|
|
log.Infof("Project ID: %s", project.Projects[i].ProjectID)
|
|
log.Infof("Project Name: %s", project.Projects[i].Name)
|
|
log.Info("========================================================================")
|
|
}
|
|
log.Infof("Please run this command to login again:\n\n%s --login --project_id <project_id>\n", os.Args[0])
|
|
}
|
|
} else {
|
|
// Log as a warning because in some cases, the CLI might still be usable
|
|
// or the user might want to retry setup later.
|
|
log.Fatalf("failed to complete user setup: %v", err)
|
|
}
|
|
} else {
|
|
auto := ts.ProjectID == ""
|
|
ts.ProjectID = projectID
|
|
err = auth.SaveTokenToFile(&ts, cfg, auto)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
}
|