release_handler.go 27 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108
  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. DockerFilePath: gitAction.DockerfilePath,
  600. FolderPath: gitAction.FolderPath,
  601. ImageRepoURL: gitAction.ImageRepoURI,
  602. BuildEnv: cEnv.Container.Env.Normal,
  603. }
  604. err = gaRunner.CreateEnvSecret()
  605. if err != nil {
  606. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  607. Code: ErrReleaseReadData,
  608. Errors: []string{"could not update github secret"},
  609. }, w)
  610. }
  611. }
  612. }
  613. w.WriteHeader(http.StatusOK)
  614. }
  615. // HandleReleaseDeployWebhook upgrades a release when a chart specific webhook is called.
  616. func (app *App) HandleReleaseDeployWebhook(w http.ResponseWriter, r *http.Request) {
  617. token := chi.URLParam(r, "token")
  618. // retrieve release by token
  619. release, err := app.Repo.Release.ReadReleaseByWebhookToken(token)
  620. if err != nil {
  621. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  622. Code: ErrReleaseReadData,
  623. Errors: []string{"release not found with given webhook"},
  624. }, w)
  625. return
  626. }
  627. params := map[string][]string{}
  628. params["cluster_id"] = []string{fmt.Sprint(release.ClusterID)}
  629. params["storage"] = []string{"secret"}
  630. params["namespace"] = []string{release.Namespace}
  631. vals, err := url.ParseQuery(r.URL.RawQuery)
  632. commit := vals["commit"][0]
  633. repository := vals["repository"][0]
  634. if err != nil {
  635. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  636. return
  637. }
  638. form := &forms.UpgradeReleaseForm{
  639. ReleaseForm: &forms.ReleaseForm{
  640. Form: &helm.Form{
  641. Repo: app.Repo,
  642. DigitalOceanOAuth: app.DOConf,
  643. },
  644. },
  645. Name: release.Name,
  646. }
  647. form.ReleaseForm.PopulateHelmOptionsFromQueryParams(
  648. params,
  649. app.Repo.Cluster,
  650. )
  651. agent, err := app.getAgentFromReleaseForm(
  652. w,
  653. r,
  654. form.ReleaseForm,
  655. )
  656. // errors are handled in app.getAgentFromBodyParams
  657. if err != nil {
  658. return
  659. }
  660. image := map[string]interface{}{}
  661. image["repository"] = repository
  662. image["tag"] = commit
  663. rel, err := agent.GetRelease(form.Name, 0)
  664. rel.Config["image"] = image
  665. if rel.Config["auto_deploy"] == false {
  666. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  667. Code: ErrReleaseDeploy,
  668. Errors: []string{"Deploy webhook is disabled for this deployment."},
  669. }, w)
  670. return
  671. }
  672. registries, err := app.Repo.Registry.ListRegistriesByProjectID(uint(form.ReleaseForm.Cluster.ProjectID))
  673. if err != nil {
  674. app.handleErrorDataRead(err, w)
  675. return
  676. }
  677. conf := &helm.UpgradeReleaseConfig{
  678. Name: form.Name,
  679. Cluster: form.ReleaseForm.Cluster,
  680. Repo: *app.Repo,
  681. Registries: registries,
  682. Values: rel.Config,
  683. }
  684. _, err = agent.UpgradeReleaseByValues(conf, app.DOConf)
  685. if err != nil {
  686. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  687. Code: ErrReleaseDeploy,
  688. Errors: []string{"error upgrading release " + err.Error()},
  689. }, w)
  690. return
  691. }
  692. client := *app.segmentClient
  693. client.Enqueue(segment.Track{
  694. UserId: "anonymous",
  695. Event: "Triggered Re-deploy via Webhook",
  696. Properties: segment.NewProperties().
  697. Set("repository", repository),
  698. })
  699. w.WriteHeader(http.StatusOK)
  700. }
  701. // HandleRollbackRelease rolls a release back to a specified revision
  702. func (app *App) HandleRollbackRelease(w http.ResponseWriter, r *http.Request) {
  703. name := chi.URLParam(r, "name")
  704. vals, err := url.ParseQuery(r.URL.RawQuery)
  705. if err != nil {
  706. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  707. return
  708. }
  709. form := &forms.RollbackReleaseForm{
  710. ReleaseForm: &forms.ReleaseForm{
  711. Form: &helm.Form{
  712. Repo: app.Repo,
  713. DigitalOceanOAuth: app.DOConf,
  714. },
  715. },
  716. Name: name,
  717. }
  718. form.ReleaseForm.PopulateHelmOptionsFromQueryParams(
  719. vals,
  720. app.Repo.Cluster,
  721. )
  722. if err := json.NewDecoder(r.Body).Decode(form); err != nil {
  723. app.handleErrorFormDecoding(err, ErrUserDecode, w)
  724. return
  725. }
  726. agent, err := app.getAgentFromReleaseForm(
  727. w,
  728. r,
  729. form.ReleaseForm,
  730. )
  731. // errors are handled in app.getAgentFromBodyParams
  732. if err != nil {
  733. return
  734. }
  735. err = agent.RollbackRelease(form.Name, form.Revision)
  736. if err != nil {
  737. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  738. Code: ErrReleaseDeploy,
  739. Errors: []string{"error rolling back release " + err.Error()},
  740. }, w)
  741. return
  742. }
  743. // get the full release data for GHA updating
  744. rel, err := agent.GetRelease(form.Name, form.Revision)
  745. if err != nil {
  746. app.sendExternalError(err, http.StatusNotFound, HTTPError{
  747. Code: ErrReleaseReadData,
  748. Errors: []string{"release not found"},
  749. }, w)
  750. return
  751. }
  752. // update the github actions env if the release exists and is built from source
  753. if cName := rel.Chart.Metadata.Name; cName == "job" || cName == "web" || cName == "worker" {
  754. clusterID, err := strconv.ParseUint(vals["cluster_id"][0], 10, 64)
  755. if err != nil {
  756. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  757. Code: ErrReleaseReadData,
  758. Errors: []string{"release not found"},
  759. }, w)
  760. }
  761. release, err := app.Repo.Release.ReadRelease(uint(clusterID), name, rel.Namespace)
  762. gitAction := release.GitActionConfig
  763. if release != nil && gitAction.ID != 0 {
  764. // parse env into build env
  765. cEnv := &ContainerEnvConfig{}
  766. rawValues, err := yaml.Marshal(rel.Config)
  767. if err != nil {
  768. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  769. Code: ErrReleaseReadData,
  770. Errors: []string{"could not get values of previous revision"},
  771. }, w)
  772. }
  773. yaml.Unmarshal(rawValues, cEnv)
  774. gr, err := app.Repo.GitRepo.ReadGitRepo(gitAction.GitRepoID)
  775. if err != nil {
  776. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  777. Code: ErrReleaseReadData,
  778. Errors: []string{"github repo integration not found"},
  779. }, w)
  780. }
  781. repoSplit := strings.Split(gitAction.GitRepo, "/")
  782. projID, err := strconv.ParseUint(chi.URLParam(r, "project_id"), 0, 64)
  783. if err != nil || projID == 0 {
  784. app.handleErrorFormDecoding(err, ErrProjectDecode, w)
  785. return
  786. }
  787. gaRunner := &actions.GithubActions{
  788. GitIntegration: gr,
  789. GitRepoName: repoSplit[1],
  790. GitRepoOwner: repoSplit[0],
  791. Repo: *app.Repo,
  792. GithubConf: app.GithubProjectConf,
  793. WebhookToken: release.WebhookToken,
  794. ProjectID: uint(projID),
  795. ReleaseName: name,
  796. DockerFilePath: gitAction.DockerfilePath,
  797. FolderPath: gitAction.FolderPath,
  798. ImageRepoURL: gitAction.ImageRepoURI,
  799. BuildEnv: cEnv.Container.Env.Normal,
  800. }
  801. err = gaRunner.CreateEnvSecret()
  802. if err != nil {
  803. app.sendExternalError(err, http.StatusInternalServerError, HTTPError{
  804. Code: ErrReleaseReadData,
  805. Errors: []string{"could not update github secret"},
  806. }, w)
  807. }
  808. }
  809. }
  810. w.WriteHeader(http.StatusOK)
  811. }
  812. // ------------------------ Release handler helper functions ------------------------ //
  813. // getAgentFromQueryParams uses the query params to populate a form, and then
  814. // passes that form to the underlying app.getAgentFromReleaseForm to create a new
  815. // Helm agent.
  816. func (app *App) getAgentFromQueryParams(
  817. w http.ResponseWriter,
  818. r *http.Request,
  819. form *forms.ReleaseForm,
  820. // populate uses the query params to populate a form
  821. populate ...func(vals url.Values, repo repository.ClusterRepository) error,
  822. ) (*helm.Agent, error) {
  823. vals, err := url.ParseQuery(r.URL.RawQuery)
  824. if err != nil {
  825. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  826. return nil, err
  827. }
  828. for _, f := range populate {
  829. err := f(vals, app.Repo.Cluster)
  830. if err != nil {
  831. return nil, err
  832. }
  833. }
  834. return app.getAgentFromReleaseForm(w, r, form)
  835. }
  836. // getAgentFromReleaseForm uses a non-validated form to construct a new Helm agent based on
  837. // the userID found in the session and the options required by the Helm agent.
  838. func (app *App) getAgentFromReleaseForm(
  839. w http.ResponseWriter,
  840. r *http.Request,
  841. form *forms.ReleaseForm,
  842. ) (*helm.Agent, error) {
  843. var err error
  844. // validate the form
  845. if err := app.validator.Struct(form); err != nil {
  846. app.handleErrorFormValidation(err, ErrReleaseValidateFields, w)
  847. return nil, err
  848. }
  849. // create a new agent
  850. var agent *helm.Agent
  851. if app.ServerConf.IsTesting {
  852. agent = app.TestAgents.HelmAgent
  853. } else {
  854. agent, err = helm.GetAgentOutOfClusterConfig(form.Form, app.Logger)
  855. }
  856. return agent, err
  857. }
  858. const veleroForm string = `tags:
  859. - hello
  860. tabs:
  861. - name: main
  862. context:
  863. type: cluster
  864. config:
  865. group: velero.io
  866. version: v1
  867. resource: backups
  868. label: Backups
  869. sections:
  870. - name: section_one
  871. contents:
  872. - type: heading
  873. label: 💾 Velero Backups
  874. - type: resource-list
  875. value: |
  876. .items[] | {
  877. name: .metadata.name,
  878. label: .metadata.namespace,
  879. status: .status.phase,
  880. timestamp: .status.completionTimestamp,
  881. message: [
  882. (if .status.volumeSnapshotsAttempted then "\(.status.volumeSnapshotsAttempted) volume snapshots attempted, \(.status.volumeSnapshotsCompleted) completed." else null end),
  883. "Finished \(.status.completionTimestamp).",
  884. "Backup expires on \(.status.expiration)."
  885. ]|join(" "),
  886. data: {
  887. "Included Namespaces": (if .spec.includedNamespaces then .spec.includedNamespaces|join(",") else "* (all)" end),
  888. "Included Resources": (if .spec.includedResources then .spec.includedResources|join(",") else "* (all)" end),
  889. "Storage Location": .spec.storageLocation
  890. }
  891. }`