0000: 2f 2f 20 43 6f 70 79 72 69 67 68 74 20 28 43 29 // Copyright (C)
0010: 20 32 30 31 39 20 20 44 61 76 69 64 20 4e 2e 20 2019 David N.
0020: 43 6f 68 65 6e 0a 0a 2f 2f 20 54 68 69 73 20 70 Cohen..// This p
0030: 72 6f 67 72 61 6d 20 69 73 20 66 72 65 65 20 73 rogram is free s
0040: 6f 66 74 77 61 72 65 3a 20 79 6f 75 20 63 61 6e oftware: you can
0050: 20 72 65 64 69 73 74 72 69 62 75 74 65 20 69 74 redistribute it
0060: 20 61 6e 64 2f 6f 72 20 6d 6f 64 69 66 79 0a 2f and/or modify./
0070: 2f 20 69 74 20 75 6e 64 65 72 20 74 68 65 20 74 / it under the t
0080: 65 72 6d 73 20 6f 66 20 74 68 65 20 47 4e 55 20 erms of the GNU
0090: 41 66 66 65 72 6f 20 47 65 6e 65 72 61 6c 20 50 Affero General P
00a0: 75 62 6c 69 63 20 4c 69 63 65 6e 73 65 20 61 73 ublic License as
00b0: 20 70 75 62 6c 69 73 68 65 64 20 62 79 0a 2f 2f published by.//
00c0: 20 74 68 65 20 46 72 65 65 20 53 6f 66 74 77 61 the Free Softwa
00d0: 72 65 20 46 6f 75 6e 64 61 74 69 6f 6e 2c 20 65 re Foundation, e
00e0: 69 74 68 65 72 20 76 65 72 73 69 6f 6e 20 33 20 ither version 3
00f0: 6f 66 20 74 68 65 20 4c 69 63 65 6e 73 65 2c 20 of the License,
0100: 6f 72 0a 2f 2f 20 28 61 74 20 79 6f 75 72 20 6f or.// (at your o
0110: 70 74 69 6f 6e 29 20 61 6e 79 20 6c 61 74 65 72 ption) any later
0120: 20 76 65 72 73 69 6f 6e 2e 0a 0a 2f 2f 20 54 68 version...// Th
0130: 69 73 20 70 72 6f 67 72 61 6d 20 69 73 20 64 69 is program is di
0140: 73 74 72 69 62 75 74 65 64 20 69 6e 20 74 68 65 stributed in the
0150: 20 68 6f 70 65 20 74 68 61 74 20 69 74 20 77 69 hope that it wi
0160: 6c 6c 20 62 65 20 75 73 65 66 75 6c 2c 0a 2f 2f ll be useful,.//
0170: 20 62 75 74 20 57 49 54 48 4f 55 54 20 41 4e 59 but WITHOUT ANY
0180: 20 57 41 52 52 41 4e 54 59 3b 20 77 69 74 68 6f WARRANTY; witho
0190: 75 74 20 65 76 65 6e 20 74 68 65 20 69 6d 70 6c ut even the impl
01a0: 69 65 64 20 77 61 72 72 61 6e 74 79 20 6f 66 0a ied warranty of.
01b0: 2f 2f 20 4d 45 52 43 48 41 4e 54 41 42 49 4c 49 // MERCHANTABILI
01c0: 54 59 20 6f 72 20 46 49 54 4e 45 53 53 20 46 4f TY or FITNESS FO
01d0: 52 20 41 20 50 41 52 54 49 43 55 4c 41 52 20 50 R A PARTICULAR P
01e0: 55 52 50 4f 53 45 2e 20 20 53 65 65 20 74 68 65 URPOSE. See the
01f0: 0a 2f 2f 20 47 4e 55 20 41 66 66 65 72 6f 20 47 .// GNU Affero G
0200: 65 6e 65 72 61 6c 20 50 75 62 6c 69 63 20 4c 69 eneral Public Li
0210: 63 65 6e 73 65 20 66 6f 72 20 6d 6f 72 65 20 64 cense for more d
0220: 65 74 61 69 6c 73 2e 0a 0a 2f 2f 20 59 6f 75 20 etails...// You
0230: 73 68 6f 75 6c 64 20 68 61 76 65 20 72 65 63 65 should have rece
0240: 69 76 65 64 20 61 20 63 6f 70 79 20 6f 66 20 74 ived a copy of t
0250: 68 65 20 47 4e 55 20 41 66 66 65 72 6f 20 47 65 he GNU Affero Ge
0260: 6e 65 72 61 6c 20 50 75 62 6c 69 63 20 4c 69 63 neral Public Lic
0270: 65 6e 73 65 0a 2f 2f 20 61 6c 6f 6e 67 20 77 69 ense.// along wi
0280: 74 68 20 74 68 69 73 20 70 72 6f 67 72 61 6d 2e th this program.
0290: 20 20 49 66 20 6e 6f 74 2c 20 73 65 65 20 3c 68 If not, see <h
02a0: 74 74 70 73 3a 2f 2f 77 77 77 2e 67 6e 75 2e 6f ttps://www.gnu.o
02b0: 72 67 2f 6c 69 63 65 6e 73 65 73 2f 3e 2e 0a 0a rg/licenses/>...
02c0: 70 61 63 6b 61 67 65 20 6d 61 69 6e 0a 0a 69 6d package main..im
02d0: 70 6f 72 74 20 28 0a 09 22 65 72 72 6f 72 73 22 port (.."errors"
02e0: 0a 09 22 66 6d 74 22 0a 09 22 6d 61 74 68 2f 62 .."fmt".."math/b
02f0: 69 67 22 0a 09 22 73 74 72 69 6e 67 73 22 0a 09 ig".."strings"..
0300: 22 74 69 6d 65 22 0a 0a 09 22 73 72 63 2e 64 31 "time"..."src.d1
0310: 30 2e 64 65 76 2f 63 6f 6d 6d 61 6e 64 22 0a 29 0.dev/command".)
0320: 0a 0a 66 75 6e 63 20 69 6e 69 74 28 29 20 7b 0a ..func init() {.
0330: 09 63 6f 6d 6d 61 6e 64 2e 52 65 67 69 73 74 65 .command.Registe
0340: 72 4f 70 65 72 61 74 69 6f 6e 28 63 6f 6d 6d 61 rOperation(comma
0350: 6e 64 2e 4f 70 65 72 61 74 69 6f 6e 7b 0a 09 09 nd.Operation{...
0360: 48 61 6e 64 6c 65 72 3a 20 20 20 20 20 62 61 73 Handler: bas
0370: 65 4d 61 69 6e 2c 0a 09 09 4e 61 6d 65 3a 20 20 eMain,...Name:
0380: 20 20 20 20 20 20 22 62 61 73 65 22 2c 0a 09 09 "base",...
0390: 53 79 6e 74 61 78 3a 20 20 20 20 20 20 22 62 61 Syntax: "ba
03a0: 73 65 20 5b 2d 62 3d 3c 62 65 67 69 6e 20 64 61 se [-b=<begin da
03b0: 74 65 3e 5d 22 2c 0a 09 09 44 65 73 63 72 69 70 te>]",...Descrip
03c0: 74 69 6f 6e 3a 20 22 43 6f 6e 76 65 72 74 20 70 tion: "Convert p
03d0: 72 69 63 65 2f 63 6f 73 74 20 69 6e 66 6f 72 6d rice/cost inform
03e0: 61 74 69 6f 6e 20 74 6f 20 62 61 73 65 20 63 75 ation to base cu
03f0: 72 72 65 6e 63 79 20 28 75 73 69 6e 67 20 6c 65 rrency (using le
0400: 64 67 65 72 2d 63 6c 69 20 70 72 69 63 65 20 64 dger-cli price d
0410: 61 74 61 29 2e 22 2c 0a 09 7d 29 0a 7d 0a 0a 66 ata).",..}).}..f
0420: 75 6e 63 20 62 61 73 65 4d 61 69 6e 28 29 20 65 unc baseMain() e
0430: 72 72 6f 72 20 7b 0a 09 2f 2f 20 64 65 66 69 6e rror {..// defin
0440: 65 20 66 6c 61 67 73 0a 09 62 65 67 69 6e 46 6c e flags..beginFl
0450: 61 67 20 3a 3d 20 63 6f 6d 6d 61 6e 64 2e 4f 70 ag := command.Op
0460: 65 72 61 74 69 6f 6e 46 6c 61 67 53 65 74 2e 53 erationFlagSet.S
0470: 74 72 69 6e 67 28 22 62 22 2c 20 22 22 2c 20 22 tring("b", "", "
0480: 62 65 67 69 6e 20 64 61 74 65 22 29 0a 0a 09 65 begin date")...e
0490: 72 72 20 3a 3d 20 63 6f 6d 6d 61 6e 64 2e 4f 70 rr := command.Op
04a0: 65 72 61 74 69 6f 6e 46 6c 61 67 53 65 74 2e 50 erationFlagSet.P
04b0: 61 72 73 65 28 63 6f 6d 6d 61 6e 64 2e 41 72 67 arse(command.Arg
04c0: 73 28 29 5b 31 3a 5d 29 0a 09 69 66 20 65 72 72 s()[1:])..if err
04d0: 20 21 3d 20 6e 69 6c 20 7b 0a 09 09 72 65 74 75 != nil {...retu
04e0: 72 6e 20 65 72 72 0a 09 7d 0a 0a 09 2f 2f 20 76 rn err..}...// v
04f0: 61 6c 69 64 61 74 65 20 66 6c 61 67 73 0a 09 69 alidate flags..i
0500: 66 20 62 61 73 65 20 3d 3d 20 22 22 20 7b 0a 09 f base == "" {..
0510: 09 72 65 74 75 72 6e 20 65 72 72 6f 72 73 2e 4e .return errors.N
0520: 65 77 28 22 41 20 62 61 73 65 20 63 75 72 72 65 ew("A base curre
0530: 6e 63 79 20 69 73 20 72 65 71 75 69 72 65 64 2c ncy is required,
0540: 20 69 2e 65 2e 20 60 2d 62 61 73 65 3d 55 53 44 i.e. `-base=USD
0550: 60 2e 22 29 0a 09 7d 0a 0a 09 76 61 72 20 62 65 `.")..}...var be
0560: 67 69 6e 20 74 69 6d 65 2e 54 69 6d 65 0a 09 69 gin time.Time..i
0570: 66 20 2a 62 65 67 69 6e 46 6c 61 67 20 21 3d 20 f *beginFlag !=
0580: 22 22 20 7b 0a 09 09 62 65 67 69 6e 2c 20 65 72 "" {...begin, er
0590: 72 20 3d 20 74 69 6d 65 2e 50 61 72 73 65 28 22 r = time.Parse("
05a0: 32 30 30 36 2f 30 31 2f 30 32 22 2c 20 2a 62 65 2006/01/02", *be
05b0: 67 69 6e 46 6c 61 67 29 0a 09 09 69 66 20 65 72 ginFlag)...if er
05c0: 72 20 21 3d 20 6e 69 6c 20 7b 0a 09 09 09 63 6f r != nil {....co
05d0: 6d 6d 61 6e 64 2e 43 68 65 63 6b 28 66 6d 74 2e mmand.Check(fmt.
05e0: 45 72 72 6f 72 66 28 22 62 61 64 20 62 65 67 69 Errorf("bad begi
05f0: 6e 20 64 61 74 65 20 28 25 71 29 3a 20 25 77 22 n date (%q): %w"
0600: 2c 20 2a 62 65 67 69 6e 46 6c 61 67 2c 20 65 72 , *beginFlag, er
0610: 72 29 29 0a 09 09 7d 0a 09 7d 0a 0a 09 66 6f 72 r))...}..}...for
0620: 20 73 63 61 6e 6e 65 72 2e 53 63 61 6e 28 29 20 scanner.Scan()
0630: 7b 0a 09 09 74 78 4c 69 6e 65 73 20 3a 3d 20 73 {...txLines := s
0640: 63 61 6e 6e 65 72 2e 4c 69 6e 65 73 28 29 0a 0a canner.Lines()..
0650: 09 09 2f 2f 20 6f 62 73 65 72 76 65 20 70 72 69 ..// observe pri
0660: 63 65 20 69 6e 66 6f 72 6d 61 74 69 6f 6e 2c 20 ce information,
0670: 69 66 20 61 6e 79 0a 09 09 2f 2f 20 68 74 74 70 if any...// http
0680: 73 3a 2f 2f 77 77 77 2e 6c 65 64 67 65 72 2d 63 s://www.ledger-c
0690: 6c 69 2e 6f 72 67 2f 33 2e 30 2f 64 6f 63 2f 6c li.org/3.0/doc/l
06a0: 65 64 67 65 72 33 2e 68 74 6d 6c 23 43 6f 6d 6d edger3.html#Comm
06b0: 6f 64 69 74 79 2d 70 72 69 63 65 2d 68 69 73 74 odity-price-hist
06c0: 6f 72 69 65 73 0a 09 09 70 72 69 63 65 48 69 73 ories...priceHis
06d0: 74 6f 72 79 20 3a 3d 20 6d 61 6b 65 28 6d 61 70 tory := make(map
06e0: 5b 73 74 72 69 6e 67 5d 2a 62 69 67 2e 52 61 74 [string]*big.Rat
06f0: 29 0a 0a 09 09 66 6f 72 20 5f 2c 20 6c 69 6e 65 )....for _, line
0700: 20 3a 3d 20 72 61 6e 67 65 20 74 78 4c 69 6e 65 := range txLine
0710: 73 2e 4c 69 6e 65 20 7b 0a 09 09 09 2f 2f 20 77 s.Line {....// w
0720: 65 27 72 65 20 6c 6f 6f 6b 69 6e 67 20 66 6f 72 e're looking for
0730: 2c 20 69 2e 65 2e 20 22 50 20 32 30 30 34 2f 30 , i.e. "P 2004/0
0740: 36 2f 32 31 20 30 32 3a 31 37 3a 35 38 20 54 57 6/21 02:17:58 TW
0750: 43 55 58 20 32 37 2e 37 36 20 55 53 44 22 0a 09 CUX 27.76 USD"..
0760: 09 09 69 66 20 73 74 72 69 6e 67 73 2e 48 61 73 ..if strings.Has
0770: 50 72 65 66 69 78 28 6c 69 6e 65 2c 20 22 50 20 Prefix(line, "P
0780: 22 29 20 7b 0a 09 09 09 09 63 6f 6d 6d 61 6e 64 ") {.....command
0790: 2e 56 28 32 29 2e 49 6e 66 6f 28 22 5c 74 22 2c .V(2).Info("\t",
07a0: 20 6c 69 6e 65 29 20 2f 2f 20 64 65 62 75 67 0a line) // debug.
07b0: 0a 09 09 09 09 66 69 65 6c 64 20 3a 3d 20 73 74 .....field := st
07c0: 72 69 6e 67 73 2e 46 69 65 6c 64 73 28 6c 69 6e rings.Fields(lin
07d0: 65 29 0a 09 09 09 09 64 61 74 65 2c 20 65 72 72 e).....date, err
07e0: 20 3a 3d 20 74 69 6d 65 2e 50 61 72 73 65 28 22 := time.Parse("
07f0: 32 30 30 36 2f 30 31 2f 30 32 20 31 35 3a 30 34 2006/01/02 15:04
0800: 3a 30 35 22 2c 20 73 74 72 69 6e 67 73 2e 4a 6f :05", strings.Jo
0810: 69 6e 28 66 69 65 6c 64 5b 31 3a 33 5d 2c 20 22 in(field[1:3], "
0820: 20 22 29 29 0a 09 09 09 09 69 66 20 65 72 72 20 ")).....if err
0830: 21 3d 20 6e 69 6c 20 7b 0a 09 09 09 09 09 63 6f != nil {......co
0840: 6d 6d 61 6e 64 2e 43 68 65 63 6b 28 66 6d 74 2e mmand.Check(fmt.
0850: 45 72 72 6f 72 66 28 22 66 61 69 6c 65 64 20 74 Errorf("failed t
0860: 6f 20 70 61 72 73 65 20 68 69 73 74 6f 72 69 63 o parse historic
0870: 61 6c 20 70 72 69 63 65 20 28 25 71 29 3a 20 25 al price (%q): %
0880: 77 22 2c 20 6c 69 6e 65 2c 20 65 72 72 29 29 0a w", line, err)).
0890: 09 09 09 09 7d 0a 0a 09 09 09 09 69 66 20 66 69 ....}......if fi
08a0: 65 6c 64 5b 35 5d 20 21 3d 20 73 74 72 69 6e 67 eld[5] != string
08b0: 28 62 61 73 65 29 20 7b 0a 09 09 09 09 09 63 6f (base) {......co
08c0: 6d 6d 61 6e 64 2e 56 28 31 29 2e 49 6e 66 6f 66 mmand.V(1).Infof
08d0: 28 22 69 67 6e 6f 72 69 6e 67 20 6e 6f 6e 2d 62 ("ignoring non-b
08e0: 61 73 65 20 70 72 69 63 65 20 28 25 71 29 22 2c ase price (%q)",
08f0: 20 6c 69 6e 65 29 0a 09 09 09 09 09 63 6f 6e 74 line)......cont
0900: 69 6e 75 65 0a 09 09 09 09 7d 0a 0a 09 09 09 09 inue.....}......
0910: 70 72 69 63 65 2c 20 6f 6b 20 3a 3d 20 6e 65 77 price, ok := new
0920: 28 62 69 67 2e 52 61 74 29 2e 53 65 74 53 74 72 (big.Rat).SetStr
0930: 69 6e 67 28 66 69 65 6c 64 5b 34 5d 29 0a 09 09 ing(field[4])...
0940: 09 09 69 66 20 21 6f 6b 20 7b 0a 09 09 09 09 09 ..if !ok {......
0950: 63 6f 6d 6d 61 6e 64 2e 43 68 65 63 6b 28 66 6d command.Check(fm
0960: 74 2e 45 72 72 6f 72 66 28 22 66 61 69 6c 65 64 t.Errorf("failed
0970: 20 74 6f 20 70 61 72 73 65 20 68 69 73 74 6f 72 to parse histor
0980: 69 63 61 6c 20 70 72 69 63 65 20 28 25 71 29 22 ical price (%q)"
0990: 2c 20 6c 69 6e 65 29 29 0a 09 09 09 09 7d 0a 0a , line)).....}..
09a0: 09 09 09 09 6b 65 79 20 3a 3d 20 68 69 73 74 6f ....key := histo
09b0: 72 79 4b 65 79 28 64 61 74 65 2c 20 41 73 73 65 ryKey(date, Asse
09c0: 74 28 66 69 65 6c 64 5b 33 5d 29 29 0a 09 09 09 t(field[3]))....
09d0: 09 6f 6c 64 2c 20 6f 6b 20 3a 3d 20 70 72 69 63 .old, ok := pric
09e0: 65 48 69 73 74 6f 72 79 5b 6b 65 79 5d 0a 09 09 eHistory[key]...
09f0: 09 09 69 66 20 6f 6b 20 7b 0a 09 09 09 09 09 63 ..if ok {......c
0a00: 6f 6d 6d 61 6e 64 2e 56 28 31 29 2e 49 6e 66 6f ommand.V(1).Info
0a10: 66 28 22 75 70 64 61 74 69 6e 67 20 70 72 69 63 f("updating pric
0a20: 65 20 68 69 73 74 6f 72 79 20 28 77 61 73 20 25 e history (was %
0a30: 66 2c 20 6e 6f 77 20 25 66 29 5c 6e 5c 74 25 73 f, now %f)\n\t%s
0a40: 22 2c 20 6f 6c 64 2c 20 70 72 69 63 65 2c 20 6c ", old, price, l
0a50: 69 6e 65 29 0a 09 09 09 09 7d 0a 09 09 09 09 70 ine).....}.....p
0a60: 72 69 63 65 48 69 73 74 6f 72 79 5b 6b 65 79 5d riceHistory[key]
0a70: 20 3d 20 70 72 69 63 65 0a 09 09 09 7d 0a 09 09 = price....}...
0a80: 7d 20 2f 2f 20 65 6e 64 20 63 6f 6c 6c 65 63 74 } // end collect
0a90: 20 70 72 69 63 65 20 68 69 73 74 6f 72 79 0a 0a price history..
0aa0: 09 09 70 61 79 65 65 2c 20 70 61 79 65 65 49 6e ..payee, payeeIn
0ab0: 64 65 78 20 3a 3d 20 74 78 4c 69 6e 65 73 2e 50 dex := txLines.P
0ac0: 61 79 65 65 28 29 0a 09 09 69 66 20 70 61 79 65 ayee()...if paye
0ad0: 65 49 6e 64 65 78 20 3d 3d 20 50 61 79 65 65 4e eIndex == PayeeN
0ae0: 6f 74 46 6f 75 6e 64 20 7b 0a 09 09 09 2f 2f 20 otFound {....//
0af0: 6e 6f 74 20 61 20 74 72 61 6e 73 61 63 74 69 6f not a transactio
0b00: 6e 20 28 6d 61 79 62 65 20 61 20 63 6f 6d 6d 65 n (maybe a comme
0b10: 6e 74 29 0a 09 09 09 77 72 69 74 65 4c 69 6e 65 nt)....writeLine
0b20: 73 28 61 70 70 65 6e 64 28 74 78 4c 69 6e 65 73 s(append(txLines
0b30: 2e 4c 69 6e 65 2c 20 22 22 29 29 20 2f 2f 20 77 .Line, "")) // w
0b40: 69 74 68 20 61 20 62 6c 61 6e 6b 0a 09 09 09 63 ith a blank....c
0b50: 6f 6e 74 69 6e 75 65 0a 09 09 7d 0a 09 09 69 66 ontinue...}...if
0b60: 20 62 65 67 69 6e 2e 41 66 74 65 72 28 74 78 4c begin.After(txL
0b70: 69 6e 65 73 2e 44 61 74 65 29 20 7b 0a 09 09 09 ines.Date) {....
0b80: 77 72 69 74 65 4c 69 6e 65 73 28 61 70 70 65 6e writeLines(appen
0b90: 64 28 74 78 4c 69 6e 65 73 2e 4c 69 6e 65 2c 20 d(txLines.Line,
0ba0: 22 22 29 29 20 2f 2f 20 77 69 74 68 20 61 20 62 "")) // with a b
0bb0: 6c 61 6e 6b 0a 09 09 09 63 6f 6e 74 69 6e 75 65 lank....continue
0bc0: 0a 09 09 7d 0a 0a 09 09 63 6f 6d 6d 61 6e 64 2e ...}....command.
0bd0: 56 28 32 29 2e 49 6e 66 6f 28 22 5c 74 22 2c 20 V(2).Info("\t",
0be0: 70 61 79 65 65 29 20 2f 2f 20 64 65 62 75 67 0a payee) // debug.
0bf0: 0a 09 09 2f 2f 20 66 69 72 73 74 20 70 61 73 73 ...// first pass
0c00: 2c 20 66 69 6e 64 20 63 6f 6e 76 65 72 73 69 6f , find conversio
0c10: 6e 73 20 74 6f 20 62 61 73 65 0a 09 09 63 6f 6e ns to base...con
0c20: 76 65 72 73 69 6f 6e 20 3a 3d 20 6d 61 6b 65 28 version := make(
0c30: 6d 61 70 5b 73 74 72 69 6e 67 5d 41 6d 6f 75 6e map[string]Amoun
0c40: 74 29 0a 09 09 66 6f 72 20 5f 2c 20 6c 69 6e 65 t)...for _, line
0c50: 20 3a 3d 20 72 61 6e 67 65 20 74 78 4c 69 6e 65 := range txLine
0c60: 73 2e 4c 69 6e 65 5b 70 61 79 65 65 49 6e 64 65 s.Line[payeeInde
0c70: 78 2b 31 3a 5d 20 7b 0a 09 09 09 73 70 6c 69 74 x+1:] {....split
0c80: 2c 20 6f 6b 20 3a 3d 20 70 61 72 73 65 53 70 6c , ok := parseSpl
0c90: 69 74 28 6c 69 6e 65 29 0a 09 09 09 69 66 20 21 it(line)....if !
0ca0: 6f 6b 20 7b 0a 09 09 09 09 69 66 20 21 73 74 72 ok {.....if !str
0cb0: 69 6e 67 73 2e 48 61 73 50 72 65 66 69 78 28 73 ings.HasPrefix(s
0cc0: 74 72 69 6e 67 73 2e 54 72 69 6d 4c 65 66 74 28 trings.TrimLeft(
0cd0: 6c 69 6e 65 2c 20 22 20 5c 74 22 29 2c 20 22 3b line, " \t"), ";
0ce0: 22 29 20 7b 20 2f 2f 20 63 68 65 63 6b 20 63 6f ") { // check co
0cf0: 6d 6d 65 6e 74 0a 09 09 09 09 09 63 6f 6d 6d 61 mment......comma
0d00: 6e 64 2e 43 68 65 63 6b 28 66 6d 74 2e 45 72 72 nd.Check(fmt.Err
0d10: 6f 72 66 28 22 66 61 69 6c 65 64 20 74 6f 20 70 orf("failed to p
0d20: 61 72 73 65 20 74 72 61 6e 73 61 63 74 69 6f 6e arse transaction
0d30: 20 73 70 6c 69 74 3a 20 25 71 22 2c 20 6c 69 6e split: %q", lin
0d40: 65 29 29 0a 09 09 09 09 7d 0a 09 09 09 09 63 6f e)).....}.....co
0d50: 6e 74 69 6e 75 65 20 2f 2f 20 63 6f 6d 6d 65 6e ntinue // commen
0d60: 74 20 69 73 20 6e 6f 6f 70 0a 09 09 09 7d 0a 0a t is noop....}..
0d70: 09 09 09 69 66 20 73 70 6c 69 74 2e 63 6f 73 74 ...if split.cost
0d80: 20 3d 3d 20 6e 69 6c 20 26 26 20 73 70 6c 69 74 == nil && split
0d90: 2e 70 72 69 63 65 20 3d 3d 20 6e 69 6c 20 7b 0a .price == nil {.
0da0: 09 09 09 09 2f 2f 20 6e 6f 20 70 72 69 63 65 20 ....// no price
0db0: 6f 72 20 63 6f 73 74 20 74 6f 20 63 6f 6e 76 65 or cost to conve
0dc0: 72 74 0a 09 09 09 09 63 6f 6e 74 69 6e 75 65 0a rt.....continue.
0dd0: 09 09 09 7d 0a 0a 09 09 09 63 6f 73 74 20 3a 3d ...}.....cost :=
0de0: 20 73 70 6c 69 74 2e 43 6f 73 74 28 29 0a 09 09 split.Cost()...
0df0: 09 69 66 20 63 6f 73 74 20 3d 3d 20 6e 69 6c 20 .if cost == nil
0e00: 7c 7c 20 63 6f 73 74 2e 41 73 73 65 74 20 3d 3d || cost.Asset ==
0e10: 20 62 61 73 65 20 7b 0a 09 09 09 09 63 6f 6e 74 base {.....cont
0e20: 69 6e 75 65 0a 09 09 09 7d 0a 0a 09 09 09 2f 2f inue....}.....//
0e30: 20 68 65 72 65 20 77 65 20 68 61 76 65 20 61 20 here we have a
0e40: 63 6f 73 74 20 74 68 61 74 20 6d 75 73 74 20 62 cost that must b
0e50: 65 20 63 6f 6e 76 65 72 74 65 64 20 69 6e 74 6f e converted into
0e60: 20 62 61 73 65 20 63 75 72 72 65 6e 63 79 0a 0a base currency..
0e70: 09 09 09 6b 65 79 20 3a 3d 20 68 69 73 74 6f 72 ...key := histor
0e80: 79 4b 65 79 28 74 78 4c 69 6e 65 73 2e 44 61 74 yKey(txLines.Dat
0e90: 65 2c 20 63 6f 73 74 2e 41 73 73 65 74 29 0a 09 e, cost.Asset)..
0ea0: 09 09 70 72 69 63 65 2c 20 6f 6b 20 3a 3d 20 70 ..price, ok := p
0eb0: 72 69 63 65 48 69 73 74 6f 72 79 5b 6b 65 79 5d riceHistory[key]
0ec0: 0a 09 09 09 69 66 20 21 6f 6b 20 7b 0a 09 09 09 ....if !ok {....
0ed0: 09 2f 2f 20 73 74 72 69 63 74 0a 09 09 09 09 63 .// strict.....c
0ee0: 6f 6d 6d 61 6e 64 2e 43 68 65 63 6b 28 66 6d 74 ommand.Check(fmt
0ef0: 2e 45 72 72 6f 72 66 28 22 70 72 69 63 65 20 6f .Errorf("price o
0f00: 66 20 25 73 20 6f 6e 20 25 73 20 75 6e 6b 6e 6f f %s on %s unkno
0f10: 77 6e 22 2c 20 63 6f 73 74 2c 20 74 78 4c 69 6e wn", cost, txLin
0f20: 65 73 2e 44 61 74 65 2e 46 6f 72 6d 61 74 28 22 es.Date.Format("
0f30: 32 30 30 36 2f 30 31 2f 30 32 22 29 29 29 0a 09 2006/01/02")))..
0f40: 09 09 7d 0a 0a 09 09 09 74 6d 70 20 3a 3d 20 6e ..}.....tmp := n
0f50: 65 77 28 62 69 67 2e 52 61 74 29 2e 4d 75 6c 28 ew(big.Rat).Mul(
0f60: 70 72 69 63 65 2c 20 63 6f 73 74 2e 52 61 74 29 price, cost.Rat)
0f70: 0a 09 09 09 62 61 73 69 73 20 3a 3d 20 4e 65 77 ....basis := New
0f80: 41 6d 6f 75 6e 74 28 62 61 73 65 2c 20 2a 74 6d Amount(base, *tm
0f90: 70 29 0a 09 09 09 63 6f 6d 6d 61 6e 64 2e 56 28 p)....command.V(
0fa0: 31 29 2e 49 6e 66 6f 66 28 22 25 73 20 40 40 20 1).Infof("%s @@
0fb0: 25 73 22 2c 20 63 6f 73 74 2c 20 62 61 73 69 73 %s", cost, basis
0fc0: 29 0a 09 09 09 63 6f 6e 76 65 72 73 69 6f 6e 5b )....conversion[
0fd0: 63 6f 73 74 2e 53 74 72 69 6e 67 28 29 5d 20 3d cost.String()] =
0fe0: 20 62 61 73 69 73 0a 09 09 7d 20 2f 2f 20 65 6e basis...} // en
0ff0: 64 20 66 69 72 73 74 20 70 61 73 73 0a 0a 09 09 d first pass....
1000: 2f 2f 20 73 65 63 6f 6e 64 20 70 61 73 73 2c 20 // second pass,
1010: 61 6c 74 65 72 0a 09 09 66 6f 72 20 69 6e 64 65 alter...for inde
1020: 78 2c 20 6c 69 6e 65 20 3a 3d 20 72 61 6e 67 65 x, line := range
1030: 20 74 78 4c 69 6e 65 73 2e 4c 69 6e 65 5b 70 61 txLines.Line[pa
1040: 79 65 65 49 6e 64 65 78 2b 31 3a 5d 20 7b 0a 09 yeeIndex+1:] {..
1050: 09 09 73 70 6c 69 74 2c 20 6f 6b 20 3a 3d 20 70 ..split, ok := p
1060: 61 72 73 65 53 70 6c 69 74 28 6c 69 6e 65 29 0a arseSplit(line).
1070: 09 09 09 69 66 20 21 6f 6b 20 7b 0a 09 09 09 09 ...if !ok {.....
1080: 63 6f 6e 74 69 6e 75 65 20 2f 2f 20 63 6f 6d 6d continue // comm
1090: 65 6e 74 20 69 73 20 6e 6f 6f 70 0a 09 09 09 7d ent is noop....}
10a0: 0a 0a 09 09 09 69 66 20 73 70 6c 69 74 2e 63 6f .....if split.co
10b0: 73 74 20 21 3d 20 6e 69 6c 20 7c 7c 20 73 70 6c st != nil || spl
10c0: 69 74 2e 70 72 69 63 65 20 21 3d 20 6e 69 6c 20 it.price != nil
10d0: 7b 0a 09 09 09 09 62 61 73 69 73 2c 20 6f 6b 20 {.....basis, ok
10e0: 3a 3d 20 63 6f 6e 76 65 72 73 69 6f 6e 5b 73 70 := conversion[sp
10f0: 6c 69 74 2e 43 6f 73 74 28 29 2e 53 74 72 69 6e lit.Cost().Strin
1100: 67 28 29 5d 0a 09 09 09 09 69 66 20 6f 6b 20 7b g()].....if ok {
1110: 0a 09 09 09 09 09 2f 2f 20 72 65 70 6c 61 63 65 ......// replace
1120: 20 65 78 69 73 74 69 6e 67 20 63 6f 73 74 2f 70 existing cost/p
1130: 72 69 63 65 20 77 69 74 68 20 62 61 73 69 73 0a rice with basis.
1140: 09 09 09 09 09 74 78 4c 69 6e 65 73 2e 4c 69 6e .....txLines.Lin
1150: 65 5b 70 61 79 65 65 49 6e 64 65 78 2b 31 2b 69 e[payeeIndex+1+i
1160: 6e 64 65 78 5d 20 3d 20 73 74 72 69 6e 67 73 2e ndex] = strings.
1170: 52 65 70 6c 61 63 65 28 6c 69 6e 65 2c 20 22 40 Replace(line, "@
1180: 22 2c 20 66 6d 74 2e 53 70 72 69 6e 74 66 28 22 ", fmt.Sprintf("
1190: 40 40 20 25 73 20 3b 20 40 22 2c 20 62 61 73 69 @@ %s ; @", basi
11a0: 73 29 2c 20 31 29 0a 09 09 09 09 7d 0a 09 09 09 s), 1).....}....
11b0: 7d 20 65 6c 73 65 20 69 66 20 73 70 6c 69 74 2e } else if split.
11c0: 64 65 6c 74 61 20 21 3d 20 6e 69 6c 20 7b 0a 09 delta != nil {..
11d0: 09 09 09 64 65 6c 74 61 53 74 72 20 3a 3d 20 73 ...deltaStr := s
11e0: 70 6c 69 74 2e 64 65 6c 74 61 2e 4e 65 67 43 6c plit.delta.NegCl
11f0: 6f 6e 65 28 29 2e 53 74 72 69 6e 67 28 29 0a 09 one().String()..
1200: 09 09 09 62 61 73 69 73 2c 20 6f 6b 20 3a 3d 20 ...basis, ok :=
1210: 63 6f 6e 76 65 72 73 69 6f 6e 5b 64 65 6c 74 61 conversion[delta
1220: 53 74 72 5d 0a 09 09 09 09 69 66 20 6f 6b 20 7b Str].....if ok {
1230: 0a 09 09 09 09 09 2f 2f 20 61 64 64 20 62 61 73 ......// add bas
1240: 69 73 20 77 68 65 72 65 20 74 68 65 72 65 20 6d is where there m
1250: 61 79 20 62 65 20 6e 6f 20 70 72 69 63 65 0a 09 ay be no price..
1260: 09 09 09 09 74 78 4c 69 6e 65 73 2e 4c 69 6e 65 ....txLines.Line
1270: 5b 70 61 79 65 65 49 6e 64 65 78 2b 31 2b 69 6e [payeeIndex+1+in
1280: 64 65 78 5d 20 3d 20 73 74 72 69 6e 67 73 2e 52 dex] = strings.R
1290: 65 70 6c 61 63 65 28 6c 69 6e 65 2c 20 64 65 6c eplace(line, del
12a0: 74 61 53 74 72 2c 20 66 6d 74 2e 53 70 72 69 6e taStr, fmt.Sprin
12b0: 74 66 28 22 25 73 20 40 40 20 25 73 20 3b 20 22 tf("%s @@ %s ; "
12c0: 2c 20 64 65 6c 74 61 53 74 72 2c 20 62 61 73 69 , deltaStr, basi
12d0: 73 29 2c 20 31 29 0a 09 09 09 09 7d 0a 09 09 09 s), 1).....}....
12e0: 7d 0a 0a 09 09 7d 20 2f 2f 20 65 6e 64 20 73 65 }....} // end se
12f0: 63 6f 6e 64 20 70 61 73 73 0a 0a 09 09 2f 2f 20 cond pass....//
1300: 77 72 69 74 65 20 74 78 4c 69 6e 65 73 20 28 77 write txLines (w
1310: 68 69 63 68 20 6d 61 79 20 68 61 76 65 20 62 65 hich may have be
1320: 65 6e 20 6d 6f 64 69 66 69 65 64 20 61 62 6f 76 en modified abov
1330: 65 29 0a 09 09 77 72 69 74 65 4c 69 6e 65 73 28 e)...writeLines(
1340: 74 78 4c 69 6e 65 73 2e 4c 69 6e 65 29 0a 09 09 txLines.Line)...
1350: 66 6d 74 2e 50 72 69 6e 74 6c 6e 28 22 22 29 20 fmt.Println("")
1360: 2f 2f 20 62 6c 61 6e 6b 20 6c 69 6e 65 20 62 65 // blank line be
1370: 74 77 65 65 6e 20 74 72 61 6e 73 61 63 74 69 6f tween transactio
1380: 6e 73 0a 0a 09 7d 20 2f 2f 20 65 6e 64 20 73 63 ns...} // end sc
1390: 61 6e 20 6c 6f 6f 70 0a 0a 09 72 65 74 75 72 6e an loop...return
13a0: 20 6e 69 6c 0a 7d 0a 0a 66 75 6e 63 20 68 69 73 nil.}..func his
13b0: 74 6f 72 79 4b 65 79 28 64 61 74 65 20 74 69 6d toryKey(date tim
13c0: 65 2e 54 69 6d 65 2c 20 61 73 73 65 74 20 41 73 e.Time, asset As
13d0: 73 65 74 29 20 73 74 72 69 6e 67 20 7b 0a 09 72 set) string {..r
13e0: 65 74 75 72 6e 20 66 6d 74 2e 53 70 72 69 6e 74 eturn fmt.Sprint
13f0: 66 28 22 25 73 20 25 73 22 2c 20 64 61 74 65 2e f("%s %s", date.
1400: 46 6f 72 6d 61 74 28 22 32 30 30 36 2f 30 31 2f Format("2006/01/
1410: 30 32 22 29 2c 20 61 73 73 65 74 29 0a 7d 0a 02"), asset).}.