master
Bel LaPointe 2022-01-14 08:40:37 -05:00
parent 003388c847
commit c1cdc6dc0c
2 changed files with 15 additions and 7 deletions

View File

@ -188,8 +188,18 @@ func (ntg NTGVision) Search(states []config.State) ([]Job, error) {
return jobs, err return jobs, err
} }
func getNTGToken() string {
return config.Get().Brokers.NTG.Token
}
func setNTGToken(token string) {
conf := config.Get()
conf.Brokers.NTG.Token = token
config.Set(*conf)
}
func (ntg NTGVision) search(states []config.State) (io.ReadCloser, error) { func (ntg NTGVision) search(states []config.State) (io.ReadCloser, error) {
if config.Get().Brokers.NTG.Token == "" { if getNTGToken() == "" {
if err := ntg.refreshAuth(); err != nil { if err := ntg.refreshAuth(); err != nil {
return nil, err return nil, err
} }
@ -232,9 +242,7 @@ func (ntg NTGVision) refreshAuth() error {
if len(v.Token) == 0 { if len(v.Token) == 0 {
return errors.New("failed to get token from login call") return errors.New("failed to get token from login call")
} }
conf := config.Get() setNTGToken(v.Token)
conf.Brokers.NTG.Token = v.Token
config.Set(*conf)
return nil return nil
} }
@ -283,7 +291,7 @@ func (ntg NTGVision) newRequest(states []config.State) (*http.Request, error) {
return nil, err return nil, err
} }
setNTGHeaders(request) setNTGHeaders(request)
request.Header.Set("Authorization", "Bearer "+config.Get().Brokers.NTG.Token) request.Header.Set("Authorization", "Bearer "+getNTGToken())
return request, nil return request, nil
} }
@ -294,7 +302,7 @@ func setNTGHeaders(request *http.Request) {
request.Header.Set("Accept-Language", "en-US,en;q=0.5") request.Header.Set("Accept-Language", "en-US,en;q=0.5")
request.Header.Set("Accept-Encoding", "gzip, deflate, br") request.Header.Set("Accept-Encoding", "gzip, deflate, br")
request.Header.Set("Content-Type", "application/json;charset=utf-8") request.Header.Set("Content-Type", "application/json;charset=utf-8")
//request.Header.Set("Authorization", "Bearer "+config.Get().Brokers.NTG.Token) //request.Header.Set("Authorization", "Bearer "+getNTGToken())
request.Header.Set("Origin", "https://ntgvision.com") request.Header.Set("Origin", "https://ntgvision.com")
request.Header.Set("DNT", "1") request.Header.Set("DNT", "1")
request.Header.Set("Connection", "keep-alive") request.Header.Set("Connection", "keep-alive")

View File

@ -1,5 +1,4 @@
todo: todo:
- cache on disk jobinfo
- tokens in db, not in config - tokens in db, not in config
- mark jobs no longer avail by modifying - mark jobs no longer avail by modifying
- write-as for clients so ma can write to pa by default - write-as for clients so ma can write to pa by default
@ -24,6 +23,7 @@ todo:
- banlist criteria like vendors, brokers, metadata - banlist criteria like vendors, brokers, metadata
- set up copy for caleb, broc - set up copy for caleb, broc
done: done:
- cache on disk jobinfo
- link to view more - link to view more
- map with to-from line - map with to-from line
- TO CONFLUENT.RS OR W/E - TO CONFLUENT.RS OR W/E