release_handler.go 27 KB

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