release_handler.go 26 KB

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