@@ -19,16 +19,57 @@ import (
1919
2020// Downloader downloads a binary
2121type Downloader struct {
22- OS string `json:"os"`
23- Arch string `json:"arch"`
24- URL string `json:"url"`
25- Checksum string `json:"checksum,omitempty" yaml:",omitempty"`
26- ArchivePath string `json:"archive_path,omitempty" yaml:"archive_path,omitempty"`
27- Link bool `json:"link,omitempty" yaml:",omitempty"`
28- BinName string `json:"bin,omitempty" yaml:"bin,omitempty"`
22+ OS string `json:"os"`
23+ Arch string `json:"arch"`
24+ URL string `json:"url"`
25+ ArchivePath string `json:"archive_path,omitempty" yaml:"archive_path,omitempty"`
26+ BinName string `json:"bin,omitempty" yaml:"bin,omitempty"`
27+ Link bool `json:"link,omitempty" yaml:",omitempty"`
28+ Checksum string `json:"checksum,omitempty" yaml:",omitempty"`
29+ Vars map [string ]string `json:"vars,omitempty" yaml:"vars,omitempty"`
30+
31+ //set to true by applyTemplates
32+ tmplApplied bool
33+ }
34+
35+ func (d * Downloader ) clone () * Downloader {
36+ c := new (Downloader )
37+ * c = * d
38+ if c .Vars != nil {
39+ c .Vars = util .CopyStringMap (c .Vars )
40+ }
41+ return c
42+ }
43+
44+ func (d * Downloader ) requireApplyTemplates () {
45+ if ! d .tmplApplied {
46+ panic ("templates not applied" )
47+ }
48+ }
49+
50+ func (d * Downloader ) applyTemplates () error {
51+ executeTemplate := func (tmpl string ) (string , error ) {
52+ return util .ExecuteTemplate (tmpl , d .OS , d .Arch , d .Vars )
53+ }
54+ var err error
55+ d .URL , err = executeTemplate (d .URL )
56+ if err != nil {
57+ return err
58+ }
59+ d .ArchivePath , err = executeTemplate (d .ArchivePath )
60+ if err != nil {
61+ return err
62+ }
63+ d .BinName , err = executeTemplate (d .BinName )
64+ if err != nil {
65+ return err
66+ }
67+ d .tmplApplied = true
68+ return nil
2969}
3070
3171func (d * Downloader ) downloadableName () (string , error ) {
72+ d .requireApplyTemplates ()
3273 u , err := url .Parse (d .URL )
3374 if err != nil {
3475 return "" , err
@@ -37,6 +78,7 @@ func (d *Downloader) downloadableName() (string, error) {
3778}
3879
3980func (d * Downloader ) downloadablePath (targetDir string ) (string , error ) {
81+ d .requireApplyTemplates ()
4082 name , err := d .downloadableName ()
4183 if err != nil {
4284 return "" , err
@@ -45,6 +87,7 @@ func (d *Downloader) downloadablePath(targetDir string) (string, error) {
4587}
4688
4789func (d * Downloader ) binPath (targetDir string ) string {
90+ d .requireApplyTemplates ()
4891 return filepath .Join (targetDir , d .BinName )
4992}
5093
@@ -53,6 +96,7 @@ func (d *Downloader) chmod(targetDir string) error {
5396}
5497
5598func (d * Downloader ) moveOrLinkBin (targetDir , extractDir string ) error {
99+ d .requireApplyTemplates ()
56100 archivePath := filepath .FromSlash (d .ArchivePath )
57101 if archivePath == "" {
58102 archivePath = filepath .FromSlash (d .BinName )
@@ -103,6 +147,7 @@ func (d *Downloader) moveOrLinkBin(targetDir, extractDir string) error {
103147}
104148
105149func (d * Downloader ) extract (downloadDir , extractDir string ) error {
150+ d .requireApplyTemplates ()
106151 dlName , err := d .downloadableName ()
107152 if err != nil {
108153 return err
@@ -124,6 +169,7 @@ func (d *Downloader) extract(downloadDir, extractDir string) error {
124169}
125170
126171func (d * Downloader ) download (downloadDir string ) error {
172+ d .requireApplyTemplates ()
127173 dlPath , err := d .downloadablePath (downloadDir )
128174 if err != nil {
129175 return err
@@ -149,6 +195,7 @@ func (d *Downloader) setDefaultBinName(defaultName string) {
149195}
150196
151197func (d * Downloader ) validateChecksum (targetDir string ) error {
198+ d .requireApplyTemplates ()
152199 targetFile , err := d .downloadablePath (targetDir )
153200 if err != nil {
154201 return err
@@ -179,12 +226,26 @@ type UpdateChecksumOpts struct {
179226
180227//UpdateChecksum updates the checksum based on a fresh download
181228func (d * Downloader ) UpdateChecksum (opts UpdateChecksumOpts ) error {
229+ sum , err := d .getUpdatedChecksum (opts )
230+ if err != nil {
231+ return err
232+ }
233+ d .Checksum = sum
234+ return nil
235+ }
236+
237+ //getUpdatedChecksum downloads the archive and returns its actual checksum.
238+ func (d * Downloader ) getUpdatedChecksum (opts UpdateChecksumOpts ) (string , error ) {
239+ d = d .clone ()
240+ err := d .applyTemplates ()
241+ if err != nil {
242+ return "" , err
243+ }
182244 cellarDir := opts .CellarDir
183- var err error
184245 if cellarDir == "" {
185246 cellarDir , err = ioutil .TempDir ("" , "bindown" )
186247 if err != nil {
187- return err
248+ return "" , err
188249 }
189250 defer func () {
190251 _ = os .RemoveAll (cellarDir ) //nolint:errcheck
@@ -196,21 +257,15 @@ func (d *Downloader) UpdateChecksum(opts UpdateChecksumOpts) error {
196257 err = d .download (downloadDir )
197258 if err != nil {
198259 log .Printf ("error downloading: %v" , err )
199- return err
260+ return "" , err
200261 }
201262
202263 dlPath , err := d .downloadablePath (downloadDir )
203264 if err != nil {
204- return err
205- }
206-
207- checkSum , err := fileChecksum (dlPath )
208- if err != nil {
209- return err
265+ return "" , err
210266 }
211267
212- d .Checksum = checkSum
213- return nil
268+ return fileChecksum (dlPath )
214269}
215270
216271//InstallOpts options for Install
@@ -235,6 +290,10 @@ func (d *Downloader) extractsSubName() string {
235290
236291//Install downloads and installs a bin
237292func (d * Downloader ) Install (opts InstallOpts ) error {
293+ err := d .applyTemplates ()
294+ if err != nil {
295+ return err
296+ }
238297 d .setDefaultBinName (opts .DownloaderName )
239298 cellarDir := opts .CellarDir
240299 if cellarDir == "" {
@@ -245,13 +304,13 @@ func (d *Downloader) Install(opts InstallOpts) error {
245304 extractDir := filepath .Join (cellarDir , "extracts" , d .extractsSubName ())
246305
247306 if opts .Force {
248- err : =os .RemoveAll (downloadDir )
307+ err = os .RemoveAll (downloadDir )
249308 if err != nil {
250309 return err
251310 }
252311 }
253312
254- err : =d .download (downloadDir )
313+ err = d .download (downloadDir )
255314 if err != nil {
256315 log .Printf ("error downloading: %v" , err )
257316 return err
@@ -295,6 +354,10 @@ type ValidateOpts struct {
295354//Validate installs the downloader to a temporary directory and returns an error if it was unsuccessful.
296355// If cellarDir is "", it will use a temp directory
297356func (d * Downloader ) Validate (opts ValidateOpts ) error {
357+ err := d .applyTemplates ()
358+ if err != nil {
359+ return err
360+ }
298361 tmpDir , err := ioutil .TempDir ("" , "bindown" )
299362 if err != nil {
300363 return err
0 commit comments