k8s_handler.go 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209
  1. package api
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "net/http"
  6. "net/url"
  7. "github.com/go-chi/chi"
  8. "github.com/gorilla/schema"
  9. "github.com/gorilla/websocket"
  10. "github.com/porter-dev/porter/internal/forms"
  11. "github.com/porter-dev/porter/internal/kubernetes"
  12. "github.com/porter-dev/porter/internal/kubernetes/prometheus"
  13. v1 "k8s.io/api/core/v1"
  14. "k8s.io/client-go/tools/clientcmd"
  15. )
  16. // Enumeration of k8s API error codes, represented as int64
  17. const (
  18. ErrK8sDecode ErrorCode = iota + 600
  19. ErrK8sValidate
  20. ErrEnvDecode
  21. )
  22. var upgrader = websocket.Upgrader{
  23. ReadBufferSize: 1024,
  24. WriteBufferSize: 1024,
  25. }
  26. // HandleListNamespaces retrieves a list of namespaces
  27. func (app *App) HandleListNamespaces(w http.ResponseWriter, r *http.Request) {
  28. vals, err := url.ParseQuery(r.URL.RawQuery)
  29. if err != nil {
  30. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  31. return
  32. }
  33. // get the filter options
  34. form := &forms.K8sForm{
  35. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  36. Repo: app.Repo,
  37. DigitalOceanOAuth: app.DOConf,
  38. },
  39. }
  40. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  41. // validate the form
  42. if err := app.validator.Struct(form); err != nil {
  43. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  44. return
  45. }
  46. // create a new agent
  47. var agent *kubernetes.Agent
  48. if app.ServerConf.IsTesting {
  49. agent = app.TestAgents.K8sAgent
  50. } else {
  51. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  52. }
  53. namespaces, err := agent.ListNamespaces()
  54. if err != nil {
  55. app.handleErrorDataRead(err, w)
  56. return
  57. }
  58. if err := json.NewEncoder(w).Encode(namespaces); err != nil {
  59. app.handleErrorFormDecoding(err, ErrK8sDecode, w)
  60. return
  61. }
  62. }
  63. // HandleCreateNamespace creates a new namespace given the name.
  64. func (app *App) HandleCreateNamespace(w http.ResponseWriter, r *http.Request) {
  65. vals, err := url.ParseQuery(r.URL.RawQuery)
  66. if err != nil {
  67. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  68. return
  69. }
  70. // get the filter options
  71. form := &forms.K8sForm{
  72. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  73. Repo: app.Repo,
  74. DigitalOceanOAuth: app.DOConf,
  75. },
  76. }
  77. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  78. // validate the form
  79. if err := app.validator.Struct(form); err != nil {
  80. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  81. return
  82. }
  83. // create a new agent
  84. var agent *kubernetes.Agent
  85. if app.ServerConf.IsTesting {
  86. agent = app.TestAgents.K8sAgent
  87. } else {
  88. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  89. }
  90. namespace, err := agent.CreateNamespace(vals["name"][0])
  91. if err != nil {
  92. app.handleErrorInternal(err, w)
  93. return
  94. }
  95. if err := json.NewEncoder(w).Encode(namespace); err != nil {
  96. app.handleErrorFormDecoding(err, ErrK8sDecode, w)
  97. return
  98. }
  99. w.WriteHeader(http.StatusOK)
  100. return
  101. }
  102. // HandleDeleteNamespace deletes a namespace given the name.
  103. func (app *App) HandleDeleteNamespace(w http.ResponseWriter, r *http.Request) {
  104. vals, err := url.ParseQuery(r.URL.RawQuery)
  105. if err != nil {
  106. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  107. return
  108. }
  109. // get the filter options
  110. form := &forms.K8sForm{
  111. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  112. Repo: app.Repo,
  113. DigitalOceanOAuth: app.DOConf,
  114. },
  115. }
  116. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  117. // validate the form
  118. if err := app.validator.Struct(form); err != nil {
  119. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  120. return
  121. }
  122. // create a new agent
  123. var agent *kubernetes.Agent
  124. if app.ServerConf.IsTesting {
  125. agent = app.TestAgents.K8sAgent
  126. } else {
  127. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  128. }
  129. if err != nil {
  130. app.handleErrorInternal(err, w)
  131. return
  132. }
  133. err = agent.DeleteNamespace(vals["name"][0])
  134. if err != nil {
  135. app.handleErrorInternal(err, w)
  136. return
  137. }
  138. w.WriteHeader(http.StatusOK)
  139. return
  140. }
  141. // HandleListPodEvents retrieves all events tied to a pod.
  142. func (app *App) HandleListPodEvents(w http.ResponseWriter, r *http.Request) {
  143. vals, err := url.ParseQuery(r.URL.RawQuery)
  144. // get path parameters
  145. namespace := chi.URLParam(r, "namespace")
  146. name := chi.URLParam(r, "name")
  147. if err != nil {
  148. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  149. return
  150. }
  151. // get the filter options
  152. form := &forms.K8sForm{
  153. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  154. Repo: app.Repo,
  155. DigitalOceanOAuth: app.DOConf,
  156. },
  157. }
  158. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  159. // validate the form
  160. if err := app.validator.Struct(form); err != nil {
  161. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  162. return
  163. }
  164. // create a new agent
  165. var agent *kubernetes.Agent
  166. if app.ServerConf.IsTesting {
  167. agent = app.TestAgents.K8sAgent
  168. } else {
  169. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  170. }
  171. events, err := agent.ListEvents(name, namespace)
  172. if err != nil {
  173. app.handleErrorDataRead(err, w)
  174. return
  175. }
  176. if err := json.NewEncoder(w).Encode(events); err != nil {
  177. app.handleErrorFormDecoding(err, ErrK8sDecode, w)
  178. return
  179. }
  180. }
  181. // HandleCreateConfigMap deletes the pod given the name and namespace.
  182. func (app *App) HandleCreateConfigMap(w http.ResponseWriter, r *http.Request) {
  183. vals, err := url.ParseQuery(r.URL.RawQuery)
  184. if err != nil {
  185. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  186. return
  187. }
  188. // get the filter options
  189. form := &forms.K8sForm{
  190. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  191. Repo: app.Repo,
  192. DigitalOceanOAuth: app.DOConf,
  193. },
  194. }
  195. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  196. // validate the form
  197. if err := app.validator.Struct(form); err != nil {
  198. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  199. return
  200. }
  201. // create a new agent
  202. var agent *kubernetes.Agent
  203. if app.ServerConf.IsTesting {
  204. agent = app.TestAgents.K8sAgent
  205. } else {
  206. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  207. }
  208. configMap := &forms.ConfigMapForm{}
  209. if err := json.NewDecoder(r.Body).Decode(configMap); err != nil {
  210. app.handleErrorFormDecoding(err, ErrUserDecode, w)
  211. return
  212. }
  213. secretData := make(map[string][]byte)
  214. for key, rawValue := range configMap.SecretEnvVariables {
  215. // encodedValue := base64.StdEncoding.EncodeToString([]byte(rawValue))
  216. // if err != nil {
  217. // app.handleErrorInternal(err, w)
  218. // return
  219. // }
  220. secretData[key] = []byte(rawValue)
  221. }
  222. // create secret first
  223. _, err = agent.CreateLinkedSecret(configMap.Name, configMap.Namespace, configMap.Name, secretData)
  224. if err != nil {
  225. app.handleErrorInternal(err, w)
  226. return
  227. }
  228. // add all secret env variables to configmap with value PORTERSECRET_${configmap_name}
  229. for key, _ := range configMap.SecretEnvVariables {
  230. configMap.EnvVariables[key] = fmt.Sprintf("PORTERSECRET_%s", configMap.Name)
  231. }
  232. _, err = agent.CreateConfigMap(configMap.Name, configMap.Namespace, configMap.EnvVariables)
  233. if err != nil {
  234. app.handleErrorInternal(err, w)
  235. return
  236. }
  237. if err := json.NewEncoder(w).Encode(configMap); err != nil {
  238. app.handleErrorFormDecoding(err, ErrEnvDecode, w)
  239. return
  240. }
  241. w.WriteHeader(http.StatusOK)
  242. return
  243. }
  244. // HandleListConfigMaps lists all configmaps in a namespace.
  245. func (app *App) HandleListConfigMaps(w http.ResponseWriter, r *http.Request) {
  246. vals, err := url.ParseQuery(r.URL.RawQuery)
  247. if err != nil {
  248. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  249. return
  250. }
  251. // get the filter options
  252. form := &forms.K8sForm{
  253. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  254. Repo: app.Repo,
  255. DigitalOceanOAuth: app.DOConf,
  256. },
  257. }
  258. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  259. // validate the form
  260. if err := app.validator.Struct(form); err != nil {
  261. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  262. return
  263. }
  264. // create a new agent
  265. var agent *kubernetes.Agent
  266. if app.ServerConf.IsTesting {
  267. agent = app.TestAgents.K8sAgent
  268. } else {
  269. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  270. }
  271. configMaps, err := agent.ListConfigMaps(vals["namespace"][0])
  272. if err != nil {
  273. app.handleErrorInternal(err, w)
  274. return
  275. }
  276. if err := json.NewEncoder(w).Encode(configMaps); err != nil {
  277. app.handleErrorFormDecoding(err, ErrEnvDecode, w)
  278. return
  279. }
  280. w.WriteHeader(http.StatusOK)
  281. return
  282. }
  283. // HandleGetConfigMap retreives the configmap given the name and namespace.
  284. func (app *App) HandleGetConfigMap(w http.ResponseWriter, r *http.Request) {
  285. vals, err := url.ParseQuery(r.URL.RawQuery)
  286. if err != nil {
  287. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  288. return
  289. }
  290. // get the filter options
  291. form := &forms.K8sForm{
  292. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  293. Repo: app.Repo,
  294. DigitalOceanOAuth: app.DOConf,
  295. },
  296. }
  297. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  298. // validate the form
  299. if err := app.validator.Struct(form); err != nil {
  300. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  301. return
  302. }
  303. // create a new agent
  304. var agent *kubernetes.Agent
  305. if app.ServerConf.IsTesting {
  306. agent = app.TestAgents.K8sAgent
  307. } else {
  308. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  309. }
  310. configMap, err := agent.GetConfigMap(vals["name"][0], vals["namespace"][0])
  311. if err != nil {
  312. app.handleErrorInternal(err, w)
  313. return
  314. }
  315. if err := json.NewEncoder(w).Encode(configMap); err != nil {
  316. app.handleErrorFormDecoding(err, ErrEnvDecode, w)
  317. return
  318. }
  319. w.WriteHeader(http.StatusOK)
  320. return
  321. }
  322. // HandleDeleteConfigMap deletes the pod given the name and namespace.
  323. func (app *App) HandleDeleteConfigMap(w http.ResponseWriter, r *http.Request) {
  324. vals, err := url.ParseQuery(r.URL.RawQuery)
  325. if err != nil {
  326. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  327. return
  328. }
  329. // get the filter options
  330. form := &forms.K8sForm{
  331. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  332. Repo: app.Repo,
  333. DigitalOceanOAuth: app.DOConf,
  334. },
  335. }
  336. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  337. // validate the form
  338. if err := app.validator.Struct(form); err != nil {
  339. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  340. return
  341. }
  342. // create a new agent
  343. var agent *kubernetes.Agent
  344. if app.ServerConf.IsTesting {
  345. agent = app.TestAgents.K8sAgent
  346. } else {
  347. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  348. }
  349. err = agent.DeleteLinkedSecret(vals["name"][0], vals["namespace"][0])
  350. if err != nil {
  351. app.handleErrorInternal(err, w)
  352. return
  353. }
  354. err = agent.DeleteConfigMap(vals["name"][0], vals["namespace"][0])
  355. if err != nil {
  356. app.handleErrorInternal(err, w)
  357. return
  358. }
  359. w.WriteHeader(http.StatusOK)
  360. return
  361. }
  362. // HandleUpdateConfigMap deletes the pod given the name and namespace.
  363. func (app *App) HandleUpdateConfigMap(w http.ResponseWriter, r *http.Request) {
  364. vals, err := url.ParseQuery(r.URL.RawQuery)
  365. if err != nil {
  366. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  367. return
  368. }
  369. // get the filter options
  370. form := &forms.K8sForm{
  371. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  372. Repo: app.Repo,
  373. DigitalOceanOAuth: app.DOConf,
  374. },
  375. }
  376. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  377. // validate the form
  378. if err := app.validator.Struct(form); err != nil {
  379. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  380. return
  381. }
  382. // create a new agent
  383. var agent *kubernetes.Agent
  384. if app.ServerConf.IsTesting {
  385. agent = app.TestAgents.K8sAgent
  386. } else {
  387. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  388. }
  389. configMap := &forms.ConfigMapForm{}
  390. if err := json.NewDecoder(r.Body).Decode(configMap); err != nil {
  391. app.handleErrorFormDecoding(err, ErrEnvDecode, w)
  392. return
  393. }
  394. secretData := make(map[string][]byte)
  395. for key, rawValue := range configMap.SecretEnvVariables {
  396. // encodedValue, err := base64.StdEncoding.DecodeString(rawValue)
  397. // if err != nil {
  398. // app.handleErrorInternal(err, w)
  399. // return
  400. // }
  401. secretData[key] = []byte(rawValue)
  402. }
  403. // create secret first
  404. err = agent.UpdateLinkedSecret(configMap.Name, configMap.Namespace, configMap.Name, secretData)
  405. if err != nil {
  406. app.handleErrorInternal(err, w)
  407. return
  408. }
  409. // add all secret env variables to configmap with value PORTERSECRET_${configmap_name}
  410. for key, val := range configMap.SecretEnvVariables {
  411. // if val is empty and key does not exist in configmap already, set to empty
  412. if _, found := configMap.EnvVariables[key]; val == "" && !found {
  413. configMap.EnvVariables[key] = ""
  414. } else if val != "" {
  415. configMap.EnvVariables[key] = fmt.Sprintf("PORTERSECRET_%s", configMap.Name)
  416. }
  417. }
  418. err = agent.UpdateConfigMap(configMap.Name, configMap.Namespace, configMap.EnvVariables)
  419. if err != nil {
  420. app.handleErrorInternal(err, w)
  421. return
  422. }
  423. if err := json.NewEncoder(w).Encode(configMap); err != nil {
  424. app.handleErrorFormDecoding(err, ErrEnvDecode, w)
  425. return
  426. }
  427. w.WriteHeader(http.StatusOK)
  428. return
  429. }
  430. // HandleGetPodLogs returns real-time logs of the pod via websockets
  431. // TODO: Refactor repeated calls.
  432. func (app *App) HandleGetPodLogs(w http.ResponseWriter, r *http.Request) {
  433. // get session to retrieve correct kubeconfig
  434. _, err := app.Store.Get(r, app.ServerConf.CookieName)
  435. // get path parameters
  436. namespace := chi.URLParam(r, "namespace")
  437. podName := chi.URLParam(r, "name")
  438. if err != nil {
  439. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  440. return
  441. }
  442. vals, err := url.ParseQuery(r.URL.RawQuery)
  443. if err != nil {
  444. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  445. return
  446. }
  447. // get the filter options
  448. form := &forms.K8sForm{
  449. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  450. Repo: app.Repo,
  451. DigitalOceanOAuth: app.DOConf,
  452. },
  453. }
  454. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  455. // validate the form
  456. if err := app.validator.Struct(form); err != nil {
  457. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  458. return
  459. }
  460. // create a new agent
  461. var agent *kubernetes.Agent
  462. if app.ServerConf.IsTesting {
  463. agent = app.TestAgents.K8sAgent
  464. } else {
  465. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  466. }
  467. upgrader.CheckOrigin = func(r *http.Request) bool { return true }
  468. // upgrade to websocket.
  469. conn, err := upgrader.Upgrade(w, r, nil)
  470. if err != nil {
  471. app.handleErrorUpgradeWebsocket(err, w)
  472. }
  473. err = agent.GetPodLogs(namespace, podName, conn)
  474. if err != nil {
  475. app.handleErrorWebsocketWrite(err, w)
  476. return
  477. }
  478. }
  479. // HandleDeletePod deletes the pod given the name and namespace.
  480. func (app *App) HandleDeletePod(w http.ResponseWriter, r *http.Request) {
  481. // get path parameters
  482. namespace := chi.URLParam(r, "namespace")
  483. name := chi.URLParam(r, "name")
  484. vals, err := url.ParseQuery(r.URL.RawQuery)
  485. if err != nil {
  486. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  487. return
  488. }
  489. // get the filter options
  490. form := &forms.K8sForm{
  491. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  492. Repo: app.Repo,
  493. DigitalOceanOAuth: app.DOConf,
  494. },
  495. }
  496. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  497. // validate the form
  498. if err := app.validator.Struct(form); err != nil {
  499. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  500. return
  501. }
  502. // create a new agent
  503. var agent *kubernetes.Agent
  504. if app.ServerConf.IsTesting {
  505. agent = app.TestAgents.K8sAgent
  506. } else {
  507. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  508. }
  509. err = agent.DeletePod(namespace, name)
  510. if err != nil {
  511. app.handleErrorInternal(err, w)
  512. return
  513. }
  514. w.WriteHeader(http.StatusOK)
  515. return
  516. }
  517. // HandleGetIngress returns the ingress object given the name and namespace.
  518. func (app *App) HandleGetIngress(w http.ResponseWriter, r *http.Request) {
  519. // get session to retrieve correct kubeconfig
  520. _, err := app.Store.Get(r, app.ServerConf.CookieName)
  521. // get path parameters
  522. namespace := chi.URLParam(r, "namespace")
  523. name := chi.URLParam(r, "name")
  524. if err != nil {
  525. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  526. return
  527. }
  528. vals, err := url.ParseQuery(r.URL.RawQuery)
  529. if err != nil {
  530. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  531. return
  532. }
  533. // get the filter options
  534. form := &forms.K8sForm{
  535. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  536. Repo: app.Repo,
  537. DigitalOceanOAuth: app.DOConf,
  538. },
  539. }
  540. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  541. // validate the form
  542. if err := app.validator.Struct(form); err != nil {
  543. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  544. return
  545. }
  546. // create a new agent
  547. var agent *kubernetes.Agent
  548. if app.ServerConf.IsTesting {
  549. agent = app.TestAgents.K8sAgent
  550. } else {
  551. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  552. }
  553. ingress, err := agent.GetIngress(namespace, name)
  554. if err != nil {
  555. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  556. return
  557. }
  558. if err := json.NewEncoder(w).Encode(ingress); err != nil {
  559. app.handleErrorFormDecoding(err, ErrK8sDecode, w)
  560. return
  561. }
  562. }
  563. // HandleListPods returns all pods that match the given selectors
  564. // TODO: Refactor repeated calls.
  565. func (app *App) HandleListPods(w http.ResponseWriter, r *http.Request) {
  566. // get session to retrieve correct kubeconfig
  567. _, err := app.Store.Get(r, app.ServerConf.CookieName)
  568. if err != nil {
  569. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  570. return
  571. }
  572. vals, err := url.ParseQuery(r.URL.RawQuery)
  573. if err != nil {
  574. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  575. return
  576. }
  577. // get the filter options
  578. form := &forms.K8sForm{
  579. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  580. Repo: app.Repo,
  581. DigitalOceanOAuth: app.DOConf,
  582. },
  583. }
  584. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  585. // validate the form
  586. if err := app.validator.Struct(form); err != nil {
  587. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  588. return
  589. }
  590. // create a new agent
  591. var agent *kubernetes.Agent
  592. if app.ServerConf.IsTesting {
  593. agent = app.TestAgents.K8sAgent
  594. } else {
  595. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  596. }
  597. namespace := vals.Get("namespace")
  598. pods := []v1.Pod{}
  599. for _, selector := range vals["selectors"] {
  600. podsList, err := agent.GetPodsByLabel(selector, namespace)
  601. if err != nil {
  602. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  603. return
  604. }
  605. for _, pod := range podsList.Items {
  606. pods = append(pods, pod)
  607. }
  608. }
  609. if err := json.NewEncoder(w).Encode(pods); err != nil {
  610. app.handleErrorFormDecoding(err, ErrK8sDecode, w)
  611. return
  612. }
  613. }
  614. // HandleListJobsByChart lists all jobs belonging to a specific Helm chart
  615. func (app *App) HandleListJobsByChart(w http.ResponseWriter, r *http.Request) {
  616. // get path parameters
  617. namespace := chi.URLParam(r, "namespace")
  618. chart := chi.URLParam(r, "chart")
  619. releaseName := chi.URLParam(r, "release_name")
  620. vals, err := url.ParseQuery(r.URL.RawQuery)
  621. if err != nil {
  622. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  623. return
  624. }
  625. // get the filter options
  626. form := &forms.K8sForm{
  627. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  628. Repo: app.Repo,
  629. DigitalOceanOAuth: app.DOConf,
  630. },
  631. }
  632. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  633. // validate the form
  634. if err := app.validator.Struct(form); err != nil {
  635. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  636. return
  637. }
  638. // create a new agent
  639. var agent *kubernetes.Agent
  640. if app.ServerConf.IsTesting {
  641. agent = app.TestAgents.K8sAgent
  642. } else {
  643. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  644. }
  645. jobs, err := agent.ListJobsByLabel(namespace, kubernetes.Label{
  646. Key: "helm.sh/chart",
  647. Val: chart,
  648. }, kubernetes.Label{
  649. Key: "meta.helm.sh/release-name",
  650. Val: releaseName,
  651. })
  652. if err != nil {
  653. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  654. return
  655. }
  656. if err := json.NewEncoder(w).Encode(jobs); err != nil {
  657. app.handleErrorFormDecoding(err, ErrK8sDecode, w)
  658. return
  659. }
  660. }
  661. // HandleStopJob stops a running job
  662. func (app *App) HandleStopJob(w http.ResponseWriter, r *http.Request) {
  663. // get path parameters
  664. namespace := chi.URLParam(r, "namespace")
  665. name := chi.URLParam(r, "name")
  666. vals, err := url.ParseQuery(r.URL.RawQuery)
  667. if err != nil {
  668. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  669. return
  670. }
  671. // get the filter options
  672. form := &forms.K8sForm{
  673. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  674. Repo: app.Repo,
  675. DigitalOceanOAuth: app.DOConf,
  676. },
  677. }
  678. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  679. // validate the form
  680. if err := app.validator.Struct(form); err != nil {
  681. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  682. return
  683. }
  684. // create a new agent
  685. var agent *kubernetes.Agent
  686. if app.ServerConf.IsTesting {
  687. agent = app.TestAgents.K8sAgent
  688. } else {
  689. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  690. }
  691. err = agent.StopJobWithJobSidecar(namespace, name)
  692. if err != nil {
  693. app.handleErrorInternal(err, w)
  694. return
  695. }
  696. w.WriteHeader(http.StatusOK)
  697. return
  698. }
  699. // HandleListJobPods lists all pods belonging to a specific job
  700. func (app *App) HandleListJobPods(w http.ResponseWriter, r *http.Request) {
  701. // get path parameters
  702. namespace := chi.URLParam(r, "namespace")
  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. // get the filter options
  710. form := &forms.K8sForm{
  711. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  712. Repo: app.Repo,
  713. DigitalOceanOAuth: app.DOConf,
  714. },
  715. }
  716. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  717. // validate the form
  718. if err := app.validator.Struct(form); err != nil {
  719. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  720. return
  721. }
  722. // create a new agent
  723. var agent *kubernetes.Agent
  724. if app.ServerConf.IsTesting {
  725. agent = app.TestAgents.K8sAgent
  726. } else {
  727. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  728. }
  729. pods, err := agent.GetJobPods(namespace, name)
  730. if err != nil {
  731. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  732. return
  733. }
  734. if err := json.NewEncoder(w).Encode(pods); err != nil {
  735. app.handleErrorFormDecoding(err, ErrK8sDecode, w)
  736. return
  737. }
  738. }
  739. // HandleStreamControllerStatus test calls
  740. // TODO: Refactor repeated calls.
  741. func (app *App) HandleStreamControllerStatus(w http.ResponseWriter, r *http.Request) {
  742. // get session to retrieve correct kubeconfig
  743. _, err := app.Store.Get(r, app.ServerConf.CookieName)
  744. if err != nil {
  745. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  746. return
  747. }
  748. vals, err := url.ParseQuery(r.URL.RawQuery)
  749. if err != nil {
  750. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  751. return
  752. }
  753. // get the filter options
  754. form := &forms.K8sForm{
  755. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  756. Repo: app.Repo,
  757. DigitalOceanOAuth: app.DOConf,
  758. },
  759. }
  760. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  761. // validate the form
  762. if err := app.validator.Struct(form); err != nil {
  763. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  764. return
  765. }
  766. // create a new agent
  767. var agent *kubernetes.Agent
  768. if app.ServerConf.IsTesting {
  769. agent = app.TestAgents.K8sAgent
  770. } else {
  771. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  772. }
  773. upgrader.CheckOrigin = func(r *http.Request) bool { return true }
  774. // upgrade to websocket.
  775. conn, err := upgrader.Upgrade(w, r, nil)
  776. if err != nil {
  777. app.handleErrorUpgradeWebsocket(err, w)
  778. }
  779. // get path parameters
  780. kind := chi.URLParam(r, "kind")
  781. err = agent.StreamControllerStatus(conn, kind)
  782. if err != nil {
  783. app.handleErrorWebsocketWrite(err, w)
  784. return
  785. }
  786. }
  787. // HandleDetectPrometheusInstalled detects a prometheus installation in the target cluster
  788. func (app *App) HandleDetectPrometheusInstalled(w http.ResponseWriter, r *http.Request) {
  789. vals, err := url.ParseQuery(r.URL.RawQuery)
  790. if err != nil {
  791. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  792. return
  793. }
  794. // get the filter options
  795. form := &forms.K8sForm{
  796. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  797. Repo: app.Repo,
  798. DigitalOceanOAuth: app.DOConf,
  799. },
  800. }
  801. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  802. // validate the form
  803. if err := app.validator.Struct(form); err != nil {
  804. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  805. return
  806. }
  807. // create a new agent
  808. var agent *kubernetes.Agent
  809. if app.ServerConf.IsTesting {
  810. agent = app.TestAgents.K8sAgent
  811. } else {
  812. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  813. }
  814. // detect prometheus service
  815. _, found, err := prometheus.GetPrometheusService(agent.Clientset)
  816. if !found {
  817. http.NotFound(w, r)
  818. return
  819. }
  820. w.WriteHeader(http.StatusOK)
  821. return
  822. }
  823. // HandleListNGINXIngresses lists all NGINX ingresses in a target cluster
  824. func (app *App) HandleListNGINXIngresses(w http.ResponseWriter, r *http.Request) {
  825. vals, err := url.ParseQuery(r.URL.RawQuery)
  826. if err != nil {
  827. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  828. return
  829. }
  830. // get the filter options
  831. form := &forms.K8sForm{
  832. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  833. Repo: app.Repo,
  834. DigitalOceanOAuth: app.DOConf,
  835. },
  836. }
  837. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  838. // validate the form
  839. if err := app.validator.Struct(form); err != nil {
  840. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  841. return
  842. }
  843. // create a new agent
  844. var agent *kubernetes.Agent
  845. if app.ServerConf.IsTesting {
  846. agent = app.TestAgents.K8sAgent
  847. } else {
  848. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  849. }
  850. res, err := prometheus.GetIngressesWithNGINXAnnotation(agent.Clientset)
  851. if err != nil {
  852. app.handleErrorInternal(err, w)
  853. return
  854. }
  855. w.WriteHeader(http.StatusOK)
  856. if err := json.NewEncoder(w).Encode(res); err != nil {
  857. app.handleErrorFormDecoding(err, ErrK8sDecode, w)
  858. return
  859. }
  860. }
  861. func (app *App) HandleGetPodMetrics(w http.ResponseWriter, r *http.Request) {
  862. vals, err := url.ParseQuery(r.URL.RawQuery)
  863. if err != nil {
  864. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  865. return
  866. }
  867. // get the filter options
  868. form := &forms.MetricsQueryForm{
  869. K8sForm: &forms.K8sForm{
  870. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  871. Repo: app.Repo,
  872. DigitalOceanOAuth: app.DOConf,
  873. },
  874. },
  875. QueryOpts: &prometheus.QueryOpts{},
  876. }
  877. form.K8sForm.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  878. // decode from JSON to form value
  879. decoder := schema.NewDecoder()
  880. decoder.IgnoreUnknownKeys(true)
  881. if err := decoder.Decode(form.QueryOpts, vals); err != nil {
  882. app.handleErrorFormDecoding(err, ErrProjectDecode, w)
  883. return
  884. }
  885. // validate the form
  886. if err := app.validator.Struct(form); err != nil {
  887. app.handleErrorFormValidation(err, ErrK8sValidate, w)
  888. return
  889. }
  890. // create a new agent
  891. var agent *kubernetes.Agent
  892. if app.ServerConf.IsTesting {
  893. agent = app.TestAgents.K8sAgent
  894. } else {
  895. agent, err = kubernetes.GetAgentOutOfClusterConfig(form.OutOfClusterConfig)
  896. }
  897. // get prometheus service
  898. promSvc, found, err := prometheus.GetPrometheusService(agent.Clientset)
  899. if err != nil {
  900. app.handleErrorInternal(err, w)
  901. return
  902. }
  903. if !found {
  904. app.handleErrorInternal(err, w)
  905. return
  906. }
  907. rawQuery, err := prometheus.QueryPrometheus(agent.Clientset, promSvc, form.QueryOpts)
  908. if err != nil {
  909. app.handleErrorInternal(err, w)
  910. return
  911. }
  912. fmt.Fprint(w, string(rawQuery))
  913. }
  914. type KubeconfigResponse struct {
  915. Kubeconfig []byte `json:"kubeconfig"`
  916. }
  917. func (app *App) HandleGetTemporaryKubeconfig(w http.ResponseWriter, r *http.Request) {
  918. vals, err := url.ParseQuery(r.URL.RawQuery)
  919. if err != nil {
  920. app.handleErrorFormDecoding(err, ErrReleaseDecode, w)
  921. return
  922. }
  923. // get the filter options
  924. form := &forms.K8sForm{
  925. OutOfClusterConfig: &kubernetes.OutOfClusterConfig{
  926. Repo: app.Repo,
  927. DigitalOceanOAuth: app.DOConf,
  928. },
  929. }
  930. form.PopulateK8sOptionsFromQueryParams(vals, app.Repo.Cluster)
  931. // get the API config
  932. apiConf, err := form.OutOfClusterConfig.CreateRawConfigFromCluster()
  933. if err != nil {
  934. app.handleErrorFormDecoding(err, ErrProjectDecode, w)
  935. return
  936. }
  937. bytes, err := clientcmd.Write(*apiConf)
  938. res := &KubeconfigResponse{
  939. Kubeconfig: bytes,
  940. }
  941. if err := json.NewEncoder(w).Encode(res); err != nil {
  942. app.handleErrorFormDecoding(err, ErrK8sDecode, w)
  943. return
  944. }
  945. }