From f2a45dc6b60901227a5a1bc125a6e1b77f53e134 Mon Sep 17 00:00:00 2001
From: Aaron Lindsay <aaron@aclindsay.com>
Date: Sun, 5 Nov 2017 20:43:32 -0500
Subject: [PATCH] Make SecurityType its own type

---
 internal/handlers/ofx.go        |  2 +-
 internal/handlers/securities.go | 16 +++++++++-------
 2 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/internal/handlers/ofx.go b/internal/handlers/ofx.go
index aaf78d4..1acd120 100644
--- a/internal/handlers/ofx.go
+++ b/internal/handlers/ofx.go
@@ -22,7 +22,7 @@ func (i *OFXImport) GetSecurity(ofxsecurityid int64) (*Security, error) {
 	return &i.Securities[ofxsecurityid], nil
 }
 
-func (i *OFXImport) GetSecurityAlternateId(alternateid string, securityType int64) (*Security, error) {
+func (i *OFXImport) GetSecurityAlternateId(alternateid string, securityType SecurityType) (*Security, error) {
 	for _, security := range i.Securities {
 		if alternateid == security.AlternateId && securityType == security.Type {
 			return &security, nil
diff --git a/internal/handlers/securities.go b/internal/handlers/securities.go
index 1944a47..e1ff9bd 100644
--- a/internal/handlers/securities.go
+++ b/internal/handlers/securities.go
@@ -13,12 +13,14 @@ import (
 	"strings"
 )
 
+type SecurityType int64
+
 const (
-	Currency int64 = 1
-	Stock          = 2
+	Currency SecurityType = 1
+	Stock                 = 2
 )
 
-func GetSecurityType(typestring string) int64 {
+func GetSecurityType(typestring string) SecurityType {
 	if strings.EqualFold(typestring, "currency") {
 		return Currency
 	} else if strings.EqualFold(typestring, "stock") {
@@ -37,7 +39,7 @@ type Security struct {
 	// Number of decimal digits (to the right of the decimal point) this
 	// security is precise to
 	Precision int
-	Type      int64
+	Type      SecurityType
 	// AlternateId is CUSIP for Type=Stock, ISO4217 for Type=Currency
 	AlternateId string
 }
@@ -66,7 +68,7 @@ func (sl *SecurityList) Write(w http.ResponseWriter) error {
 	return enc.Encode(sl)
 }
 
-func SearchSecurityTemplates(search string, _type int64, limit int64) []*Security {
+func SearchSecurityTemplates(search string, _type SecurityType, limit int64) []*Security {
 	upperSearch := strings.ToUpper(search)
 	var results []*Security
 	for i, security := range SecurityTemplates {
@@ -84,7 +86,7 @@ func SearchSecurityTemplates(search string, _type int64, limit int64) []*Securit
 	return results
 }
 
-func FindSecurityTemplate(name string, _type int64) *Security {
+func FindSecurityTemplate(name string, _type SecurityType) *Security {
 	for _, security := range SecurityTemplates {
 		if name == security.Name && _type == security.Type {
 			return &security
@@ -349,7 +351,7 @@ func SecurityTemplateHandler(w http.ResponseWriter, r *http.Request) {
 		var limit int64 = -1
 		search := query.Get("search")
 
-		var _type int64 = 0
+		var _type SecurityType = 0
 		typestring := query.Get("type")
 		if len(typestring) > 0 {
 			_type = GetSecurityType(typestring)