Add command-line client

cmd/ofx/get_accounts.go

@@ -0,0 +1,70 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"github.com/aclindsa/ofxgo"
+	"os"
+	"time"
+)
+
+var getAccountsCommand = Command{
+	Name:        "get-accounts",
+	Description: "List accounts at your financial institution",
+	Flags:       flag.NewFlagSet("get-accounts", flag.ExitOnError),
+	CheckFlags:  checkServerFlags,
+	Do:          getAccounts,
+}
+
+func init() {
+	defineServerFlags(getAccountsCommand.Flags)
+}
+
+func getAccounts() {
+	client, query := NewRequest()
+
+	uid, err := ofxgo.RandomUID()
+	if err != nil {
+		fmt.Println("Error creating uid for transaction:", err)
+		os.Exit(1)
+	}
+
+	acctInfo := ofxgo.AcctInfoRequest{
+		TrnUID:    *uid,
+		DtAcctUp:  ofxgo.Date(time.Unix(0, 0)),
+		CltCookie: 1,
+	}
+	query.Signup = append(query.Signup, &acctInfo)
+
+	signupResponse, err := client.Request(query)
+	if err != nil {
+		fmt.Println("Error requesting account information:", err)
+		os.Exit(1)
+	}
+
+	if signupResponse.Signon.Status.Code != 0 {
+		fmt.Printf("Nonzero signon status with message: %s\n", signupResponse.Signon.Status.Message)
+		os.Exit(1)
+	}
+
+	if len(signupResponse.Signup) < 1 {
+		fmt.Println("No signup messages received")
+		return
+	}
+
+	fmt.Println("\nFound the following accounts:\n")
+
+	if acctinfo, ok := signupResponse.Signup[0].(ofxgo.AcctInfoResponse); ok {
+		for _, acct := range acctinfo.AcctInfo {
+			if acct.BankAcctInfo != nil {
+				fmt.Printf("Bank Account:\n\tBankId: \"%s\"\n\tAcctId: \"%s\"\n\tAcctType: %s\n", acct.BankAcctInfo.BankAcctFrom.BankId, acct.BankAcctInfo.BankAcctFrom.AcctId, acct.BankAcctInfo.BankAcctFrom.AcctType)
+			} else if acct.CCAcctInfo != nil {
+				fmt.Printf("Credit card:\n\tAcctId: \"%s\"\n", acct.CCAcctInfo.CCAcctFrom.AcctId)
+			} else if acct.InvAcctInfo != nil {
+				fmt.Printf("Investment account:\n\tBrokerId: \"%s\"\n\tAcctId: \"%s\"\n", acct.InvAcctInfo.InvAcctFrom.BrokerId, acct.InvAcctInfo.InvAcctFrom.AcctId)
+			} else {
+				fmt.Printf("Unknown type: %s %s\n", acct.Name, acct.Desc)
+			}
+		}
+	}
+}