Compare commits
30 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a6c1b8505a | ||
|
|
c755aa88fb | ||
|
|
b451ed93bf | ||
|
|
76b7211d6c | ||
|
|
31a608d7f8 | ||
|
|
0c3419a1fb | ||
|
|
6ea4d4700c | ||
|
|
451f741f5a | ||
|
|
ecf22c3a3d | ||
|
|
ced1afff88 | ||
|
|
ffa33ea299 | ||
|
|
92b6019052 | ||
|
|
6a2b2f38d0 | ||
|
|
d4c1e20230 | ||
|
|
f2c9602d70 | ||
|
|
744365b9d3 | ||
|
|
16bdc174d4 | ||
|
|
d3381749f7 | ||
|
|
55848d6c7d | ||
|
|
f9819350ad | ||
|
|
7062094234 | ||
|
|
bdaa647923 | ||
|
|
d37f60bb3a | ||
|
|
a6d5ae606a | ||
|
|
a38a627f5a | ||
|
|
73ccc22fd5 | ||
|
|
e660f2ef9f | ||
|
|
937f91bbf6 | ||
|
|
bda1fd6f15 | ||
|
|
6f972d71ba |
1
.gitignore
vendored
1
.gitignore
vendored
@@ -4,3 +4,4 @@ cmd/cmd
|
|||||||
cmd/cli
|
cmd/cli
|
||||||
cmd/pttodo/pttodo
|
cmd/pttodo/pttodo
|
||||||
/truckstop
|
/truckstop
|
||||||
|
/exec-truckstop
|
||||||
|
|||||||
@@ -1,9 +1,10 @@
|
|||||||
package broker
|
package broker
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/base64"
|
||||||
"fmt"
|
"fmt"
|
||||||
"local/truckstop/config"
|
"local/truckstop/config"
|
||||||
"log"
|
"local/truckstop/logtr"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
@@ -16,7 +17,8 @@ type Job struct {
|
|||||||
Weight int
|
Weight int
|
||||||
Miles int
|
Miles int
|
||||||
Meta string
|
Meta string
|
||||||
secrets func() interface{} `json:"-"`
|
Pays string
|
||||||
|
secrets func(j *Job) `json:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type JobLocation struct {
|
type JobLocation struct {
|
||||||
@@ -25,16 +27,23 @@ type JobLocation struct {
|
|||||||
State string
|
State string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (j Job) UID() string {
|
||||||
|
return fmt.Sprintf(
|
||||||
|
"%v-%s-%s-%s-%s-%v",
|
||||||
|
j.ID,
|
||||||
|
j.Pickup.State,
|
||||||
|
base64.StdEncoding.EncodeToString([]byte(j.Pickup.City)),
|
||||||
|
j.Dropoff.State,
|
||||||
|
base64.StdEncoding.EncodeToString([]byte(j.Dropoff.City)),
|
||||||
|
j.Pickup.Date.Unix(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
func (j *Job) Secrets() {
|
func (j *Job) Secrets() {
|
||||||
if j.secrets == nil {
|
if j.secrets == nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
v := j.secrets()
|
j.secrets(j)
|
||||||
j.secrets = nil
|
|
||||||
if v == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
j.Meta = fmt.Sprintf("%s %+v", j.Meta, v)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (j Job) String() string {
|
func (j Job) String() string {
|
||||||
@@ -53,6 +62,19 @@ func (j JobLocation) String() string {
|
|||||||
return fmt.Sprintf("%s, %s @ %s", j.City, j.State, j.Date.Format("Monday Jan 02"))
|
return fmt.Sprintf("%s, %s @ %s", j.City, j.State, j.Date.Format("Monday Jan 02"))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (j Job) FormatMultilineTextDead() string {
|
||||||
|
return fmt.Sprintf(
|
||||||
|
"no longer available: %s,%s => %s,%s for $%v @%s",
|
||||||
|
j.Pickup.City,
|
||||||
|
j.Pickup.State,
|
||||||
|
j.Dropoff.City,
|
||||||
|
j.Dropoff.State,
|
||||||
|
j.Pays,
|
||||||
|
j.URI,
|
||||||
|
)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
func (j Job) FormatMultilineText() string {
|
func (j Job) FormatMultilineText() string {
|
||||||
foo := func(client string) string {
|
foo := func(client string) string {
|
||||||
return fmt.Sprintf(
|
return fmt.Sprintf(
|
||||||
@@ -65,7 +87,7 @@ func (j Job) FormatMultilineText() string {
|
|||||||
out := ""
|
out := ""
|
||||||
clients := config.Clients(j.Pickup.Date)
|
clients := config.Clients(j.Pickup.Date)
|
||||||
for k := range clients {
|
for k := range clients {
|
||||||
log.Printf("job multiline: %+v contains %s then use %v", clients[k].States, j.Pickup.State, k)
|
logtr.Debugf("job multiline: %+v contains %s then use %v", clients[k].States, j.Pickup.State, k)
|
||||||
if strings.Contains(fmt.Sprint(clients[k].States), j.Pickup.State) {
|
if strings.Contains(fmt.Sprint(clients[k].States), j.Pickup.State) {
|
||||||
if len(out) > 0 {
|
if len(out) > 0 {
|
||||||
out += "\n"
|
out += "\n"
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ import (
|
|||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"local/truckstop/config"
|
"local/truckstop/config"
|
||||||
"log"
|
"local/truckstop/logtr"
|
||||||
"net/http"
|
"net/http"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
@@ -16,6 +16,7 @@ import (
|
|||||||
type NTGVision struct {
|
type NTGVision struct {
|
||||||
searcher interface {
|
searcher interface {
|
||||||
search(states []config.State) (io.ReadCloser, error)
|
search(states []config.State) (io.ReadCloser, error)
|
||||||
|
searchJob(id int64) (io.ReadCloser, error)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -43,13 +44,14 @@ type ntgVisionJobInfo struct {
|
|||||||
Instructions string `json:"instructions"`
|
Instructions string `json:"instructions"`
|
||||||
IsDropTrailer bool `json:"isDropTrailer"`
|
IsDropTrailer bool `json:"isDropTrailer"`
|
||||||
} `json:"stopinfos"`
|
} `json:"stopinfos"`
|
||||||
PayUpTo float32 `json:"payUpTo"`
|
PayUpTo float32 `json:"payUpTo"`
|
||||||
LoadState string `json:"loadStatus"`
|
TotalCarrierRate float32 `json:"totalCarrierRate"`
|
||||||
|
LoadState string `json:"loadStatus"`
|
||||||
//CanBidNow bool `json:"canBidNow"`
|
//CanBidNow bool `json:"canBidNow"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ji ntgVisionJobInfo) IsZero() bool {
|
func (ji ntgVisionJobInfo) IsZero() bool {
|
||||||
return len(ji.StopsInfo) == 0 && ji.PayUpTo == 0 && ji.LoadState == ""
|
return len(ji.StopsInfo) == 0 && ji.TotalCarrierRate == 0 && ji.PayUpTo == 0 && ji.LoadState == ""
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ji ntgVisionJobInfo) String() string {
|
func (ji ntgVisionJobInfo) String() string {
|
||||||
@@ -60,6 +62,9 @@ func (ji ntgVisionJobInfo) String() string {
|
|||||||
if ji.PayUpTo != 0 {
|
if ji.PayUpTo != 0 {
|
||||||
out = fmt.Sprintf("\nPAYS:%v\n%s", ji.PayUpTo, out)
|
out = fmt.Sprintf("\nPAYS:%v\n%s", ji.PayUpTo, out)
|
||||||
}
|
}
|
||||||
|
if ji.TotalCarrierRate != 0 {
|
||||||
|
out = fmt.Sprintf("%s Total_Carrier_Rate:%v", out, ji.TotalCarrierRate)
|
||||||
|
}
|
||||||
if ji.LoadState != "" {
|
if ji.LoadState != "" {
|
||||||
out = fmt.Sprintf("%s Auction:%s", out, ji.LoadState)
|
out = fmt.Sprintf("%s Auction:%s", out, ji.LoadState)
|
||||||
}
|
}
|
||||||
@@ -90,13 +95,8 @@ func (ntgJob *ntgVisionJob) JobInfo() (ntgVisionJobInfo, error) {
|
|||||||
} else if err := json.Unmarshal(b, &ntgJob.jobinfo); err == nil {
|
} else if err := json.Unmarshal(b, &ntgJob.jobinfo); err == nil {
|
||||||
return ntgJob.jobinfo, nil
|
return ntgJob.jobinfo, nil
|
||||||
}
|
}
|
||||||
ji, err := ntgJob.jobInfo()
|
ntg := NewNTGVision()
|
||||||
if err == ErrNoAuth {
|
ji, err := ntg.SearchJob(ntgJob.ID)
|
||||||
if err := NewNTGVision().refreshAuth(); err != nil {
|
|
||||||
return ntgVisionJobInfo{}, err
|
|
||||||
}
|
|
||||||
ji, err = ntgJob.jobInfo()
|
|
||||||
}
|
|
||||||
if err == nil {
|
if err == nil {
|
||||||
ntgJob.jobinfo = ji
|
ntgJob.jobinfo = ji
|
||||||
b, err := json.Marshal(ntgJob.jobinfo)
|
b, err := json.Marshal(ntgJob.jobinfo)
|
||||||
@@ -107,26 +107,21 @@ func (ntgJob *ntgVisionJob) JobInfo() (ntgVisionJobInfo, error) {
|
|||||||
return ji, err
|
return ji, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ntgJob *ntgVisionJob) jobInfo() (ntgVisionJobInfo, error) {
|
func (ntg NTGVision) searchJob(id int64) (io.ReadCloser, error) {
|
||||||
time.Sleep(config.Get().Interval.JobInfo.Get())
|
time.Sleep(config.Get().Interval.JobInfo.Get())
|
||||||
request, err := http.NewRequest(http.MethodGet, fmt.Sprintf(`https://ntgvision.com/api/v1/load/LoadDetails?loadId=%v`, ntgJob.ID), nil)
|
request, err := http.NewRequest(http.MethodGet, fmt.Sprintf(config.Get().Brokers.NTG.LoadPageAPIURIFormat, id), nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return ntgVisionJobInfo{}, err
|
return nil, err
|
||||||
}
|
}
|
||||||
setNTGHeaders(request)
|
setNTGHeaders(request)
|
||||||
resp, err := do(request)
|
resp, err := do(request)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return ntgVisionJobInfo{}, err
|
return nil, err
|
||||||
}
|
}
|
||||||
defer resp.Body.Close()
|
defer resp.Body.Close()
|
||||||
b, _ := ioutil.ReadAll(resp.Body)
|
b, _ := ioutil.ReadAll(resp.Body)
|
||||||
log.Printf("fetch ntg job info %+v: %d: %s", request, resp.StatusCode, b)
|
logtr.Debugf("fetch ntg job info %+v: %d: %s", request, resp.StatusCode, b)
|
||||||
if resp.StatusCode > 400 && resp.StatusCode < 500 && resp.StatusCode != 404 && resp.StatusCode != 410 {
|
return io.NopCloser(bytes.NewReader(b)), nil
|
||||||
return ntgVisionJobInfo{}, ErrNoAuth
|
|
||||||
}
|
|
||||||
var result ntgVisionJobInfo
|
|
||||||
err = json.Unmarshal(b, &result)
|
|
||||||
return result, err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ntgJob *ntgVisionJob) Job() Job {
|
func (ntgJob *ntgVisionJob) Job() Job {
|
||||||
@@ -148,13 +143,14 @@ func (ntgJob *ntgVisionJob) Job() Job {
|
|||||||
Miles: ntgJob.Miles,
|
Miles: ntgJob.Miles,
|
||||||
Weight: ntgJob.Weight,
|
Weight: ntgJob.Weight,
|
||||||
Meta: fmt.Sprintf("equipment:%s", ntgJob.Equipment),
|
Meta: fmt.Sprintf("equipment:%s", ntgJob.Equipment),
|
||||||
secrets: func() interface{} {
|
secrets: func(j *Job) {
|
||||||
jobInfo, err := ntgJob.JobInfo()
|
jobInfo, err := ntgJob.JobInfo()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("failed to get jobinfo: %v", err)
|
logtr.Errorf("failed to get jobinfo: %v", err)
|
||||||
return nil
|
return
|
||||||
}
|
}
|
||||||
return jobInfo.String()
|
j.Meta = jobInfo.String()
|
||||||
|
j.Pays = fmt.Sprint(jobInfo.PayUpTo)
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -162,6 +158,9 @@ func (ntgJob *ntgVisionJob) Job() Job {
|
|||||||
func NewNTGVision() NTGVision {
|
func NewNTGVision() NTGVision {
|
||||||
ntgv := NTGVision{}
|
ntgv := NTGVision{}
|
||||||
ntgv.searcher = ntgv
|
ntgv.searcher = ntgv
|
||||||
|
if config.Get().Brokers.NTG.Mock {
|
||||||
|
ntgv = ntgv.WithMock()
|
||||||
|
}
|
||||||
return ntgv
|
return ntgv
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -170,6 +169,24 @@ func (ntg NTGVision) WithMock() NTGVision {
|
|||||||
return ntg
|
return ntg
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (ntg NTGVision) SearchJob(id int64) (ntgVisionJobInfo, error) {
|
||||||
|
rc, err := ntg.searcher.searchJob(id)
|
||||||
|
if err != nil {
|
||||||
|
return ntgVisionJobInfo{}, err
|
||||||
|
}
|
||||||
|
defer rc.Close()
|
||||||
|
b, err := ioutil.ReadAll(rc)
|
||||||
|
if err != nil {
|
||||||
|
return ntgVisionJobInfo{}, fmt.Errorf("failed to readall search job result: %w", err)
|
||||||
|
}
|
||||||
|
var result ntgVisionJobInfo
|
||||||
|
err = json.Unmarshal(b, &result)
|
||||||
|
if err != nil {
|
||||||
|
err = fmt.Errorf("failed to parse job info: %w: %s", err, b)
|
||||||
|
}
|
||||||
|
return result, err
|
||||||
|
}
|
||||||
|
|
||||||
func (ntg NTGVision) Search(states []config.State) ([]Job, error) {
|
func (ntg NTGVision) Search(states []config.State) ([]Job, error) {
|
||||||
rc, err := ntg.searcher.search(states)
|
rc, err := ntg.searcher.search(states)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -182,7 +199,7 @@ func (ntg NTGVision) Search(states []config.State) ([]Job, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Printf("ntg search for %+v: %s", states, b)
|
logtr.Debugf("ntg search for %+v: %s", states, b)
|
||||||
|
|
||||||
var ntgjobs []ntgVisionJob
|
var ntgjobs []ntgVisionJob
|
||||||
err = json.Unmarshal(b, &ntgjobs)
|
err = json.Unmarshal(b, &ntgjobs)
|
||||||
@@ -226,7 +243,15 @@ func (ntg NTGVision) search(states []config.State) (io.ReadCloser, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (ntg NTGVision) refreshAuth() error {
|
func (ntg NTGVision) refreshAuth() error {
|
||||||
log.Printf("refreshing ntg auth...")
|
err := ntg._refreshAuth()
|
||||||
|
if err != nil {
|
||||||
|
logtr.SOSf("failed to refresh ntg auth: %v", err)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ntg NTGVision) _refreshAuth() error {
|
||||||
|
logtr.Infof("refreshing ntg auth...")
|
||||||
b, _ := json.Marshal(map[string]string{
|
b, _ := json.Marshal(map[string]string{
|
||||||
"username": config.Get().Brokers.NTG.Username,
|
"username": config.Get().Brokers.NTG.Username,
|
||||||
"password": config.Get().Brokers.NTG.Password,
|
"password": config.Get().Brokers.NTG.Password,
|
||||||
@@ -280,7 +305,7 @@ func (ntg NTGVision) _search(states []config.State) (io.ReadCloser, error) {
|
|||||||
|
|
||||||
func (ntg NTGVision) newRequest(states []config.State) (*http.Request, error) {
|
func (ntg NTGVision) newRequest(states []config.State) (*http.Request, error) {
|
||||||
body, err := json.Marshal(map[string]interface{}{
|
body, err := json.Marshal(map[string]interface{}{
|
||||||
"OriginFromDate": time.Now().UTC().Format("2006-01-02T15:04:05.000Z"),
|
"OriginFromDate": time.Now().Add(time.Hour * -24).UTC().Format("2006-01-02T15:04:05.000Z"),
|
||||||
"OriginToDate": time.Now().UTC().Add(time.Hour * 24 * 30).Format("2006-01-02T15:04:05.000Z"),
|
"OriginToDate": time.Now().UTC().Add(time.Hour * 24 * 30).Format("2006-01-02T15:04:05.000Z"),
|
||||||
"DestinationFromDate": nil,
|
"DestinationFromDate": nil,
|
||||||
"DestinationToDate": nil,
|
"DestinationToDate": nil,
|
||||||
|
|||||||
@@ -20,3 +20,9 @@ func (ntgm NTGVisionMock) search(states []config.State) (io.ReadCloser, error) {
|
|||||||
b, err := ioutil.ReadFile(path)
|
b, err := ioutil.ReadFile(path)
|
||||||
return io.NopCloser(bytes.NewReader(b)), err
|
return io.NopCloser(bytes.NewReader(b)), err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (ntgm NTGVisionMock) searchJob(id int64) (io.ReadCloser, error) {
|
||||||
|
path := path.Join(os.Getenv("GOPATH"), "src", "local", "truckstop", "broker", "testdata", "ntgvision_jobinfo_response.json")
|
||||||
|
b, err := ioutil.ReadFile(path)
|
||||||
|
return io.NopCloser(bytes.NewReader(b)), err
|
||||||
|
}
|
||||||
|
|||||||
84
broker/testdata/ntgvision_jobinfo_response.json
vendored
Normal file
84
broker/testdata/ntgvision_jobinfo_response.json
vendored
Normal file
@@ -0,0 +1,84 @@
|
|||||||
|
{
|
||||||
|
"headerSummary": "A to B",
|
||||||
|
"loadId": 123,
|
||||||
|
"miles": 123,
|
||||||
|
"weight": 123,
|
||||||
|
"temp": null,
|
||||||
|
"equipment": "equipment",
|
||||||
|
"categoryName": "category",
|
||||||
|
"categoryLabel": "category generic",
|
||||||
|
"cargoInformation": [
|
||||||
|
"carto info"
|
||||||
|
],
|
||||||
|
"loadRequirements": [
|
||||||
|
"load requirements"
|
||||||
|
],
|
||||||
|
"stopInfos": [
|
||||||
|
{
|
||||||
|
"loadInfoIds": [
|
||||||
|
123
|
||||||
|
],
|
||||||
|
"stopDateTime": "0001-01-01T00:00:00",
|
||||||
|
"isPickup": true,
|
||||||
|
"location": "A",
|
||||||
|
"stopDate": "Friday, 01/14/22",
|
||||||
|
"stopHours": "10:00 - 12:00",
|
||||||
|
"appointmentTime": "",
|
||||||
|
"appointmentType": "FCFS",
|
||||||
|
"instructions": "instruction",
|
||||||
|
"isDropTrailer": false,
|
||||||
|
"shipperName": "client",
|
||||||
|
"processedStopDate": ""
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"loadInfoIds": [
|
||||||
|
123
|
||||||
|
],
|
||||||
|
"stopDateTime": "0001-01-01T00:00:00",
|
||||||
|
"isPickup": false,
|
||||||
|
"location": "B",
|
||||||
|
"stopDate": "Monday, 01/17/22",
|
||||||
|
"stopHours": "09:00 - 10:00",
|
||||||
|
"appointmentTime": "09:00",
|
||||||
|
"appointmentType": "APPT",
|
||||||
|
"instructions": "instructions",
|
||||||
|
"isDropTrailer": false,
|
||||||
|
"shipperName": "client",
|
||||||
|
"processedStopDate": ""
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"drayStopInfos": null,
|
||||||
|
"rateInfo": null,
|
||||||
|
"isDray": false,
|
||||||
|
"equipmentGroupId": 8,
|
||||||
|
"totalCarrierRate": 2600,
|
||||||
|
"payUpTo": 2700,
|
||||||
|
"firstLoadInfoId": 123,
|
||||||
|
"loadStatus": "ACTIVE",
|
||||||
|
"hasBlockingAlert": false,
|
||||||
|
"customerLoadBlocked": false,
|
||||||
|
"canSeeRate": false,
|
||||||
|
"canBookNow": false,
|
||||||
|
"canBidNow": true,
|
||||||
|
"buttonData": {
|
||||||
|
"useBidDialog": false,
|
||||||
|
"lastBidAmount": null,
|
||||||
|
"remainingBids": 1,
|
||||||
|
"carrierPhoneNumber": null,
|
||||||
|
"carrierPhoneExtension": null
|
||||||
|
},
|
||||||
|
"stopData": [
|
||||||
|
{
|
||||||
|
"addr": "IDX NORTH CAROLINA",
|
||||||
|
"city": "Washington",
|
||||||
|
"state": "NC",
|
||||||
|
"zip": "27889"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"addr": "hibbett sports",
|
||||||
|
"city": "Abilene",
|
||||||
|
"state": "TX",
|
||||||
|
"zip": "79606"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
21
broker/testdata/ntgvision_response.json
vendored
21
broker/testdata/ntgvision_response.json
vendored
@@ -1,21 +1,4 @@
|
|||||||
[
|
[
|
||||||
{
|
|
||||||
"id": 4650337,
|
|
||||||
"sDate": "01/12/22",
|
|
||||||
"sCity": "Columbus",
|
|
||||||
"sState": "OH",
|
|
||||||
"sdh": null,
|
|
||||||
"cDate": "01/13/22",
|
|
||||||
"cCity": "Jamaica",
|
|
||||||
"cState": "NY",
|
|
||||||
"cdh": null,
|
|
||||||
"stopCnt": 2,
|
|
||||||
"miles": 578,
|
|
||||||
"weight": 5000,
|
|
||||||
"equip": "Str Truck W/ Lift Gate",
|
|
||||||
"temp": "",
|
|
||||||
"alertReasons": []
|
|
||||||
},
|
|
||||||
{
|
{
|
||||||
"id": 4650338,
|
"id": 4650338,
|
||||||
"sDate": "01/12/22",
|
"sDate": "01/12/22",
|
||||||
@@ -23,8 +6,8 @@
|
|||||||
"sState": "NC",
|
"sState": "NC",
|
||||||
"sdh": null,
|
"sdh": null,
|
||||||
"cDate": "01/13/22",
|
"cDate": "01/13/22",
|
||||||
"cCity": "Atlanta",
|
"cCity": "Winston-Salem",
|
||||||
"cState": "GA",
|
"cState": "NC",
|
||||||
"cdh": null,
|
"cdh": null,
|
||||||
"stopCnt": 2,
|
"stopCnt": 2,
|
||||||
"miles": 378,
|
"miles": 378,
|
||||||
|
|||||||
53
config.json
53
config.json
@@ -1,4 +1,17 @@
|
|||||||
{
|
{
|
||||||
|
"Log": {
|
||||||
|
"Path": "/tmp/truckstop.log",
|
||||||
|
"Level": "debug",
|
||||||
|
"SOSMatrix": {
|
||||||
|
"ReceiveEnabled": true,
|
||||||
|
"Mock": false,
|
||||||
|
"Homeserver": "https://m.bltrucks.top",
|
||||||
|
"Username": "@bot.m.bltrucks.top",
|
||||||
|
"Token": "mvDWB96KXMF8XhOam8EC5XVdQvSEw0CDeClcSWocBcYkwZX3FPNWZ5uOnQk2EmT1cjpzfeuD7gDYPPjOuyZlI3bE9TE35UjNOlZgi0Tugm25s91iVsbIF6kMZsCIhVMSmEf6w3jxX6wQYOWvmDZ4mu6f5c8wr221EMDcOpEzQV09d1zuBSWgKLBgjqAkYHJZ5dTRIWpEDpPgujhOFZa2ld1HiAOxrJKlIrlfDBN0CUsTlGOGplujDAr4VtpFzNRS",
|
||||||
|
"Device": "TGNIOGKATZ",
|
||||||
|
"Room": "!OYZqtInrBCn1cyz90D:m.bltrucks.top"
|
||||||
|
}
|
||||||
|
},
|
||||||
"Interval": {
|
"Interval": {
|
||||||
"Input": "5s..10s",
|
"Input": "5s..10s",
|
||||||
"OK": "6h0m0s..6h0m0s",
|
"OK": "6h0m0s..6h0m0s",
|
||||||
@@ -17,41 +30,28 @@
|
|||||||
"UploadMethod": "POST"
|
"UploadMethod": "POST"
|
||||||
},
|
},
|
||||||
"Maps": {
|
"Maps": {
|
||||||
"DirectionsURIFormat": "https://maps.googleapis.com/maps/api/directions/json?origin=%s\u0026destination=%s\u0026mode=driving\u0026key=AIzaSyBkACm-LQkoSfsTO5_XAzBVZE9-JQzcNkg",
|
|
||||||
"PathedURIFormat": "https://maps.googleapis.com/maps/api/staticmap?size=250x250\u0026path=%s\u0026format=jpeg\u0026maptype=roadmap\u0026key=AIzaSyBkACm-LQkoSfsTO5_XAzBVZE9-JQzcNkg\u0026size=250x250\u0026markers=%s|%s\u0026zoom=5",
|
|
||||||
"URIFormat": "https://maps.googleapis.com/maps/api/staticmap?center=%s\u0026markers=label=A|%s\u0026zoom=5\u0026size=250x250\u0026scale=1\u0026format=jpeg\u0026maptype=roadmap\u0026key=AIzaSyBkACm-LQkoSfsTO5_XAzBVZE9-JQzcNkg",
|
"URIFormat": "https://maps.googleapis.com/maps/api/staticmap?center=%s\u0026markers=label=A|%s\u0026zoom=5\u0026size=250x250\u0026scale=1\u0026format=jpeg\u0026maptype=roadmap\u0026key=AIzaSyBkACm-LQkoSfsTO5_XAzBVZE9-JQzcNkg",
|
||||||
"Pathed": false,
|
|
||||||
"Pickup": false,
|
"Pickup": false,
|
||||||
"Dropoff": false
|
"Dropoff": false,
|
||||||
|
"Pathed": {
|
||||||
|
"Enabled": true,
|
||||||
|
"DirectionsURIFormat": "https://maps.googleapis.com/maps/api/directions/json?origin=%s\u0026destination=%s\u0026mode=driving\u0026key=AIzaSyBkACm-LQkoSfsTO5_XAzBVZE9-JQzcNkg",
|
||||||
|
"PathedURIFormat": "https://maps.googleapis.com/maps/api/staticmap?size=250x250\u0026path=%s\u0026format=jpeg\u0026maptype=roadmap\u0026key=AIzaSyBkACm-LQkoSfsTO5_XAzBVZE9-JQzcNkg\u0026size=250x250\u0026markers=%s|%s",
|
||||||
|
"Zoom": {
|
||||||
|
"AcceptableLatLngDelta": 2,
|
||||||
|
"Override": 5
|
||||||
|
}
|
||||||
|
}
|
||||||
},
|
},
|
||||||
"Clients": {
|
"Clients": {
|
||||||
"bel": {
|
"bel": {
|
||||||
"States": [
|
"States": [
|
||||||
"IL"
|
"NC"
|
||||||
],
|
],
|
||||||
"IDs": {
|
"IDs": {
|
||||||
"Matrix": "@bel:m.bltrucks.top"
|
"Matrix": "@bel:m.bltrucks.top"
|
||||||
},
|
},
|
||||||
"Available": 1512328400
|
"Available": 1512328400
|
||||||
},
|
|
||||||
"broc": {
|
|
||||||
"States": [
|
|
||||||
"FL",
|
|
||||||
"NC"
|
|
||||||
],
|
|
||||||
"IDs": {
|
|
||||||
"Matrix": "@broc:m.bltrucks.top"
|
|
||||||
},
|
|
||||||
"Available": 5642452800
|
|
||||||
},
|
|
||||||
"pa": {
|
|
||||||
"States": [
|
|
||||||
"NC"
|
|
||||||
],
|
|
||||||
"IDs": {
|
|
||||||
"Matrix": "@ron:m.bltrucks.top"
|
|
||||||
},
|
|
||||||
"Available": -62135596800
|
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"Storage": [
|
"Storage": [
|
||||||
@@ -75,8 +75,9 @@
|
|||||||
"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",
|
||||||
"Username": "noeasyrunstrucking@gmail.com",
|
"Username": "noeasyrunstrucking@gmail.com",
|
||||||
"Password": "thumper123"
|
"Password": "thumper1234"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,12 +4,28 @@ import (
|
|||||||
"encoding/json"
|
"encoding/json"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"local/storage"
|
"local/storage"
|
||||||
|
"local/truckstop/logtr"
|
||||||
"os"
|
"os"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type Matrix struct {
|
||||||
|
ReceiveEnabled bool
|
||||||
|
Mock bool
|
||||||
|
Homeserver string
|
||||||
|
Username string
|
||||||
|
Token string
|
||||||
|
Device string
|
||||||
|
Room string
|
||||||
|
}
|
||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
|
Log struct {
|
||||||
|
Path string
|
||||||
|
Level logtr.Level
|
||||||
|
SOSMatrix Matrix
|
||||||
|
}
|
||||||
Interval struct {
|
Interval struct {
|
||||||
Input Duration
|
Input Duration
|
||||||
OK Duration
|
OK Duration
|
||||||
@@ -28,34 +44,33 @@ type Config struct {
|
|||||||
UploadMethod string
|
UploadMethod string
|
||||||
}
|
}
|
||||||
Maps struct {
|
Maps struct {
|
||||||
DirectionsURIFormat string
|
URIFormat string
|
||||||
PathedURIFormat string
|
Pickup bool
|
||||||
URIFormat string
|
Dropoff bool
|
||||||
Pathed bool
|
Pathed struct {
|
||||||
Pickup bool
|
Enabled bool
|
||||||
Dropoff bool
|
DirectionsURIFormat string
|
||||||
|
PathedURIFormat string
|
||||||
|
Zoom struct {
|
||||||
|
AcceptableLatLngDelta float32
|
||||||
|
Override int
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
Clients map[string]Client
|
Clients map[string]Client
|
||||||
Storage []string
|
Storage []string
|
||||||
Message struct {
|
Message struct {
|
||||||
Matrix struct {
|
Matrix Matrix
|
||||||
ReceiveEnabled bool
|
|
||||||
Mock bool
|
|
||||||
Homeserver string
|
|
||||||
Username string
|
|
||||||
Token string
|
|
||||||
Device string
|
|
||||||
Room string
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
Once bool
|
Once bool
|
||||||
Brokers struct {
|
Brokers struct {
|
||||||
NTG struct {
|
NTG struct {
|
||||||
JobInfo bool
|
JobInfo bool
|
||||||
Mock bool
|
Mock bool
|
||||||
LoadPageURIFormat string
|
LoadPageURIFormat string
|
||||||
Username string
|
LoadPageAPIURIFormat string
|
||||||
Password string
|
Username string
|
||||||
|
Password string
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -106,7 +121,7 @@ func AllStates() []State {
|
|||||||
return states
|
return states
|
||||||
}
|
}
|
||||||
|
|
||||||
func Refresh() error {
|
func Refresh(soser func() logtr.SOSer) error {
|
||||||
b, err := ioutil.ReadFile(configPath())
|
b, err := ioutil.ReadFile(configPath())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@@ -115,10 +130,15 @@ func Refresh() error {
|
|||||||
if err := json.Unmarshal(b, &c); err != nil {
|
if err := json.Unmarshal(b, &c); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
logtr.SetLogpath(c.Log.Path)
|
||||||
|
logtr.SetLevel(c.Log.Level)
|
||||||
if live.db != nil {
|
if live.db != nil {
|
||||||
live.db.Close()
|
live.db.Close()
|
||||||
}
|
}
|
||||||
live = c
|
live = c
|
||||||
|
if soser != nil {
|
||||||
|
logtr.SetSOSer(soser())
|
||||||
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -132,7 +152,7 @@ func Set(other Config) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
ioutil.WriteFile(configPath(), b, os.ModePerm)
|
ioutil.WriteFile(configPath(), b, os.ModePerm)
|
||||||
Refresh()
|
Refresh(nil)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) DB() storage.DB {
|
func (c *Config) DB() storage.DB {
|
||||||
|
|||||||
1
go.mod
1
go.mod
@@ -30,6 +30,7 @@ require (
|
|||||||
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
|
||||||
|
|||||||
138
logtr/log.go
Normal file
138
logtr/log.go
Normal file
@@ -0,0 +1,138 @@
|
|||||||
|
package logtr
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Level int
|
||||||
|
|
||||||
|
const (
|
||||||
|
VERBOSE = Level(7)
|
||||||
|
DEBUG = Level(8)
|
||||||
|
INFO = Level(9)
|
||||||
|
ERROR = Level(10)
|
||||||
|
SOS = Level(15)
|
||||||
|
)
|
||||||
|
|
||||||
|
type SOSer interface {
|
||||||
|
Send(string) error
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l Level) MarshalJSON() ([]byte, error) {
|
||||||
|
return json.Marshal(l.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Level) UnmarshalJSON(b []byte) error {
|
||||||
|
var s string
|
||||||
|
if err := json.Unmarshal(b, &s); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
s = strings.TrimSpace(s)
|
||||||
|
s = strings.ToUpper(s)
|
||||||
|
if len(s) > 3 {
|
||||||
|
s = s[:3]
|
||||||
|
}
|
||||||
|
for i := 0; i < int(SOS)+5; i++ {
|
||||||
|
l2 := Level(i)
|
||||||
|
if l2.String() == s {
|
||||||
|
*l = l2
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return errors.New("unknown log level: " + s)
|
||||||
|
}
|
||||||
|
|
||||||
|
var logger io.Writer = os.Stderr
|
||||||
|
var loggerPath string = ""
|
||||||
|
var lock = &sync.Mutex{}
|
||||||
|
var level Level = INFO
|
||||||
|
var ansoser SOSer = nil
|
||||||
|
|
||||||
|
func SetLogpath(p string) {
|
||||||
|
lock.Lock()
|
||||||
|
defer lock.Unlock()
|
||||||
|
if p == loggerPath {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
f, err := os.OpenFile(p, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
logger = f
|
||||||
|
loggerPath = p
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetSOSer(another SOSer) {
|
||||||
|
lock.Lock()
|
||||||
|
defer lock.Unlock()
|
||||||
|
ansoser = another
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetLevel(l Level) {
|
||||||
|
lock.Lock()
|
||||||
|
defer lock.Unlock()
|
||||||
|
level = l
|
||||||
|
}
|
||||||
|
|
||||||
|
func logf(l Level, format string, args []interface{}) {
|
||||||
|
format = fmt.Sprintf("%v: %v: %s\n", time.Now().Format("15:04:05"), l.String(), strings.TrimSpace(format))
|
||||||
|
cLevel := level
|
||||||
|
cAnsoser := ansoser
|
||||||
|
if l >= cLevel {
|
||||||
|
fmt.Fprintf(os.Stderr, format, args...)
|
||||||
|
}
|
||||||
|
fmt.Fprintf(logger, format, args...)
|
||||||
|
if l == SOS && cAnsoser != nil {
|
||||||
|
if err := cAnsoser.Send(fmt.Sprintf(format, args...)); err != nil {
|
||||||
|
Errorf("failed to SOS: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SOSf(format string, args ...interface{}) {
|
||||||
|
logf(SOS, format, args)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Infof(format string, args ...interface{}) {
|
||||||
|
logf(INFO, format, args)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Printf(format string, args ...interface{}) {
|
||||||
|
Infof(format, args...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Debugf(format string, args ...interface{}) {
|
||||||
|
logf(DEBUG, format, args)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Verbosef(format string, args ...interface{}) {
|
||||||
|
logf(VERBOSE, format, args)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Errorf(format string, args ...interface{}) {
|
||||||
|
logf(ERROR, format, args)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l Level) String() string {
|
||||||
|
switch l {
|
||||||
|
case ERROR:
|
||||||
|
return "ERR"
|
||||||
|
case INFO:
|
||||||
|
return "INF"
|
||||||
|
case DEBUG:
|
||||||
|
return "DEB"
|
||||||
|
case VERBOSE:
|
||||||
|
return "VER"
|
||||||
|
case SOS:
|
||||||
|
return "SOS"
|
||||||
|
default:
|
||||||
|
return "?"
|
||||||
|
}
|
||||||
|
}
|
||||||
199
main.go
199
main.go
@@ -7,8 +7,8 @@ import (
|
|||||||
"local/storage"
|
"local/storage"
|
||||||
"local/truckstop/broker"
|
"local/truckstop/broker"
|
||||||
"local/truckstop/config"
|
"local/truckstop/config"
|
||||||
|
"local/truckstop/logtr"
|
||||||
"local/truckstop/message"
|
"local/truckstop/message"
|
||||||
"log"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
"regexp"
|
"regexp"
|
||||||
@@ -21,11 +21,12 @@ import (
|
|||||||
var stateFinder = regexp.MustCompile(`[A-Za-z]+`)
|
var stateFinder = regexp.MustCompile(`[A-Za-z]+`)
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
if err := config.Refresh(); err != nil {
|
if err := config.Refresh(message.NewSOSMatrix); err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
if config.Get().Message.Matrix.ReceiveEnabled {
|
if config.Get().Message.Matrix.ReceiveEnabled {
|
||||||
if err := matrixrecv(); err != nil {
|
if err := matrixrecv(); err != nil {
|
||||||
|
logtr.SOSf("failed to recv matrix on boot: %v", err)
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -33,13 +34,13 @@ func main() {
|
|||||||
go func() {
|
go func() {
|
||||||
for {
|
for {
|
||||||
time.Sleep(config.Get().Interval.Input.Get())
|
time.Sleep(config.Get().Interval.Input.Get())
|
||||||
if err := config.Refresh(); err != nil {
|
if err := config.Refresh(message.NewSOSMatrix); err != nil {
|
||||||
log.Println(err)
|
logtr.Errorf("failed parsing config: %v", err)
|
||||||
} else {
|
} else {
|
||||||
if config.Get().Message.Matrix.ReceiveEnabled {
|
if config.Get().Message.Matrix.ReceiveEnabled {
|
||||||
lock.Lock()
|
lock.Lock()
|
||||||
if err := matrixrecv(); err != nil {
|
if err := matrixrecv(); err != nil {
|
||||||
log.Print(err)
|
logtr.Errorf("failed receiving and parsing matrix: %v", err)
|
||||||
}
|
}
|
||||||
lock.Unlock()
|
lock.Unlock()
|
||||||
}
|
}
|
||||||
@@ -47,21 +48,22 @@ func main() {
|
|||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
if err := _main(); err != nil {
|
if err := _main(); err != nil {
|
||||||
|
logtr.SOSf("failed _main: %v", err)
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
lock.Lock()
|
lock.Lock()
|
||||||
}
|
}
|
||||||
|
|
||||||
func matrixrecv() error {
|
func matrixrecv() error {
|
||||||
log.Printf("checking matrix...")
|
logtr.Debugf("checking matrix...")
|
||||||
defer log.Printf("/checking matrix...")
|
defer logtr.Debugf("/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() {
|
||||||
log.Printf("looking for help")
|
logtr.Debugf("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") {
|
||||||
@@ -71,26 +73,26 @@ func matrixrecv() error {
|
|||||||
db := config.Get().DB()
|
db := config.Get().DB()
|
||||||
if !printed {
|
if !printed {
|
||||||
if _, err := db.Get(key); err == storage.ErrNotFound {
|
if _, err := db.Get(key); err == storage.ErrNotFound {
|
||||||
log.Printf("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`")
|
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`")
|
||||||
if err := sender.Send(help); err != nil {
|
if err := sender.Send(help); err != nil {
|
||||||
log.Printf("failed to send help: %v", err)
|
logtr.Errorf("failed to send help: %v", err)
|
||||||
} else {
|
} else {
|
||||||
printed = true
|
printed = true
|
||||||
if err := db.Set(key, []byte{'k'}); err != nil {
|
if err := db.Set(key, []byte{'k'}); err != nil {
|
||||||
log.Printf("failed to mark help given @%s: %v", key, err)
|
logtr.Errorf("failed to mark help given @%s: %v", key, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if err := db.Set(key, []byte{'k'}); err != nil {
|
if err := db.Set(key, []byte{'k'}); err != nil {
|
||||||
log.Printf("failed to mark help given @%s: %v", key, err)
|
logtr.Errorf("failed to mark help given @%s: %v", key, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
func() {
|
func() {
|
||||||
log.Printf("looking for states")
|
logtr.Debugf("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 {
|
||||||
@@ -108,13 +110,13 @@ func matrixrecv() error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
if err := db.Set(key, []byte{'k'}); err != nil {
|
if err := db.Set(key, []byte{'k'}); err != nil {
|
||||||
log.Printf("failed to mark state gathered @%s: %v", key, err)
|
logtr.Errorf("failed to mark state gathered @%s: %v", key, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
setNewStates(states)
|
setNewStates(states)
|
||||||
}()
|
}()
|
||||||
func() {
|
func() {
|
||||||
log.Printf("looking for pauses")
|
logtr.Debugf("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 {
|
||||||
@@ -136,7 +138,7 @@ func matrixrecv() error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
if err := db.Set(key, []byte{'k'}); err != nil {
|
if err := db.Set(key, []byte{'k'}); err != nil {
|
||||||
log.Printf("failed to mark state gathered @%s: %v", key, err)
|
logtr.Errorf("failed to mark state gathered @%s: %v", key, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
setNewPauses(pauses)
|
setNewPauses(pauses)
|
||||||
@@ -149,7 +151,7 @@ func setNewPauses(pauses map[string]time.Time) {
|
|||||||
if len(pauses) == 0 {
|
if len(pauses) == 0 {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
log.Printf("set new pauses: %+v", pauses)
|
logtr.Debugf("set new pauses: %+v", pauses)
|
||||||
conf := *config.Get()
|
conf := *config.Get()
|
||||||
changed := map[string]time.Time{}
|
changed := map[string]time.Time{}
|
||||||
for client, pause := range pauses {
|
for client, pause := range pauses {
|
||||||
@@ -164,11 +166,11 @@ func setNewPauses(pauses map[string]time.Time) {
|
|||||||
if len(changed) == 0 {
|
if len(changed) == 0 {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
log.Printf("updating config new pauses: %+v", conf)
|
logtr.Infof("updating config new pauses: %+v", conf)
|
||||||
config.Set(conf)
|
config.Set(conf)
|
||||||
for client, pause := range changed {
|
for client, pause := range changed {
|
||||||
if err := sendNewPause(client, pause); err != nil {
|
if err := sendNewPause(client, pause); err != nil {
|
||||||
log.Printf("failed to send new pause %s/%+v: %v", client, pause, err)
|
logtr.Errorf("failed to send new pause %s/%+v: %v", client, pause, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -199,11 +201,11 @@ func setNewStates(states map[string]map[config.State]struct{}) {
|
|||||||
if len(changed) == 0 {
|
if len(changed) == 0 {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
log.Printf("updating config new states: %+v", conf)
|
logtr.Infof("updating config new states: %+v", conf)
|
||||||
config.Set(conf)
|
config.Set(conf)
|
||||||
for client, states := range changed {
|
for client, states := range changed {
|
||||||
if err := sendNewStates(client, states); err != nil {
|
if err := sendNewStates(client, states); err != nil {
|
||||||
log.Printf("failed to send new states %s/%+v: %v", client, states, err)
|
logtr.Errorf("failed to send new states %s/%+v: %v", client, states, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -230,10 +232,10 @@ func _main() error {
|
|||||||
for {
|
for {
|
||||||
err := _mainOne()
|
err := _mainOne()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println(err)
|
logtr.Errorf("failed _main: %v", err)
|
||||||
}
|
}
|
||||||
if config.Get().Once {
|
if config.Get().Once {
|
||||||
time.Sleep(time.Second)
|
time.Sleep(3 * time.Second)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -246,15 +248,17 @@ func _main() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func _mainOne() error {
|
func _mainOne() error {
|
||||||
log.Println("config.refreshing...")
|
logtr.Debugf("config.refreshing...")
|
||||||
if err := config.Refresh(); err != nil {
|
if err := config.Refresh(message.NewSOSMatrix); err != nil {
|
||||||
|
logtr.SOSf("bad config: %v", err)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
log.Println("once...")
|
logtr.Debugf("once...")
|
||||||
if err := once(); err != nil {
|
if err := once(); err != nil {
|
||||||
|
logtr.SOSf("failed once(): %v", err)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
log.Println("/_mainOne")
|
logtr.Debugf("/_mainOne")
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -263,27 +267,37 @@ func once() error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
log.Printf("once: all jobs: %+v", alljobs)
|
logtr.Debugf("once: update dead jobs: %+v", alljobs)
|
||||||
|
err = updateDeadJobs(alljobs)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
logtr.Debugf("once: all jobs: %+v", alljobs)
|
||||||
newjobs, err := dropStaleJobs(alljobs)
|
newjobs, err := dropStaleJobs(alljobs)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
log.Printf("once: new jobs: %+v", newjobs)
|
logtr.Debugf("once: new jobs: %+v", newjobs)
|
||||||
jobs, err := dropBanlistJobs(newjobs)
|
jobs, err := dropBanlistJobs(newjobs)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
log.Printf("once: loading job secrets: %+v", jobs)
|
logtr.Debugf("found jobs: %+v", jobs)
|
||||||
|
if len(jobs) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
logtr.Debugf("once: loading job secrets: %+v", jobs)
|
||||||
for i := range jobs {
|
for i := range jobs {
|
||||||
jobs[i].Secrets()
|
jobs[i].Secrets()
|
||||||
}
|
}
|
||||||
log.Printf("once: sending jobs: %+v", jobs)
|
logtr.Infof("once: sending jobs: %+v", jobs)
|
||||||
|
db := config.Get().DB()
|
||||||
for i := range jobs {
|
for i := range jobs {
|
||||||
if ok, err := sendJob(jobs[i]); err != nil {
|
if ok, err := sendJob(jobs[i]); err != nil {
|
||||||
return err
|
return err
|
||||||
} else if ok {
|
} else if ok {
|
||||||
log.Println("sent job", jobs[i])
|
logtr.Debugf("sent job", jobs[i])
|
||||||
if err := config.Get().DB().Set(jobs[i].ID, []byte(`sent`)); err != nil {
|
if err := db.Set(jobs[i].UID(), []byte(`sent`)); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -294,9 +308,6 @@ func once() error {
|
|||||||
func getJobs() ([]broker.Job, error) {
|
func getJobs() ([]broker.Job, error) {
|
||||||
states := config.AllStates()
|
states := config.AllStates()
|
||||||
ntg := broker.NewNTGVision()
|
ntg := broker.NewNTGVision()
|
||||||
if config.Get().Brokers.NTG.Mock {
|
|
||||||
ntg = ntg.WithMock()
|
|
||||||
}
|
|
||||||
brokers := []broker.Broker{ntg}
|
brokers := []broker.Broker{ntg}
|
||||||
jobs := []broker.Job{}
|
jobs := []broker.Job{}
|
||||||
for _, broker := range brokers {
|
for _, broker := range brokers {
|
||||||
@@ -309,10 +320,58 @@ func getJobs() ([]broker.Job, error) {
|
|||||||
return jobs, nil
|
return jobs, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type recordedJob struct {
|
||||||
|
Job broker.Job
|
||||||
|
SentNS int64
|
||||||
|
MatrixID string
|
||||||
|
MatrixImageIDs []string
|
||||||
|
}
|
||||||
|
|
||||||
|
func updateDeadJobs(jobs []broker.Job) error {
|
||||||
|
db := config.Get().DB()
|
||||||
|
list, err := db.List([]string{}, "sent_job_", "sent_job_}}")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, listEntry := range list {
|
||||||
|
wouldBe := strings.TrimPrefix(listEntry, "sent_job_")
|
||||||
|
found := false
|
||||||
|
for i := range jobs {
|
||||||
|
found = found || jobs[i].ID == wouldBe
|
||||||
|
}
|
||||||
|
logtr.Debugf("found job %s to be still alive==%v", wouldBe, found)
|
||||||
|
if !found {
|
||||||
|
logtr.Debugf("updating dead job %+v", listEntry)
|
||||||
|
b, err := db.Get(listEntry)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
var recorded recordedJob
|
||||||
|
if err := json.Unmarshal(b, &recorded); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
/* // TODO this beeps on fluffychat
|
||||||
|
if err := message.NewMatrix().Update(recorded.MatrixID, recorded.Job.FormatMultilineTextDead()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
if err := db.Set(listEntry, nil); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, imageid := range recorded.MatrixImageIDs {
|
||||||
|
if err := message.NewMatrix().Remove(imageid); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func dropStaleJobs(jobs []broker.Job) ([]broker.Job, error) {
|
func dropStaleJobs(jobs []broker.Job) ([]broker.Job, error) {
|
||||||
db := config.Get().DB()
|
db := config.Get().DB()
|
||||||
for i := len(jobs) - 1; i >= 0; i-- {
|
for i := len(jobs) - 1; i >= 0; i-- {
|
||||||
if _, err := db.Get(jobs[i].ID); err == storage.ErrNotFound {
|
if _, err := db.Get(jobs[i].UID()); err == storage.ErrNotFound {
|
||||||
} else if err != nil {
|
} else if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
} else {
|
} else {
|
||||||
@@ -330,18 +389,36 @@ func dropBanlistJobs(jobs []broker.Job) ([]broker.Job, error) {
|
|||||||
func sendJob(job broker.Job) (bool, error) {
|
func sendJob(job broker.Job) (bool, error) {
|
||||||
sender := message.NewMatrix()
|
sender := message.NewMatrix()
|
||||||
payload := job.FormatMultilineText()
|
payload := job.FormatMultilineText()
|
||||||
log.Printf("once: send job %s if nonzero: %s", job.String(), payload)
|
logtr.Debugf("once: send job %s if nonzero: %s", job.String(), payload)
|
||||||
if len(payload) == 0 {
|
if len(payload) == 0 {
|
||||||
return false, nil
|
return false, nil
|
||||||
}
|
}
|
||||||
if err := sender.Send(payload); err != nil {
|
id, err := sender.SendTracked(payload)
|
||||||
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
|
recordedJob := recordedJob{
|
||||||
|
Job: job,
|
||||||
|
SentNS: time.Now().UnixNano(),
|
||||||
|
MatrixID: id,
|
||||||
|
}
|
||||||
|
defer func() {
|
||||||
|
db := config.Get().DB()
|
||||||
|
b, err := json.Marshal(recordedJob)
|
||||||
|
if err != nil {
|
||||||
|
logtr.Errorf("failed to marshal recorded job: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := db.Set("sent_job_"+job.ID, b); err != nil {
|
||||||
|
logtr.Errorf("failed to set recorded job: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}()
|
||||||
maps := config.Get().Maps
|
maps := config.Get().Maps
|
||||||
pickup := fmt.Sprintf("%s,%s", url.QueryEscape(job.Pickup.City), job.Pickup.State)
|
pickup := fmt.Sprintf("%s,%s", url.QueryEscape(job.Pickup.City), job.Pickup.State)
|
||||||
dropoff := fmt.Sprintf("%s,%s", url.QueryEscape(job.Dropoff.City), job.Dropoff.State)
|
dropoff := fmt.Sprintf("%s,%s", url.QueryEscape(job.Dropoff.City), job.Dropoff.State)
|
||||||
if maps.Pathed {
|
if maps.Pathed.Enabled {
|
||||||
directionsURI := fmt.Sprintf(maps.DirectionsURIFormat, pickup, dropoff)
|
directionsURI := fmt.Sprintf(maps.Pathed.DirectionsURIFormat, pickup, dropoff)
|
||||||
resp, err := http.Get(directionsURI)
|
resp, err := http.Get(directionsURI)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return true, err
|
return true, err
|
||||||
@@ -368,30 +445,56 @@ func sendJob(job broker.Job) (bool, error) {
|
|||||||
if len(directionsResp.Routes) < 1 || len(directionsResp.Routes[0].Legs) < 1 || len(directionsResp.Routes[0].Legs[0].Steps) < 1 {
|
if len(directionsResp.Routes) < 1 || len(directionsResp.Routes[0].Legs) < 1 || len(directionsResp.Routes[0].Legs[0].Steps) < 1 {
|
||||||
} else {
|
} else {
|
||||||
latLngPath := make([]string, 0)
|
latLngPath := make([]string, 0)
|
||||||
|
minLat := directionsResp.Routes[0].Legs[0].Steps[0].StartLocation.Lat
|
||||||
|
maxLat := directionsResp.Routes[0].Legs[0].Steps[0].StartLocation.Lat
|
||||||
|
minLng := directionsResp.Routes[0].Legs[0].Steps[0].StartLocation.Lng
|
||||||
|
maxLng := directionsResp.Routes[0].Legs[0].Steps[0].StartLocation.Lng
|
||||||
for _, v := range directionsResp.Routes[0].Legs[0].Steps {
|
for _, v := range directionsResp.Routes[0].Legs[0].Steps {
|
||||||
|
if v.StartLocation.Lng < minLng {
|
||||||
|
minLng = v.StartLocation.Lng
|
||||||
|
}
|
||||||
|
if v.StartLocation.Lng > maxLng {
|
||||||
|
maxLng = v.StartLocation.Lng
|
||||||
|
}
|
||||||
|
if v.StartLocation.Lat < minLat {
|
||||||
|
minLat = v.StartLocation.Lat
|
||||||
|
}
|
||||||
|
if v.StartLocation.Lat > maxLat {
|
||||||
|
maxLat = v.StartLocation.Lat
|
||||||
|
}
|
||||||
latLngPath = append(latLngPath, fmt.Sprintf("%.9f,%.9f", v.StartLocation.Lat, v.StartLocation.Lng))
|
latLngPath = append(latLngPath, fmt.Sprintf("%.9f,%.9f", v.StartLocation.Lat, v.StartLocation.Lng))
|
||||||
}
|
}
|
||||||
pathQuery := strings.Join(latLngPath, "|")
|
pathQuery := strings.Join(latLngPath, "|")
|
||||||
uri := fmt.Sprintf(maps.PathedURIFormat, pathQuery, pickup, dropoff)
|
uri := fmt.Sprintf(maps.Pathed.PathedURIFormat, pathQuery, pickup, dropoff)
|
||||||
log.Printf("sending pathed image: %s", uri)
|
// if the bigger delta is <acceptable, override zoom
|
||||||
if err := sender.SendImage(uri); err != nil {
|
if maxLat-minLat <= maps.Pathed.Zoom.AcceptableLatLngDelta && maxLng-minLng <= maps.Pathed.Zoom.AcceptableLatLngDelta {
|
||||||
|
uri = fmt.Sprintf("%s&zoom=%d", uri, maps.Pathed.Zoom.Override)
|
||||||
|
}
|
||||||
|
logtr.Debugf("sending pathed image: %s", uri)
|
||||||
|
pathedid, err := sender.SendImageTracked(uri)
|
||||||
|
if err != nil {
|
||||||
return true, err
|
return true, err
|
||||||
}
|
}
|
||||||
|
recordedJob.MatrixImageIDs = append(recordedJob.MatrixImageIDs, pathedid)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if maps.Pickup {
|
if maps.Pickup {
|
||||||
uri := fmt.Sprintf(maps.URIFormat, pickup, pickup)
|
uri := fmt.Sprintf(maps.URIFormat, pickup, pickup)
|
||||||
log.Printf("sending pickup image: %s", uri)
|
logtr.Debugf("sending pickup image: %s", uri)
|
||||||
if err := sender.SendImage(uri); err != nil {
|
pickupid, err := sender.SendImageTracked(uri)
|
||||||
|
if err != nil {
|
||||||
return true, err
|
return true, err
|
||||||
}
|
}
|
||||||
|
recordedJob.MatrixImageIDs = append(recordedJob.MatrixImageIDs, pickupid)
|
||||||
}
|
}
|
||||||
if maps.Dropoff {
|
if maps.Dropoff {
|
||||||
uri := fmt.Sprintf(maps.URIFormat, dropoff, dropoff)
|
uri := fmt.Sprintf(maps.URIFormat, dropoff, dropoff)
|
||||||
log.Printf("sending dropoff image: %s", uri)
|
logtr.Debugf("sending dropoff image: %s", uri)
|
||||||
if err := sender.SendImage(uri); err != nil {
|
dropid, err := sender.SendImageTracked(uri)
|
||||||
|
if err != nil {
|
||||||
return true, err
|
return true, err
|
||||||
}
|
}
|
||||||
|
recordedJob.MatrixImageIDs = append(recordedJob.MatrixImageIDs, dropid)
|
||||||
}
|
}
|
||||||
return true, nil
|
return true, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"local/truckstop/config"
|
"local/truckstop/config"
|
||||||
"log"
|
"local/truckstop/logtr"
|
||||||
"mime/multipart"
|
"mime/multipart"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
@@ -40,7 +40,7 @@ func uploadImage(b []byte) (string, error) {
|
|||||||
} else if s, ok := u.Query()["name"]; !ok {
|
} else if s, ok := u.Query()["name"]; !ok {
|
||||||
} else {
|
} else {
|
||||||
name = s[0]
|
name = s[0]
|
||||||
log.Printf("found name in upload uri: %s", name)
|
logtr.Debugf("found name in upload uri: %s", name)
|
||||||
}
|
}
|
||||||
part, err := writer.CreateFormFile("image", name)
|
part, err := writer.CreateFormFile("image", name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ func TestImageUpload(t *testing.T) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
if err := config.Refresh(); err != nil {
|
if err := config.Refresh(nil); err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
got, err := UploadImage(b)
|
got, err := UploadImage(b)
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"local/truckstop/config"
|
"local/truckstop/config"
|
||||||
"log"
|
"local/truckstop/logtr"
|
||||||
"net/http"
|
"net/http"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
@@ -23,15 +23,24 @@ type Matrix struct {
|
|||||||
continuation string
|
continuation string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func NewSOSMatrix() logtr.SOSer {
|
||||||
|
conf := config.Get().Log.SOSMatrix
|
||||||
|
return newMatrix(conf, "0")
|
||||||
|
}
|
||||||
|
|
||||||
func NewMatrix() Matrix {
|
func NewMatrix() Matrix {
|
||||||
conf := config.Get().Message.Matrix
|
conf := config.Get().Message.Matrix
|
||||||
|
return newMatrix(conf, GetMatrixContinuation())
|
||||||
|
}
|
||||||
|
|
||||||
|
func newMatrix(conf config.Matrix, cont string) Matrix {
|
||||||
return Matrix{
|
return Matrix{
|
||||||
homeserver: conf.Homeserver,
|
homeserver: conf.Homeserver,
|
||||||
username: conf.Username,
|
username: conf.Username,
|
||||||
token: conf.Token,
|
token: conf.Token,
|
||||||
room: conf.Room,
|
room: conf.Room,
|
||||||
mock: conf.Mock,
|
mock: conf.Mock,
|
||||||
continuation: GetMatrixContinuation(),
|
continuation: cont,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -45,10 +54,10 @@ func (m Matrix) Continuation() string {
|
|||||||
|
|
||||||
func (m *Matrix) Receive() ([]Message, error) {
|
func (m *Matrix) Receive() ([]Message, error) {
|
||||||
if m.mock {
|
if m.mock {
|
||||||
log.Printf("matrix.Receive()")
|
logtr.Infof("matrix.Receive()")
|
||||||
messages := make([]Message, 0)
|
messages := make([]Message, 0)
|
||||||
for k := range config.Get().Clients {
|
for k := range config.Get().Clients {
|
||||||
messages = append(messages, Message{Timestamp: time.Now(), Sender: k, Content: "!state OH"})
|
messages = append(messages, Message{Timestamp: time.Now(), Sender: k, Content: "!state nc"})
|
||||||
if k == "bel" {
|
if k == "bel" {
|
||||||
messages = append(messages, Message{Timestamp: time.Now(), Sender: k, Content: "!help"})
|
messages = append(messages, Message{Timestamp: time.Now(), Sender: k, Content: "!help"})
|
||||||
}
|
}
|
||||||
@@ -75,10 +84,10 @@ func (m *Matrix) Receive() ([]Message, error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
log.Printf("%s => {Start:%s End:%v};; %v, (%d)", m.Continuation(), result.Start, result.End, err, len(result.Chunk))
|
logtr.Debugf("%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 {
|
||||||
//log.Printf("%+v", event)
|
logtr.Verbosef("matrix event: %+v", event)
|
||||||
if _, ok := matrixIDs[event.Sender]; !ok {
|
if _, ok := matrixIDs[event.Sender]; !ok {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@@ -91,7 +100,7 @@ func (m *Matrix) Receive() ([]Message, error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
clientChange := regexp.MustCompile("@[a-z]+$")
|
clientChange := regexp.MustCompile("@[a-z]+$")
|
||||||
log.Printf("rewriting messages based on @abc")
|
logtr.Debugf("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, ""))
|
||||||
@@ -105,7 +114,7 @@ func (m *Matrix) Receive() ([]Message, error) {
|
|||||||
}
|
}
|
||||||
messages[i].Content = strings.TrimSpace(messages[i].Content)
|
messages[i].Content = strings.TrimSpace(messages[i].Content)
|
||||||
}
|
}
|
||||||
log.Printf("rewriting messages based on ! CoMmAnD ...")
|
logtr.Debugf("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:])
|
||||||
@@ -116,50 +125,114 @@ func (m *Matrix) Receive() ([]Message, error) {
|
|||||||
return messages, nil
|
return messages, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m Matrix) Send(text string) error {
|
func (m Matrix) Remove(id string) error {
|
||||||
if m.mock {
|
if m.mock {
|
||||||
log.Printf("matrix.Send(%s)", text)
|
logtr.Infof("matrix.Remove(%s)", id)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
c, err := m.getclient()
|
c, err := m.getclient()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
_, err = c.SendText(m.room, text)
|
_, err = c.RedactEvent(m.room, id, &gomatrix.ReqRedact{Reason: "stale"})
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m Matrix) SendImage(uri string) error {
|
func (m Matrix) Update(id, text string) error {
|
||||||
if m.mock {
|
if m.mock {
|
||||||
log.Printf("matrix.SendImage(%s)", uri)
|
logtr.Infof("matrix.Update(%s)", text)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
c, err := m.getclient()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
type MRelatesTo struct {
|
||||||
|
EventID string `json:"event_id"`
|
||||||
|
RelType string `json:"rel_type"`
|
||||||
|
}
|
||||||
|
type NewContent struct {
|
||||||
|
Body string `json:"body"`
|
||||||
|
MsgType string `json:"msgtype"`
|
||||||
|
}
|
||||||
|
type RelatesToRoomMessage struct {
|
||||||
|
Body string `json:"body"`
|
||||||
|
MsgType string `json:"msgtype"`
|
||||||
|
MRelatesTo MRelatesTo `json:"m.relates_to"`
|
||||||
|
MNewContent NewContent `json:"m.new_content"`
|
||||||
|
}
|
||||||
|
_, err = c.SendMessageEvent(m.room, "m.room.message", RelatesToRoomMessage{
|
||||||
|
Body: "",
|
||||||
|
MsgType: "m.text",
|
||||||
|
MNewContent: NewContent{
|
||||||
|
Body: text,
|
||||||
|
MsgType: "m.text",
|
||||||
|
},
|
||||||
|
MRelatesTo: MRelatesTo{
|
||||||
|
EventID: id,
|
||||||
|
RelType: "m.replace",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m Matrix) Send(text string) error {
|
||||||
|
_, err := m.SendTracked(text)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m Matrix) SendTracked(text string) (string, error) {
|
||||||
|
if m.mock {
|
||||||
|
logtr.Infof("matrix.SendTracked(%s)", text)
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
c, err := m.getclient()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
resp, err := c.SendText(m.room, text)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return resp.EventID, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m Matrix) SendImage(uri string) error {
|
||||||
|
_, err := m.SendImageTracked(uri)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m Matrix) SendImageTracked(uri string) (string, error) {
|
||||||
|
if m.mock {
|
||||||
|
logtr.Infof("matrix.SendImage(%s)", uri)
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
response, err := http.Get(uri)
|
response, err := http.Get(uri)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return "", err
|
||||||
}
|
}
|
||||||
if response.StatusCode != http.StatusOK {
|
if response.StatusCode != http.StatusOK {
|
||||||
b, _ := ioutil.ReadAll(response.Body)
|
b, _ := ioutil.ReadAll(response.Body)
|
||||||
response.Body.Close()
|
response.Body.Close()
|
||||||
return fmt.Errorf("failed to get %s: (%d) %s", uri, response.StatusCode, b)
|
return "", fmt.Errorf("failed to get %s: (%d) %s", uri, response.StatusCode, b)
|
||||||
}
|
}
|
||||||
b, err := ioutil.ReadAll(response.Body)
|
b, err := ioutil.ReadAll(response.Body)
|
||||||
response.Body.Close()
|
response.Body.Close()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return "", err
|
||||||
}
|
}
|
||||||
c, err := m.getclient()
|
c, err := m.getclient()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return "", err
|
||||||
}
|
}
|
||||||
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)
|
||||||
log.Printf("sent image %s => %s: %+v", uri, publicURI, resp)
|
logtr.Debugf("sent image %s => %s: %+v", uri, publicURI, resp)
|
||||||
return err
|
return resp.EventID, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func SetMatrixContinuation(continuation string) {
|
func SetMatrixContinuation(continuation string) {
|
||||||
|
|||||||
@@ -1,57 +1,65 @@
|
|||||||
package message
|
package message
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"local/truckstop/config"
|
"local/truckstop/config"
|
||||||
"os"
|
"os"
|
||||||
|
"path"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestMatrixSend(t *testing.T) {
|
func TestMatrixSendDel(t *testing.T) {
|
||||||
if len(os.Getenv("INTEGRATION")) == 0 {
|
sender := testMatrix(t)
|
||||||
t.Skip("$INTEGRATION not set")
|
if id, err := sender.SendTracked("hello world from unittest"); err != nil {
|
||||||
}
|
|
||||||
var c config.Config
|
|
||||||
b, err := ioutil.ReadFile("../config.json")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
} else if err := sender.Remove(id); err != nil {
|
||||||
if err := json.Unmarshal(b, &c); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
var sender Sender = &Matrix{
|
|
||||||
homeserver: c.Message.Matrix.Homeserver,
|
|
||||||
username: c.Message.Matrix.Username,
|
|
||||||
token: c.Message.Matrix.Token,
|
|
||||||
room: c.Message.Matrix.Room,
|
|
||||||
}
|
|
||||||
if err := sender.Send("hello world from unittest"); err != nil {
|
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestMatrixReceive(t *testing.T) {
|
func TestMatrixUpdate(t *testing.T) {
|
||||||
if len(os.Getenv("INTEGRATION")) == 0 {
|
// 19:32:34: VER: matrix event: {StateKey:<nil> Sender:@bot:m.bltrucks.top Type:m.room.message Timestamp:1642471594729 ID:$n0ln9TFZrko_cBNFXeh8YyICZ3fFm17Jhz3bmZcqig4 RoomID:!OYZqtInrBCn1cyz90D:m.bltrucks.top Redacts: Unsigned:map[transaction_id:go1642471594571852423] Content:map[body:hello world from unittest format: formatted_body: msgtype:m.text] PrevContent:map[]}
|
||||||
t.Skip("$INTEGRATION not set")
|
// func (m Matrix) Update(id, text string) error {
|
||||||
}
|
m := testMatrix(t)
|
||||||
var c config.Config
|
uid := uuid.New().String()[:3]
|
||||||
b, err := ioutil.ReadFile("../config.json")
|
id, err := m.SendTracked("hello, heheheh from test matrix update " + uid)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
if err := json.Unmarshal(b, &c); err != nil {
|
err = m.Update(id, "heheheh i updated from test matrix update "+uid)
|
||||||
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
var sender Sender = &Matrix{
|
m.Receive()
|
||||||
homeserver: c.Message.Matrix.Homeserver,
|
}
|
||||||
username: c.Message.Matrix.Username,
|
|
||||||
token: c.Message.Matrix.Token,
|
func TestMatrixReceive(t *testing.T) {
|
||||||
room: c.Message.Matrix.Room,
|
sender := testMatrix(t)
|
||||||
}
|
|
||||||
if msgs, err := sender.Receive(); err != nil {
|
if msgs, err := sender.Receive(); err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
} else {
|
} else {
|
||||||
t.Logf("%+v", msgs)
|
t.Logf("%+v", msgs)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func testMatrix(t *testing.T) Matrix {
|
||||||
|
if len(os.Getenv("INTEGRATION")) == 0 {
|
||||||
|
t.Skip("$INTEGRATION not set")
|
||||||
|
}
|
||||||
|
d := t.TempDir()
|
||||||
|
f := path.Join(d, "config.test.json")
|
||||||
|
b, err := ioutil.ReadFile("../config.json")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if err := ioutil.WriteFile(f, b, os.ModePerm); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
os.Setenv("CONFIG", f)
|
||||||
|
if err := config.Refresh(nil); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
return NewMatrix()
|
||||||
|
}
|
||||||
|
|||||||
@@ -4,6 +4,8 @@ import "time"
|
|||||||
|
|
||||||
type Sender interface {
|
type Sender interface {
|
||||||
Send(string) error
|
Send(string) error
|
||||||
|
SendTracked(string) (string, error)
|
||||||
|
Update(string, string) (string, error)
|
||||||
Receive() ([]Message, error)
|
Receive() ([]Message, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
14
todo.yaml
14
todo.yaml
@@ -1,10 +1,11 @@
|
|||||||
todo:
|
todo:
|
||||||
- mark jobs no longer avail by modifying
|
- !states emits current state
|
||||||
- write-as for clients so ma can write to pa by default
|
- TEST. Just like, refactor and test to shit.
|
||||||
|
- try search ntg by autoinc?
|
||||||
|
- test each !command callbacks to matrix
|
||||||
|
- recv-as for clients so pa receives mas commands as writes
|
||||||
- continuation is garbo, but I can still do better client side to avoid get-set high level
|
- continuation is garbo, but I can still do better client side to avoid get-set high level
|
||||||
- no hard code jpeg or have it in multiple places
|
- no hard code jpeg or have it in multiple places
|
||||||
- TEST its falling apart
|
|
||||||
- test each !command callbacks to matrix
|
|
||||||
- change matrix so I test my custom logic even if I dont fetch remote
|
- change matrix so I test my custom logic even if I dont fetch remote
|
||||||
- warn/err/etc. on clobbering ids.matrix since clients can mess with one another
|
- warn/err/etc. on clobbering ids.matrix since clients can mess with one another
|
||||||
- modify old items once no longer available; drop stale jobs good candidate but requires new matrix interaction
|
- modify old items once no longer available; drop stale jobs good candidate but requires new matrix interaction
|
||||||
@@ -14,6 +15,11 @@ 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:
|
||||||
|
- mark jobs no longer avail by modifying in matrix;; save matrix ID over dummy payload
|
||||||
|
- TEST its falling apart
|
||||||
|
- help() log on truckstop for stuff like perma 403
|
||||||
|
- figure out zoom on maps;; is there like an auto-zoom I can leverage?
|
||||||
|
- mock is nigh useless
|
||||||
- mark consumed;; save scroll id for matrix
|
- mark consumed;; save scroll id for matrix
|
||||||
- todo: switch house to selfhosted for rate limit
|
- todo: switch house to selfhosted for rate limit
|
||||||
tags: now
|
tags: now
|
||||||
|
|||||||
Reference in New Issue
Block a user