Compare commits

..

43 Commits

Author SHA1 Message Date
Bel LaPointe 1d2f661bb0 good enough 2023-04-10 11:17:23 -06:00
Bel LaPointe ee06ab7511 tar 2022-10-03 08:31:24 -06:00
Bel LaPointe 0046c92155 also http client close idle 2022-03-16 16:51:04 -06:00
bel bd71b4ecbc ensure http bodies closed, matrix client some kinda closed 2022-03-16 16:42:25 -06:00
bel 55bd95cc9a up matrix timeout in http client 2022-02-25 21:59:54 -07:00
bel 3700ed2c36 update image with not avail over del 2022-02-24 09:26:45 -07:00
Bel LaPointe 4d16b0d5cc do not delete old jobs text 2022-02-24 09:23:22 -07:00
bel ddcdac24ef bad matrix cert ok 2022-02-02 07:12:18 -07:00
bel 3125d9debb SH 2022-01-31 09:35:56 -07:00
bel d9ed3c886f NO 2022-01-31 09:31:40 -07:00
bel cf34889bef little 2022-01-30 08:43:57 -07:00
bel 4a25e7dc0f make working hours, weekdays configurable by impl ntg 2022-01-30 08:34:14 -07:00
bel 93d74fb9e1 add ntg configurable work hours, weekdays 2022-01-30 08:29:06 -07:00
bel 0ded04f072 cache last ntg result for off hours 2022-01-29 18:59:16 -07:00
bel ff1f62def9 Merge branch 'master' of http://gogs.scratch.com:59515/bel/truckstop 2022-01-29 18:44:21 -07:00
bel 72c6ef9619 fuck backwards bool 2022-01-29 18:44:13 -07:00
bel 546805cd71 log 2022-01-29 18:40:13 -07:00
Bel LaPointe 02bbfd6fd9 todo 2022-01-27 20:18:13 -07:00
Bel LaPointe 9e24ca3c65 fix offset to not need system timezone 2022-01-27 20:04:31 -07:00
Bel LaPointe 6551fb906d only do ntgvision searches 9-4 EST because 417s mean unauth until forced pw reset 2022-01-27 19:53:50 -07:00
Bel LaPointe 5d4677ab5c accept !radius for miles radius search 2022-01-27 19:21:03 -07:00
Bel LaPointe c40b99a8d8 include one very nice log 2022-01-27 19:00:40 -07:00
Bel LaPointe d1ca69f533 remove unwanted from help, only print !zip or !state as configured 2022-01-27 18:53:56 -07:00
Bel LaPointe 6f5a928e73 fix typo in config.json for == in query 2022-01-27 18:49:05 -07:00
Bel LaPointe 521dc4ce77 debug to verbose logs 2022-01-27 18:46:59 -07:00
Bel LaPointe dbf4a7304f whoosp backwards if body 2022-01-27 18:41:58 -07:00
Bel LaPointe 8768857fbe job filters by zip or state depending on config 2022-01-27 18:35:40 -07:00
Bel LaPointe 2b420252d5 support logging zip changes 2022-01-27 18:27:01 -07:00
Bel LaPointe 64f1488a91 main will search zips if Broker.UseZips, ntg does its own filtering by city-state dist from search-zips 2022-01-27 18:20:52 -07:00
Bel LaPointe 00368c50ff ntgvision maps zips to states then does same query 2022-01-27 18:14:49 -07:00
Bel LaPointe a3b8d5a6b2 impl and test zip.Get, MilesTo, GetStatesWithin, FromCityState 2022-01-27 18:11:11 -07:00
Bel LaPointe ae7b834599 impl zip.Get for translating zip to info and calc dist 2022-01-27 17:58:00 -07:00
Bel LaPointe a41ea7d501 move radius miles to broker level 2022-01-27 17:39:20 -07:00
Bel LaPointe d38555a555 move usezips to brokers 2022-01-27 17:37:36 -07:00
Bel LaPointe d7d523b0a7 impl fastexact SearchZips and dedupe diff-id same-job for fastexact 2022-01-27 17:36:17 -07:00
Bel LaPointe 649bae91f2 get zip dump including lat lng 2022-01-27 17:29:41 -07:00
Bel LaPointe 5406250af3 states are explicit 2022-01-27 17:27:08 -07:00
Bel LaPointe e19cd7095d add zip to configs 2022-01-27 17:23:33 -07:00
Bel LaPointe 0013db850c Merge branch 'master' of http://192.168.0.86:59515/bel/truckstop 2022-01-27 17:20:39 -07:00
Bel LaPointe d978a398b1 todo 2022-01-27 17:15:10 -07:00
bel e7782d4ff5 todo zip code wish liist 2022-01-27 17:12:05 -07:00
Bel LaPointe 5174a876ef maybe cookies are lowkey one header but delimited 2022-01-27 16:59:42 -07:00
bel 2ef92d25db Merge branch 'master' of http://gogs.scratch.com:59515/bel/truckstop 2022-01-27 16:35:08 -07:00
24 changed files with 668 additions and 158 deletions

View File

@ -3,9 +3,9 @@ package broker
import ( import (
"context" "context"
"encoding/json" "encoding/json"
"local/storage" "gogs.inhome.blapointe.com/local/storage"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"local/truckstop/logtr" "gogs.inhome.blapointe.com/local/truckstop/logtr"
"net/http" "net/http"
"strings" "strings"
"time" "time"
@ -20,7 +20,8 @@ var authlimiter = rate.NewLimiter(rate.Limit(1.0/60.0), 1)
var limiter = rate.NewLimiter(rate.Limit(1.0/20.0), 1) var limiter = rate.NewLimiter(rate.Limit(1.0/20.0), 1)
type Broker interface { type Broker interface {
Search([]config.State) ([]Job, error) SearchStates([]config.State) ([]Job, error)
SearchZips([]string) ([]Job, error)
} }
func do(r *http.Request) (*http.Response, error) { func do(r *http.Request) (*http.Response, error) {
@ -42,9 +43,14 @@ func _do(db storage.DB, r *http.Request) (*http.Response, error) {
logtr.Errorf("failed to get cookies: %v", err) logtr.Errorf("failed to get cookies: %v", err)
} else { } else {
logtr.Verbosef("got cookies for %s: %+v", cookieJarKey, cookies) logtr.Verbosef("got cookies for %s: %+v", cookieJarKey, cookies)
cookieV := strings.Join(r.Header["Cookie"], "; ")
for _, cookie := range cookies { for _, cookie := range cookies {
r.Header.Add("Cookie", cookie) if len(cookieV) > 0 {
cookieV += "; "
} }
cookieV += cookie
}
r.Header.Set("Cookie", cookieV)
} }
logtr.Verbosef("_do: %+v", r) logtr.Verbosef("_do: %+v", r)
resp, err := client.Do(r) resp, err := client.Do(r)
@ -75,13 +81,15 @@ func setCookies(db storage.DB, host string, resp *http.Response) error {
continue continue
} }
name := strings.Split(value, "=")[0] name := strings.Split(value, "=")[0]
value := strings.Split(value, ";")[0]
nameValues = append(nameValues, [2]string{name, value}) nameValues = append(nameValues, [2]string{name, value})
} }
for _, value := range resp.Header["Set-Cookie"] { for _, cookie := range resp.Header["Set-Cookie"] {
if len(value) == 0 { if len(cookie) == 0 {
continue continue
} }
name := strings.Split(value, "=")[0] name := strings.Split(cookie, "=")[0]
value := strings.Split(cookie, ";")[0]
found := false found := false
for i := range nameValues { for i := range nameValues {
if nameValues[i][0] == name { if nameValues[i][0] == name {

View File

@ -2,8 +2,8 @@ package broker
import ( import (
"fmt" "fmt"
"local/storage" "gogs.inhome.blapointe.com/local/storage"
"local/truckstop/logtr" "gogs.inhome.blapointe.com/local/truckstop/logtr"
"net/http" "net/http"
"net/http/httptest" "net/http/httptest"
"strconv" "strconv"
@ -32,9 +32,9 @@ func TestDoCookies(t *testing.T) {
if calls != 1 { if calls != 1 {
if !strings.Contains(fmt.Sprint(r.Header["Cookie"]), "name=value"+strconv.Itoa(calls-1)) { if !strings.Contains(fmt.Sprint(r.Header["Cookie"]), "name=value"+strconv.Itoa(calls-1)) {
w.WriteHeader(http.StatusBadRequest) w.WriteHeader(http.StatusBadRequest)
t.Error("cookie not set as latest") t.Error("cookie not set as latest: " + fmt.Sprint(r.Header["Cookie"]))
} }
if !strings.Contains(fmt.Sprint(r.Header["Cookie"]), "Expires") { if strings.Contains(fmt.Sprint(r.Header["Cookie"]), "Expires") {
w.WriteHeader(http.StatusBadRequest) w.WriteHeader(http.StatusBadRequest)
t.Error("cookie not expiration: ", r.Header["Cookie"]) t.Error("cookie not expiration: ", r.Header["Cookie"])
} }

Binary file not shown.

View File

@ -7,10 +7,11 @@ import (
"fmt" "fmt"
"io" "io"
"io/ioutil" "io/ioutil"
"local/storage" "gogs.inhome.blapointe.com/local/storage"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"local/truckstop/logtr" "gogs.inhome.blapointe.com/local/truckstop/logtr"
"net/http" "net/http"
"sort"
"strconv" "strconv"
"strings" "strings"
"time" "time"
@ -38,13 +39,24 @@ func (fe FastExact) WithMock() FastExact {
return fe return fe
} }
func (fe FastExact) Search(states []config.State) ([]Job, error) { func (fe FastExact) SearchZips(zips []string) ([]Job, error) {
jobs, err := fe.search(states) jobs, err := fe.searchZips(zips)
if err == ErrNoAuth { if err == ErrNoAuth {
if err := fe.login(); err != nil { if err := fe.login(); err != nil {
return nil, err return nil, err
} }
jobs, err = fe.search(states) jobs, err = fe.searchZips(zips)
}
return jobs, err
}
func (fe FastExact) SearchStates(states []config.State) ([]Job, error) {
jobs, err := fe.searchStates(states)
if err == ErrNoAuth {
if err := fe.login(); err != nil {
return nil, err
}
jobs, err = fe.searchStates(states)
} }
return jobs, err return jobs, err
} }
@ -75,7 +87,7 @@ func (fe FastExact) _login(username, password string, db storage.DB) error {
return err return err
} }
b, _ := ioutil.ReadAll(resp.Body) b, _ := ioutil.ReadAll(resp.Body)
resp.Body.Close() defer resp.Body.Close()
if resp.StatusCode != http.StatusOK { if resp.StatusCode != http.StatusOK {
return fmt.Errorf("bad status logging into fast exact: %d: %s", resp.StatusCode, b) return fmt.Errorf("bad status logging into fast exact: %d: %s", resp.StatusCode, b)
@ -91,28 +103,47 @@ func (fe FastExact) setHeaders(req *http.Request) {
req.Header.Set("Content-Type", "application/x-www-form-urlencoded") req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
} }
func (fe FastExact) search(states []config.State) ([]Job, error) { func (fe FastExact) searchZips(zips []string) ([]Job, error) {
var jobs []Job var jobs []Job
for _, state := range states { for _, zip := range zips {
subjobs, err := fe.searchOne(state) subjobs, err := fe.searchOneZip(zip)
if err != nil { if err != nil {
return nil, err return nil, err
} }
jobs = append(jobs, subjobs...) jobs = append(jobs, subjobs...)
} }
return fe.dedupeJobs(jobs), nil
}
func (fe FastExact) searchStates(states []config.State) ([]Job, error) {
var jobs []Job
for _, state := range states {
subjobs, err := fe.searchOneState(state)
if err != nil {
return nil, err
}
jobs = append(jobs, subjobs...)
}
return fe.dedupeJobs(jobs), nil
}
func (fe FastExact) dedupeJobs(jobs []Job) []Job {
sort.Slice(jobs, func(i, j int) bool {
return jobs[i].UID() < jobs[j].UID()
})
dedupeJobs := map[string]Job{} dedupeJobs := map[string]Job{}
for _, job := range jobs { for _, job := range jobs {
dedupeJobs[job.UID()] = job dedupeJobs[strings.ReplaceAll(job.UID(), job.ID, "")] = job
} }
result := []Job{} result := []Job{}
for _, job := range dedupeJobs { for _, job := range dedupeJobs {
result = append(result, job) result = append(result, job)
} }
return result, nil return result
} }
func (fe FastExact) searchOne(state config.State) ([]Job, error) { func (fe FastExact) searchOneZip(zip string) ([]Job, error) {
req, err := fe.newRequest(state) req, err := fe.newRequest(zip)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -121,17 +152,36 @@ func (fe FastExact) searchOne(state config.State) ([]Job, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
defer resp.Body.Close()
return fe.parse(resp) return fe.parse(resp)
} }
func (fe FastExact) newRequest(state config.State) (*http.Request, error) { func (fe FastExact) searchOneState(state config.State) ([]Job, error) {
req, err := fe.newRequestWithState(state)
if err != nil {
return nil, err
}
resp, err := fe.doer.doRequest(req)
logtr.Verbosef("req: %+v => resp: %+v", req, resp)
if err != nil {
return nil, err
}
defer resp.Body.Close()
return fe.parse(resp)
}
func (fe FastExact) newRequestWithState(state config.State) (*http.Request, error) {
zip, ok := config.States[state] zip, ok := config.States[state]
if !ok { if !ok {
return nil, fmt.Errorf("no configured zip for %s", state) return nil, fmt.Errorf("no configured zip for %s", state)
} }
return fe.newRequest(zip)
}
func (fe FastExact) newRequest(zip string) (*http.Request, error) {
req, err := http.NewRequest( req, err := http.NewRequest(
http.MethodGet, http.MethodGet,
"https://www.fastexact.com/secure/index.php?page=ajaxListJobs&action=ajax&zipcode="+zip+"&records_per_page=50&distance="+strconv.Itoa(config.Get().Brokers.FastExact.RadiusMiles)+"&st_loc_zip=8", "https://www.fastexact.com/secure/index.php?page=ajaxListJobs&action=ajax&zipcode="+zip+"&records_per_page=50&distance="+strconv.Itoa(config.Get().Brokers.RadiusMiles)+"&st_loc_zip=8",
nil, nil,
) )
if err != nil { if err != nil {
@ -244,7 +294,7 @@ func (mock mockFastExactDoer) doRequest(req *http.Request) (*http.Response, erro
if req.URL.Query().Get("records_per_page") != "50" { if req.URL.Query().Get("records_per_page") != "50" {
return nil, errors.New("bad query: records_per_page should be 50") return nil, errors.New("bad query: records_per_page should be 50")
} }
if req.URL.Query().Get("distance") != strconv.Itoa(config.Get().Brokers.FastExact.RadiusMiles) { if req.URL.Query().Get("distance") != strconv.Itoa(config.Get().Brokers.RadiusMiles) {
return nil, errors.New("bad query: distance should be as configured") return nil, errors.New("bad query: distance should be as configured")
} }
if req.URL.Query().Get("zipcode") == "" { if req.URL.Query().Get("zipcode") == "" {

View File

@ -1,9 +1,9 @@
package broker package broker
import ( import (
"local/storage" "gogs.inhome.blapointe.com/local/storage"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"local/truckstop/logtr" "gogs.inhome.blapointe.com/local/truckstop/logtr"
"os" "os"
"path" "path"
"testing" "testing"
@ -34,7 +34,7 @@ func TestFastExactReal(t *testing.T) {
if err := fe.login(); err != nil { if err := fe.login(); err != nil {
t.Fatal(err) t.Fatal(err)
} }
jobs, err := fe.search(states) jobs, err := fe.searchStates(states)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -56,16 +56,16 @@ func TestFastExactLogin(t *testing.T) {
} }
} }
func TestFastExactSearch(t *testing.T) { func TestFastExactSearchStates(t *testing.T) {
logtr.SetLevel(logtr.SOS) logtr.SetLevel(logtr.SOS)
limiter = rate.NewLimiter(rate.Limit(60.0), 1) limiter = rate.NewLimiter(rate.Limit(60.0), 1)
fe := NewFastExact().WithMock() fe := NewFastExact().WithMock()
_ = fe _ = fe
db := storage.NewMap() db := storage.NewMap()
_ = db _ = db
if jobs, err := fe.search([]config.State{config.State("NC"), config.State("SC")}); err != nil { if jobs, err := fe.searchStates([]config.State{config.State("NC"), config.State("SC")}); err != nil {
t.Fatal(err) t.Fatal(err)
} else if len(jobs) != 10 { } else if len(jobs) != 9 {
t.Fatal(len(jobs)) t.Fatal(len(jobs))
} else { } else {
for _, job := range jobs { for _, job := range jobs {

View File

@ -3,8 +3,8 @@ package broker
import ( import (
"encoding/base64" "encoding/base64"
"fmt" "fmt"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"local/truckstop/logtr" "gogs.inhome.blapointe.com/local/truckstop/zip"
"strings" "strings"
"time" "time"
) )
@ -86,9 +86,17 @@ func (j Job) FormatMultilineText() string {
} }
out := "" out := ""
clients := config.Clients(j.Pickup.Date) clients := config.Clients(j.Pickup.Date)
useZip := config.Get().Brokers.UseZips
zipRadius := config.Get().Brokers.RadiusMiles
jobZip := zip.FromCityState(j.Pickup.City, j.Pickup.State)
for k := range clients { for k := range clients {
logtr.Debugf("job multiline: %+v contains %s then use %v", clients[k].States, j.Pickup.State, k) should := strings.Contains(fmt.Sprint(clients[k].States), j.Pickup.State)
if strings.Contains(fmt.Sprint(clients[k].States), j.Pickup.State) { if useZip {
for _, z := range clients[k].Zips {
should = should || zip.Get(z).MilesTo(jobZip) <= zipRadius
}
}
if should {
if len(out) > 0 { if len(out) > 0 {
out += "\n" out += "\n"
} }

View File

@ -7,15 +7,16 @@ import (
"fmt" "fmt"
"io" "io"
"io/ioutil" "io/ioutil"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"local/truckstop/logtr" "gogs.inhome.blapointe.com/local/truckstop/logtr"
"gogs.inhome.blapointe.com/local/truckstop/zip"
"net/http" "net/http"
"time" "time"
) )
type NTGVision struct { type NTGVision struct {
searcher interface { searcher interface {
search(states []config.State) (io.ReadCloser, error) searchStates(states []config.State) (io.ReadCloser, error)
searchJobReadCloser(id int64) (io.ReadCloser, error) searchJobReadCloser(id int64) (io.ReadCloser, error)
} }
} }
@ -187,8 +188,77 @@ func (ntg NTGVision) searchJob(id int64) (ntgVisionJobInfo, error) {
return result, err return result, err
} }
func (ntg NTGVision) Search(states []config.State) ([]Job, error) { func (ntg NTGVision) SearchZips(zips []string) ([]Job, error) {
rc, err := ntg.searcher.search(states) radius := config.Get().Brokers.RadiusMiles
statesm := map[string]struct{}{}
for _, z := range zips {
somestates := zip.GetStatesWithin(zip.Get(z), radius)
for _, state := range somestates {
statesm[state] = struct{}{}
}
}
states := make([]config.State, 0, len(statesm))
for state := range statesm {
states = append(states, config.State(state))
}
if len(states) == 0 {
return nil, fmt.Errorf("failed to map zipcodes %+v to any states", zips)
}
jobs, err := ntg.SearchStates(states)
if err != nil {
return nil, err
}
for i := len(jobs) - 1; i >= 0; i-- {
ok := false
for _, z := range zips {
ok = ok || zip.Get(z).MilesTo(zip.FromCityState(jobs[i].Pickup.City, jobs[i].Pickup.State)) <= radius
}
if !ok {
jobs[i], jobs[len(jobs)-1] = jobs[len(jobs)-1], jobs[i]
jobs = jobs[:len(jobs)-1]
}
}
return jobs, nil
}
func (ntg NTGVision) workingHours(now time.Time) bool {
// TODO assert M-F 9-4 EST
now = now.In(time.FixedZone("EST", -5*60*60))
working := config.Get().Brokers.NTG.Working
logtr.Debugf("ntg.workingHours: now=%s, weekday=%v, hour=%v (ok=%+v)", now.String(), now.Weekday(), now.Hour(), working)
if ok := func() bool {
for _, hr := range working.Hours {
if now.Hour() == hr {
return true
}
}
return false
}(); !ok {
return false
}
if ok := func() bool {
for _, weekday := range working.Weekdays {
if now.Weekday() == time.Weekday(weekday) {
return true
}
}
return false
}(); !ok {
return false
}
return true
}
func (ntg NTGVision) SearchStates(states []config.State) ([]Job, error) {
if !ntg.workingHours(time.Now()) {
lastNtgB, _ := config.Get().DB().Get("ntg_last_search_states")
var jobs []Job
json.Unmarshal(lastNtgB, &jobs)
logtr.Verbosef("ntg.SearchStates: outside of working hours so returning ntg_last_search_states: %+v", jobs)
return jobs, nil
}
rc, err := ntg.searcher.searchStates(states)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -203,12 +273,22 @@ func (ntg NTGVision) Search(states []config.State) ([]Job, error) {
var ntgjobs []ntgVisionJob var ntgjobs []ntgVisionJob
err = json.Unmarshal(b, &ntgjobs) err = json.Unmarshal(b, &ntgjobs)
if err != nil {
return nil, err
}
jobs := make([]Job, len(ntgjobs)) jobs := make([]Job, len(ntgjobs))
for i := range jobs { for i := range jobs {
jobs[i] = ntgjobs[i].Job() jobs[i] = ntgjobs[i].Job()
} }
return jobs, err
jobsB, err := json.Marshal(jobs)
if err == nil {
config.Get().DB().Set("ntg_last_search_states", jobsB)
logtr.Verbosef("ntg.SearchStates: in working hours so setting ntg_last_search_states: %+v", jobs)
}
return jobs, nil
} }
func getNTGTokenKey() string { func getNTGTokenKey() string {
@ -226,20 +306,20 @@ func setNTGToken(token string) {
db.Set(getNTGTokenKey(), []byte(token)) db.Set(getNTGTokenKey(), []byte(token))
} }
func (ntg NTGVision) search(states []config.State) (io.ReadCloser, error) { func (ntg NTGVision) searchStates(states []config.State) (io.ReadCloser, error) {
if getNTGToken() == "" { if getNTGToken() == "" {
logtr.Debugf("NTG token is empty, refreshing ntg auth") logtr.Debugf("NTG token is empty, refreshing ntg auth")
if err := ntg.refreshAuth(); err != nil { if err := ntg.refreshAuth(); err != nil {
return nil, err return nil, err
} }
} }
rc, err := ntg._search(states) rc, err := ntg._searchStates(states)
if err == ErrNoAuth { if err == ErrNoAuth {
logtr.Debugf("err no auth on search, refreshing ntg auth") logtr.Debugf("err no auth on search, refreshing ntg auth")
if err := ntg.refreshAuth(); err != nil { if err := ntg.refreshAuth(); err != nil {
return nil, err return nil, err
} }
rc, err = ntg._search(states) rc, err = ntg._searchStates(states)
} }
return rc, err return rc, err
} }
@ -285,7 +365,7 @@ func (ntg NTGVision) _refreshAuth() error {
return nil return nil
} }
func (ntg NTGVision) _search(states []config.State) (io.ReadCloser, error) { func (ntg NTGVision) _searchStates(states []config.State) (io.ReadCloser, error) {
request, err := ntg.newRequest(states) request, err := ntg.newRequest(states)
if err != nil { if err != nil {
return nil, err return nil, err
@ -294,13 +374,16 @@ func (ntg NTGVision) _search(states []config.State) (io.ReadCloser, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
defer resp.Body.Close()
b, _ := ioutil.ReadAll(resp.Body)
resp.Body = io.NopCloser(bytes.NewReader(b))
if resp.StatusCode != http.StatusOK { if resp.StatusCode != http.StatusOK {
b, _ := ioutil.ReadAll(resp.Body) b, _ := ioutil.ReadAll(resp.Body)
resp.Body.Close() resp.Body.Close()
request2, _ := ntg.newRequest(states) request2, _ := ntg.newRequest(states)
requestb, _ := ioutil.ReadAll(request2.Body) requestb, _ := ioutil.ReadAll(request2.Body)
logtr.Debugf("ntg auth bad status: url=%s, status=%v, body=%s, headers=%+v, request=%+v, requestb=%s", request.URL.String(), resp.StatusCode, b, resp.Header, request2, requestb) logtr.Debugf("ntg auth bad status: url=%s, status=%v, body=%s, headers=%+v, request=%+v, requestb=%s", request.URL.String(), resp.StatusCode, b, resp.Header, request2, requestb)
if resp.StatusCode > 400 && resp.StatusCode < 404 { if resp.StatusCode > 400 && resp.StatusCode < 404 || resp.StatusCode == 417 { // TODO wtf is 417 for
logtr.Debugf("ntg auth bad status: err no auth") logtr.Debugf("ntg auth bad status: err no auth")
return nil, ErrNoAuth return nil, ErrNoAuth
} }

View File

@ -4,7 +4,7 @@ import (
"bytes" "bytes"
"io" "io"
"io/ioutil" "io/ioutil"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"os" "os"
"path" "path"
) )
@ -15,7 +15,7 @@ func NewNTGVisionMock() NTGVisionMock {
return NTGVisionMock{} return NTGVisionMock{}
} }
func (ntgm NTGVisionMock) search(states []config.State) (io.ReadCloser, error) { func (ntgm NTGVisionMock) searchStates(states []config.State) (io.ReadCloser, error) {
path := path.Join(os.Getenv("GOPATH"), "src", "local", "truckstop", "broker", "testdata", "ntgvision_response.json") path := path.Join(os.Getenv("GOPATH"), "src", "local", "truckstop", "broker", "testdata", "ntgvision_response.json")
b, err := ioutil.ReadFile(path) b, err := ioutil.ReadFile(path)
return io.NopCloser(bytes.NewReader(b)), err return io.NopCloser(bytes.NewReader(b)), err

22
broker/ntgvision_test.go Normal file
View File

@ -0,0 +1,22 @@
package broker
import (
"gogs.inhome.blapointe.com/local/truckstop/config"
"os"
"testing"
"time"
)
func TestWorkingHoursNTG(t *testing.T) {
os.Setenv("CONFIG", "../config.json")
if err := config.Refresh(nil); err != nil {
t.Fatal(err)
}
ntg := NTGVision{}
if !ntg.workingHours(time.Date(2022, 1, 27, 12, 0, 0, 0, time.FixedZone("MST", -7*60*60))) {
t.Fatal("noon MST not ok")
}
if ntg.workingHours(time.Date(2022, 1, 27, 23, 0, 0, 0, time.FixedZone("MST", -7*60*60))) {
t.Fatal("midnight MST ok")
}
}

View File

@ -1,7 +1,7 @@
{ {
"Log": { "Log": {
"Path": "/tmp/truckstop.log", "Path": "/tmp/truckstop.log",
"Level": "debug", "Level": "DEB",
"SOSMatrix": { "SOSMatrix": {
"ReceiveEnabled": true, "ReceiveEnabled": true,
"Mock": false, "Mock": false,
@ -48,6 +48,11 @@
"States": [ "States": [
"NC" "NC"
], ],
"Zips": [
"27006",
"73044",
"84058"
],
"IDs": { "IDs": {
"Matrix": "@bel:m.bltrucks.top" "Matrix": "@bel:m.bltrucks.top"
}, },
@ -71,21 +76,26 @@
}, },
"Once": false, "Once": false,
"Brokers": { "Brokers": {
"FastExact": { "UseZips": true,
"RadiusMiles": 100, "RadiusMiles": 200,
"Enabled": true,
"Mock": true,
"Username": "birdman",
"Password": "166647"
},
"NTG": { "NTG": {
"Working": {
"Hours": [6, 7, 8, 9, 10, 11, 12, 13, 14, 15],
"Weekdays": [1, 2, 3, 4, 5]
},
"Enabled": false, "Enabled": false,
"JobInfo": true, "JobInfo": true,
"Mock": true, "Mock": true,
"LoadPageURIFormat": "https://ntgvision.com/LoadDetails?loadId=%d", "LoadPageURIFormat": "https://ntgvision.com/LoadDetails?loadId=%d",
"LoadPageAPIURIFormat": "https://ntgvision.com/api/v1/load/LoadDetails?loadId==%d", "LoadPageAPIURIFormat": "https://ntgvision.com/api/v1/load/LoadDetails?loadId=%d",
"Username": "noeasyrunstrucking@gmail.com", "Username": "noeasyrunstrucking@gmail.com",
"Password": "thumper12345" "Password": "thumper12345"
},
"FastExact": {
"Enabled": true,
"Mock": true,
"Username": "birdman",
"Password": "166647"
} }
} }
} }

View File

@ -48,6 +48,9 @@
"States": [ "States": [
"NC" "NC"
], ],
"Zips": [
"27006"
],
"IDs": { "IDs": {
"Matrix": "@bel:m.bltrucks.top" "Matrix": "@bel:m.bltrucks.top"
}, },
@ -71,19 +74,24 @@
}, },
"Once": true, "Once": true,
"Brokers": { "Brokers": {
"FastExact": { "UseZips": true,
"RadiusMiles": 100, "RadiusMiles": 100,
"FastExact": {
"Enabled": true, "Enabled": true,
"Mock": true, "Mock": true,
"Username": "u", "Username": "u",
"Password": "p" "Password": "p"
}, },
"NTG": { "NTG": {
"Working": {
"Hours": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23],
"Weekdays": [0, 1, 2, 3, 4, 5, 6]
},
"Enabled": true, "Enabled": true,
"JobInfo": true, "JobInfo": true,
"Mock": true, "Mock": true,
"LoadPageURIFormat": "https://ntgvision.com/LoadDetails?loadId=%d", "LoadPageURIFormat": "https://ntgvision.com/LoadDetails?loadId=%d",
"LoadPageAPIURIFormat": "https://ntgvision.com/api/v1/load/LoadDetails?loadId==%d", "LoadPageAPIURIFormat": "https://ntgvision.com/api/v1/load/LoadDetails?loadId=%d",
"Username": "noeasyrunstrucking@gmail.com", "Username": "noeasyrunstrucking@gmail.com",
"Password": "thumper1234" "Password": "thumper1234"
} }

View File

@ -3,8 +3,8 @@ package config
import ( import (
"encoding/json" "encoding/json"
"io/ioutil" "io/ioutil"
"local/storage" "gogs.inhome.blapointe.com/local/storage"
"local/truckstop/logtr" "gogs.inhome.blapointe.com/local/truckstop/logtr"
"os" "os"
"sync" "sync"
"time" "time"
@ -64,7 +64,13 @@ type Config struct {
} }
Once bool Once bool
Brokers struct { Brokers struct {
UseZips bool
RadiusMiles int
NTG struct { NTG struct {
Working struct {
Hours []int
Weekdays []int
}
Enabled bool Enabled bool
JobInfo bool JobInfo bool
Mock bool Mock bool
@ -75,7 +81,6 @@ type Config struct {
} }
FastExact struct { FastExact struct {
Enabled bool Enabled bool
RadiusMiles int
Mock bool Mock bool
Username string Username string
Password string Password string
@ -88,6 +93,7 @@ type Config struct {
type Client struct { type Client struct {
States []State States []State
Zips []string
IDs struct { IDs struct {
Matrix string Matrix string
} }
@ -115,6 +121,20 @@ func Clients(t time.Time) map[string]Client {
return result return result
} }
func AllZips() []string {
zipm := map[string]struct{}{}
for _, v := range Clients(time.Now().Add(time.Hour * 24 * 365)) {
for _, state := range v.Zips {
zipm[state] = struct{}{}
}
}
zips := make([]string, 0, len(zipm)+1)
for k := range zipm {
zips = append(zips, k)
}
return zips
}
func AllStates() []State { func AllStates() []State {
statem := map[State]struct{}{} statem := map[State]struct{}{}
for _, v := range Clients(time.Now().Add(time.Hour * 24 * 365)) { for _, v := range Clients(time.Now().Add(time.Hour * 24 * 365)) {

27
go.mod
View File

@ -1,38 +1,26 @@
module local/truckstop module gogs.inhome.blapointe.com/local/truckstop
go 1.17 go 1.17
replace local/storage => ../storage
replace local/logb => ../logb
replace local/sandbox/contact/contact => ../sandbox/contact/contact
require ( require (
github.com/PuerkitoBio/goquery v1.8.0
github.com/google/uuid v1.3.0
github.com/matrix-org/gomatrix v0.0.0-20210324163249-be2af5ef2e16 github.com/matrix-org/gomatrix v0.0.0-20210324163249-be2af5ef2e16
local/sandbox/contact/contact v0.0.0-00010101000000-000000000000 gogs.inhome.blapointe.com/local/storage v0.0.0-20230410162102-db39d7b02e29
local/storage v0.0.0-00010101000000-000000000000 golang.org/x/time v0.0.0-20181108054448-85acf8d2951c
) )
require ( require (
cloud.google.com/go v0.33.1 // indirect cloud.google.com/go v0.33.1 // indirect
github.com/PuerkitoBio/goquery v1.8.0 // indirect
github.com/Unknwon/goconfig v0.0.0-20181105214110-56bd8ab18619 // indirect github.com/Unknwon/goconfig v0.0.0-20181105214110-56bd8ab18619 // indirect
github.com/abbot/go-http-auth v0.4.0 // indirect github.com/abbot/go-http-auth v0.4.0 // indirect
github.com/andybalholm/cascadia v1.3.1 // indirect github.com/andybalholm/cascadia v1.3.1 // indirect
github.com/aws/aws-sdk-go v1.15.81 // indirect github.com/aws/aws-sdk-go v1.15.81 // indirect
github.com/boltdb/bolt v1.3.1 // indirect github.com/boltdb/bolt v1.3.1 // indirect
github.com/bradfitz/gomemcache v0.0.0-20190913173617-a41fca850d0b // indirect
github.com/buraksezer/consistent v0.9.0 // indirect
github.com/bytbox/go-pop3 v0.0.0-20120201222208-3046caf0763e // indirect
github.com/cespare/xxhash v1.1.0 // indirect
github.com/emersion/go-imap v1.2.0 // indirect
github.com/emersion/go-sasl v0.0.0-20200509203442-7bfe0ed36a21 // indirect
github.com/go-stack/stack v1.8.0 // indirect github.com/go-stack/stack v1.8.0 // indirect
github.com/golang/protobuf v1.2.0 // indirect github.com/golang/protobuf v1.2.0 // indirect
github.com/golang/snappy v0.0.1 // indirect github.com/golang/snappy v0.0.1 // indirect
github.com/gomodule/redigo v1.8.5 // indirect github.com/gomodule/redigo v1.8.5 // indirect
github.com/google/uuid v1.3.0 // indirect
github.com/jmespath/go-jmespath v0.0.0-20180206201540-c2b33e8439af // indirect github.com/jmespath/go-jmespath v0.0.0-20180206201540-c2b33e8439af // indirect
github.com/json-iterator/go v1.1.9 // indirect github.com/json-iterator/go v1.1.9 // indirect
github.com/klauspost/compress v1.9.5 // indirect github.com/klauspost/compress v1.9.5 // indirect
@ -57,17 +45,16 @@ require (
github.com/xdg-go/stringprep v1.0.2 // indirect github.com/xdg-go/stringprep v1.0.2 // indirect
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect
go.mongodb.org/mongo-driver v1.7.2 // indirect go.mongodb.org/mongo-driver v1.7.2 // indirect
gogs.inhome.blapointe.com/local/logb v0.0.0-20230410154319-880efa39d871 // indirect
golang.org/x/crypto v0.0.0-20211215153901-e495a2d5b3d3 // indirect golang.org/x/crypto v0.0.0-20211215153901-e495a2d5b3d3 // indirect
golang.org/x/net v0.0.0-20211216030914-fe4d6282115f // indirect golang.org/x/net v0.0.0-20211216030914-fe4d6282115f // indirect
golang.org/x/oauth2 v0.0.0-20181120190819-8f65e3013eba // indirect golang.org/x/oauth2 v0.0.0-20181120190819-8f65e3013eba // indirect
golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e // indirect golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e // indirect
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1 // indirect golang.org/x/sys v0.0.0-20220503163025-988cb79eb6c6 // indirect
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1 // indirect golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1 // indirect
golang.org/x/text v0.3.7 // indirect golang.org/x/text v0.3.7 // indirect
golang.org/x/time v0.0.0-20181108054448-85acf8d2951c // indirect
google.golang.org/api v0.0.0-20181120235003-faade3cbb06a // indirect google.golang.org/api v0.0.0-20181120235003-faade3cbb06a // indirect
google.golang.org/appengine v1.3.0 // indirect google.golang.org/appengine v1.3.0 // indirect
gopkg.in/ini.v1 v1.42.0 // indirect gopkg.in/ini.v1 v1.42.0 // indirect
gopkg.in/yaml.v2 v2.4.0 // indirect gopkg.in/yaml.v2 v2.4.0 // indirect
local/logb v0.0.0-00010101000000-000000000000 // indirect
) )

25
go.sum
View File

@ -5,8 +5,6 @@ github.com/Azure/azure-pipeline-go v0.1.8/go.mod h1:XA1kFWRVhSK+KNFiOhfv83Fv8L9a
github.com/Azure/azure-storage-blob-go v0.0.0-20181023070848-cf01652132cc/go.mod h1:oGfmITT1V6x//CswqY2gtAHND+xIP64/qL7a5QJix0Y= github.com/Azure/azure-storage-blob-go v0.0.0-20181023070848-cf01652132cc/go.mod h1:oGfmITT1V6x//CswqY2gtAHND+xIP64/qL7a5QJix0Y=
github.com/Azure/go-ansiterm v0.0.0-20170929234023-d6e3b3328b78/go.mod h1:LmzpDX56iTiv29bbRTIsUNlaFfuhWRQBWjQdVyAevI8= github.com/Azure/go-ansiterm v0.0.0-20170929234023-d6e3b3328b78/go.mod h1:LmzpDX56iTiv29bbRTIsUNlaFfuhWRQBWjQdVyAevI8=
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
github.com/OneOfOne/xxhash v1.2.2 h1:KMrpdQIwFcEqXDklaen+P1axHaj9BSKzvpUUfnHldSE=
github.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU=
github.com/PuerkitoBio/goquery v1.8.0 h1:PJTF7AmFCFKk1N6V6jmKfrNH9tV5pNE6lZMkG0gta/U= github.com/PuerkitoBio/goquery v1.8.0 h1:PJTF7AmFCFKk1N6V6jmKfrNH9tV5pNE6lZMkG0gta/U=
github.com/PuerkitoBio/goquery v1.8.0/go.mod h1:ypIiRMtY7COPGk+I/YbZLbxsxn9g5ejnI2HSMtkjZvI= github.com/PuerkitoBio/goquery v1.8.0/go.mod h1:ypIiRMtY7COPGk+I/YbZLbxsxn9g5ejnI2HSMtkjZvI=
github.com/Unknwon/goconfig v0.0.0-20181105214110-56bd8ab18619 h1:6X8iB881g299aNEv6KXrcjL31iLOH7yA6NXoQX+MbDg= github.com/Unknwon/goconfig v0.0.0-20181105214110-56bd8ab18619 h1:6X8iB881g299aNEv6KXrcjL31iLOH7yA6NXoQX+MbDg=
@ -22,14 +20,6 @@ github.com/aws/aws-sdk-go v1.15.81/go.mod h1:E3/ieXAlvM0XWO57iftYVDLLvQ824smPP3A
github.com/billziss-gh/cgofuse v1.1.0/go.mod h1:LJjoaUojlVjgo5GQoEJTcJNqZJeRU0nCR84CyxKt2YM= github.com/billziss-gh/cgofuse v1.1.0/go.mod h1:LJjoaUojlVjgo5GQoEJTcJNqZJeRU0nCR84CyxKt2YM=
github.com/boltdb/bolt v1.3.1 h1:JQmyP4ZBrce+ZQu0dY660FMfatumYDLun9hBCUVIkF4= github.com/boltdb/bolt v1.3.1 h1:JQmyP4ZBrce+ZQu0dY660FMfatumYDLun9hBCUVIkF4=
github.com/boltdb/bolt v1.3.1/go.mod h1:clJnj/oiGkjum5o1McbSZDSLxVThjynRyGBgiAx27Ps= github.com/boltdb/bolt v1.3.1/go.mod h1:clJnj/oiGkjum5o1McbSZDSLxVThjynRyGBgiAx27Ps=
github.com/bradfitz/gomemcache v0.0.0-20190913173617-a41fca850d0b h1:L/QXpzIa3pOvUGt1D1lA5KjYhPBAN/3iWdP7xeFS9F0=
github.com/bradfitz/gomemcache v0.0.0-20190913173617-a41fca850d0b/go.mod h1:H0wQNHz2YrLsuXOZozoeDmnHXkNCRmMW0gwFWDfEZDA=
github.com/buraksezer/consistent v0.9.0 h1:Zfs6bX62wbP3QlbPGKUhqDw7SmNkOzY5bHZIYXYpR5g=
github.com/buraksezer/consistent v0.9.0/go.mod h1:6BrVajWq7wbKZlTOUPs/XVfR8c0maujuPowduSpZqmw=
github.com/bytbox/go-pop3 v0.0.0-20120201222208-3046caf0763e h1:mQTN05gz0rDZSABqKMzAPMb5ATWcvvdMljRzEh0LjBo=
github.com/bytbox/go-pop3 v0.0.0-20120201222208-3046caf0763e/go.mod h1:alXX+s7a4cKaIprgjeEboqi4Tm7XR/HXEwUTxUV/ywU=
github.com/cespare/xxhash v1.1.0 h1:a6HrQnmkObjyL+Gs60czilIUGqrzKutQD6XZog3p+ko=
github.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc=
github.com/coreos/bbolt v0.0.0-20180318001526-af9db2027c98/go.mod h1:iRUV2dpdMOn7Bo10OQBFzIJO9kkE559Wcmn+qkEiiKk= github.com/coreos/bbolt v0.0.0-20180318001526-af9db2027c98/go.mod h1:iRUV2dpdMOn7Bo10OQBFzIJO9kkE559Wcmn+qkEiiKk=
github.com/cpuguy83/go-md2man v1.0.8/go.mod h1:N6JayAiVKtlHSnuTCeuLSQVs75hb8q+dYQLjr7cDsKY= github.com/cpuguy83/go-md2man v1.0.8/go.mod h1:N6JayAiVKtlHSnuTCeuLSQVs75hb8q+dYQLjr7cDsKY=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
@ -38,12 +28,6 @@ github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSs
github.com/djherbis/times v1.1.0/go.mod h1:CGMZlo255K5r4Yw0b9RRfFQpM2y7uOmxg4jm9HsaVf8= github.com/djherbis/times v1.1.0/go.mod h1:CGMZlo255K5r4Yw0b9RRfFQpM2y7uOmxg4jm9HsaVf8=
github.com/dropbox/dropbox-sdk-go-unofficial v5.4.0+incompatible/go.mod h1:lr+LhMM3F6Y3lW1T9j2U5l7QeuWm87N9+PPXo3yH4qY= github.com/dropbox/dropbox-sdk-go-unofficial v5.4.0+incompatible/go.mod h1:lr+LhMM3F6Y3lW1T9j2U5l7QeuWm87N9+PPXo3yH4qY=
github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk= github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
github.com/emersion/go-imap v1.2.0 h1:lyUQ3+EVM21/qbWE/4Ya5UG9r5+usDxlg4yfp3TgHFA=
github.com/emersion/go-imap v1.2.0/go.mod h1:Qlx1FSx2FTxjnjWpIlVNEuX+ylerZQNFE5NsmKFSejY=
github.com/emersion/go-message v0.15.0/go.mod h1:wQUEfE+38+7EW8p8aZ96ptg6bAb1iwdgej19uXASlE4=
github.com/emersion/go-sasl v0.0.0-20200509203442-7bfe0ed36a21 h1:OJyUGMJTzHTd1XQp98QTaHernxMYzRaOasRir9hUlFQ=
github.com/emersion/go-sasl v0.0.0-20200509203442-7bfe0ed36a21/go.mod h1:iL2twTeMvZnrg54ZoPDNfJaJaqy0xIQFuBdrLsmspwQ=
github.com/emersion/go-textwrapper v0.0.0-20200911093747-65d896831594/go.mod h1:aqO8z8wPrjkscevZJFVE1wXJrLpC5LtJG7fqLOsPb2U=
github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo= github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
github.com/go-stack/stack v1.8.0 h1:5SgMzNM5HxrEjV0ww2lTmX6E2Izsfxas4+YHWRs3Lsk= github.com/go-stack/stack v1.8.0 h1:5SgMzNM5HxrEjV0ww2lTmX6E2Izsfxas4+YHWRs3Lsk=
github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY= github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=
@ -178,8 +162,6 @@ github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1
github.com/smartystreets/goconvey v0.0.0-20181108003508-044398e4856c/go.mod h1:XDJAKZRPZ1CvBcN2aX5YOUTYGHki24fSF0Iv48Ibg0s= github.com/smartystreets/goconvey v0.0.0-20181108003508-044398e4856c/go.mod h1:XDJAKZRPZ1CvBcN2aX5YOUTYGHki24fSF0Iv48Ibg0s=
github.com/smartystreets/goconvey v0.0.0-20190330032615-68dc04aab96a h1:pa8hGb/2YqsZKovtsgrwcDH1RZhVbTKCjLp47XpqCDs= github.com/smartystreets/goconvey v0.0.0-20190330032615-68dc04aab96a h1:pa8hGb/2YqsZKovtsgrwcDH1RZhVbTKCjLp47XpqCDs=
github.com/smartystreets/goconvey v0.0.0-20190330032615-68dc04aab96a/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA= github.com/smartystreets/goconvey v0.0.0-20190330032615-68dc04aab96a/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA=
github.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72 h1:qLC7fQah7D6K1B0ujays3HV9gkFtllcxhzImRR7ArPQ=
github.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=
github.com/spf13/cobra v0.0.3/go.mod h1:1l0Ry5zgKvJasoi3XT1TypsSe7PqH0Sj9dhYf7v3XqQ= github.com/spf13/cobra v0.0.3/go.mod h1:1l0Ry5zgKvJasoi3XT1TypsSe7PqH0Sj9dhYf7v3XqQ=
github.com/spf13/pflag v1.0.3 h1:zPAT6CGy6wXeQ7NtTnaTerfKOsV6V6F8agHXFiazDkg= github.com/spf13/pflag v1.0.3 h1:zPAT6CGy6wXeQ7NtTnaTerfKOsV6V6F8agHXFiazDkg=
github.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4= github.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4=
@ -209,6 +191,10 @@ github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d/go.mod h1:rHwXgn7Jul
github.com/yunify/qingstor-sdk-go v2.2.15+incompatible/go.mod h1:w6wqLDQ5bBTzxGJ55581UrSwLrsTAsdo9N6yX/8d9RY= github.com/yunify/qingstor-sdk-go v2.2.15+incompatible/go.mod h1:w6wqLDQ5bBTzxGJ55581UrSwLrsTAsdo9N6yX/8d9RY=
go.mongodb.org/mongo-driver v1.7.2 h1:pFttQyIiJUHEn50YfZgC9ECjITMT44oiN36uArf/OFg= go.mongodb.org/mongo-driver v1.7.2 h1:pFttQyIiJUHEn50YfZgC9ECjITMT44oiN36uArf/OFg=
go.mongodb.org/mongo-driver v1.7.2/go.mod h1:Q4oFMbo1+MSNqICAdYMlC/zSTrwCogR4R8NzkI+yfU8= go.mongodb.org/mongo-driver v1.7.2/go.mod h1:Q4oFMbo1+MSNqICAdYMlC/zSTrwCogR4R8NzkI+yfU8=
gogs.inhome.blapointe.com/local/logb v0.0.0-20230410154319-880efa39d871 h1:cMGPiwvK/QGg4TfW8VasO6SsS/O7UQmwyKDErV/ozoA=
gogs.inhome.blapointe.com/local/logb v0.0.0-20230410154319-880efa39d871/go.mod h1:E0pLNvMLzY0Kth1W078y+06z1AUyVMWnChMpRFf4w2Q=
gogs.inhome.blapointe.com/local/storage v0.0.0-20230410162102-db39d7b02e29 h1:SPSz7yQsEfScqyLlBS5uNSOGeT203BkIkFCL8jrm/FA=
gogs.inhome.blapointe.com/local/storage v0.0.0-20230410162102-db39d7b02e29/go.mod h1:zk8Fe2Ezc2f6oOe2yllsbEhXqssUU1K2faoS0eQ9alY=
golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
golang.org/x/crypto v0.0.0-20190131182504-b8fe1690c613/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= golang.org/x/crypto v0.0.0-20190131182504-b8fe1690c613/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
@ -247,8 +233,9 @@ golang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7w
golang.org/x/sys v0.0.0-20190531175056-4c3a928424d2/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20190531175056-4c3a928424d2/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1 h1:SrN+KX8Art/Sf4HNj6Zcz06G7VEz+7w9tdXTPOZ7+l4=
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20220503163025-988cb79eb6c6 h1:nonptSpoQ4vQjyraW20DXPAglgQfVnM9ZC6MmNLMR60=
golang.org/x/sys v0.0.0-20220503163025-988cb79eb6c6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1 h1:v+OssWQX+hTHEmOBgwxdZxK4zHq3yOs8F9J7mk0PY8E= golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1 h1:v+OssWQX+hTHEmOBgwxdZxK4zHq3yOs8F9J7mk0PY8E=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=

216
main.go
View File

@ -4,21 +4,23 @@ import (
"bytes" "bytes"
"encoding/json" "encoding/json"
"fmt" "fmt"
"local/storage" "gogs.inhome.blapointe.com/local/storage"
"local/truckstop/broker" "gogs.inhome.blapointe.com/local/truckstop/broker"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"local/truckstop/logtr" "gogs.inhome.blapointe.com/local/truckstop/logtr"
"local/truckstop/message" "gogs.inhome.blapointe.com/local/truckstop/message"
"net/http" "net/http"
"net/url" "net/url"
"regexp" "regexp"
"sort" "sort"
"strconv"
"strings" "strings"
"sync" "sync"
"time" "time"
) )
var stateFinder = regexp.MustCompile(`[A-Za-z]+`) var stateFinder = regexp.MustCompile(`[A-Za-z]+`)
var zipFinder = regexp.MustCompile(`[0-9]{5}[0-9]*`)
func main() { func main() {
if err := _main(); err != nil { if err := _main(); err != nil {
@ -62,15 +64,15 @@ func _main() error {
} }
func matrixrecv() error { func matrixrecv() error {
logtr.Debugf("checking matrix...") logtr.Verbosef("checking matrix...")
defer logtr.Debugf("/checking matrix...") defer logtr.Verbosef("/checking matrix...")
sender := message.NewMatrix() sender := message.NewMatrix()
messages, err := sender.Receive() messages, err := sender.Receive()
if err != nil { if err != nil {
return err return err
} }
func() { func() {
logtr.Debugf("looking for help") logtr.Verbosef("looking for help")
printed := false printed := false
for _, msg := range messages { for _, msg := range messages {
if !strings.HasPrefix(msg.Content, "!help") { if !strings.HasPrefix(msg.Content, "!help") {
@ -81,7 +83,11 @@ func matrixrecv() error {
if !printed { if !printed {
if _, err := db.Get(key); err == storage.ErrNotFound { if _, err := db.Get(key); err == storage.ErrNotFound {
logtr.Debugf("sending help") logtr.Debugf("sending help")
help := fmt.Sprintf("commands:\n...`!help` print this help\n...`!state nc NC nC Nc` set states for self\n...`!available 2022-12-31` set date self is available for work\n\nrun a command for someone else: `!state ga @caleb`") locationHelp := "`!state nc NC nC Nc` set states for self"
if config.Get().Brokers.UseZips {
locationHelp = "...`!zip 27006 84058` to set zip codes for self\n...`!radius 100` to set miles radius to 100, 200, or 300"
}
help := fmt.Sprintf("commands:\n...`!help` print this help\n%s\n\nrun a command for someone else: `!zip 2022-12-31 @caleb`", locationHelp)
if err := sender.Send(help); err != nil { if err := sender.Send(help); err != nil {
logtr.Errorf("failed to send help: %v", err) logtr.Errorf("failed to send help: %v", err)
} else { } else {
@ -99,7 +105,33 @@ func matrixrecv() error {
} }
}() }()
func() { func() {
logtr.Debugf("looking for states") logtr.Verbosef("looking for zips")
db := config.Get().DB()
zips := map[string]map[string]struct{}{}
for _, msg := range messages {
key := fmt.Sprintf("zips_%d", msg.Timestamp.Unix())
if !strings.HasPrefix(msg.Content, "!zip") {
continue
}
if _, ok := zips[msg.Sender]; ok {
continue
}
if _, err := db.Get(key); err == storage.ErrNotFound {
zips[msg.Sender] = map[string]struct{}{}
for _, zip := range parseOutZips([]byte(msg.Content)) {
zips[msg.Sender][zip] = struct{}{}
}
}
if err := db.Set(key, []byte{'k'}); err != nil {
logtr.Errorf("failed to mark state gathered @%s: %v", key, err)
}
}
if config.Get().Brokers.UseZips {
setNewZips(zips)
}
}()
func() {
logtr.Verbosef("looking for states")
db := config.Get().DB() db := config.Get().DB()
states := map[string]map[config.State]struct{}{} states := map[string]map[config.State]struct{}{}
for _, msg := range messages { for _, msg := range messages {
@ -120,10 +152,37 @@ func matrixrecv() error {
logtr.Errorf("failed to mark state gathered @%s: %v", key, err) logtr.Errorf("failed to mark state gathered @%s: %v", key, err)
} }
} }
if !config.Get().Brokers.UseZips {
setNewStates(states) setNewStates(states)
}
}() }()
func() { func() {
logtr.Debugf("looking for pauses") logtr.Verbosef("looking for radius")
db := config.Get().DB()
var radius *int
for _, msg := range messages {
key := fmt.Sprintf("radius_%d", msg.Timestamp.Unix())
if !strings.HasPrefix(msg.Content, "!radius ") {
continue
}
if _, err := db.Get(key); err == storage.ErrNotFound {
cmd := msg.Content
for strings.HasPrefix(cmd, "!radius ") {
cmd = strings.TrimPrefix(cmd, "!radius ")
}
n, err := strconv.Atoi(strings.TrimSpace(cmd))
if err == nil {
radius = &n
}
}
if err := db.Set(key, []byte{'k'}); err != nil {
logtr.Errorf("failed to mark radius gathered @%s: %v", key, err)
}
}
setNewRadius(radius)
}()
func() {
logtr.Verbosef("looking for pauses")
db := config.Get().DB() db := config.Get().DB()
pauses := map[string]time.Time{} pauses := map[string]time.Time{}
for _, msg := range messages { for _, msg := range messages {
@ -145,7 +204,7 @@ func matrixrecv() error {
} }
} }
if err := db.Set(key, []byte{'k'}); err != nil { if err := db.Set(key, []byte{'k'}); err != nil {
logtr.Errorf("failed to mark state gathered @%s: %v", key, err) logtr.Errorf("failed to mark pause gathered @%s: %v", key, err)
} }
} }
setNewPauses(pauses) setNewPauses(pauses)
@ -154,6 +213,33 @@ func matrixrecv() error {
return nil return nil
} }
func setNewRadius(radius *int) {
if radius == nil {
return
}
sender := message.NewMatrix()
logtr.Debugf("set new radius: %d", *radius)
acceptable := map[int]struct{}{
100: struct{}{},
200: struct{}{},
300: struct{}{},
}
if _, ok := acceptable[*radius]; !ok {
sender.Send(fmt.Sprintf("radius of %v is not among acceptable %+v", *radius, acceptable))
logtr.Debugf("bad radius, not setting: %d", *radius)
return
}
conf := *config.Get()
if conf.Brokers.RadiusMiles == *radius {
logtr.Debugf("noop radius, not setting: %d", *radius)
return
}
conf.Brokers.RadiusMiles = *radius
logtr.Infof("updating config new pauses: %+v", conf)
config.Set(conf)
sender.Send(fmt.Sprintf("now using radius of %d mi for zip code search", *radius))
}
func setNewPauses(pauses map[string]time.Time) { func setNewPauses(pauses map[string]time.Time) {
if len(pauses) == 0 { if len(pauses) == 0 {
return return
@ -182,6 +268,41 @@ func setNewPauses(pauses map[string]time.Time) {
} }
} }
func setNewZips(zips map[string]map[string]struct{}) {
if len(zips) == 0 {
return
}
conf := *config.Get()
changed := map[string][]string{}
for client, clientZips := range zips {
newzips := []string{}
for k := range clientZips {
newzips = append(newzips, k)
}
sort.Slice(newzips, func(i, j int) bool {
return newzips[i] < newzips[j]
})
clientconf := conf.Clients[client]
if fmt.Sprint(newzips) == fmt.Sprint(clientconf.Zips) {
message.NewMatrix().Send(fmt.Sprintf("%s: still searching for %+v", client, newzips))
continue
}
clientconf.Zips = newzips
conf.Clients[client] = clientconf
changed[client] = newzips
}
if len(changed) == 0 {
return
}
logtr.Infof("updating config new zips: %+v", conf)
config.Set(conf)
for client, zips := range changed {
if err := sendNewZips(client, zips); err != nil {
logtr.Errorf("failed to send new zips %s/%+v: %v", client, zips, err)
}
}
}
func setNewStates(states map[string]map[config.State]struct{}) { func setNewStates(states map[string]map[config.State]struct{}) {
if len(states) == 0 { if len(states) == 0 {
return return
@ -217,6 +338,18 @@ func setNewStates(states map[string]map[config.State]struct{}) {
} }
} }
func parseOutZips(b []byte) []string {
var zips []string
candidates := zipFinder.FindAll(b, -1)
for _, candidate := range candidates {
if len(candidate) != 5 {
continue
}
zips = append(zips, string(candidate))
}
return zips
}
func parseOutStates(b []byte) []config.State { func parseOutStates(b []byte) []config.State {
var states []config.State var states []config.State
var state config.State var state config.State
@ -242,7 +375,7 @@ func __main() error {
logtr.Errorf("failed _main: %v", err) logtr.Errorf("failed _main: %v", err)
} }
if config.Get().Once { if config.Get().Once {
time.Sleep(3 * time.Second) time.Sleep(10 * time.Second)
return err return err
} }
if err != nil { if err != nil {
@ -313,7 +446,39 @@ func once() error {
} }
func getJobs() ([]broker.Job, error) { func getJobs() ([]broker.Job, error) {
if config.Get().Brokers.UseZips {
return getJobsZips()
}
return getJobsStates()
}
func getJobsZips() ([]broker.Job, error) {
zips := config.AllZips()
jobs := []broker.Job{}
for _, broker := range getBrokers() {
somejobs, err := broker.SearchZips(zips)
if err != nil {
return nil, err
}
jobs = append(jobs, somejobs...)
}
return jobs, nil
}
func getJobsStates() ([]broker.Job, error) {
states := config.AllStates() states := config.AllStates()
jobs := []broker.Job{}
for _, broker := range getBrokers() {
somejobs, err := broker.SearchStates(states)
if err != nil {
return nil, err
}
jobs = append(jobs, somejobs...)
}
return jobs, nil
}
func getBrokers() []broker.Broker {
brokers := []broker.Broker{} brokers := []broker.Broker{}
if config.Get().Brokers.NTG.Enabled { if config.Get().Brokers.NTG.Enabled {
logtr.Debugf("NTG enabled") logtr.Debugf("NTG enabled")
@ -323,16 +488,7 @@ func getJobs() ([]broker.Job, error) {
logtr.Debugf("FastExact enabled") logtr.Debugf("FastExact enabled")
brokers = append(brokers, broker.NewFastExact()) brokers = append(brokers, broker.NewFastExact())
} }
logtr.Debugf("brokers=%+v", brokers) return brokers
jobs := []broker.Job{}
for _, broker := range brokers {
somejobs, err := broker.Search(states)
if err != nil {
return nil, err
}
jobs = append(jobs, somejobs...)
}
return jobs, nil
} }
type recordedJob struct { type recordedJob struct {
@ -370,15 +526,12 @@ func updateDeadJobs(jobs []broker.Job) error {
return err return err
} }
*/ */
if err := message.NewMatrix().Remove(recorded.MatrixID); err != nil {
logtr.Debugf("failed to remove matrix: %v: %v", recorded.MatrixID, err)
}
if err := db.Set(listEntry, nil); err != nil { if err := db.Set(listEntry, nil); err != nil {
logtr.Debugf("failed to remove db: %v: %v", listEntry, err) logtr.Debugf("failed to remove db: %v: %v", listEntry, err)
} }
for _, imageid := range recorded.MatrixImageIDs { for _, imageid := range recorded.MatrixImageIDs {
if err := message.NewMatrix().Remove(imageid); err != nil { if err := message.NewMatrix().Update(imageid, "<job no longer available>"); err != nil {
logtr.Debugf("failed to remove matrix image: %v: %v", imageid, err) logtr.Debugf("failed to update matrix image: %v: %v", imageid, err)
} }
} }
} }
@ -517,9 +670,14 @@ func sendJob(job broker.Job) (bool, error) {
return true, nil return true, nil
} }
func sendNewZips(client string, zips []string) error {
sender := message.NewMatrix()
return sender.Send(fmt.Sprintf("%s: now searching for loads from zip codes: %+v", client, zips))
}
func sendNewStates(client string, states []config.State) error { func sendNewStates(client string, states []config.State) error {
sender := message.NewMatrix() sender := message.NewMatrix()
return sender.Send(fmt.Sprintf("%s: now searching for loads from: %+v", client, states)) return sender.Send(fmt.Sprintf("%s: now searching for loads from states: %+v", client, states))
} }
func sendNewPause(client string, pause time.Time) error { func sendNewPause(client string, pause time.Time) error {

View File

@ -2,7 +2,7 @@ package main
import ( import (
"fmt" "fmt"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"os" "os"
"testing" "testing"
) )

View File

@ -6,8 +6,8 @@ import (
"errors" "errors"
"fmt" "fmt"
"io/ioutil" "io/ioutil"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"local/truckstop/logtr" "gogs.inhome.blapointe.com/local/truckstop/logtr"
"mime/multipart" "mime/multipart"
"net/http" "net/http"
"net/url" "net/url"
@ -64,7 +64,7 @@ func uploadImage(b []byte) (string, error) {
request.Header.Set("Authorization", "Bearer "+images.AccessToken) request.Header.Set("Authorization", "Bearer "+images.AccessToken)
request.Header.Set("Content-Type", writer.FormDataContentType()) request.Header.Set("Content-Type", writer.FormDataContentType())
c := &http.Client{Timeout: time.Minute} c := &http.Client{Timeout: time.Minute * 5}
response, err := c.Do(request) response, err := c.Do(request)
if err != nil { if err != nil {
return "", err return "", err

View File

@ -2,7 +2,7 @@ package message
import ( import (
"io/ioutil" "io/ioutil"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"os" "os"
"testing" "testing"
) )

View File

@ -2,10 +2,11 @@ package message
import ( import (
"bytes" "bytes"
"crypto/tls"
"fmt" "fmt"
"io/ioutil" "io/ioutil"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"local/truckstop/logtr" "gogs.inhome.blapointe.com/local/truckstop/logtr"
"net/http" "net/http"
"regexp" "regexp"
"strings" "strings"
@ -44,8 +45,28 @@ func newMatrix(conf config.Matrix, cont string) Matrix {
} }
} }
func (m Matrix) closeclient(client *gomatrix.Client) {
go func() {
client.Client.CloseIdleConnections()
client.StopSync()
client.Logout()
}()
}
func (m Matrix) getclient() (*gomatrix.Client, error) { func (m Matrix) getclient() (*gomatrix.Client, error) {
return gomatrix.NewClient(m.homeserver, m.username, m.token) client, err := gomatrix.NewClient(m.homeserver, m.username, m.token)
if err != nil {
return nil, err
}
client.Client = &http.Client{
Timeout: time.Minute * 5,
Transport: &http.Transport{
TLSClientConfig: &tls.Config{
InsecureSkipVerify: true,
},
},
}
return client, nil
} }
func (m Matrix) Continuation() string { func (m Matrix) Continuation() string {
@ -79,12 +100,13 @@ func (m *Matrix) Receive() ([]Message, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
defer m.closeclient(c)
messages := make([]Message, 0) messages := make([]Message, 0)
result, err := c.Messages(m.room, "999999999999999999", m.Continuation(), 'b', 50) result, err := c.Messages(m.room, "999999999999999999", m.Continuation(), 'b', 50)
if err != nil { if err != nil {
return nil, err return nil, err
} }
logtr.Debugf("%s => {Start:%s End:%v};; %v, (%d)", m.Continuation(), result.Start, result.End, err, len(result.Chunk)) logtr.Verbosef("%s => {Start:%s End:%v};; %v, (%d)", m.Continuation(), result.Start, result.End, err, len(result.Chunk))
m.continuation = result.End m.continuation = result.End
for _, event := range result.Chunk { for _, event := range result.Chunk {
logtr.Verbosef("matrix event: %+v", event) logtr.Verbosef("matrix event: %+v", event)
@ -100,7 +122,7 @@ func (m *Matrix) Receive() ([]Message, error) {
} }
} }
clientChange := regexp.MustCompile("@[a-z]+$") clientChange := regexp.MustCompile("@[a-z]+$")
logtr.Debugf("rewriting messages based on @abc") logtr.Verbosef("rewriting messages based on @abc")
for i := range messages { for i := range messages {
if found := clientChange.FindString(messages[i].Content); found != "" { if found := clientChange.FindString(messages[i].Content); found != "" {
messages[i].Content = strings.TrimSpace(strings.ReplaceAll(messages[i].Content, found, "")) messages[i].Content = strings.TrimSpace(strings.ReplaceAll(messages[i].Content, found, ""))
@ -114,7 +136,7 @@ func (m *Matrix) Receive() ([]Message, error) {
} }
messages[i].Content = strings.TrimSpace(messages[i].Content) messages[i].Content = strings.TrimSpace(messages[i].Content)
} }
logtr.Debugf("rewriting messages based on ! CoMmAnD ...") logtr.Verbosef("rewriting messages based on ! CoMmAnD ...")
for i := range messages { for i := range messages {
if strings.HasPrefix(messages[i].Content, "!") { if strings.HasPrefix(messages[i].Content, "!") {
messages[i].Content = "!" + strings.TrimSpace(messages[i].Content[1:]) messages[i].Content = "!" + strings.TrimSpace(messages[i].Content[1:])
@ -134,6 +156,7 @@ func (m Matrix) Remove(id string) error {
if err != nil { if err != nil {
return err return err
} }
defer m.closeclient(c)
_, err = c.RedactEvent(m.room, id, &gomatrix.ReqRedact{Reason: "stale"}) _, err = c.RedactEvent(m.room, id, &gomatrix.ReqRedact{Reason: "stale"})
return err return err
} }
@ -147,6 +170,7 @@ func (m Matrix) Update(id, text string) error {
if err != nil { if err != nil {
return err return err
} }
defer m.closeclient(c)
type MRelatesTo struct { type MRelatesTo struct {
EventID string `json:"event_id"` EventID string `json:"event_id"`
RelType string `json:"rel_type"` RelType string `json:"rel_type"`
@ -190,6 +214,7 @@ func (m Matrix) SendTracked(text string) (string, error) {
if err != nil { if err != nil {
return "", err return "", err
} }
defer m.closeclient(c)
resp, err := c.SendText(m.room, text) resp, err := c.SendText(m.room, text)
if err != nil { if err != nil {
return "", err return "", err
@ -225,13 +250,14 @@ func (m Matrix) SendImageTracked(uri string) (string, error) {
if err != nil { if err != nil {
return "", err return "", err
} }
defer m.closeclient(c)
mediaUpload, err := c.UploadToContentRepo(bytes.NewReader(b), "image/jpeg", int64(len(b))) mediaUpload, err := c.UploadToContentRepo(bytes.NewReader(b), "image/jpeg", int64(len(b)))
if err != nil { if err != nil {
return "", err return "", err
} }
publicURI := mediaUpload.ContentURI publicURI := mediaUpload.ContentURI
resp, err := c.SendImage(m.room, "img", publicURI) resp, err := c.SendImage(m.room, "img", publicURI)
logtr.Debugf("sent image %s => %s: %+v", uri, publicURI, resp) logtr.Verbosef("sent image %s => %s: %+v", uri, publicURI, resp)
return resp.EventID, err return resp.EventID, err
} }

View File

@ -2,7 +2,7 @@ package message
import ( import (
"io/ioutil" "io/ioutil"
"local/truckstop/config" "gogs.inhome.blapointe.com/local/truckstop/config"
"os" "os"
"path" "path"
"testing" "testing"

View File

@ -1,6 +1,4 @@
todo: todo:
- fast exact is dumb or...?
- modify old items once no longer available; drop stale jobs good candidate but requires new matrix interaction
- more than NTG; blue one - more than NTG; blue one
- !states emits current state - !states emits current state
- test each !command callbacks to matrix - test each !command callbacks to matrix
@ -13,7 +11,12 @@ todo:
subtasks: subtasks:
- banlist criteria like vendors, brokers, metadata - banlist criteria like vendors, brokers, metadata
- set up copy for caleb, broc - set up copy for caleb, broc
- move from main() and make more functions
done: done:
- modify old items once no longer available; drop stale jobs good candidate but requires new matrix interaction
- from states to zip codes w/ range
- fast exact does not use ID in UID because they spammy
- fast exact is dumb or...?
- try search ntg by autoinc? - try search ntg by autoinc?
- TEST. Just like, refactor and test to shit. - TEST. Just like, refactor and test to shit.
- mark jobs no longer avail by modifying in matrix;; save matrix ID over dummy payload - mark jobs no longer avail by modifying in matrix;; save matrix ID over dummy payload
@ -42,7 +45,7 @@ done:
- pause until => !busy until - pause until => !busy until
- todo: maps of to+from to get location within state via api - todo: maps of to+from to get location within state via api
details: | details: |
curl 'https://maps.googleapis.com/maps/api/staticmap?center=Advance,NC&markers=label=A|Advance,NC&zoom=5&size=250x250&scale=2&format=jpeg&maptype=roadmap&key=AIzaSyBkACm-LQkoSfsTO5_XAzBVZE9-JQzcNkg' -H 'User-Agent: Mozilla/5.0 (Windows NT 10.0; rv:91.0) Gecko/20100101 Firefox/91.0' -H 'Accept: image/avif,image/webp,*/*' -H 'Accept-Language: en-US,en;q=0.5' -H 'Accept-Encoding: gzip, deflate, br' -H 'DNT: 1' -H 'Alt-Used: maps.googleapis.com' -H 'Connection: keep-alive' -H 'Sec-Fetch-Dest: image' -H 'Sec-Fetch-Mode: no-cors' -H 'Sec-Fetch-Site: cross-site' -H 'Pragma: no-cache' -H 'Cache-Control: no-cache' -H 'TE: trailers' > whatever.jpg; open whatever.jpg~/Go/src/local/truckstop curl 'https://maps.googleapis.com/maps/api/staticmap?center=Advance,NC&markers=label=A|Advance,NC&zoom=5&size=250x250&scale=2&format=jpeg&maptype=roadmap&key=AIzaSyBkACm-LQkoSfsTO5_XAzBVZE9-JQzcNkg' -H 'User-Agent: Mozilla/5.0 (Windows NT 10.0; rv:91.0) Gecko/20100101 Firefox/91.0' -H 'Accept: image/avif,image/webp,*/*' -H 'Accept-Language: en-US,en;q=0.5' -H 'Accept-Encoding: gzip, deflate, br' -H 'DNT: 1' -H 'Alt-Used: maps.googleapis.com' -H 'Connection: keep-alive' -H 'Sec-Fetch-Dest: image' -H 'Sec-Fetch-Mode: no-cors' -H 'Sec-Fetch-Site: cross-site' -H 'Pragma: no-cache' -H 'Cache-Control: no-cache' -H 'TE: trailers' > whatever.jpg; open whatever.jpg~/Go/src/gogs.inhome.blapointe.com/local/truckstop
subtasks: subtasks:
- DONE; yandex; key 9baa3e42-c6e5-4eb5-a891-05ffcede6a25 yandex maps - DONE; yandex; key 9baa3e42-c6e5-4eb5-a891-05ffcede6a25 yandex maps
- google; key AIzaSyBkACm-LQkoSfsTO5_XAzBVZE9-JQzcNkg - google; key AIzaSyBkACm-LQkoSfsTO5_XAzBVZE9-JQzcNkg
@ -72,4 +75,4 @@ done:
- parse jobs - parse jobs
- gather jobs - gather jobs
- read states - read states
- read email to state file from local/sandbox/contact/contact - read email to state file from gogs.inhome.blapointe.com/local/sandbox/contact/contact

Binary file not shown.

112
zip/zip.go Normal file
View File

@ -0,0 +1,112 @@
package zip
import (
_ "embed"
"gogs.inhome.blapointe.com/local/truckstop/logtr"
"math"
"strconv"
"strings"
)
//go:embed testdata/simplemaps_uszips_basicv1.79/uszips.csv
var csv string
type Zip struct {
Lat float64
Lng float64
City string
State string
}
var zips map[string]Zip
func init() {
if len(zips) > 0 {
return
}
zips = map[string]Zip{}
trim := func(s string) string {
return strings.Trim(s, `"`)
}
for _, line := range strings.Split(csv, "\n")[1:] {
strs := strings.Split(line, ",")
if len(strs) < 5 {
continue
}
zip := trim(strs[0])
lat, _ := strconv.ParseFloat(trim(strs[1]), 32)
lng, _ := strconv.ParseFloat(trim(strs[2]), 32)
city := trim(alphafy(strs[3]))
state := strings.ToUpper(trim(strs[4]))
zips[zip] = Zip{
Lat: lat,
Lng: lng,
City: city,
State: state,
}
}
logtr.Infof("loaded %d zip codes", len(zips))
}
func alphafy(s string) string {
bs := make([]byte, 0, len(s)+1)
for i := range s {
if (s[i] < 'a' || s[i] > 'z') && (s[i] < 'A' || s[i] > 'Z') {
} else {
bs = append(bs, s[i])
}
}
return strings.ToLower(string(bs))
}
func (zip Zip) MilesTo(other Zip) int {
// c**2 = a**2 + b**2
// 69.2 mi per lat
// 60.0 mi per lng
return int(math.Sqrt(
math.Pow(69.2*(math.Abs(zip.Lat)-math.Abs(other.Lat)), 2) +
math.Pow(60.0*(math.Abs(zip.Lng)-math.Abs(other.Lng)), 2),
))
}
func Get(zip string) Zip {
if z, ok := zips[zip]; ok {
return z
}
zipAsI, err := strconv.Atoi(strings.Split(zip, "-")[0])
if err != nil {
return Zip{}
}
for i := 0; i < 5; i++ {
j := i - 2
if z2, ok := zips[strconv.Itoa(zipAsI+j)]; ok {
return z2
}
}
return Zip{}
}
func GetStatesWithin(zip Zip, miles int) []string {
states := map[string]struct{}{}
for _, other := range zips {
if zip.MilesTo(other) <= miles {
states[other.State] = struct{}{}
}
}
result := []string{}
for state := range states {
result = append(result, state)
}
return result
}
func FromCityState(city, state string) Zip {
city = alphafy(city)
state = strings.ToUpper(state)
for _, z := range zips {
if z.City == city && z.State == state {
return z
}
}
return Zip{}
}

28
zip/zip_test.go Normal file
View File

@ -0,0 +1,28 @@
package zip
import (
"fmt"
"sort"
"testing"
)
func TestZip(t *testing.T) {
_ = true
zip27006 := Get("27006")
zip84059 := Get("84059")
t.Logf("27006 = %+v", zip27006)
t.Logf("84059 = %+v", zip84059)
if zip27006.MilesTo(zip84059) < 1800 {
t.Error("failed to compute miles from advance to ut")
}
sorted := func(s []string) []string {
sort.Strings(s)
return s
}
if got := GetStatesWithin(Get("27006"), 100); fmt.Sprint(sorted(got)) != fmt.Sprint(sorted([]string{"NC", "SC", "TN", "VA"})) {
t.Error(got)
}
if got := FromCityState("ADVANCE", "nc"); got != zip27006 {
t.Error(got)
}
}