152 lines
2.6 KiB
Go
152 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/base32"
|
|
"log"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
restful "github.com/emicklei/go-restful"
|
|
"m.cluseau.fr/go/cow"
|
|
)
|
|
|
|
type DownloadSpec struct {
|
|
Kind string
|
|
Name string
|
|
Assets []string
|
|
|
|
createdAt time.Time
|
|
}
|
|
|
|
func wsAuthorizeDownload(req *restful.Request, resp *restful.Response) {
|
|
var spec DownloadSpec
|
|
|
|
if err := req.ReadEntity(&spec); err != nil {
|
|
wsError(resp, err)
|
|
return
|
|
}
|
|
|
|
if spec.Kind == "" || spec.Name == "" || len(spec.Assets) == 0 {
|
|
resp.WriteErrorString(http.StatusBadRequest, "missing data")
|
|
return
|
|
}
|
|
|
|
randBytes := make([]byte, 32)
|
|
_, err := rand.Read(randBytes)
|
|
if err != nil {
|
|
wsError(resp, err)
|
|
return
|
|
}
|
|
|
|
token := base32.StdEncoding.WithPadding(base32.NoPadding).EncodeToString(randBytes)
|
|
|
|
spec.createdAt = time.Now()
|
|
|
|
wState.Change(func(v *State) {
|
|
cow.MapSet(&v.Downloads, token, spec)
|
|
})
|
|
|
|
log.Printf("download token created for %s %q, assets %q", spec.Kind, spec.Name, spec.Assets)
|
|
|
|
resp.WriteAsJson(token)
|
|
}
|
|
|
|
func wsDownload(req *restful.Request, resp *restful.Response) {
|
|
token := req.PathParameter("token")
|
|
asset := req.PathParameter("asset")
|
|
|
|
if token == "" || asset == "" {
|
|
wsNotFound(resp)
|
|
return
|
|
}
|
|
|
|
var spec DownloadSpec
|
|
found := false
|
|
wState.Change(func(v *State) {
|
|
var ok bool
|
|
spec, ok = v.Downloads[token]
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
newAssets := make([]string, 0, len(spec.Assets))
|
|
for _, a := range spec.Assets {
|
|
if a == asset {
|
|
found = true
|
|
} else {
|
|
newAssets = append(newAssets, a)
|
|
}
|
|
}
|
|
|
|
if !found {
|
|
wsNotFound(resp)
|
|
return
|
|
}
|
|
|
|
cow.Map(&v.Downloads)
|
|
|
|
if len(newAssets) == 0 {
|
|
delete(v.Downloads, token)
|
|
} else {
|
|
spec.Assets = newAssets
|
|
v.Downloads[token] = spec
|
|
}
|
|
})
|
|
|
|
if !found {
|
|
wsNotFound(resp)
|
|
return
|
|
}
|
|
|
|
log.Printf("download via token: %s %q asset %q", spec.Kind, spec.Name, asset)
|
|
|
|
cfg, err := readConfig()
|
|
if err != nil {
|
|
wsError(resp, err)
|
|
return
|
|
}
|
|
|
|
setHeader := func(ext string) {
|
|
resp.AddHeader("Content-Disposition", "attachment; filename="+strconv.Quote(spec.Kind+"_"+spec.Name+"_"+asset+ext))
|
|
}
|
|
|
|
switch spec.Kind {
|
|
case "cluster":
|
|
cluster := cfg.ClusterByName(spec.Name)
|
|
if cluster == nil {
|
|
wsNotFound(resp)
|
|
return
|
|
}
|
|
|
|
switch asset {
|
|
case "addons":
|
|
setHeader(".yaml")
|
|
resp.Write([]byte(cluster.Addons))
|
|
|
|
default:
|
|
wsNotFound(resp)
|
|
}
|
|
|
|
case "host":
|
|
host := hostOrTemplate(cfg, spec.Name)
|
|
if host == nil {
|
|
wsNotFound(resp)
|
|
return
|
|
}
|
|
|
|
switch asset {
|
|
case "config", "bootstrap-config":
|
|
setHeader(".yaml")
|
|
default:
|
|
setHeader("")
|
|
}
|
|
|
|
renderHost(resp.ResponseWriter, req.Request, asset, host, cfg)
|
|
|
|
default:
|
|
wsNotFound(resp)
|
|
}
|
|
}
|