lots of cleanups to token handling. protobuf fixes

Signed-off-by: Jeff Carr <jcarr@wit.com>
This commit is contained in:
Jeff Carr 2019-05-26 08:04:12 +00:00
parent b16b4c84bb
commit 2d3a0a900a
2 changed files with 29 additions and 22 deletions

View File

@ -43,9 +43,16 @@ func fetchJSON(GETorPOST string, token string, a string) []byte {
} }
func getToken(ge *pb.Event) string { func getToken(ge *pb.Event) string {
if ge == nil {return ""} if ge == nil {
log.Println("getToken() has ge = nil")
return ""
}
if ge.Account == nil {
log.Println("getToken() has ge.Account = nil")
return ""
}
url := ge.Account.URL + "/auth/login" + "?email=" + ge.Email + "&password=" + ge.Password url := ge.Account.URL + "/auth/login" + "?email=" + ge.Account.Email + "&password=" + ge.Account.Password
log.Println("URL =", url) log.Println("URL =", url)
var jsonStr = []byte(`{}`) var jsonStr = []byte(`{}`)

View File

@ -26,7 +26,7 @@ type network struct {
func checkLogin(ge *pb.Event) string { func checkLogin(ge *pb.Event) string {
if (origLoginCheck(ge) == true) { if (origLoginCheck(ge) == true) {
log.Println("origLoginCheck() LOGIN WORKED") log.Println("origLoginCheck() LOGIN WORKED")
return ge.Token return ge.Account.Token
} }
log.Println("origLoginCheck() LOGIN FAILED. ATTEMPT getToken(ge)") log.Println("origLoginCheck() LOGIN FAILED. ATTEMPT getToken(ge)")
log.Println("origLoginCheck() LOGIN FAILED. ATTEMPT getToken(ge)") log.Println("origLoginCheck() LOGIN FAILED. ATTEMPT getToken(ge)")
@ -226,13 +226,13 @@ func main() {
func processGetEvent(ge *pb.Event) *pb.Event { func processGetEvent(ge *pb.Event) *pb.Event {
log.Println("processGetEvent() START") log.Println("processGetEvent() START")
log.Println("processGetEvent() ge.Token =", ge.Token) log.Println("processGetEvent() ge.Account.Token =", ge.Account.Token)
log.Println("processGetEvent() ge.Username =", ge.Username) log.Println("processGetEvent() ge.Account.Username =", ge.Account.Username)
log.Println("processGetEvent() ge.Email =", ge.Email) log.Println("processGetEvent() ge.Account.Email =", ge.Account.Email)
e := pb.Event{} e := pb.Event{}
networks := fetchNetworks(ge.Token) networks := fetchNetworks(ge.Account.Token)
for i, network := range(networks) { for i, network := range(networks) {
if i >= MAX_NETS { if i >= MAX_NETS {
break break
@ -244,7 +244,7 @@ func processGetEvent(ge *pb.Event) *pb.Event {
e.Networks = append(e.Networks, &enetwork) e.Networks = append(e.Networks, &enetwork)
vms := fetchVMs(ge.Token, network.Subdomain) vms := fetchVMs(ge.Account.Token, network.Subdomain)
for j, vm := range(vms) { for j, vm := range(vms) {
if j >= MAX_VMS { if j >= MAX_VMS {
break break
@ -266,7 +266,7 @@ func processGetEvent(ge *pb.Event) *pb.Event {
func processLoginEvent(ge *pb.Event) *pb.Event { func processLoginEvent(ge *pb.Event) *pb.Event {
log.Println("processLoginEvent() START") log.Println("processLoginEvent() START")
log.Println("processLoginEvent() ge.Token =", ge.Token) log.Println("processLoginEvent() ge.Account.Token =", ge.Account.Token)
log.Println("processLoginEvent() NEED TO LOGIN HERE OR CHECK TOKEN WORKS") log.Println("processLoginEvent() NEED TO LOGIN HERE OR CHECK TOKEN WORKS")
@ -276,29 +276,29 @@ func processLoginEvent(ge *pb.Event) *pb.Event {
e.Comment = "LOGIN FAILED" e.Comment = "LOGIN FAILED"
return e return e
} }
log.Println("processLoginEvent() LOGIN WORKED") log.Println("processLoginEvent() LOGIN WORKED")
log.Println("processLoginEvent() LOGIN WORKED") log.Println("processLoginEvent() LOGIN WORKED")
log.Println("processLoginEvent() LOGIN WORKED") log.Println("processLoginEvent() LOGIN WORKED")
e := pb.MakeOkResponse() e := pb.MakeOkResponse()
e.Token = token e.Account = ge.Account
if (e.Account.Token != token) {
log.Println("processLoginEvent() GENERATED NEW TOKEN old =", e.Account.Token)
log.Println("processLoginEvent() GENERATED NEW TOKEN new =", token)
}
e.Account.Token = token
e.Comment = "LOGIN WORKED" e.Comment = "LOGIN WORKED"
return e return e
} }
func processAddEvent(ge *pb.Event) *pb.Event { func processAddEvent(ge *pb.Event) *pb.Event {
log.Println("processAddEvent() START") log.Println("processAddEvent() START")
log.Println("processAddLoginEvent() ge.Token =", ge.Token) log.Println("processAddLoginEvent() ge.Account.Token =", ge.Account.Token)
log.Println("processAddEvent() END") log.Println("processAddEvent() END")
return nil return nil
} }
/* /*
var login string
login = "auth/login?email=jcarr@wit.com&password=yowzayowza"
raw = fetchPOST("badtoken", login)
log.Println("raw =", string(raw))
*/
func jsonAddEvent(ge *pb.Event) bool { func jsonAddEvent(ge *pb.Event) bool {
var url string var url string
var httpType string var httpType string
@ -306,7 +306,6 @@ func jsonAddEvent(ge *pb.Event) bool {
if ge.Type == pb.Event_LOGIN { if ge.Type == pb.Event_LOGIN {
url = API + "me" url = API + "me"
} else if ge.Type == pb.Event_ADD { } else if ge.Type == pb.Event_ADD {
// stackapi:4000/vms/jcarr.com?count=1&cpu=2&ram=512&disk=25
url = API + "vms/jcarr.com?count=1&cpu=2&ram=512&disk=25" url = API + "vms/jcarr.com?count=1&cpu=2&ram=512&disk=25"
httpType = "POST" httpType = "POST"
} else if ge.Type == pb.Event_MIGRATE { } else if ge.Type == pb.Event_MIGRATE {
@ -352,6 +351,7 @@ func jsonAddEvent(ge *pb.Event) bool {
} }
return false return false
} }
*/
// This is the 'orig' code // This is the 'orig' code
// TODO: phase this out for more common code // TODO: phase this out for more common code
@ -367,7 +367,7 @@ func origLoginCheck(ge *pb.Event) bool {
log.Println("can't make new req") log.Println("can't make new req")
return false return false
} }
req.Header.Set("X-Wit-Auth", ge.Token) req.Header.Set("X-Wit-Auth", ge.Account.Token)
client := &http.Client{} client := &http.Client{}
resp, err := client.Do(req) resp, err := client.Do(req)
if err != nil { if err != nil {
@ -379,7 +379,7 @@ func origLoginCheck(ge *pb.Event) bool {
log.Println("can't read resp") log.Println("can't read resp")
return false return false
} }
log.Printf("RESPONSE =", body) log.Printf("RESPONSE =", string(body))
if body == nil { if body == nil {
log.Println("RESPONSE is nil") log.Println("RESPONSE is nil")
return false return false
@ -389,8 +389,8 @@ func origLoginCheck(ge *pb.Event) bool {
log.Println("body len = ", len) log.Println("body len = ", len)
email := gjson.Get(string(body), "email") email := gjson.Get(string(body), "email")
log.Println("email = ", email.String()) log.Println("email = ", email.String())
log.Println("ge.Username = ", ge.Username) log.Println("ge.Account.Username = ", ge.Account.Username)
if (email.String() == ge.Username) { if (email.String() == ge.Account.Username) {
log.Println("\tTHIS TOTALLY WORKED MAN") log.Println("\tTHIS TOTALLY WORKED MAN")
return true return true
} }