55 "encoding/xml"
66 "errors"
77 "fmt"
8+ ggr "github.com/aerokube/ggr/config"
89 "github.com/spf13/cobra"
910 "io/ioutil"
1011 "os"
@@ -53,8 +54,8 @@ func init() {
5354 generateCmd .PersistentFlags ().BoolVar (& dryRun , "dryRun" , false , "whether to send output to stdout instead of writing files" )
5455}
5556
56- func convert (input JsonInput ) map [string ]XmlBrowsers {
57- ret := make (map [string ]XmlBrowsers )
57+ func convert (input Input ) map [string ]ggr. Browsers {
58+ ret := make (map [string ]ggr. Browsers )
5859 hostsMap := input .Hosts
5960 quotaMap := input .Quota
6061 aliasesMap := input .Aliases
@@ -74,14 +75,14 @@ func convert(input JsonInput) map[string]XmlBrowsers {
7475 return ret
7576}
7677
77- func createQuota (quotaName string , hostsMap JsonHosts , quota JsonQuota ) XmlBrowsers {
78- browsers := [] XmlBrowser {}
78+ func createQuota (quotaName string , hostsMap Hosts , quota Quota ) ggr. Browsers {
79+ var browsers []ggr. Browser
7980 for browserName , browser := range quota {
80- xmlVersions := [] XmlVersion {}
81+ var xmlVersions []ggr. Version
8182 for versionName , hostsRef := range browser .Versions {
8283 regions := hostsMap [hostsRef ]
8384 if regions != nil {
84- xmlVersion := XmlVersion {
85+ xmlVersion := ggr. Version {
8586 Number : versionName ,
8687 Regions : jsonRegionsToXmlRegions (regions ),
8788 }
@@ -91,27 +92,26 @@ func createQuota(quotaName string, hostsMap JsonHosts, quota JsonQuota) XmlBrows
9192 os .Exit (1 )
9293 }
9394 }
94- xmlBrowser := XmlBrowser {
95+ xmlBrowser := ggr. Browser {
9596 Name : browserName ,
9697 DefaultVersion : browser .DefaultVersion ,
9798 Versions : xmlVersions ,
9899 }
99100 browsers = append (browsers , xmlBrowser )
100101 }
101- return XmlBrowsers {
102+ return ggr. Browsers {
102103 Browsers : browsers ,
103- XmlNS : "urn:config.gridrouter.qatools.ru" ,
104104 }
105105}
106106
107- func jsonRegionsToXmlRegions (regions JsonRegions ) []XmlRegion {
108- xmlRegions := [] XmlRegion {}
107+ func jsonRegionsToXmlRegions (regions Regions ) []ggr. Region {
108+ var xmlRegions []ggr. Region
109109 for regionName , region := range regions {
110- xmlHosts := XmlHosts {}
110+ xmlHosts := ggr. Hosts {}
111111 for hostPattern , host := range region {
112112 hostNames := parseHostPattern (hostPattern )
113113 for _ , hostName := range hostNames {
114- xmlHosts = append (xmlHosts , XmlHost {
114+ xmlHosts = append (xmlHosts , ggr. Host {
115115 Name : hostName ,
116116 Port : host .Port ,
117117 Count : host .Count ,
@@ -120,31 +120,31 @@ func jsonRegionsToXmlRegions(regions JsonRegions) []XmlRegion {
120120 })
121121 }
122122 }
123- xmlRegions = append (xmlRegions , XmlRegion {
123+ xmlRegions = append (xmlRegions , ggr. Region {
124124 Name : regionName ,
125125 Hosts : xmlHosts ,
126126 })
127127 }
128128 return xmlRegions
129129}
130130
131- func parseInputFile (filePath string ) (* JsonInput , error ) {
131+ func parseInputFile (filePath string ) (* Input , error ) {
132132 bytes , err := ioutil .ReadFile (filePath )
133133 if err != nil {
134134 return nil , errors .New (fmt .Sprintf ("error reading input file [%s]: %v" , filePath , err ))
135135 }
136- input := new (JsonInput )
136+ input := new (Input )
137137 if err := json .Unmarshal (bytes , input ); err != nil {
138138 return nil , errors .New (fmt .Sprintf ("error parsing input file [%s]: %v" , filePath , err ))
139139 }
140140 return input , nil
141141}
142142
143- func marshalBrowsers (browsers XmlBrowsers ) ([]byte , error ) {
143+ func marshalBrowsers (browsers ggr. Browsers ) ([]byte , error ) {
144144 return xml .MarshalIndent (browsers , "" , " " )
145145}
146146
147- func output (quotaName string , browsers XmlBrowsers , outputDirectory string ) error {
147+ func output (quotaName string , browsers ggr. Browsers , outputDirectory string ) error {
148148 filePath := path .Join (outputDirectory , quotaName + ".xml" )
149149 if dryRun {
150150 return printOutputFile (filePath , browsers )
@@ -153,7 +153,7 @@ func output(quotaName string, browsers XmlBrowsers, outputDirectory string) erro
153153 }
154154}
155155
156- func printOutputFile (filePath string , browsers XmlBrowsers ) error {
156+ func printOutputFile (filePath string , browsers ggr. Browsers ) error {
157157 bytes , err := marshalBrowsers (browsers )
158158 if err != nil {
159159 return err
@@ -165,7 +165,7 @@ func printOutputFile(filePath string, browsers XmlBrowsers) error {
165165 return nil
166166}
167167
168- func saveOutputFile (filePath string , browsers XmlBrowsers ) error {
168+ func saveOutputFile (filePath string , browsers ggr. Browsers ) error {
169169 bytes , err := marshalBrowsers (browsers )
170170 if err != nil {
171171 return err
@@ -195,4 +195,4 @@ func parseHostPattern(pattern string) []string {
195195 }
196196 }
197197 return []string {pattern }
198- }
198+ }
0 commit comments