Impl uccu

bank.go

@@ -5,6 +5,7 @@ type Bank int
 const (
 	Chase Bank = iota + 1
 	Citi  Bank = iota + 1
+	UCCU  Bank = iota + 1
 )
 
 func (b Bank) String() string {
@@ -13,6 +14,8 @@ func (b Bank) String() string {
 		return "Chase"
 	case Citi:
 		return "Citi"
+	case UCCU:
+		return "UCCU"
 	}
 	return "?"
 }

scrape.go

@@ -7,6 +7,7 @@ import (
 	"io/ioutil"
 	"net/mail"
 	"regexp"
+	"strconv"
 	"strings"
 )
 
@@ -16,6 +17,7 @@ type scraper interface {
 
 type chaseScraper struct{}
 type citiScraper struct{}
+type uccuScraper struct{}
 
 func Scrape(m *mail.Message) ([]*Transaction, error) {
 	scraper, err := buildScraper(m)
@@ -37,6 +39,9 @@ func buildScraper(m *mail.Message) (scraper, error) {
 	if strings.Contains(from, "Citi") {
 		return newCitiScraper(), nil
 	}
+	if strings.Contains(from, "Notifications@uccu.com") {
+		return newUCCUScraper(), nil
+	}
 	return nil, errors.New("unknown sender: " + from)
 }
 
@@ -44,6 +49,10 @@ func newChaseScraper() scraper {
 	return &chaseScraper{}
 }
 
+func newUCCUScraper() scraper {
+	return &uccuScraper{}
+}
+
 func newCitiScraper() scraper {
 	return &citiScraper{}
 }
@@ -84,6 +93,27 @@ func (c *chaseScraper) scrape(m *mail.Message) ([]*Transaction, error) {
 }
 
 func (c *citiScraper) scrape(m *mail.Message) ([]*Transaction, error) {
+	date := fmt.Sprint(m.Header["Date"])
+	b, err := ioutil.ReadAll(m.Body)
+	if err != nil {
+		return nil, err
+	}
+	re := regexp.MustCompile(`Citi Alert: A \$[0-9][0-9]*\.[0-9][0-9] transaction was made at .* on card ending in`)
+	match := re.Find(b)
+	if len(match) == 0 {
+		return nil, nil
+	}
+
+	rePrice := regexp.MustCompile(`[0-9][0-9]*\.[0-9][0-9]`)
+	price := rePrice.Find(match)
+
+	vendor := bytes.Split(bytes.Split(match, []byte(" on card ending in"))[0], []byte("transaction was made at "))[1]
+
+	transaction := NewTransaction(string(price), string(vendor), date, Citi)
+
+	return []*Transaction{transaction}, nil
+	//Citi Alert: A $598.14 transaction was made at REMIX MUSIC SPRINGDA on card ending in 3837
+	/*
 		b, err := ioutil.ReadAll(m.Body)
 		if err != nil {
 			return nil, err
@@ -121,4 +151,28 @@ func (c *citiScraper) scrape(m *mail.Message) ([]*Transaction, error) {
 		}
 
 		return transactions, nil
+	*/
+}
+
+func (c *uccuScraper) scrape(m *mail.Message) ([]*Transaction, error) {
+	b, err := ioutil.ReadAll(m.Body)
+	if err != nil {
+		return nil, err
+	}
+	regexp := regexp.MustCompile(`\$([0-9]+,?)+\.[0-9][0-9]`)
+	match := regexp.Find(b)
+	if len(match) == 0 {
+		return nil, fmt.Errorf("no matches found")
+	}
+	match = match[1:]
+	match = bytes.ReplaceAll(match, []byte(","), []byte{})
+	f, err := strconv.ParseFloat(string(match), 10)
+	if err != nil {
+		return nil, err
+	}
+	if !bytes.Contains(b, []byte("credit")) {
+		f *= -1.0
+	}
+	transaction := NewTransaction(fmt.Sprintf("%.2f", f), "?", fmt.Sprint(m.Header["Date"]), UCCU)
+	return []*Transaction{transaction}, nil
 }