mirror of
https://github.com/aclindsa/ofxgo.git
synced 2024-11-16 09:50:04 -05:00
Aaron Lindsay
f185d78d29
This removes the many decodeXXXMessageSet() functions and replaces them with a large map and a single generic decodeMessageSet() function. Also change Responses to satisfy the Message interface as pointer types (instead of the raw types), add the full set of top-level message sets (though most of them still lack any message-parsing ability), adjust the message set names to more closely mirror their OFX names, and fixup tests and the command-line client to match the above changes.
85 lines
1.8 KiB
Go
85 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"github.com/aclindsa/ofxgo"
|
|
"io"
|
|
"os"
|
|
)
|
|
|
|
var invDownloadCommand = Command{
|
|
Name: "download-inv",
|
|
Description: "Download a investment account statement to a file",
|
|
Flags: flag.NewFlagSet("download-inv", flag.ExitOnError),
|
|
CheckFlags: invDownloadCheckFlags,
|
|
Do: invDownload,
|
|
}
|
|
|
|
var brokerId string
|
|
|
|
func init() {
|
|
defineServerFlags(invDownloadCommand.Flags)
|
|
invDownloadCommand.Flags.StringVar(&filename, "filename", "./download.ofx", "The file to save to")
|
|
invDownloadCommand.Flags.StringVar(&acctId, "acctid", "", "AcctId (from `get-accounts` subcommand)")
|
|
invDownloadCommand.Flags.StringVar(&brokerId, "brokerid", "", "BrokerId (from `get-accounts` subcommand)")
|
|
}
|
|
|
|
func invDownloadCheckFlags() bool {
|
|
ret := checkServerFlags()
|
|
|
|
if len(filename) == 0 {
|
|
fmt.Println("Error: Filename empty")
|
|
return false
|
|
}
|
|
|
|
return ret
|
|
}
|
|
|
|
func invDownload() {
|
|
client, query := NewRequest()
|
|
|
|
uid, err := ofxgo.RandomUID()
|
|
if err != nil {
|
|
fmt.Println("Error creating uid for transaction:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
statementRequest := ofxgo.InvStatementRequest{
|
|
TrnUID: *uid,
|
|
InvAcctFrom: ofxgo.InvAcct{
|
|
BrokerId: ofxgo.String(brokerId),
|
|
AcctId: ofxgo.String(acctId),
|
|
},
|
|
Include: true,
|
|
IncludeOO: true,
|
|
IncludePos: true,
|
|
IncludeBalance: true,
|
|
Include401K: true,
|
|
Include401KBal: true,
|
|
}
|
|
query.InvStmt = append(query.InvStmt, &statementRequest)
|
|
|
|
response, err := client.RequestNoParse(query)
|
|
|
|
if err != nil {
|
|
fmt.Println("Error requesting account statement:", err)
|
|
|
|
os.Exit(1)
|
|
}
|
|
defer response.Body.Close()
|
|
|
|
file, err := os.Create(filename)
|
|
if err != nil {
|
|
fmt.Println("Error creating file to write to:", err)
|
|
os.Exit(1)
|
|
}
|
|
defer file.Close()
|
|
|
|
_, err = io.Copy(file, response.Body)
|
|
if err != nil {
|
|
fmt.Println("Error writing response to file:", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|