From d7d523b0a7843e9f1a9421a5add8e4e84e1efe35 Mon Sep 17 00:00:00 2001
From: Bel LaPointe <breel@qualtrics.com>
Date: Thu, 27 Jan 2022 17:36:17 -0700
Subject: [PATCH] impl fastexact SearchZips and dedupe diff-id same-job for
 fastexact

---
 broker/broker.go         |  1 +
 broker/fastexact.go      | 64 +++++++++++++++++++++++++++++++++++-----
 broker/fastexact_test.go |  2 +-
 broker/ntgvision.go      |  4 +++
 4 files changed, 62 insertions(+), 9 deletions(-)

diff --git a/broker/broker.go b/broker/broker.go
index 430a06f..35d490c 100644
--- a/broker/broker.go
+++ b/broker/broker.go
@@ -21,6 +21,7 @@ var limiter = rate.NewLimiter(rate.Limit(1.0/20.0), 1)
 
 type Broker interface {
 	SearchStates([]config.State) ([]Job, error)
+	SearchZips([]string) ([]Job, error)
 }
 
 func do(r *http.Request) (*http.Response, error) {
diff --git a/broker/fastexact.go b/broker/fastexact.go
index 12961f9..25cf24c 100644
--- a/broker/fastexact.go
+++ b/broker/fastexact.go
@@ -11,6 +11,7 @@ import (
 	"local/truckstop/config"
 	"local/truckstop/logtr"
 	"net/http"
+	"sort"
 	"strconv"
 	"strings"
 	"time"
@@ -38,6 +39,17 @@ func (fe FastExact) WithMock() FastExact {
 	return fe
 }
 
+func (fe FastExact) SearchZips(zips []string) ([]Job, error) {
+	jobs, err := fe.searchZips(zips)
+	if err == ErrNoAuth {
+		if err := fe.login(); err != nil {
+			return nil, err
+		}
+		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 {
@@ -91,28 +103,47 @@ func (fe FastExact) setHeaders(req *http.Request) {
 	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
 }
 
-func (fe FastExact) searchStates(states []config.State) ([]Job, error) {
+func (fe FastExact) searchZips(zips []string) ([]Job, error) {
 	var jobs []Job
-	for _, state := range states {
-		subjobs, err := fe.searchOne(state)
+	for _, zip := range zips {
+		subjobs, err := fe.searchOneZip(zip)
 		if err != nil {
 			return nil, err
 		}
 		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{}
 	for _, job := range jobs {
-		dedupeJobs[job.UID()] = job
+		dedupeJobs[strings.ReplaceAll(job.UID(), job.ID, "")] = job
 	}
 	result := []Job{}
 	for _, job := range dedupeJobs {
 		result = append(result, job)
 	}
-	return result, nil
+	return result
 }
 
-func (fe FastExact) searchOne(state config.State) ([]Job, error) {
-	req, err := fe.newRequest(state)
+func (fe FastExact) searchOneZip(zip string) ([]Job, error) {
+	req, err := fe.newRequest(zip)
 	if err != nil {
 		return nil, err
 	}
@@ -124,11 +155,28 @@ func (fe FastExact) searchOne(state config.State) ([]Job, error) {
 	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
+	}
+	return fe.parse(resp)
+}
+
+func (fe FastExact) newRequestWithState(state config.State) (*http.Request, error) {
 	zip, ok := config.States[state]
 	if !ok {
 		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(
 		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",
diff --git a/broker/fastexact_test.go b/broker/fastexact_test.go
index 7d1dce4..6e16d9a 100644
--- a/broker/fastexact_test.go
+++ b/broker/fastexact_test.go
@@ -65,7 +65,7 @@ func TestFastExactSearchStates(t *testing.T) {
 	_ = db
 	if jobs, err := fe.searchStates([]config.State{config.State("NC"), config.State("SC")}); err != nil {
 		t.Fatal(err)
-	} else if len(jobs) != 10 {
+	} else if len(jobs) != 9 {
 		t.Fatal(len(jobs))
 	} else {
 		for _, job := range jobs {
diff --git a/broker/ntgvision.go b/broker/ntgvision.go
index 8abd732..f95f365 100644
--- a/broker/ntgvision.go
+++ b/broker/ntgvision.go
@@ -187,6 +187,10 @@ func (ntg NTGVision) searchJob(id int64) (ntgVisionJobInfo, error) {
 	return result, err
 }
 
+func (ntg NTGVision) SearchZips(zips []string) ([]Job, error) {
+	return nil, errors.New("not impl: ntg search zips")
+}
+
 func (ntg NTGVision) SearchStates(states []config.State) ([]Job, error) {
 	rc, err := ntg.searcher.searchStates(states)
 	if err != nil {