Skip to content

Commit

Permalink
Merge branch 'master' of https://github.com/unarxiv/CVPM into bug/run…
Browse files Browse the repository at this point in the history
…ning-port#5
  • Loading branch information
xzyaoi committed Oct 18, 2018
2 parents 7cfac71 + b07cc32 commit 0e793db
Show file tree
Hide file tree
Showing 4 changed files with 32 additions and 24 deletions.
3 changes: 1 addition & 2 deletions .gitignore
Expand Up @@ -6,5 +6,4 @@ build
dist
.env
.idea
.DS_Store
*.go
.DS_Store
4 changes: 4 additions & 0 deletions cli/handler.go
Expand Up @@ -149,3 +149,7 @@ func ConfigHandler(c *cli.Context) {
nextConfig.Local.Pip = newPipLocation
writeConfig(nextConfig)
}

func InitHandler(c *cli.Context) {

}
6 changes: 0 additions & 6 deletions cli/main.go
Expand Up @@ -8,14 +8,8 @@ import (
)

func main() {
initRaven()
validateConfig()
// sessionToken := getCache("session-token")
var currentUser User
// if sessionToken != "" {
// currentUser = User{"", "", sessionToken}
// currentUser.become()
// }
cvpm := cli.NewApp()
cvpm.Name = "CVPM"
cvpm.Usage = "Computer Vision Package Manager"
Expand Down
43 changes: 27 additions & 16 deletions cli/user.go
Expand Up @@ -3,7 +3,13 @@ package main
import (
"encoding/json"
"fmt"
"time"
"github.com/fatih/color"
"github.com/alexrudd/cognito-srp"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/aws/external"
"github.com/aws/aws-sdk-go-v2/aws/endpoints"
cip "github.com/aws/aws-sdk-go-v2/service/cognitoidentityprovider"
"github.com/levigross/grequests"
"log"
)
Expand All @@ -15,24 +21,29 @@ type User struct {
}

func (u *User) login() User {
var respUser User
loginURL := apiURL + "user/login"
loginRequestHeader := &grequests.RequestOptions{
JSON: map[string]string{"username": u.Username, "password": u.Password},
IsAjax: true,
}
resp, err := grequests.Post(loginURL, loginRequestHeader)
if err != nil {
log.Fatal(err)
}
if resp.Ok != true {
fmt.Println("\nLogin Failed!")
fmt.Println(u.Username)
csrp, _:= cognitosrp.NewCognitoSRP(u.Username, u.Password, "us-east-1_IYJ3FvCKZ", "1jinmsd412vcs8pkhqg5u0gjd2", nil)
cfg, _ := external.LoadDefaultAWSConfig()
cfg.Region = endpoints.UsEast1RegionID
cfg.Credentials = aws.AnonymousCredentials
svc := cip.New(cfg)
req := svc.InitiateAuthRequest(&cip.InitiateAuthInput{
AuthFlow: cip.AuthFlowTypeUserSrpAuth,
ClientId: aws.String(csrp.GetClientId()),
AuthParameters: csrp.GetAuthParams(),
})
resp, _:= req.Send()
fmt.Println("123456")
fmt.Println(resp.ChallengeName)
if resp.ChallengeName == cip.ChallengeNameTypePasswordVerifier {
challengeInput, _:= csrp.PasswordVerifierChallenge(resp.ChallengeParameters, time.Now())
chal := svc.RespondToAuthChallengeRequest(challengeInput)
resp, _:= chal.Send()
fmt.Println(resp.AuthenticationResult)
} else {
_ = json.Unmarshal(resp.Bytes(), &respUser)
setCache("session-token", respUser.SessionToken)
color.Green("\nLogin Successfully")

}
return respUser
return *u
}

func (u *User) become() User {
Expand Down

0 comments on commit 0e793db

Please sign in to comment.