public_git
/
unpoller_unpoller
mirror of https://github.com/unpoller/unpoller.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

A start.

This commit is contained in:
DN2 2019-01-10 00:53:58 -08:00
parent b78aaca852
commit ace03c02f0
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
integrations/promunifi/unidev/unidev.go
View File

@ -3,6 +3,7 @@ package unidev
import ( import (
"bytes" "bytes"
"crypto/tls" "crypto/tls"
"log"
"net/http" "net/http"
"net/http/cookiejar" "net/http/cookiejar"
@ -29,6 +30,9 @@ type AuthedReq struct {
baseURL string baseURL string
} }
// StringInt is used to unmarshal quoted integers in JSON responses.
type StringInt int
// AuthController creates a http.Client with authenticated cookies. // AuthController creates a http.Client with authenticated cookies.
// Used to make additional, authenticated requests to the APIs. // Used to make additional, authenticated requests to the APIs.
func AuthController(user, pass, url string) (*AuthedReq, error) { func AuthController(user, pass, url string) (*AuthedReq, error) {
@ -41,11 +45,20 @@ func AuthController(user, pass, url string) (*AuthedReq, error) {
Transport: &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: true}}, Transport: &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: true}},
Jar: jar, Jar: jar,
}, url} }, url}
if req, err := authReq.UniReq(LoginPath, json); err != nil { req, err := authReq.UniReq(LoginPath, json)
if err != nil {
return nil, errors.Wrap(err, "UniReq(LoginPath, json)") return nil, errors.Wrap(err, "UniReq(LoginPath, json)")
} else if resp, err := authReq.Do(req); err != nil { }
resp, err := authReq.Do(req)
if err != nil {
return nil, errors.Wrap(err, "authReq.Do(req)") return nil, errors.Wrap(err, "authReq.Do(req)")
} else if resp.StatusCode != http.StatusOK { }
defer func() {
if err := resp.Body.Close(); err != nil {
log.Println("resp.Body.Close():", err) // Not fatal. Just log it.
}
}()
if resp.StatusCode != http.StatusOK {
return nil, errors.Errorf("authentication failed (%v): %v (status: %v/%v)", return nil, errors.Errorf("authentication failed (%v): %v (status: %v/%v)",
user, url+LoginPath, resp.StatusCode, resp.Status) user, url+LoginPath, resp.StatusCode, resp.Status)
} }