release_handler.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110
  1. package api
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "net/http"
  6. "net/url"
  7. "strconv"
  8. "strings"
  9. "github.com/porter-dev/porter/internal/kubernetes/prometheus"
  10. "github.com/porter-dev/porter/internal/models"
  11. "github.com/porter-dev/porter/internal/templater/parser"
  12. "helm.sh/helm/v3/pkg/release"
  13. v1 "k8s.io/api/core/v1"
  14. metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  15. "github.com/go-chi/chi"
  16. "github.com/porter-dev/porter/internal/forms"
  17. "github.com/porter-dev/porter/internal/helm"
  18. "github.com/porter-dev/porter/internal/helm/grapher"
  19. "github.com/porter-dev/porter/internal/integrations/ci/actions"
  20. "github.com/porter-dev/porter/internal/kubernetes"
  21. "github.com/porter-dev/porter/internal/repository"
  22. segment "gopkg.in/segmentio/analytics-go.v3"
  23. "gopkg.in/yaml.v2"
  24. )
  25. // Enumeration of release API error codes, represented as int64
  26. const (
  27. ErrReleaseDecode ErrorCode = iota + 600
  28. ErrReleaseValidateFields
  29. ErrReleaseReadData
  30. ErrReleaseDeploy
  31. )
  32. // HandleListReleases retrieves a list of releases for a cluster
  33. // with various filter options
  34. func (app *App) HandleListReleases(w http.ResponseWriter, r *http.Request) {
  35. form := &forms.ListReleaseForm{
  36. ReleaseForm: &forms.ReleaseForm{
  37. Form: &helm.Form{
  38. Repo: app.Repo,
  39. DigitalOceanOAuth: app.DOConf,
  40. },
  41. },
  42. ListFilter: &helm.ListFilter{},
  43. }
  44. agent, err := app.getAgentFromQueryParams(
  45. w,
  46. r,
  47. form.ReleaseForm,
  48. form.ReleaseForm.PopulateHelmOptionsFromQueryParams,
  49. form.PopulateListFromQueryParams,
  50. )
  51. // errors are handled in app.getAgentFromQueryParams
  52. if err != nil {
  53. return
  54. }
  55. releases, err := agent.ListReleases(form.Namespace, form.ListFilter)
  56. if err != nil {
  57. app.handleErrorRead(err, ErrReleaseReadData, w)
  58. return
  59. }
  60. if err := json.NewEncoder(w).Encode(releases); err != nil {
  61. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  62. return
  63. }
  64. }
  65. // PorterRelease is a helm release with a form attached
  66. type PorterRelease struct {
  67. *release.Release
  68. Form *models.FormYAML `json:"form"`
  69. HasMetrics bool `json:"has_metrics"`
  70. }
  71. // HandleGetRelease retrieves a single release based on a name and revision
  72. func (app *App) HandleGetRelease(w http.ResponseWriter, r *http.Request) {
  73. name := chi.URLParam(r, "name")
  74. revision, err := strconv.ParseUint(chi.URLParam(r, "revision"), 0, 64)
  75. form := &forms.GetReleaseForm{
  76. ReleaseForm: &forms.ReleaseForm{
  77. Form: &helm.Form{
  78. Repo: app.Repo,
  79. DigitalOceanOAuth: app.DOConf,
  80. },
  81. },
  82. Name: name,
  83. Revision: int(revision),
  84. }
  85. agent, err := app.getAgentFromQueryParams(
  86. w,
  87. r,
  88. form.ReleaseForm,
  89. form.ReleaseForm.PopulateHelmOptionsFromQueryParams,
  90. )
  91. // errors are handled in app.getAgentFromQueryParams
  92. if err != nil {
  93. return
  94. }
  95. release, err := agent.GetRelease(form.Name, form.Revision)
  96. if err != nil {
  97. app.sendExternalError(err, http.StatusNotFound, HTTPError{
  98. Code: ErrReleaseReadData,
  99. Errors: []string{"release not found"},
  100. }, w)
  101. return
  102. }
  103. // get the filter options
  104. k8sForm := &forms.K8sForm{
  105. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  106. Repo: app.Repo,
  107. DigitalOceanOAuth: app.DOConf,
  108. },
  109. }
  110. vals, err := url.ParseQuery(r.URL.RawQuery)
  111. if err != nil {
  112. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  113. return
  114. }
  115. k8sForm.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  116. k8sForm.DefaultNamespace = form.ReleaseForm.Namespace
  117. // validate the form
  118. if err := app.validator.Struct(k8sForm); err != nil {
  119. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  120. return
  121. }
  122. // create a new dynamic client
  123. dynClient, err := kubernetes.GetDynamicClientOutOfClusterConfig(k8sForm.OutOfClusterConfig)
  124. if err != nil {
  125. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  126. return
  127. }
  128. parserDef := &parser.ClientConfigDefault{
  129. DynamicClient: dynClient,
  130. HelmChart: release.Chart,
  131. HelmRelease: release,
  132. }
  133. res := &PorterRelease{release, nil, false}
  134. for _, file := range release.Chart.Files {
  135. if strings.Contains(file.Name, "form.yaml") {
  136. formYAML, err := parser.FormYAMLFromBytes(parserDef, file.Data, "")
  137. if err != nil {
  138. break
  139. }
  140. res.Form = formYAML
  141. break
  142. }
  143. }
  144. // if form not populated, detect common charts
  145. if res.Form == nil {
  146. // for now just case by name
  147. if res.Release.Chart.Name() == "velero" {
  148. formYAML, err := parser.FormYAMLFromBytes(parserDef, []byte(veleroForm), "")
  149. if err == nil {
  150. res.Form = formYAML
  151. }
  152. }
  153. }
  154. // get prometheus service
  155. _, found, err := prometheus.GetPrometheusService(agent.K8sAgent.Clientset)
  156. if err != nil {
  157. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  158. return
  159. }
  160. res.HasMetrics = found
  161. if err := json.NewEncoder(w).Encode(res); err != nil {
  162. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  163. return
  164. }
  165. }
  166. // HandleGetReleaseComponents retrieves kubernetes objects listed in a release identified by name and revision
  167. func (app *App) HandleGetReleaseComponents(w http.ResponseWriter, r *http.Request) {
  168. name := chi.URLParam(r, "name")
  169. revision, err := strconv.ParseUint(chi.URLParam(r, "revision"), 0, 64)
  170. form := &forms.GetReleaseForm{
  171. ReleaseForm: &forms.ReleaseForm{
  172. Form: &helm.Form{
  173. Repo: app.Repo,
  174. DigitalOceanOAuth: app.DOConf,
  175. },
  176. },
  177. Name: name,
  178. Revision: int(revision),
  179. }
  180. agent, err := app.getAgentFromQueryParams(
  181. w,
  182. r,
  183. form.ReleaseForm,
  184. form.ReleaseForm.PopulateHelmOptionsFromQueryParams,
  185. )
  186. // errors are handled in app.getAgentFromQueryParams
  187. if err != nil {
  188. return
  189. }
  190. release, err := agent.GetRelease(form.Name, form.Revision)
  191. if err != nil {
  192. app.sendExternalError(err, http.StatusNotFound, HTTPError{
  193. Code: ErrReleaseReadData,
  194. Errors: []string{"release not found"},
  195. }, w)
  196. return
  197. }
  198. yamlArr := grapher.ImportMultiDocYAML([]byte(release.Manifest))
  199. objects := grapher.ParseObjs(yamlArr)
  200. parsed := grapher.ParsedObjs{
  201. Objects: objects,
  202. }
  203. parsed.GetControlRel()
  204. parsed.GetLabelRel()
  205. parsed.GetSpecRel()
  206. if err := json.NewEncoder(w).Encode(parsed); err != nil {
  207. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  208. return
  209. }
  210. }
  211. // HandleGetReleaseControllers retrieves controllers that belong to a release.
  212. // Used to display status of charts.
  213. func (app *App) HandleGetReleaseControllers(w http.ResponseWriter, r *http.Request) {
  214. name := chi.URLParam(r, "name")
  215. revision, err := strconv.ParseUint(chi.URLParam(r, "revision"), 0, 64)
  216. form := &forms.GetReleaseForm{
  217. ReleaseForm: &forms.ReleaseForm{
  218. Form: &helm.Form{
  219. Repo: app.Repo,
  220. DigitalOceanOAuth: app.DOConf,
  221. },
  222. },
  223. Name: name,
  224. Revision: int(revision),
  225. }
  226. agent, err := app.getAgentFromQueryParams(
  227. w,
  228. r,
  229. form.ReleaseForm,
  230. form.ReleaseForm.PopulateHelmOptionsFromQueryParams,
  231. )
  232. // errors are handled in app.getAgentFromQueryParams
  233. if err != nil {
  234. return
  235. }
  236. release, err := agent.GetRelease(form.Name, form.Revision)
  237. if err != nil {
  238. app.sendExternalError(err, http.StatusNotFound, HTTPError{
  239. Code: ErrReleaseReadData,
  240. Errors: []string{"release not found"},
  241. }, w)
  242. return
  243. }
  244. vals, err := url.ParseQuery(r.URL.RawQuery)
  245. if err != nil {
  246. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  247. return
  248. }
  249. // get the filter options
  250. k8sForm := &forms.K8sForm{
  251. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  252. Repo: app.Repo,
  253. DigitalOceanOAuth: app.DOConf,
  254. },
  255. }
  256. k8sForm.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  257. k8sForm.DefaultNamespace = form.ReleaseForm.Namespace
  258. // validate the form
  259. if err := app.validator.Struct(k8sForm); err != nil {
  260. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  261. return
  262. }
  263. // create a new kubernetes agent
  264. var k8sAgent *kubernetes.Agent
  265. if app.ServerConf.IsTesting {
  266. k8sAgent = app.TestAgents.K8sAgent
  267. } else {
  268. k8sAgent, err = kubernetes.GetAgentOutOfClusterConfig(k8sForm.OutOfClusterConfig)
  269. }
  270. yamlArr := grapher.ImportMultiDocYAML([]byte(release.Manifest))
  271. controllers := grapher.ParseControllers(yamlArr)
  272. retrievedControllers := []interface{}{}
  273. // get current status of each controller
  274. // TODO: refactor with type assertion
  275. for _, c := range controllers {
  276. c.Namespace = form.ReleaseForm.Form.Namespace
  277. switch c.Kind {
  278. case "Deployment":
  279. rc, err := k8sAgent.GetDeployment(c)
  280. if err != nil {
  281. app.handleErrorDataRead(err, w)
  282. return
  283. }
  284. rc.Kind = c.Kind
  285. retrievedControllers = append(retrievedControllers, rc)
  286. case "StatefulSet":
  287. rc, err := k8sAgent.GetStatefulSet(c)
  288. if err != nil {
  289. app.handleErrorDataRead(err, w)
  290. return
  291. }
  292. rc.Kind = c.Kind
  293. retrievedControllers = append(retrievedControllers, rc)
  294. case "DaemonSet":
  295. rc, err := k8sAgent.GetDaemonSet(c)
  296. if err != nil {
  297. app.handleErrorDataRead(err, w)
  298. return
  299. }
  300. rc.Kind = c.Kind
  301. retrievedControllers = append(retrievedControllers, rc)
  302. case "ReplicaSet":
  303. rc, err := k8sAgent.GetReplicaSet(c)
  304. if err != nil {
  305. app.handleErrorDataRead(err, w)
  306. return
  307. }
  308. rc.Kind = c.Kind
  309. retrievedControllers = append(retrievedControllers, rc)
  310. case "CronJob":
  311. rc, err := k8sAgent.GetCronJob(c)
  312. if err != nil {
  313. app.handleErrorDataRead(err, w)
  314. return
  315. }
  316. rc.Kind = c.Kind
  317. retrievedControllers = append(retrievedControllers, rc)
  318. }
  319. }
  320. if err := json.NewEncoder(w).Encode(retrievedControllers); err != nil {
  321. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  322. return
  323. }
  324. }
  325. // HandleGetReleaseAllPods retrieves all pods that are associated with a given release.
  326. func (app *App) HandleGetReleaseAllPods(w http.ResponseWriter, r *http.Request) {
  327. name := chi.URLParam(r, "name")
  328. revision, err := strconv.ParseUint(chi.URLParam(r, "revision"), 0, 64)
  329. form := &forms.GetReleaseForm{
  330. ReleaseForm: &forms.ReleaseForm{
  331. Form: &helm.Form{
  332. Repo: app.Repo,
  333. DigitalOceanOAuth: app.DOConf,
  334. },
  335. },
  336. Name: name,
  337. Revision: int(revision),
  338. }
  339. agent, err := app.getAgentFromQueryParams(
  340. w,
  341. r,
  342. form.ReleaseForm,
  343. form.ReleaseForm.PopulateHelmOptionsFromQueryParams,
  344. )
  345. // errors are handled in app.getAgentFromQueryParams
  346. if err != nil {
  347. return
  348. }
  349. release, err := agent.GetRelease(form.Name, form.Revision)
  350. if err != nil {
  351. app.sendExternalError(err, http.StatusNotFound, HTTPError{
  352. Code: ErrReleaseReadData,
  353. Errors: []string{"release not found"},
  354. }, w)
  355. return
  356. }
  357. vals, err := url.ParseQuery(r.URL.RawQuery)
  358. if err != nil {
  359. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  360. return
  361. }
  362. // get the filter options
  363. k8sForm := &forms.K8sForm{
  364. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  365. Repo: app.Repo,
  366. DigitalOceanOAuth: app.DOConf,
  367. },
  368. }
  369. k8sForm.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  370. k8sForm.DefaultNamespace = form.ReleaseForm.Namespace
  371. // validate the form
  372. if err := app.validator.Struct(k8sForm); err != nil {
  373. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  374. return
  375. }
  376. // create a new kubernetes agent
  377. var k8sAgent *kubernetes.Agent
  378. if app.ServerConf.IsTesting {
  379. k8sAgent = app.TestAgents.K8sAgent
  380. } else {
  381. k8sAgent, err = kubernetes.GetAgentOutOfClusterConfig(k8sForm.OutOfClusterConfig)
  382. }
  383. yamlArr := grapher.ImportMultiDocYAML([]byte(release.Manifest))
  384. controllers := grapher.ParseControllers(yamlArr)
  385. pods := make([]v1.Pod, 0)
  386. // get current status of each controller
  387. for _, c := range controllers {
  388. var selector *metav1.LabelSelector
  389. switch c.Kind {
  390. case "Deployment":
  391. rc, err := k8sAgent.GetDeployment(c)
  392. if err != nil {
  393. app.handleErrorDataRead(err, w)
  394. return
  395. }
  396. selector = rc.Spec.Selector
  397. case "StatefulSet":
  398. rc, err := k8sAgent.GetStatefulSet(c)
  399. if err != nil {
  400. app.handleErrorDataRead(err, w)
  401. return
  402. }
  403. selector = rc.Spec.Selector
  404. case "DaemonSet":
  405. rc, err := k8sAgent.GetDaemonSet(c)
  406. if err != nil {
  407. app.handleErrorDataRead(err, w)
  408. return
  409. }
  410. selector = rc.Spec.Selector
  411. case "ReplicaSet":
  412. rc, err := k8sAgent.GetReplicaSet(c)
  413. if err != nil {
  414. app.handleErrorDataRead(err, w)
  415. return
  416. }
  417. selector = rc.Spec.Selector
  418. case "CronJob":
  419. rc, err := k8sAgent.GetCronJob(c)
  420. if err != nil {
  421. app.handleErrorDataRead(err, w)
  422. return
  423. }
  424. selector = rc.Spec.JobTemplate.Spec.Selector
  425. }
  426. selectors := make([]string, 0)
  427. for key, val := range selector.MatchLabels {
  428. selectors = append(selectors, key+"="+val)
  429. }
  430. podList, err := k8sAgent.GetPodsByLabel(strings.Join(selectors, ","))
  431. if err != nil {
  432. app.handleErrorDataRead(err, w)
  433. return
  434. }
  435. pods = append(pods, podList.Items...)
  436. }
  437. if err := json.NewEncoder(w).Encode(pods); err != nil {
  438. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  439. return
  440. }
  441. }
  442. // HandleListReleaseHistory retrieves a history of releases based on a release name
  443. func (app *App) HandleListReleaseHistory(w http.ResponseWriter, r *http.Request) {
  444. name := chi.URLParam(r, "name")
  445. form := &forms.ListReleaseHistoryForm{
  446. ReleaseForm: &forms.ReleaseForm{
  447. Form: &helm.Form{
  448. Repo: app.Repo,
  449. DigitalOceanOAuth: app.DOConf,
  450. },
  451. },
  452. Name: name,
  453. }
  454. agent, err := app.getAgentFromQueryParams(
  455. w,
  456. r,
  457. form.ReleaseForm,
  458. form.ReleaseForm.PopulateHelmOptionsFromQueryParams,
  459. )
  460. // errors are handled in app.getAgentFromQueryParams
  461. if err != nil {
  462. return
  463. }
  464. release, err := agent.GetReleaseHistory(form.Name)
  465. if err != nil {
  466. app.sendExternalError(err, http.StatusNotFound, HTTPError{
  467. Code: ErrReleaseReadData,
  468. Errors: []string{"release not found"},
  469. }, w)
  470. return
  471. }
  472. if err := json.NewEncoder(w).Encode(release); err != nil {
  473. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  474. return
  475. }
  476. }
  477. // HandleGetReleaseToken retrieves the webhook token of a specific release.
  478. func (app *App) HandleGetReleaseToken(w http.ResponseWriter, r *http.Request) {
  479. name := chi.URLParam(r, "name")
  480. vals, err := url.ParseQuery(r.URL.RawQuery)
  481. namespace := vals["namespace"][0]
  482. clusterID, err := strconv.ParseUint(vals["cluster_id"][0], 10, 64)
  483. if err != nil {
  484. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  485. Code: ErrReleaseReadData,
  486. Errors: []string{"release not found"},
  487. }, w)
  488. }
  489. release, err := app.Repo.Release.ReadRelease(uint(clusterID), name, namespace)
  490. if err != nil {
  491. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  492. Code: ErrReleaseReadData,
  493. Errors: []string{"release not found"},
  494. }, w)
  495. }
  496. releaseExt := release.Externalize()
  497. if err := json.NewEncoder(w).Encode(releaseExt); err != nil {
  498. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  499. return
  500. }
  501. }
  502. type ContainerEnvConfig struct {
  503. Container struct {
  504. Env struct {
  505. Normal map[string]string `yaml:"normal"`
  506. } `yaml:"env"`
  507. } `yaml:"container"`
  508. }
  509. // HandleUpgradeRelease upgrades a release with new values.yaml
  510. func (app *App) HandleUpgradeRelease(w http.ResponseWriter, r *http.Request) {
  511. projID, err := strconv.ParseUint(chi.URLParam(r, "project_id"), 0, 64)
  512. if err != nil || projID == 0 {
  513. app.handleErrorFormDecoding(err, ErrProjectDecode, w)
  514. return
  515. }
  516. name := chi.URLParam(r, "name")
  517. vals, err := url.ParseQuery(r.URL.RawQuery)
  518. if err != nil {
  519. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  520. return
  521. }
  522. form := &forms.UpgradeReleaseForm{
  523. ReleaseForm: &forms.ReleaseForm{
  524. Form: &helm.Form{
  525. Repo: app.Repo,
  526. DigitalOceanOAuth: app.DOConf,
  527. },
  528. },
  529. Name: name,
  530. }
  531. form.ReleaseForm.PopulateHelmOptionsFromQueryParams(
  532. vals,
  533. app.Repo.Cluster,
  534. )
  535. if err := json.NewDecoder(r.Body).Decode(form); err != nil {
  536. app.handleErrorFormDecoding(err, ErrUserDecode, w)
  537. return
  538. }
  539. agent, err := app.getAgentFromReleaseForm(
  540. w,
  541. r,
  542. form.ReleaseForm,
  543. )
  544. // errors are handled in app.getAgentFromBodyParams
  545. if err != nil {
  546. return
  547. }
  548. registries, err := app.Repo.Registry.ListRegistriesByProjectID(uint(projID))
  549. if err != nil {
  550. app.handleErrorDataRead(err, w)
  551. return
  552. }
  553. conf := &helm.UpgradeReleaseConfig{
  554. Name: form.Name,
  555. Cluster: form.ReleaseForm.Cluster,
  556. Repo: *app.Repo,
  557. Registries: registries,
  558. }
  559. rel, err := agent.UpgradeRelease(conf, form.Values, app.DOConf)
  560. if err != nil {
  561. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  562. Code: ErrReleaseDeploy,
  563. Errors: []string{"error upgrading release " + err.Error()},
  564. }, w)
  565. return
  566. }
  567. // update the github actions env if the release exists and is built from source
  568. if cName := rel.Chart.Metadata.Name; cName == "job" || cName == "web" || cName == "worker" {
  569. clusterID, err := strconv.ParseUint(vals["cluster_id"][0], 10, 64)
  570. if err != nil {
  571. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  572. Code: ErrReleaseReadData,
  573. Errors: []string{"release not found"},
  574. }, w)
  575. }
  576. release, err := app.Repo.Release.ReadRelease(uint(clusterID), name, rel.Namespace)
  577. gitAction := release.GitActionConfig
  578. if release != nil && gitAction.ID != 0 {
  579. // parse env into build env
  580. cEnv := &ContainerEnvConfig{}
  581. yaml.Unmarshal([]byte(form.Values), cEnv)
  582. gr, err := app.Repo.GitRepo.ReadGitRepo(gitAction.GitRepoID)
  583. if err != nil {
  584. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  585. Code: ErrReleaseReadData,
  586. Errors: []string{"github repo integration not found"},
  587. }, w)
  588. }
  589. repoSplit := strings.Split(gitAction.GitRepo, "/")
  590. gaRunner := &actions.GithubActions{
  591. GitIntegration: gr,
  592. GitRepoName: repoSplit[1],
  593. GitRepoOwner: repoSplit[0],
  594. Repo: *app.Repo,
  595. GithubConf: app.GithubProjectConf,
  596. WebhookToken: release.WebhookToken,
  597. ProjectID: uint(projID),
  598. ReleaseName: name,
  599. GitBranch: gitAction.GitBranch,
  600. DockerFilePath: gitAction.DockerfilePath,
  601. FolderPath: gitAction.FolderPath,
  602. ImageRepoURL: gitAction.ImageRepoURI,
  603. BuildEnv: cEnv.Container.Env.Normal,
  604. }
  605. err = gaRunner.CreateEnvSecret()
  606. if err != nil {
  607. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  608. Code: ErrReleaseReadData,
  609. Errors: []string{"could not update github secret"},
  610. }, w)
  611. }
  612. }
  613. }
  614. w.WriteHeader(http.StatusOK)
  615. }
  616. // HandleReleaseDeployWebhook upgrades a release when a chart specific webhook is called.
  617. func (app *App) HandleReleaseDeployWebhook(w http.ResponseWriter, r *http.Request) {
  618. token := chi.URLParam(r, "token")
  619. // retrieve release by token
  620. release, err := app.Repo.Release.ReadReleaseByWebhookToken(token)
  621. if err != nil {
  622. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  623. Code: ErrReleaseReadData,
  624. Errors: []string{"release not found with given webhook"},
  625. }, w)
  626. return
  627. }
  628. params := map[string][]string{}
  629. params["cluster_id"] = []string{fmt.Sprint(release.ClusterID)}
  630. params["storage"] = []string{"secret"}
  631. params["namespace"] = []string{release.Namespace}
  632. vals, err := url.ParseQuery(r.URL.RawQuery)
  633. commit := vals["commit"][0]
  634. repository := vals["repository"][0]
  635. if err != nil {
  636. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  637. return
  638. }
  639. form := &forms.UpgradeReleaseForm{
  640. ReleaseForm: &forms.ReleaseForm{
  641. Form: &helm.Form{
  642. Repo: app.Repo,
  643. DigitalOceanOAuth: app.DOConf,
  644. },
  645. },
  646. Name: release.Name,
  647. }
  648. form.ReleaseForm.PopulateHelmOptionsFromQueryParams(
  649. params,
  650. app.Repo.Cluster,
  651. )
  652. agent, err := app.getAgentFromReleaseForm(
  653. w,
  654. r,
  655. form.ReleaseForm,
  656. )
  657. // errors are handled in app.getAgentFromBodyParams
  658. if err != nil {
  659. return
  660. }
  661. image := map[string]interface{}{}
  662. image["repository"] = repository
  663. image["tag"] = commit
  664. rel, err := agent.GetRelease(form.Name, 0)
  665. rel.Config["image"] = image
  666. if rel.Config["auto_deploy"] == false {
  667. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  668. Code: ErrReleaseDeploy,
  669. Errors: []string{"Deploy webhook is disabled for this deployment."},
  670. }, w)
  671. return
  672. }
  673. registries, err := app.Repo.Registry.ListRegistriesByProjectID(uint(form.ReleaseForm.Cluster.ProjectID))
  674. if err != nil {
  675. app.handleErrorDataRead(err, w)
  676. return
  677. }
  678. conf := &helm.UpgradeReleaseConfig{
  679. Name: form.Name,
  680. Cluster: form.ReleaseForm.Cluster,
  681. Repo: *app.Repo,
  682. Registries: registries,
  683. Values: rel.Config,
  684. }
  685. _, err = agent.UpgradeReleaseByValues(conf, app.DOConf)
  686. if err != nil {
  687. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  688. Code: ErrReleaseDeploy,
  689. Errors: []string{"error upgrading release " + err.Error()},
  690. }, w)
  691. return
  692. }
  693. client := *app.segmentClient
  694. client.Enqueue(segment.Track{
  695. UserId: "anonymous",
  696. Event: "Triggered Re-deploy via Webhook",
  697. Properties: segment.NewProperties().
  698. Set("repository", repository),
  699. })
  700. w.WriteHeader(http.StatusOK)
  701. }
  702. // HandleRollbackRelease rolls a release back to a specified revision
  703. func (app *App) HandleRollbackRelease(w http.ResponseWriter, r *http.Request) {
  704. name := chi.URLParam(r, "name")
  705. vals, err := url.ParseQuery(r.URL.RawQuery)
  706. if err != nil {
  707. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  708. return
  709. }
  710. form := &forms.RollbackReleaseForm{
  711. ReleaseForm: &forms.ReleaseForm{
  712. Form: &helm.Form{
  713. Repo: app.Repo,
  714. DigitalOceanOAuth: app.DOConf,
  715. },
  716. },
  717. Name: name,
  718. }
  719. form.ReleaseForm.PopulateHelmOptionsFromQueryParams(
  720. vals,
  721. app.Repo.Cluster,
  722. )
  723. if err := json.NewDecoder(r.Body).Decode(form); err != nil {
  724. app.handleErrorFormDecoding(err, ErrUserDecode, w)
  725. return
  726. }
  727. agent, err := app.getAgentFromReleaseForm(
  728. w,
  729. r,
  730. form.ReleaseForm,
  731. )
  732. // errors are handled in app.getAgentFromBodyParams
  733. if err != nil {
  734. return
  735. }
  736. err = agent.RollbackRelease(form.Name, form.Revision)
  737. if err != nil {
  738. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  739. Code: ErrReleaseDeploy,
  740. Errors: []string{"error rolling back release " + err.Error()},
  741. }, w)
  742. return
  743. }
  744. // get the full release data for GHA updating
  745. rel, err := agent.GetRelease(form.Name, form.Revision)
  746. if err != nil {
  747. app.sendExternalError(err, http.StatusNotFound, HTTPError{
  748. Code: ErrReleaseReadData,
  749. Errors: []string{"release not found"},
  750. }, w)
  751. return
  752. }
  753. // update the github actions env if the release exists and is built from source
  754. if cName := rel.Chart.Metadata.Name; cName == "job" || cName == "web" || cName == "worker" {
  755. clusterID, err := strconv.ParseUint(vals["cluster_id"][0], 10, 64)
  756. if err != nil {
  757. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  758. Code: ErrReleaseReadData,
  759. Errors: []string{"release not found"},
  760. }, w)
  761. }
  762. release, err := app.Repo.Release.ReadRelease(uint(clusterID), name, rel.Namespace)
  763. gitAction := release.GitActionConfig
  764. if release != nil && gitAction.ID != 0 {
  765. // parse env into build env
  766. cEnv := &ContainerEnvConfig{}
  767. rawValues, err := yaml.Marshal(rel.Config)
  768. if err != nil {
  769. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  770. Code: ErrReleaseReadData,
  771. Errors: []string{"could not get values of previous revision"},
  772. }, w)
  773. }
  774. yaml.Unmarshal(rawValues, cEnv)
  775. gr, err := app.Repo.GitRepo.ReadGitRepo(gitAction.GitRepoID)
  776. if err != nil {
  777. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  778. Code: ErrReleaseReadData,
  779. Errors: []string{"github repo integration not found"},
  780. }, w)
  781. }
  782. repoSplit := strings.Split(gitAction.GitRepo, "/")
  783. projID, err := strconv.ParseUint(chi.URLParam(r, "project_id"), 0, 64)
  784. if err != nil || projID == 0 {
  785. app.handleErrorFormDecoding(err, ErrProjectDecode, w)
  786. return
  787. }
  788. gaRunner := &actions.GithubActions{
  789. GitIntegration: gr,
  790. GitRepoName: repoSplit[1],
  791. GitRepoOwner: repoSplit[0],
  792. Repo: *app.Repo,
  793. GithubConf: app.GithubProjectConf,
  794. WebhookToken: release.WebhookToken,
  795. ProjectID: uint(projID),
  796. ReleaseName: name,
  797. GitBranch: gitAction.GitBranch,
  798. DockerFilePath: gitAction.DockerfilePath,
  799. FolderPath: gitAction.FolderPath,
  800. ImageRepoURL: gitAction.ImageRepoURI,
  801. BuildEnv: cEnv.Container.Env.Normal,
  802. }
  803. err = gaRunner.CreateEnvSecret()
  804. if err != nil {
  805. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  806. Code: ErrReleaseReadData,
  807. Errors: []string{"could not update github secret"},
  808. }, w)
  809. }
  810. }
  811. }
  812. w.WriteHeader(http.StatusOK)
  813. }
  814. // ------------------------ Release handler helper functions ------------------------ //
  815. // getAgentFromQueryParams uses the query params to populate a form, and then
  816. // passes that form to the underlying app.getAgentFromReleaseForm to create a new
  817. // Helm agent.
  818. func (app *App) getAgentFromQueryParams(
  819. w http.ResponseWriter,
  820. r *http.Request,
  821. form *forms.ReleaseForm,
  822. // populate uses the query params to populate a form
  823. populate ...func(vals url.Values, repo repository.ClusterRepository) error,
  824. ) (*helm.Agent, error) {
  825. vals, err := url.ParseQuery(r.URL.RawQuery)
  826. if err != nil {
  827. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  828. return nil, err
  829. }
  830. for _, f := range populate {
  831. err := f(vals, app.Repo.Cluster)
  832. if err != nil {
  833. return nil, err
  834. }
  835. }
  836. return app.getAgentFromReleaseForm(w, r, form)
  837. }
  838. // getAgentFromReleaseForm uses a non-validated form to construct a new Helm agent based on
  839. // the userID found in the session and the options required by the Helm agent.
  840. func (app *App) getAgentFromReleaseForm(
  841. w http.ResponseWriter,
  842. r *http.Request,
  843. form *forms.ReleaseForm,
  844. ) (*helm.Agent, error) {
  845. var err error
  846. // validate the form
  847. if err := app.validator.Struct(form); err != nil {
  848. app.handleErrorFormValidation(err, ErrReleaseValidateFields, w)
  849. return nil, err
  850. }
  851. // create a new agent
  852. var agent *helm.Agent
  853. if app.ServerConf.IsTesting {
  854. agent = app.TestAgents.HelmAgent
  855. } else {
  856. agent, err = helm.GetAgentOutOfClusterConfig(form.Form, app.Logger)
  857. }
  858. return agent, err
  859. }
  860. const veleroForm string = `tags:
  861. - hello
  862. tabs:
  863. - name: main
  864. context:
  865. type: cluster
  866. config:
  867. group: velero.io
  868. version: v1
  869. resource: backups
  870. label: Backups
  871. sections:
  872. - name: section_one
  873. contents:
  874. - type: heading
  875. label: 💾 Velero Backups
  876. - type: resource-list
  877. value: |
  878. .items[] | {
  879. name: .metadata.name,
  880. label: .metadata.namespace,
  881. status: .status.phase,
  882. timestamp: .status.completionTimestamp,
  883. message: [
  884. (if .status.volumeSnapshotsAttempted then "\(.status.volumeSnapshotsAttempted) volume snapshots attempted, \(.status.volumeSnapshotsCompleted) completed." else null end),
  885. "Finished \(.status.completionTimestamp).",
  886. "Backup expires on \(.status.expiration)."
  887. ]|join(" "),
  888. data: {
  889. "Included Namespaces": (if .spec.includedNamespaces then .spec.includedNamespaces|join(",") else "* (all)" end),
  890. "Included Resources": (if .spec.includedResources then .spec.includedResources|join(",") else "* (all)" end),
  891. "Storage Location": .spec.storageLocation
  892. }
  893. }`