-
Notifications
You must be signed in to change notification settings - Fork 0
/
actions.go
119 lines (105 loc) · 2.93 KB
/
actions.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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package main
import (
"context"
"fmt"
"net/http"
"strings"
"github.com/qri-io/apiutil"
"github.com/qri-io/dataset"
"github.com/qri-io/qri/base"
"github.com/qri-io/qri/dsref"
"github.com/qri-io/qri/lib"
)
// SimActionHandler triggers actions on this server that simulate real-world
// behaviour
func SimActionHandler(inst *lib.Instance) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
key := strings.ToLower(r.FormValue("action"))
act, ok := simActions[key]
if !ok {
apiutil.WriteErrResponse(w, http.StatusBadRequest, fmt.Errorf("action not found: '%s'", key))
return
}
if err := act(r.Context(), inst); err != nil {
log.Errorf("running action %s: %s", key, err)
apiutil.WriteErrResponse(w, http.StatusInternalServerError, err)
return
}
apiutil.WriteResponse(w, "ok")
}
}
// simActions is a mapping of simulation actions keyed by string
var simActions = map[string]simActionFunc{
"createsynthsdataset": createSynthsDataset,
"appendsynthsdataset": appendSynthsDataset,
}
type simActionFunc func(ctx context.Context, inst *lib.Instance) error
func createSynthsDataset(ctx context.Context, inst *lib.Instance) error {
dsm := lib.NewDatasetMethods(inst)
res := dataset.Dataset{}
err := dsm.Save(&lib.SaveParams{
Ref: "me/synths",
Dataset: &dataset.Dataset{
Meta: &dataset.Meta{
Title: "synthesizers",
Description: "A list of great types of synthesizers",
},
BodyPath: "body.csv",
BodyBytes: []byte(`company,name,year_of_release,initial_cost,initial_cost_adjusted
moog,little phatty,,,,
moog,sub 37,,,,
moog,subsequent 37,,,,
`),
},
}, &res)
if err != nil {
log.Errorf("createSynthsDataset: error saving dataset: %s", res)
return err
}
ref := dsref.Ref{
Username: res.Peername,
Name: res.Name,
ProfileID: res.ProfileID,
Path: res.Path,
}
err = base.SetPublishStatus(inst.Repo(), ref, true)
if err != nil {
log.Errorf("createSynthsDataset: error setting published status: %s", res)
return err
}
log.Infof("createSynthsDataset dataset saved: %s", res)
return nil
}
func appendSynthsDataset(ctx context.Context, inst *lib.Instance) error {
dsm := lib.NewDatasetMethods(inst)
res := dataset.Dataset{}
err := dsm.Save(&lib.SaveParams{
Ref: "me/synths",
Dataset: &dataset.Dataset{
BodyPath: "body.csv",
BodyBytes: []byte(`company,name,year_of_release,initial_cost,initial_cost_adjusted
moog,little phatty,,,,
moog,sub 37,,,,
moog,subsequent 37,,,,
novation,bass station,,,,
`),
},
}, &res)
if err != nil {
log.Errorf("appendSynthsDataset: error saving dataset: %s", res)
return err
}
ref := dsref.Ref{
Username: res.Peername,
Name: res.Name,
ProfileID: res.ProfileID,
Path: res.Path,
}
err = base.SetPublishStatus(inst.Repo(), ref, true)
if err != nil {
log.Errorf("appendSynthsDataset: error setting published status: %s", res)
return err
}
log.Infof("appendSynthsDataset saved: %s", res)
return nil
}