-
-
Notifications
You must be signed in to change notification settings - Fork 77
/
translate.go
100 lines (84 loc) · 2.72 KB
/
translate.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
/*
Copyright © 2019 Ce Gao
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package cmd
import (
"log"
"github.com/spf13/cobra"
"github.com/spf13/viper"
"github.com/deb-sig/double-entry-generator/pkg/cmd/validator"
"github.com/deb-sig/double-entry-generator/pkg/compiler"
"github.com/deb-sig/double-entry-generator/pkg/config"
"github.com/deb-sig/double-entry-generator/pkg/consts"
"github.com/deb-sig/double-entry-generator/pkg/provider"
)
var (
providerName string
targetName string
appendMode bool
output string
)
var translateCmd = &cobra.Command{
Use: "translate [flags] <path to bill file>",
Short: "Translate the bills to a given format",
Long: ``,
Args: func(cmd *cobra.Command, args []string) error {
return validator.TranslateArgs(args)
},
Run: func(cmd *cobra.Command, args []string) {
run(args)
},
}
func init() {
rootCmd.AddCommand(translateCmd)
translateCmd.Flags().StringVarP(&providerName, "provider", "p", "alipay", "Bills provider (alipay)")
translateCmd.Flags().StringVarP(&targetName, "target", "t", "beancount", "Target (beancount)")
translateCmd.Flags().BoolVarP(&appendMode, "append", "a", false, "Append mode")
translateCmd.Flags().StringVarP(&output, "output", "o", "default_output.beancount", "Output file")
}
func run(args []string) {
// Get the config from viper.
c := &config.Config{}
err := viper.Unmarshal(c)
logErrorIfNotNil(err)
switch providerName {
case consts.ProviderAlipay:
fallthrough
case consts.ProviderWechat:
if c.DefaultCurrency == "" ||
c.DefaultMinusAccount == "" ||
c.DefaultPlusAccount == "" {
log.Fatalf("Failed to get default options in config")
}
case consts.ProviderHuobi:
if c.DefaultCurrency == "" ||
c.DefaultCashAccount == "" ||
c.DefaultPositionAccount == "" ||
c.DefaultCommissionAccount == "" ||
c.DefaultPnlAccount == "" {
log.Fatalf("Failed to get default options in config")
}
}
p, err := provider.New(providerName)
logErrorIfNotNil(err)
i, err := p.Translate(args[0])
logErrorIfNotNil(err)
cpl, err := compiler.New(providerName, targetName, output, appendMode, c, i)
logErrorIfNotNil(err)
err = cpl.Compile()
logErrorIfNotNil(err)
}
func logErrorIfNotNil(err error) {
if err != nil {
log.Fatalf("Failed to translate: %v", err)
}
}