/
handlers.go
289 lines (258 loc) · 19 KB
/
handlers.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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
package handlers
import (
"encoding/json"
"net/http"
"github.com/gorilla/mux"
"github.com/gorilla/websocket"
"github.com/replicatedhq/kots/kotsadm/pkg/policy"
"github.com/replicatedhq/kots/kotsadm/pkg/store"
kotsscheme "github.com/replicatedhq/kots/kotskinds/client/kotsclientset/scheme"
"github.com/replicatedhq/kots/pkg/logger"
troubleshootscheme "github.com/replicatedhq/troubleshoot/pkg/client/troubleshootclientset/scheme"
yaml "github.com/replicatedhq/yaml/v3"
veleroscheme "github.com/vmware-tanzu/velero/pkg/generated/clientset/versioned/scheme"
"k8s.io/client-go/kubernetes/scheme"
)
var _ KOTSHandler = (*Handler)(nil)
type Handler struct {
}
func init() {
kotsscheme.AddToScheme(scheme.Scheme)
troubleshootscheme.AddToScheme(scheme.Scheme)
veleroscheme.AddToScheme(scheme.Scheme)
}
func RegisterSessionAuthRoutes(r *mux.Router, kotsStore store.KOTSStore, handler KOTSHandler, middleware *policy.Middleware) {
r.Use(RequireValidSessionMiddleware(kotsStore))
// Installation
r.Name("UploadNewLicense").Path("/api/v1/license").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppCreate, handler.UploadNewLicense))
r.Name("ExchangePlatformLicense").Path("/api/v1/license/platform").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppCreate, handler.ExchangePlatformLicense))
r.Name("ResumeInstallOnline").Path("/api/v1/license/resume").Methods("PUT").
HandlerFunc(middleware.EnforceAccess(policy.AppCreate, handler.ResumeInstallOnline))
r.Name("GetOnlineInstallStatus").Path("/api/v1/app/online/status").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppCreate, handler.GetOnlineInstallStatus))
// Support Bundles
r.Name("GetSupportBundle").Path("/api/v1/troubleshoot/supportbundle/{bundleSlug}").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppSupportbundleRead, handler.GetSupportBundle)) // TODO: appSlug
r.Name("ListSupportBundles").Path("/api/v1/troubleshoot/app/{appSlug}/supportbundles").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppSupportbundleRead, handler.ListSupportBundles))
r.Name("GetSupportBundleCommand").Path("/api/v1/troubleshoot/app/{appSlug}/supportbundlecommand").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppSupportbundleRead, handler.GetSupportBundleCommand))
r.Name("GetSupportBundleFiles").Path("/api/v1/troubleshoot/supportbundle/{bundleId}/files").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppSupportbundleRead, handler.GetSupportBundleFiles)) // TODO: appSlug
r.Name("GetSupportBundleRedactions").Path("/api/v1/troubleshoot/supportbundle/{bundleId}/redactions").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppSupportbundleRead, handler.GetSupportBundleRedactions)) // TODO: appSlug
r.Name("DownloadSupportBundle").Path("/api/v1/troubleshoot/supportbundle/{bundleId}/download").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppSupportbundleRead, handler.DownloadSupportBundle)) // TODO: appSlug
r.Name("CollectSupportBundle").Path("/api/v1/troubleshoot/supportbundle/app/{appId}/cluster/{clusterId}/collect").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppSupportbundleWrite, handler.CollectSupportBundle))
// redactor routes
r.Name("UpdateRedact").Path("/api/v1/redact/set").Methods("PUT").
HandlerFunc(middleware.EnforceAccess(policy.RedactorWrite, handler.UpdateRedact))
r.Name("GetRedact").Path("/api/v1/redact/get").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.RedactorRead, handler.GetRedact))
r.Name("ListRedactors").Path("/api/v1/redacts").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.RedactorRead, handler.ListRedactors))
r.Name("GetRedactMetadataAndYaml").Path("/api/v1/redact/spec/{slug}").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.RedactorRead, handler.GetRedactMetadataAndYaml))
r.Name("SetRedactMetadataAndYaml").Path("/api/v1/redact/spec/{slug}").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.RedactorWrite, handler.SetRedactMetadataAndYaml))
r.Name("DeleteRedact").Path("/api/v1/redact/spec/{slug}").Methods("DELETE").
HandlerFunc(middleware.EnforceAccess(policy.RedactorWrite, handler.DeleteRedact))
r.Name("SetRedactEnabled").Path("/api/v1/redact/enabled/{slug}").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.RedactorWrite, handler.SetRedactEnabled))
// Kotsadm Identity Service
r.Name("ConfigureIdentityService").Path("/api/v1/identity/config").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.IdentityServiceWrite, handler.ConfigureIdentityService))
r.Name("GetIdentityServiceConfig").Path("/api/v1/identity/config").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.IdentityServiceRead, handler.GetIdentityServiceConfig))
// App Identity Service
r.Name("ConfigureAppIdentityService").Path("/api/v1/app/{appSlug}/identity/config").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppIdentityServiceWrite, handler.ConfigureAppIdentityService))
r.Name("GetAppIdentityServiceConfig").Path("/api/v1/app/{appSlug}/identity/config").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppIdentityServiceRead, handler.GetAppIdentityServiceConfig))
// Apps
r.Name("ListApps").Path("/api/v1/apps").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppList, handler.ListApps))
r.Name("GetApp").Path("/api/v1/app/{appSlug}").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppRead, handler.GetApp))
r.Name("GetAppStatus").Path("/api/v1/app/{appSlug}/status").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppStatusRead, handler.GetAppStatus))
r.Name("GetAppVersionHistory").Path("/api/v1/app/{appSlug}/versions").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamRead, handler.GetAppVersionHistory))
r.Name("GetUpdateDownloadStatus").Path("/api/v1/app/{appSlug}/task/updatedownload").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppRead, handler.GetUpdateDownloadStatus)) // NOTE: appSlug is unused
// Airgap
r.Name("AirgapBundleProgress").Path("/api/v1/app/{appSlug}/airgap/bundleprogress/{identifier}/{totalChunks}").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.AirgapBundleProgress))
r.Name("AirgapBundleExists").Path("/api/v1/app/{appSlug}/airgap/bundleexists/{identifier}/{totalChunks}").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.AirgapBundleExists))
r.Name("CreateAppFromAirgap").Path("/api/v1/app/{appSlug}/airgap/processbundle/{identifier}/{totalChunks}").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.CreateAppFromAirgap))
r.Name("UpdateAppFromAirgap").Path("/api/v1/app/{appSlug}/airgap/processbundle/{identifier}/{totalChunks}").Methods("PUT").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.UpdateAppFromAirgap))
r.Name("CheckAirgapBundleChunk").Path("/api/v1/app/{appSlug}/airgap/chunk").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.CheckAirgapBundleChunk))
r.Name("UploadAirgapBundleChunk").Path("/api/v1/app/{appSlug}/airgap/chunk").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.UploadAirgapBundleChunk))
r.Name("GetAirgapInstallStatus").Path("/api/v1/app/{appSlug}/airgap/status").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.GetAirgapInstallStatus))
r.Name("ResetAirgapInstallStatus").Path("/api/v1/app/{appSlug}/airgap/reset").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.ResetAirgapInstallStatus))
// Implemented handlers
r.Name("IgnorePreflightRBACErrors").Path("/api/v1/app/{appSlug}/sequence/{sequence}/preflight/ignore-rbac").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamPreflightWrite, handler.IgnorePreflightRBACErrors))
r.Name("StartPreflightChecks").Path("/api/v1/app/{appSlug}/sequence/{sequence}/preflight/run").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamPreflightWrite, handler.StartPreflightChecks))
r.Name("GetLatestPreflightResultsForSequenceZero").Path("/api/v1/app/{appSlug}/preflight/result").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamPreflightRead, handler.GetLatestPreflightResultsForSequenceZero))
r.Name("GetPreflightResult").Path("/api/v1/app/{appSlug}/sequence/{sequence}/preflight/result").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamPreflightRead, handler.GetPreflightResult))
r.Name("GetPreflightCommand").Path("/api/v1/app/{appSlug}/sequence/{sequence}/preflightcommand").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppRead, handler.GetPreflightCommand)) // this is intentionally policy.AppRead
r.Name("DeployAppVersion").Path("/api/v1/app/{appSlug}/sequence/{sequence}/deploy").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.DeployAppVersion))
r.Name("RedeployAppVersion").Path("/api/v1/app/{appSlug}/sequence/{sequence}/redeploy").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.RedeployAppVersion))
r.Name("GetAppRenderedContents").Path("/api/v1/app/{appSlug}/sequence/{sequence}/renderedcontents").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamFiletreeRead, handler.GetAppRenderedContents))
r.Name("GetAppContents").Path("/api/v1/app/{appSlug}/sequence/{sequence}/contents").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamFiletreeRead, handler.GetAppContents))
r.Name("GetAppDashboard").Path("/api/v1/app/{appSlug}/cluster/{clusterId}/dashboard").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppRead, handler.GetAppDashboard))
r.Name("GetDownstreamOutput").Path("/api/v1/app/{appSlug}/cluster/{clusterId}/sequence/{sequence}/downstreamoutput").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamLogsRead, handler.GetDownstreamOutput))
r.Name("GetKotsadmRegistry").Path("/api/v1/registry").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.RegistryRead, handler.GetKotsadmRegistry))
r.Name("GetImageRewriteStatusOld").Path("/api/v1/imagerewritestatus").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.RegistryRead, handler.GetImageRewriteStatus))
r.Name("UpdateAppRegistry").Path("/api/v1/app/{appSlug}/registry").Methods("PUT").
HandlerFunc(middleware.EnforceAccess(policy.AppRegistryWrite, handler.UpdateAppRegistry))
r.Name("GetAppRegistry").Path("/api/v1/app/{appSlug}/registry").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppRegistryRead, handler.GetAppRegistry))
r.Name("GetImageRewriteStatus").Path("/api/v1/app/{appSlug}/imagerewritestatus").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppRegistryRead, handler.GetImageRewriteStatus))
r.Name("ValidateAppRegistry").Path("/api/v1/app/{appSlug}/registry/validate").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppRegistryWrite, handler.ValidateAppRegistry))
r.Name("UpdateAppConfig").Path("/api/v1/app/{appSlug}/config").Methods("PUT").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamConfigWrite, handler.UpdateAppConfig))
r.Name("CurrentAppConfig").Path("/api/v1/app/{appSlug}/config/{sequence}").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamConfigRead, handler.CurrentAppConfig))
r.Name("LiveAppConfig").Path("/api/v1/app/{appSlug}/liveconfig").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamConfigWrite, handler.LiveAppConfig))
r.Name("SyncLicense").Path("/api/v1/app/{appSlug}/license").Methods("PUT").
HandlerFunc(middleware.EnforceAccess(policy.AppLicenseWrite, handler.SyncLicense))
r.Name("GetLicense").Path("/api/v1/app/{appSlug}/license").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppLicenseRead, handler.GetLicense))
r.Name("AppUpdateCheck").Path("/api/v1/app/{appSlug}/updatecheck").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.AppUpdateCheck))
r.Name("UpdateCheckerSpec").Path("/api/v1/app/{appSlug}/updatecheckerspec").Methods("PUT").
HandlerFunc(middleware.EnforceAccess(policy.AppDownstreamWrite, handler.UpdateCheckerSpec))
r.Name("RemoveApp").Path("/api/v1/app/{appSlug}/remove").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppUpdate, handler.RemoveApp))
// App snapshot routes
r.Name("CreateApplicationBackup").Path("/api/v1/app/{appSlug}/snapshot/backup").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppBackupWrite, handler.CreateApplicationBackup))
r.Name("GetRestoreStatus").Path("/api/v1/app/{appSlug}/snapshot/restore/status").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppRestoreRead, handler.GetRestoreStatus))
r.Name("CancelRestore").Path("/api/v1/app/{appSlug}/snapshot/restore").Methods("DELETE").
HandlerFunc(middleware.EnforceAccess(policy.AppRestoreWrite, handler.CancelRestore))
r.Name("CreateApplicationRestore").Path("/api/v1/app/{appSlug}/snapshot/restore/{snapshotName}").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppRestoreWrite, handler.CreateApplicationRestore))
r.Name("GetRestoreDetails").Path("/api/v1/app/{appSlug}/snapshot/restore/{restoreName}").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppRestoreRead, handler.GetRestoreDetails))
r.Name("ListBackups").Path("/api/v1/app/{appSlug}/snapshots").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppBackupRead, handler.ListBackups))
r.Name("GetSnapshotConfig").Path("/api/v1/app/{appSlug}/snapshot/config").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.AppSnapshotsettingsRead, handler.GetSnapshotConfig))
r.Name("SaveSnapshotConfig").Path("/api/v1/app/{appSlug}/snapshot/config").Methods("PUT").
HandlerFunc(middleware.EnforceAccess(policy.AppSnapshotsettingsWrite, handler.SaveSnapshotConfig))
// Global snapshot routes
r.Name("ListInstanceBackups").Path("/api/v1/snapshots").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.BackupRead, handler.ListInstanceBackups))
r.Name("CreateInstanceBackup").Path("/api/v1/snapshot/backup").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.BackupWrite, handler.CreateInstanceBackup))
r.Name("GetInstanceSnapshotConfig").Path("/api/v1/snapshot/config").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.SnapshotsettingsRead, handler.GetInstanceSnapshotConfig))
r.Name("SaveInstanceSnapshotConfig").Path("/api/v1/snapshot/config").Methods("PUT").
HandlerFunc(middleware.EnforceAccess(policy.SnapshotsettingsWrite, handler.SaveInstanceSnapshotConfig))
r.Name("GetGlobalSnapshotSettings").Path("/api/v1/snapshots/settings").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.SnapshotsettingsRead, handler.GetGlobalSnapshotSettings))
r.Name("UpdateGlobalSnapshotSettings").Path("/api/v1/snapshots/settings").Methods("PUT").
HandlerFunc(middleware.EnforceAccess(policy.SnapshotsettingsWrite, handler.UpdateGlobalSnapshotSettings))
r.Name("GetBackup").Path("/api/v1/snapshot/{snapshotName}").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.BackupRead, handler.GetBackup))
r.Name("DeleteBackup").Path("/api/v1/snapshot/{snapshotName}/delete").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.BackupWrite, handler.DeleteBackup))
r.Name("RestoreApps").Path("/api/v1/snapshot/{snapshotName}/restore-apps").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.RestoreWrite, handler.RestoreApps))
r.Name("GetRestoreAppsStatus").Path("/api/v1/snapshot/{snapshotName}/apps-restore-status").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.RestoreWrite, handler.GetRestoreAppsStatus))
r.Name("DownloadSnapshotLogs").Path("/api/v1/snapshot/{backup}/logs").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.BackupRead, handler.DownloadSnapshotLogs))
r.Name("GetVeleroStatus").Path("/api/v1/velero").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.BackupRead, handler.GetVeleroStatus))
// KURL
r.Name("Kurl").Path("/api/v1/kurl").HandlerFunc(NotImplemented) // I'm not sure why this is here
r.Name("GenerateNodeJoinCommandWorker").Path("/api/v1/kurl/generate-node-join-command-worker").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.ClusterWrite, handler.GenerateNodeJoinCommandWorker))
r.Name("GenerateNodeJoinCommandMaster").Path("/api/v1/kurl/generate-node-join-command-master").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.ClusterWrite, handler.GenerateNodeJoinCommandMaster))
r.Name("DrainNode").Path("/api/v1/kurl/nodes/{nodeName}/drain").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.ClusterWrite, handler.DrainNode))
r.Name("DeleteNode").Path("/api/v1/kurl/nodes/{nodeName}").Methods("DELETE").
HandlerFunc(middleware.EnforceAccess(policy.ClusterWrite, handler.DeleteNode))
r.Name("GetKurlNodes").Path("/api/v1/kurl/nodes").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.ClusterRead, handler.GetKurlNodes))
// Prometheus
r.Name("SetPrometheusAddress").Path("/api/v1/prometheus").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.PrometheussettingsWrite, handler.SetPrometheusAddress))
// GitOps
r.Name("UpdateAppGitOps").Path("/api/v1/gitops/app/{appId}/cluster/{clusterId}/update").Methods("PUT").
HandlerFunc(middleware.EnforceAccess(policy.AppGitopsWrite, handler.UpdateAppGitOps))
r.Name("DisableAppGitOps").Path("/api/v1/gitops/app/{appId}/cluster/{clusterId}/disable").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppGitopsWrite, handler.DisableAppGitOps))
r.Name("InitGitOpsConnection").Path("/api/v1/gitops/app/{appId}/cluster/{clusterId}/initconnection").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.AppGitopsWrite, handler.InitGitOpsConnection))
r.Name("CreateGitOps").Path("/api/v1/gitops/create").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.GitopsWrite, handler.CreateGitOps))
r.Name("ResetGitOps").Path("/api/v1/gitops/reset").Methods("POST").
HandlerFunc(middleware.EnforceAccess(policy.GitopsWrite, handler.ResetGitOps))
r.Name("GetGitOpsRepo").Path("/api/v1/gitops/get").Methods("GET").
HandlerFunc(middleware.EnforceAccess(policy.GitopsRead, handler.GetGitOpsRepo))
}
func JSON(w http.ResponseWriter, code int, payload interface{}) {
response, err := json.Marshal(payload)
if err != nil {
logger.Error(err)
w.WriteHeader(500)
return
}
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(code)
w.Write(response)
}
func StreamJSON(c *websocket.Conn, payload interface{}) {
response, err := json.Marshal(payload)
if err != nil {
logger.Error(err)
return
}
err = c.WriteMessage(websocket.TextMessage, response)
if err != nil {
logger.Error(err)
return
}
}
func YAML(w http.ResponseWriter, code int, payload interface{}) {
response, err := yaml.Marshal(payload)
if err != nil {
logger.Error(err)
w.WriteHeader(500)
return
}
w.Header().Set("Content-Type", "text/yaml")
w.WriteHeader(code)
w.Write(response)
}