agent.go 45 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047
  1. package kubernetes
  2. import (
  3. "bufio"
  4. "bytes"
  5. "compress/gzip"
  6. "context"
  7. "encoding/base64"
  8. "encoding/json"
  9. "fmt"
  10. "io"
  11. "io/ioutil"
  12. "strconv"
  13. "strings"
  14. "sync"
  15. "time"
  16. goerrors "errors"
  17. "github.com/porter-dev/porter/api/server/shared/websocket"
  18. "github.com/porter-dev/porter/internal/models"
  19. "github.com/porter-dev/porter/internal/registry"
  20. "github.com/porter-dev/porter/internal/repository"
  21. "golang.org/x/oauth2"
  22. "github.com/porter-dev/porter/internal/helm/grapher"
  23. appsv1 "k8s.io/api/apps/v1"
  24. batchv1 "k8s.io/api/batch/v1"
  25. v1 "k8s.io/api/core/v1"
  26. v1beta1 "k8s.io/api/extensions/v1beta1"
  27. netv1 "k8s.io/api/networking/v1"
  28. netv1beta1 "k8s.io/api/networking/v1beta1"
  29. "k8s.io/apimachinery/pkg/api/errors"
  30. metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  31. "k8s.io/apimachinery/pkg/fields"
  32. "k8s.io/apimachinery/pkg/runtime"
  33. "k8s.io/apimachinery/pkg/runtime/schema"
  34. "k8s.io/apimachinery/pkg/types"
  35. "k8s.io/apimachinery/pkg/watch"
  36. "k8s.io/cli-runtime/pkg/genericclioptions"
  37. "k8s.io/client-go/informers"
  38. "k8s.io/client-go/kubernetes"
  39. "k8s.io/client-go/rest"
  40. "k8s.io/client-go/tools/cache"
  41. "k8s.io/client-go/tools/remotecommand"
  42. "k8s.io/kubectl/pkg/scheme"
  43. rspb "github.com/stefanmcshane/helm/pkg/release"
  44. istiov1beta1 "istio.io/client-go/pkg/apis/networking/v1beta1"
  45. versionedclient "istio.io/client-go/pkg/clientset/versioned"
  46. )
  47. // Agent is a Kubernetes agent for performing operations that interact with the
  48. // api server
  49. type Agent struct {
  50. RESTClientGetter genericclioptions.RESTClientGetter
  51. Clientset kubernetes.Interface
  52. }
  53. type Message struct {
  54. EventType string `json:"event_type"`
  55. Object interface{}
  56. Kind string
  57. }
  58. type ListOptions struct {
  59. FieldSelector string
  60. }
  61. type AuthError struct{}
  62. func (e *AuthError) Error() string {
  63. return "Unauthorized error"
  64. }
  65. // UpdateClientset updates the Agent's Clientset (this refreshes auth tokens)
  66. func (a *Agent) UpdateClientset() error {
  67. restConf, err := a.RESTClientGetter.ToRESTConfig()
  68. if err != nil {
  69. return err
  70. }
  71. clientset, err := kubernetes.NewForConfig(restConf)
  72. if err != nil {
  73. return err
  74. }
  75. a.Clientset = clientset
  76. return nil
  77. }
  78. // CreateConfigMap creates the configmap given the key-value pairs and namespace
  79. func (a *Agent) CreateConfigMap(name string, namespace string, configMap map[string]string) (*v1.ConfigMap, error) {
  80. return a.Clientset.CoreV1().ConfigMaps(namespace).Create(
  81. context.TODO(),
  82. &v1.ConfigMap{
  83. ObjectMeta: metav1.ObjectMeta{
  84. Name: name,
  85. Namespace: namespace,
  86. Labels: map[string]string{
  87. "porter": "true",
  88. },
  89. },
  90. Data: configMap,
  91. },
  92. metav1.CreateOptions{},
  93. )
  94. }
  95. func (a *Agent) CreateVersionedConfigMap(name, namespace string, version uint, configMap map[string]string, apps ...string) (*v1.ConfigMap, error) {
  96. return a.Clientset.CoreV1().ConfigMaps(namespace).Create(
  97. context.TODO(),
  98. &v1.ConfigMap{
  99. ObjectMeta: metav1.ObjectMeta{
  100. Name: fmt.Sprintf("%s.v%d", name, version),
  101. Namespace: namespace,
  102. Labels: map[string]string{
  103. "owner": "porter",
  104. "envgroup": name,
  105. "version": fmt.Sprintf("%d", version),
  106. },
  107. Annotations: map[string]string{
  108. PorterAppAnnotationName: strings.Join(apps, ","),
  109. },
  110. },
  111. Data: configMap,
  112. },
  113. metav1.CreateOptions{},
  114. )
  115. }
  116. const PorterAppAnnotationName = "porter.run/apps"
  117. func (a *Agent) AddApplicationToVersionedConfigMap(cm *v1.ConfigMap, appName string) (*v1.ConfigMap, error) {
  118. annons := cm.Annotations
  119. if annons == nil {
  120. annons = make(map[string]string)
  121. }
  122. appAnnon, appAnnonExists := annons[PorterAppAnnotationName]
  123. if !appAnnonExists || appAnnon == "" {
  124. annons[PorterAppAnnotationName] = appName
  125. } else {
  126. appStrArr := strings.Split(appAnnon, ",")
  127. foundApp := false
  128. for _, appStr := range appStrArr {
  129. if appStr == appName {
  130. foundApp = true
  131. }
  132. }
  133. if !foundApp {
  134. annons[PorterAppAnnotationName] = fmt.Sprintf("%s,%s", appAnnon, appName)
  135. }
  136. }
  137. cm.SetAnnotations(annons)
  138. return a.Clientset.CoreV1().ConfigMaps(cm.Namespace).Update(
  139. context.TODO(),
  140. cm,
  141. metav1.UpdateOptions{},
  142. )
  143. }
  144. func (a *Agent) RemoveApplicationFromVersionedConfigMap(cm *v1.ConfigMap, appName string) (*v1.ConfigMap, error) {
  145. annons := cm.Annotations
  146. if annons == nil {
  147. annons = make(map[string]string)
  148. }
  149. appAnn, appAnnExists := annons[PorterAppAnnotationName]
  150. if !appAnnExists {
  151. return nil, IsNotFoundError
  152. }
  153. appStrArr := strings.Split(appAnn, ",")
  154. newStrArr := make([]string, 0)
  155. for _, appStr := range appStrArr {
  156. if appStr != appName {
  157. newStrArr = append(newStrArr, appStr)
  158. }
  159. }
  160. annons[PorterAppAnnotationName] = strings.Join(newStrArr, ",")
  161. cm.SetAnnotations(annons)
  162. return a.Clientset.CoreV1().ConfigMaps(cm.Namespace).Update(
  163. context.TODO(),
  164. cm,
  165. metav1.UpdateOptions{},
  166. )
  167. }
  168. func (a *Agent) CreateLinkedVersionedSecret(name, namespace, cmName string, version uint, data map[string][]byte) (*v1.Secret, error) {
  169. return a.Clientset.CoreV1().Secrets(namespace).Create(
  170. context.TODO(),
  171. &v1.Secret{
  172. ObjectMeta: metav1.ObjectMeta{
  173. Name: fmt.Sprintf("%s.v%d", name, version),
  174. Namespace: namespace,
  175. Labels: map[string]string{
  176. "owner": "porter",
  177. "envgroup": name,
  178. "version": fmt.Sprintf("%d", version),
  179. "configmap": cmName,
  180. },
  181. },
  182. Data: data,
  183. },
  184. metav1.CreateOptions{},
  185. )
  186. }
  187. // CreateLinkedSecret creates a secret given the key-value pairs and namespace. Values are
  188. // base64 encoded
  189. func (a *Agent) CreateLinkedSecret(name, namespace, cmName string, data map[string][]byte) (*v1.Secret, error) {
  190. return a.Clientset.CoreV1().Secrets(namespace).Create(
  191. context.TODO(),
  192. &v1.Secret{
  193. ObjectMeta: metav1.ObjectMeta{
  194. Name: name,
  195. Namespace: namespace,
  196. Labels: map[string]string{
  197. "porter": "true",
  198. "configmap": cmName,
  199. },
  200. },
  201. Data: data,
  202. },
  203. metav1.CreateOptions{},
  204. )
  205. }
  206. type mergeConfigMapData struct {
  207. Data map[string]*string `json:"data"`
  208. }
  209. // UpdateConfigMap updates the configmap given its name and namespace
  210. func (a *Agent) UpdateConfigMap(name string, namespace string, configMap map[string]string) (*v1.ConfigMap, error) {
  211. cmData := make(map[string]*string)
  212. for key, val := range configMap {
  213. valCopy := val
  214. cmData[key] = &valCopy
  215. if len(val) == 0 {
  216. cmData[key] = nil
  217. }
  218. }
  219. mergeCM := &mergeConfigMapData{
  220. Data: cmData,
  221. }
  222. patchBytes, err := json.Marshal(mergeCM)
  223. if err != nil {
  224. return nil, err
  225. }
  226. return a.Clientset.CoreV1().ConfigMaps(namespace).Patch(
  227. context.Background(),
  228. name,
  229. types.MergePatchType,
  230. patchBytes,
  231. metav1.PatchOptions{},
  232. )
  233. }
  234. type mergeLinkedSecretData struct {
  235. Data map[string]*[]byte `json:"data"`
  236. }
  237. // UpdateLinkedSecret updates the secret given its name and namespace
  238. func (a *Agent) UpdateLinkedSecret(name, namespace, cmName string, data map[string][]byte) error {
  239. secretData := make(map[string]*[]byte)
  240. for key, val := range data {
  241. valCopy := val
  242. secretData[key] = &valCopy
  243. if len(val) == 0 {
  244. secretData[key] = nil
  245. }
  246. }
  247. mergeSecret := &mergeLinkedSecretData{
  248. Data: secretData,
  249. }
  250. patchBytes, err := json.Marshal(mergeSecret)
  251. if err != nil {
  252. return err
  253. }
  254. _, err = a.Clientset.CoreV1().Secrets(namespace).Patch(
  255. context.TODO(),
  256. name,
  257. types.MergePatchType,
  258. patchBytes,
  259. metav1.PatchOptions{},
  260. )
  261. return err
  262. }
  263. // DeleteConfigMap deletes the configmap given its name and namespace
  264. func (a *Agent) DeleteConfigMap(name string, namespace string) error {
  265. return a.Clientset.CoreV1().ConfigMaps(namespace).Delete(
  266. context.TODO(),
  267. name,
  268. metav1.DeleteOptions{},
  269. )
  270. }
  271. // DeleteLinkedSecret deletes the secret given its name and namespace
  272. func (a *Agent) DeleteLinkedSecret(name, namespace string) error {
  273. return a.Clientset.CoreV1().Secrets(namespace).Delete(
  274. context.TODO(),
  275. name,
  276. metav1.DeleteOptions{},
  277. )
  278. }
  279. func (a *Agent) ListVersionedConfigMaps(name string, namespace string) ([]v1.ConfigMap, error) {
  280. listResp, err := a.Clientset.CoreV1().ConfigMaps(namespace).List(
  281. context.Background(),
  282. metav1.ListOptions{
  283. LabelSelector: fmt.Sprintf("envgroup=%s", name),
  284. },
  285. )
  286. if err != nil {
  287. return nil, err
  288. }
  289. return listResp.Items, nil
  290. }
  291. func (a *Agent) DeleteVersionedConfigMap(name string, namespace string) error {
  292. return a.Clientset.CoreV1().ConfigMaps(namespace).DeleteCollection(
  293. context.Background(),
  294. metav1.DeleteOptions{},
  295. metav1.ListOptions{
  296. LabelSelector: fmt.Sprintf("envgroup=%s", name),
  297. },
  298. )
  299. }
  300. func (a *Agent) DeleteVersionedSecret(name string, namespace string) error {
  301. return a.Clientset.CoreV1().Secrets(namespace).DeleteCollection(
  302. context.Background(),
  303. metav1.DeleteOptions{},
  304. metav1.ListOptions{
  305. LabelSelector: fmt.Sprintf("envgroup=%s", name),
  306. },
  307. )
  308. }
  309. func (a *Agent) ListAllVersionedConfigMaps(namespace string) ([]v1.ConfigMap, error) {
  310. listResp, err := a.Clientset.CoreV1().ConfigMaps(namespace).List(
  311. context.Background(),
  312. metav1.ListOptions{
  313. LabelSelector: fmt.Sprintf("envgroup"),
  314. },
  315. )
  316. if err != nil {
  317. return nil, err
  318. }
  319. // only keep the latest version for each configmap
  320. latestMap := make(map[string]v1.ConfigMap)
  321. for _, configmap := range listResp.Items {
  322. egName, egNameExists := configmap.Labels["envgroup"]
  323. if !egNameExists {
  324. continue
  325. }
  326. id := fmt.Sprintf("%s/%s", configmap.Namespace, egName)
  327. if currLatest, exists := latestMap[id]; exists {
  328. // get version
  329. currVersionStr, currVersionExists := currLatest.Labels["version"]
  330. versionStr, versionExists := configmap.Labels["version"]
  331. if versionExists && currVersionExists {
  332. currVersion, currErr := strconv.Atoi(currVersionStr)
  333. version, err := strconv.Atoi(versionStr)
  334. if currErr == nil && err == nil && currVersion < version {
  335. latestMap[id] = configmap
  336. }
  337. }
  338. } else {
  339. latestMap[id] = configmap
  340. }
  341. }
  342. res := make([]v1.ConfigMap, 0)
  343. for _, cm := range latestMap {
  344. res = append(res, cm)
  345. }
  346. return res, nil
  347. }
  348. // GetConfigMap retrieves the configmap given its name and namespace
  349. func (a *Agent) GetConfigMap(name string, namespace string) (*v1.ConfigMap, error) {
  350. return a.Clientset.CoreV1().ConfigMaps(namespace).Get(
  351. context.TODO(),
  352. name,
  353. metav1.GetOptions{},
  354. )
  355. }
  356. func (a *Agent) GetVersionedConfigMap(name, namespace string, version uint) (*v1.ConfigMap, error) {
  357. listResp, err := a.Clientset.CoreV1().ConfigMaps(namespace).List(
  358. context.Background(),
  359. metav1.ListOptions{
  360. LabelSelector: fmt.Sprintf("envgroup=%s,version=%d", name, version),
  361. },
  362. )
  363. if err != nil {
  364. return nil, err
  365. }
  366. if listResp.Items == nil || len(listResp.Items) == 0 {
  367. return nil, IsNotFoundError
  368. }
  369. // if the length of the list is greater than 1, return an error -- this shouldn't happen
  370. if len(listResp.Items) > 1 {
  371. return nil, fmt.Errorf("multiple configmaps found while searching for %s/%s and version %d", namespace, name, version)
  372. }
  373. return &listResp.Items[0], nil
  374. }
  375. func (a *Agent) GetLatestVersionedConfigMap(name, namespace string) (*v1.ConfigMap, uint, error) {
  376. listResp, err := a.Clientset.CoreV1().ConfigMaps(namespace).List(
  377. context.Background(),
  378. metav1.ListOptions{
  379. LabelSelector: fmt.Sprintf("envgroup=%s", name),
  380. },
  381. )
  382. if err != nil {
  383. return nil, 0, err
  384. }
  385. if listResp.Items == nil || len(listResp.Items) == 0 {
  386. return nil, 0, IsNotFoundError
  387. }
  388. // iterate through the configmaps and get the greatest version
  389. var res *v1.ConfigMap
  390. var latestVersion uint
  391. for _, configmap := range listResp.Items {
  392. if res == nil {
  393. versionStr, versionExists := configmap.Labels["version"]
  394. if !versionExists {
  395. continue
  396. }
  397. version, err := strconv.Atoi(versionStr)
  398. if err != nil {
  399. continue
  400. }
  401. latestV := configmap
  402. res = &latestV
  403. latestVersion = uint(version)
  404. } else {
  405. // get version
  406. versionStr, versionExists := configmap.Labels["version"]
  407. currVersionStr, currVersionExists := res.Labels["version"]
  408. if versionExists && currVersionExists {
  409. currVersion, currErr := strconv.Atoi(currVersionStr)
  410. version, err := strconv.Atoi(versionStr)
  411. if currErr == nil && err == nil && currVersion < version {
  412. latestV := configmap
  413. res = &latestV
  414. latestVersion = uint(version)
  415. }
  416. }
  417. }
  418. }
  419. if res == nil {
  420. return nil, 0, IsNotFoundError
  421. }
  422. return res, latestVersion, nil
  423. }
  424. func (a *Agent) GetLatestVersionedSecret(name, namespace string) (*v1.Secret, uint, error) {
  425. listResp, err := a.Clientset.CoreV1().Secrets(namespace).List(
  426. context.Background(),
  427. metav1.ListOptions{
  428. LabelSelector: fmt.Sprintf("envgroup=%s", name),
  429. },
  430. )
  431. if err != nil {
  432. return nil, 0, err
  433. }
  434. if listResp.Items == nil || len(listResp.Items) == 0 {
  435. return nil, 0, IsNotFoundError
  436. }
  437. // iterate through the configmaps and get the greatest version
  438. var res *v1.Secret
  439. var latestVersion uint
  440. for _, secret := range listResp.Items {
  441. if res == nil {
  442. versionStr, versionExists := secret.Labels["version"]
  443. if !versionExists {
  444. continue
  445. }
  446. version, err := strconv.Atoi(versionStr)
  447. if err != nil {
  448. continue
  449. }
  450. latestV := secret
  451. res = &latestV
  452. latestVersion = uint(version)
  453. } else {
  454. // get version
  455. versionStr, versionExists := secret.Labels["version"]
  456. currVersionStr, currVersionExists := res.Labels["version"]
  457. if versionExists && currVersionExists {
  458. currVersion, currErr := strconv.Atoi(currVersionStr)
  459. version, err := strconv.Atoi(versionStr)
  460. if currErr == nil && err == nil && currVersion < version {
  461. latestV := secret
  462. res = &latestV
  463. latestVersion = uint(version)
  464. }
  465. }
  466. }
  467. }
  468. if res == nil {
  469. return nil, 0, IsNotFoundError
  470. }
  471. return res, latestVersion, nil
  472. }
  473. // GetSecret retrieves the secret given its name and namespace
  474. func (a *Agent) GetSecret(name string, namespace string) (*v1.Secret, error) {
  475. return a.Clientset.CoreV1().Secrets(namespace).Get(
  476. context.TODO(),
  477. name,
  478. metav1.GetOptions{},
  479. )
  480. }
  481. // ListConfigMaps simply lists namespaces
  482. func (a *Agent) ListConfigMaps(namespace string) (*v1.ConfigMapList, error) {
  483. return a.Clientset.CoreV1().ConfigMaps(namespace).List(
  484. context.TODO(),
  485. metav1.ListOptions{
  486. LabelSelector: "porter",
  487. },
  488. )
  489. }
  490. // ListEvents lists the events of a given object.
  491. func (a *Agent) ListEvents(name string, namespace string) (*v1.EventList, error) {
  492. return a.Clientset.CoreV1().Events(namespace).List(
  493. context.TODO(),
  494. metav1.ListOptions{
  495. FieldSelector: fmt.Sprintf("involvedObject.name=%s,involvedObject.namespace=%s", name, namespace),
  496. },
  497. )
  498. }
  499. // ListNamespaces simply lists namespaces
  500. func (a *Agent) ListNamespaces() (*v1.NamespaceList, error) {
  501. return a.Clientset.CoreV1().Namespaces().List(
  502. context.TODO(),
  503. metav1.ListOptions{},
  504. )
  505. }
  506. // CreateNamespace creates a namespace with the given name.
  507. func (a *Agent) CreateNamespace(name string, labels map[string]string) (*v1.Namespace, error) {
  508. // check if namespace exists
  509. checkNS, err := a.Clientset.CoreV1().Namespaces().Get(
  510. context.TODO(),
  511. name,
  512. metav1.GetOptions{},
  513. )
  514. if err == nil && checkNS != nil {
  515. if checkNS.Status.Phase == v1.NamespaceTerminating {
  516. // edge case for when the same namespace was previously created
  517. // but was deleted and is currently in the "Terminating" phase
  518. // let us wait for a maximum of 10 seconds
  519. timeNow := time.Now().Add(10 * time.Second)
  520. stillTerminating := true
  521. for {
  522. _, err := a.Clientset.CoreV1().Namespaces().Get(
  523. context.TODO(),
  524. name,
  525. metav1.GetOptions{},
  526. )
  527. if err != nil && errors.IsNotFound(err) {
  528. stillTerminating = false
  529. break
  530. }
  531. time.Sleep(time.Second)
  532. if time.Now().After(timeNow) {
  533. break
  534. }
  535. }
  536. if stillTerminating {
  537. // the namespace has been in the "Terminating" phase
  538. return nil, fmt.Errorf("cannot create namespace %s, stuck in \"Terminating\" phase", name)
  539. }
  540. } else {
  541. return checkNS, nil
  542. }
  543. }
  544. namespace := &v1.Namespace{
  545. ObjectMeta: metav1.ObjectMeta{
  546. Name: name,
  547. },
  548. }
  549. if len(labels) > 0 {
  550. namespace.SetLabels(labels)
  551. }
  552. return a.Clientset.CoreV1().Namespaces().Create(
  553. context.TODO(),
  554. namespace,
  555. metav1.CreateOptions{},
  556. )
  557. }
  558. // GetNamespace gets the namespace given the name
  559. func (a *Agent) GetNamespace(name string) (*v1.Namespace, error) {
  560. ns, err := a.Clientset.CoreV1().Namespaces().Get(
  561. context.Background(),
  562. name,
  563. metav1.GetOptions{},
  564. )
  565. if err != nil {
  566. return nil, err
  567. }
  568. return ns, nil
  569. }
  570. // DeleteNamespace deletes the namespace given the name.
  571. func (a *Agent) DeleteNamespace(name string) error {
  572. // check if namespace exists
  573. checkNS, err := a.Clientset.CoreV1().Namespaces().Get(
  574. context.TODO(),
  575. name,
  576. metav1.GetOptions{},
  577. )
  578. // if the namespace is not found, don't return an error.
  579. if err != nil && errors.IsNotFound(err) {
  580. return nil
  581. }
  582. // if the namespace was found but is in the "Terminating" phase
  583. // we should ignore it and not return an error
  584. if checkNS != nil && checkNS.Status.Phase == v1.NamespaceTerminating {
  585. return nil
  586. }
  587. return a.Clientset.CoreV1().Namespaces().Delete(
  588. context.TODO(),
  589. name,
  590. metav1.DeleteOptions{},
  591. )
  592. }
  593. func (a *Agent) GetPorterAgent() (*appsv1.Deployment, error) {
  594. depl, err := a.Clientset.AppsV1().Deployments("porter-agent-system").Get(
  595. context.TODO(),
  596. "porter-agent-controller-manager",
  597. metav1.GetOptions{},
  598. )
  599. if err != nil && errors.IsNotFound(err) {
  600. return nil, IsNotFoundError
  601. }
  602. return depl, err
  603. }
  604. // ListJobsByLabel lists jobs in a namespace matching a label
  605. type Label struct {
  606. Key string
  607. Val string
  608. }
  609. func (a *Agent) ListJobsByLabel(namespace string, labels ...Label) ([]batchv1.Job, error) {
  610. selectors := make([]string, 0)
  611. for _, label := range labels {
  612. selectors = append(selectors, fmt.Sprintf("%s=%s", label.Key, label.Val))
  613. }
  614. resp, err := a.Clientset.BatchV1().Jobs(namespace).List(
  615. context.TODO(),
  616. metav1.ListOptions{
  617. LabelSelector: strings.Join(selectors, ","),
  618. },
  619. )
  620. if err != nil {
  621. return nil, err
  622. }
  623. return resp.Items, nil
  624. }
  625. // StreamJobs streams a list of jobs to the websocket writer, closing the connection once all jobs have been sent
  626. func (a *Agent) StreamJobs(namespace string, selectors string, rw *websocket.WebsocketSafeReadWriter) error {
  627. run := func() error {
  628. errorchan := make(chan error)
  629. ctx, cancel := context.WithCancel(context.Background())
  630. defer cancel()
  631. var wg sync.WaitGroup
  632. var once sync.Once
  633. var err error
  634. wg.Add(2)
  635. go func() {
  636. wg.Wait()
  637. close(errorchan)
  638. }()
  639. go func() {
  640. defer func() {
  641. if r := recover(); r != nil {
  642. // TODO: add method to alert on panic
  643. return
  644. }
  645. }()
  646. // listens for websocket closing handshake
  647. defer wg.Done()
  648. for {
  649. if _, _, err := rw.ReadMessage(); err != nil {
  650. errorchan <- nil
  651. return
  652. }
  653. }
  654. }()
  655. go func() {
  656. defer func() {
  657. if r := recover(); r != nil {
  658. // TODO: add method to alert on panic
  659. return
  660. }
  661. }()
  662. // listens for websocket closing handshake
  663. defer wg.Done()
  664. continueVal := ""
  665. for {
  666. if ctx.Err() != nil {
  667. errorchan <- nil
  668. return
  669. }
  670. labelSelector := "meta.helm.sh/release-name"
  671. if selectors != "" {
  672. labelSelector = selectors
  673. }
  674. jobs, err := a.Clientset.BatchV1().Jobs(namespace).List(
  675. ctx,
  676. metav1.ListOptions{
  677. Limit: 100,
  678. Continue: continueVal,
  679. LabelSelector: labelSelector,
  680. },
  681. )
  682. if err != nil {
  683. errorchan <- err
  684. return
  685. }
  686. for _, job := range jobs.Items {
  687. err := rw.WriteJSON(job)
  688. if err != nil {
  689. errorchan <- err
  690. return
  691. }
  692. }
  693. if jobs.Continue == "" {
  694. // we have reached the end of the list of jobs
  695. break
  696. } else {
  697. // start pagination
  698. continueVal = jobs.Continue
  699. }
  700. }
  701. // at this point, we can return the status finished
  702. err := rw.WriteJSON(map[string]interface{}{
  703. "streamStatus": "finished",
  704. })
  705. errorchan <- err
  706. }()
  707. for err = range errorchan {
  708. once.Do(func() {
  709. rw.Close()
  710. cancel()
  711. })
  712. }
  713. return err
  714. }
  715. return a.RunWebsocketTask(run)
  716. }
  717. // DeleteJob deletes the job in the given name and namespace.
  718. func (a *Agent) DeleteJob(name, namespace string) error {
  719. return a.Clientset.BatchV1().Jobs(namespace).Delete(
  720. context.TODO(),
  721. name,
  722. metav1.DeleteOptions{},
  723. )
  724. }
  725. // GetJobPods lists all pods belonging to a job in a namespace
  726. func (a *Agent) GetJobPods(namespace, jobName string) ([]v1.Pod, error) {
  727. resp, err := a.Clientset.CoreV1().Pods(namespace).List(
  728. context.TODO(),
  729. metav1.ListOptions{
  730. LabelSelector: fmt.Sprintf("%s=%s", "job-name", jobName),
  731. },
  732. )
  733. if err != nil {
  734. return nil, err
  735. }
  736. return resp.Items, nil
  737. }
  738. // GetIngress gets ingress given the name and namespace
  739. func (a *Agent) GetExtensionsV1Beta1Ingress(namespace string, name string) (*v1beta1.Ingress, error) {
  740. resp, err := a.Clientset.ExtensionsV1beta1().Ingresses(namespace).Get(
  741. context.TODO(),
  742. name,
  743. metav1.GetOptions{},
  744. )
  745. if err != nil && errors.IsNotFound(err) {
  746. return nil, IsNotFoundError
  747. } else if err != nil {
  748. return nil, err
  749. }
  750. return resp, nil
  751. }
  752. func (a *Agent) GetNetworkingV1Ingress(namespace string, name string) (*netv1.Ingress, error) {
  753. resp, err := a.Clientset.NetworkingV1().Ingresses(namespace).Get(
  754. context.TODO(),
  755. name,
  756. metav1.GetOptions{},
  757. )
  758. if err != nil && errors.IsNotFound(err) {
  759. return nil, IsNotFoundError
  760. } else if err != nil {
  761. return nil, err
  762. }
  763. return resp, nil
  764. }
  765. func (a *Agent) GetNetworkingV1Beta1Ingress(namespace string, name string) (*netv1beta1.Ingress, error) {
  766. resp, err := a.Clientset.NetworkingV1beta1().Ingresses(namespace).Get(
  767. context.TODO(),
  768. name,
  769. metav1.GetOptions{},
  770. )
  771. if err != nil && errors.IsNotFound(err) {
  772. return nil, IsNotFoundError
  773. } else if err != nil {
  774. return nil, err
  775. }
  776. return resp, nil
  777. }
  778. func (a *Agent) GetIstioIngress(namespace, name string) (*istiov1beta1.Gateway, error) {
  779. restConf, err := a.RESTClientGetter.ToRESTConfig()
  780. if err != nil {
  781. return nil, err
  782. }
  783. clientset, err := versionedclient.NewForConfig(restConf)
  784. if err != nil {
  785. return nil, err
  786. }
  787. gateway, err := clientset.NetworkingV1beta1().Gateways(namespace).Get(
  788. context.Background(), name, metav1.GetOptions{},
  789. )
  790. if err != nil {
  791. return nil, err
  792. }
  793. return gateway, nil
  794. }
  795. var IsNotFoundError = fmt.Errorf("not found")
  796. type BadRequestError struct {
  797. msg string
  798. }
  799. func (e *BadRequestError) Error() string {
  800. return e.msg
  801. }
  802. // GetDeployment gets the deployment given the name and namespace
  803. func (a *Agent) GetDeployment(c grapher.Object) (*appsv1.Deployment, error) {
  804. res, err := a.Clientset.AppsV1().Deployments(c.Namespace).Get(
  805. context.TODO(),
  806. c.Name,
  807. metav1.GetOptions{},
  808. )
  809. if err != nil && errors.IsNotFound(err) {
  810. return nil, IsNotFoundError
  811. } else if err != nil {
  812. return nil, err
  813. }
  814. res.Kind = c.Kind
  815. return res, nil
  816. }
  817. // GetStatefulSet gets the statefulset given the name and namespace
  818. func (a *Agent) GetStatefulSet(c grapher.Object) (*appsv1.StatefulSet, error) {
  819. res, err := a.Clientset.AppsV1().StatefulSets(c.Namespace).Get(
  820. context.TODO(),
  821. c.Name,
  822. metav1.GetOptions{},
  823. )
  824. if err != nil && errors.IsNotFound(err) {
  825. return nil, IsNotFoundError
  826. } else if err != nil {
  827. return nil, err
  828. }
  829. res.Kind = c.Kind
  830. return res, nil
  831. }
  832. // GetReplicaSet gets the replicaset given the name and namespace
  833. func (a *Agent) GetReplicaSet(c grapher.Object) (*appsv1.ReplicaSet, error) {
  834. res, err := a.Clientset.AppsV1().ReplicaSets(c.Namespace).Get(
  835. context.TODO(),
  836. c.Name,
  837. metav1.GetOptions{},
  838. )
  839. if err != nil && errors.IsNotFound(err) {
  840. return nil, IsNotFoundError
  841. } else if err != nil {
  842. return nil, err
  843. }
  844. res.Kind = c.Kind
  845. return res, nil
  846. }
  847. // GetDaemonSet gets the daemonset by name and namespace
  848. func (a *Agent) GetDaemonSet(c grapher.Object) (*appsv1.DaemonSet, error) {
  849. res, err := a.Clientset.AppsV1().DaemonSets(c.Namespace).Get(
  850. context.TODO(),
  851. c.Name,
  852. metav1.GetOptions{},
  853. )
  854. if err != nil && errors.IsNotFound(err) {
  855. return nil, IsNotFoundError
  856. } else if err != nil {
  857. return nil, err
  858. }
  859. res.Kind = c.Kind
  860. return res, nil
  861. }
  862. // GetJob gets the job by name and namespace
  863. func (a *Agent) GetJob(c grapher.Object) (*batchv1.Job, error) {
  864. res, err := a.Clientset.BatchV1().Jobs(c.Namespace).Get(
  865. context.TODO(),
  866. c.Name,
  867. metav1.GetOptions{},
  868. )
  869. if err != nil && errors.IsNotFound(err) {
  870. return nil, IsNotFoundError
  871. } else if err != nil {
  872. return nil, err
  873. }
  874. res.Kind = c.Kind
  875. return res, nil
  876. }
  877. // GetCronJob gets the CronJob by name and namespace
  878. func (a *Agent) GetCronJob(c grapher.Object) (*batchv1.CronJob, error) {
  879. res, err := a.Clientset.BatchV1().CronJobs(c.Namespace).Get(
  880. context.TODO(),
  881. c.Name,
  882. metav1.GetOptions{},
  883. )
  884. if err != nil && errors.IsNotFound(err) {
  885. return nil, IsNotFoundError
  886. } else if err != nil {
  887. return nil, err
  888. }
  889. res.Kind = c.Kind
  890. return res, nil
  891. }
  892. // GetPodsByLabel retrieves pods with matching labels
  893. func (a *Agent) GetPodsByLabel(selector string, namespace string) (*v1.PodList, error) {
  894. // Search in all namespaces for matching pods
  895. return a.Clientset.CoreV1().Pods(namespace).List(
  896. context.TODO(),
  897. metav1.ListOptions{
  898. LabelSelector: selector,
  899. },
  900. )
  901. }
  902. // GetPodByName retrieves a single instance of pod with given name
  903. func (a *Agent) GetPodByName(name string, namespace string) (*v1.Pod, error) {
  904. // Get pod by name
  905. pod, err := a.Clientset.CoreV1().Pods(namespace).Get(
  906. context.TODO(),
  907. name,
  908. metav1.GetOptions{},
  909. )
  910. if err != nil && errors.IsNotFound(err) {
  911. return nil, IsNotFoundError
  912. }
  913. if err != nil {
  914. return nil, err
  915. }
  916. return pod, nil
  917. }
  918. // DeletePod deletes a pod by name and namespace
  919. func (a *Agent) DeletePod(namespace string, name string) error {
  920. err := a.Clientset.CoreV1().Pods(namespace).Delete(
  921. context.TODO(),
  922. name,
  923. metav1.DeleteOptions{},
  924. )
  925. if err != nil && errors.IsNotFound(err) {
  926. return IsNotFoundError
  927. }
  928. return err
  929. }
  930. // GetPodLogs streams real-time logs from a given pod.
  931. func (a *Agent) GetPodLogs(namespace string, name string, selectedContainer string, rw *websocket.WebsocketSafeReadWriter) error {
  932. // get the pod to read in the list of contains
  933. pod, err := a.Clientset.CoreV1().Pods(namespace).Get(
  934. context.Background(),
  935. name,
  936. metav1.GetOptions{},
  937. )
  938. if err != nil && errors.IsNotFound(err) {
  939. return IsNotFoundError
  940. } else if err != nil {
  941. return fmt.Errorf("Cannot get logs from pod %s: %s", name, err.Error())
  942. }
  943. // see if container is ready and able to open a stream. If not, wait for container
  944. // to be ready.
  945. err, _ = a.waitForPod(pod)
  946. if err != nil && goerrors.Is(err, IsNotFoundError) {
  947. return IsNotFoundError
  948. } else if err != nil {
  949. return fmt.Errorf("Cannot get logs from pod %s: %s", name, err.Error())
  950. }
  951. container := pod.Spec.Containers[0].Name
  952. if len(selectedContainer) > 0 {
  953. container = selectedContainer
  954. }
  955. tails := int64(400)
  956. // follow logs
  957. podLogOpts := v1.PodLogOptions{
  958. Follow: true,
  959. TailLines: &tails,
  960. Container: container,
  961. }
  962. req := a.Clientset.CoreV1().Pods(namespace).GetLogs(name, &podLogOpts)
  963. podLogs, err := req.Stream(context.TODO())
  964. // in the case of bad request errors, such as if the pod is stuck in "ContainerCreating",
  965. // we'd like to pass this through to the client.
  966. if err != nil && errors.IsBadRequest(err) {
  967. return &BadRequestError{err.Error()}
  968. } else if err != nil {
  969. return fmt.Errorf("Cannot open log stream for pod %s: %s", name, err.Error())
  970. }
  971. r := bufio.NewReader(podLogs)
  972. errorchan := make(chan error)
  973. var wg sync.WaitGroup
  974. var once sync.Once
  975. wg.Add(2)
  976. go func() {
  977. wg.Wait()
  978. close(errorchan)
  979. }()
  980. go func() {
  981. defer func() {
  982. if r := recover(); r != nil {
  983. // TODO: add method to alert on panic
  984. return
  985. }
  986. }()
  987. // listens for websocket closing handshake
  988. defer wg.Done()
  989. for {
  990. if _, _, err := rw.ReadMessage(); err != nil {
  991. errorchan <- nil
  992. return
  993. }
  994. }
  995. }()
  996. go func() {
  997. defer func() {
  998. if r := recover(); r != nil {
  999. // TODO: add method to alert on panic
  1000. return
  1001. }
  1002. }()
  1003. defer wg.Done()
  1004. for {
  1005. bytes, err := r.ReadBytes('\n')
  1006. if err != nil {
  1007. errorchan <- err
  1008. return
  1009. }
  1010. if _, writeErr := rw.Write(bytes); writeErr != nil {
  1011. errorchan <- writeErr
  1012. return
  1013. }
  1014. }
  1015. }()
  1016. for err = range errorchan {
  1017. // only call these methods a single time
  1018. once.Do(func() {
  1019. rw.Close()
  1020. podLogs.Close()
  1021. })
  1022. }
  1023. return err
  1024. }
  1025. // GetPodLogs streams real-time logs from a given pod.
  1026. func (a *Agent) GetPreviousPodLogs(namespace string, name string, selectedContainer string) ([]string, error) {
  1027. // get the pod to read in the list of contains
  1028. pod, err := a.Clientset.CoreV1().Pods(namespace).Get(
  1029. context.Background(),
  1030. name,
  1031. metav1.GetOptions{},
  1032. )
  1033. if err != nil && errors.IsNotFound(err) {
  1034. return nil, IsNotFoundError
  1035. } else if err != nil {
  1036. return nil, fmt.Errorf("Cannot get logs from pod %s: %s", name, err.Error())
  1037. }
  1038. container := pod.Spec.Containers[0].Name
  1039. if len(selectedContainer) > 0 {
  1040. container = selectedContainer
  1041. }
  1042. tails := int64(400)
  1043. // follow logs
  1044. podLogOpts := v1.PodLogOptions{
  1045. Follow: true,
  1046. TailLines: &tails,
  1047. Container: container,
  1048. Previous: true,
  1049. }
  1050. req := a.Clientset.CoreV1().Pods(namespace).GetLogs(name, &podLogOpts)
  1051. podLogs, err := req.Stream(context.TODO())
  1052. // in the case of bad request errors, such as if the pod is stuck in "ContainerCreating",
  1053. // we'd like to pass this through to the client.
  1054. if err != nil && strings.Contains(err.Error(), "not found") {
  1055. return nil, IsNotFoundError
  1056. }
  1057. if err != nil && errors.IsBadRequest(err) {
  1058. return nil, &BadRequestError{err.Error()}
  1059. } else if err != nil {
  1060. return nil, fmt.Errorf("Cannot open log stream for pod %s: %s", name, err.Error())
  1061. }
  1062. defer podLogs.Close()
  1063. r := bufio.NewReader(podLogs)
  1064. var logs []string
  1065. for {
  1066. line, err := r.ReadString('\n')
  1067. logs = append(logs, line)
  1068. if err == io.EOF {
  1069. break
  1070. } else if err != nil {
  1071. return nil, err
  1072. }
  1073. }
  1074. return logs, nil
  1075. }
  1076. // StopJobWithJobSidecar sends a termination signal to a job running with a sidecar
  1077. func (a *Agent) StopJobWithJobSidecar(namespace, name string) error {
  1078. jobPods, err := a.GetJobPods(namespace, name)
  1079. if err != nil {
  1080. return err
  1081. }
  1082. podName := jobPods[0].ObjectMeta.Name
  1083. restConf, err := a.RESTClientGetter.ToRESTConfig()
  1084. restConf.GroupVersion = &schema.GroupVersion{
  1085. Group: "api",
  1086. Version: "v1",
  1087. }
  1088. restConf.NegotiatedSerializer = runtime.NewSimpleNegotiatedSerializer(runtime.SerializerInfo{})
  1089. restClient, err := rest.RESTClientFor(restConf)
  1090. if err != nil {
  1091. return err
  1092. }
  1093. req := restClient.Post().
  1094. Resource("pods").
  1095. Name(podName).
  1096. Namespace(namespace).
  1097. SubResource("exec")
  1098. req.Param("command", "./signal.sh")
  1099. req.Param("container", "sidecar")
  1100. req.Param("stdin", "true")
  1101. req.Param("stdout", "false")
  1102. req.Param("tty", "false")
  1103. exec, err := remotecommand.NewSPDYExecutor(restConf, "POST", req.URL())
  1104. if err != nil {
  1105. return err
  1106. }
  1107. return exec.Stream(remotecommand.StreamOptions{
  1108. Tty: false,
  1109. Stdin: strings.NewReader("./signal.sh"),
  1110. })
  1111. }
  1112. // RunWebsocketTask will run a websocket task. If the websocket returns an anauthorized error, it will restart
  1113. // the task some number of times until failing
  1114. func (a *Agent) RunWebsocketTask(task func() error) error {
  1115. lastTime := int64(0)
  1116. for {
  1117. if err := a.UpdateClientset(); err != nil {
  1118. return err
  1119. }
  1120. err := task()
  1121. if err == nil {
  1122. return nil
  1123. }
  1124. if !goerrors.Is(err, &AuthError{}) {
  1125. return err
  1126. }
  1127. if time.Now().Unix()-lastTime < 60 { // don't regenerate connection if too many unauthorized errors
  1128. return err
  1129. }
  1130. lastTime = time.Now().Unix()
  1131. }
  1132. }
  1133. // StreamControllerStatus streams controller status. Supports Deployment, StatefulSet, ReplicaSet, and DaemonSet
  1134. // TODO: Support Jobs
  1135. func (a *Agent) StreamControllerStatus(kind string, selectors string, rw *websocket.WebsocketSafeReadWriter) error {
  1136. run := func() error {
  1137. // selectors is an array of max length 1. StreamControllerStatus accepts calls without the selectors argument.
  1138. // selectors argument is a single string with comma separated key=value pairs. (e.g. "app=porter,porter=true")
  1139. tweakListOptionsFunc := func(options *metav1.ListOptions) {
  1140. options.LabelSelector = selectors
  1141. }
  1142. factory := informers.NewSharedInformerFactoryWithOptions(
  1143. a.Clientset,
  1144. 0,
  1145. informers.WithTweakListOptions(tweakListOptionsFunc),
  1146. )
  1147. var informer cache.SharedInformer
  1148. // Spins up an informer depending on kind. Convert to lowercase for robustness
  1149. switch strings.ToLower(kind) {
  1150. case "deployment":
  1151. informer = factory.Apps().V1().Deployments().Informer()
  1152. case "statefulset":
  1153. informer = factory.Apps().V1().StatefulSets().Informer()
  1154. case "replicaset":
  1155. informer = factory.Apps().V1().ReplicaSets().Informer()
  1156. case "daemonset":
  1157. informer = factory.Apps().V1().DaemonSets().Informer()
  1158. case "job":
  1159. informer = factory.Batch().V1().Jobs().Informer()
  1160. case "cronjob":
  1161. informer = factory.Batch().V1beta1().CronJobs().Informer()
  1162. case "namespace":
  1163. informer = factory.Core().V1().Namespaces().Informer()
  1164. case "pod":
  1165. informer = factory.Core().V1().Pods().Informer()
  1166. }
  1167. stopper := make(chan struct{})
  1168. errorchan := make(chan error)
  1169. var wg sync.WaitGroup
  1170. var once sync.Once
  1171. var err error
  1172. wg.Add(2)
  1173. go func() {
  1174. wg.Wait()
  1175. close(errorchan)
  1176. }()
  1177. go func() {
  1178. defer func() {
  1179. if r := recover(); r != nil {
  1180. // TODO: add method to alert on panic
  1181. return
  1182. }
  1183. }()
  1184. // listens for websocket closing handshake
  1185. defer wg.Done()
  1186. for {
  1187. if _, _, err := rw.ReadMessage(); err != nil {
  1188. errorchan <- nil
  1189. return
  1190. }
  1191. }
  1192. }()
  1193. go func() {
  1194. defer func() {
  1195. if r := recover(); r != nil {
  1196. // TODO: add method to alert on panic
  1197. return
  1198. }
  1199. }()
  1200. // listens for websocket closing handshake
  1201. defer wg.Done()
  1202. informer.SetWatchErrorHandler(func(r *cache.Reflector, err error) {
  1203. if strings.HasSuffix(err.Error(), ": Unauthorized") {
  1204. errorchan <- &AuthError{}
  1205. }
  1206. })
  1207. informer.AddEventHandler(cache.ResourceEventHandlerFuncs{
  1208. UpdateFunc: func(oldObj, newObj interface{}) {
  1209. msg := Message{
  1210. EventType: "UPDATE",
  1211. Object: newObj,
  1212. Kind: strings.ToLower(kind),
  1213. }
  1214. err := rw.WriteJSON(msg)
  1215. if err != nil {
  1216. errorchan <- err
  1217. }
  1218. },
  1219. AddFunc: func(obj interface{}) {
  1220. msg := Message{
  1221. EventType: "ADD",
  1222. Object: obj,
  1223. Kind: strings.ToLower(kind),
  1224. }
  1225. err := rw.WriteJSON(msg)
  1226. if err != nil {
  1227. errorchan <- err
  1228. }
  1229. },
  1230. DeleteFunc: func(obj interface{}) {
  1231. msg := Message{
  1232. EventType: "DELETE",
  1233. Object: obj,
  1234. Kind: strings.ToLower(kind),
  1235. }
  1236. err := rw.WriteJSON(msg)
  1237. if err != nil {
  1238. errorchan <- err
  1239. }
  1240. },
  1241. })
  1242. informer.Run(stopper)
  1243. }()
  1244. for err = range errorchan {
  1245. once.Do(func() {
  1246. close(stopper)
  1247. rw.Close()
  1248. })
  1249. }
  1250. return err
  1251. }
  1252. return a.RunWebsocketTask(run)
  1253. }
  1254. var b64 = base64.StdEncoding
  1255. var magicGzip = []byte{0x1f, 0x8b, 0x08}
  1256. func decodeRelease(data string) (*rspb.Release, error) {
  1257. // base64 decode string
  1258. b, err := b64.DecodeString(data)
  1259. if err != nil {
  1260. return nil, err
  1261. }
  1262. // For backwards compatibility with releases that were stored before
  1263. // compression was introduced we skip decompression if the
  1264. // gzip magic header is not found
  1265. if bytes.Equal(b[0:3], magicGzip) {
  1266. r, err := gzip.NewReader(bytes.NewReader(b))
  1267. if err != nil {
  1268. return nil, err
  1269. }
  1270. defer r.Close()
  1271. b2, err := ioutil.ReadAll(r)
  1272. if err != nil {
  1273. return nil, err
  1274. }
  1275. b = b2
  1276. }
  1277. var rls rspb.Release
  1278. // unmarshal release object bytes
  1279. if err := json.Unmarshal(b, &rls); err != nil {
  1280. return nil, err
  1281. }
  1282. return &rls, nil
  1283. }
  1284. func contains(s []string, str string) bool {
  1285. for _, v := range s {
  1286. if v == str {
  1287. return true
  1288. }
  1289. }
  1290. return false
  1291. }
  1292. func ParseSecretToHelmRelease(secret v1.Secret, chartList []string) (*rspb.Release, bool, error) {
  1293. if secret.Type != "helm.sh/release.v1" {
  1294. return nil, true, nil
  1295. }
  1296. releaseData, ok := secret.Data["release"]
  1297. if !ok {
  1298. return nil, true, fmt.Errorf("release field not found")
  1299. }
  1300. helm_object, err := decodeRelease(string(releaseData))
  1301. if err != nil {
  1302. return nil, true, err
  1303. }
  1304. if len(chartList) > 0 && !contains(chartList, helm_object.Name) {
  1305. return nil, true, nil
  1306. }
  1307. return helm_object, false, nil
  1308. }
  1309. func (a *Agent) StreamHelmReleases(namespace string, chartList []string, selectors string, rw *websocket.WebsocketSafeReadWriter) error {
  1310. run := func() error {
  1311. tweakListOptionsFunc := func(options *metav1.ListOptions) {
  1312. options.LabelSelector = selectors
  1313. }
  1314. factory := informers.NewSharedInformerFactoryWithOptions(
  1315. a.Clientset,
  1316. 0,
  1317. informers.WithTweakListOptions(tweakListOptionsFunc),
  1318. informers.WithNamespace(namespace),
  1319. )
  1320. informer := factory.Core().V1().Secrets().Informer()
  1321. stopper := make(chan struct{})
  1322. errorchan := make(chan error)
  1323. var wg sync.WaitGroup
  1324. var once sync.Once
  1325. var err error
  1326. wg.Add(2)
  1327. go func() {
  1328. wg.Wait()
  1329. close(errorchan)
  1330. }()
  1331. go func() {
  1332. defer func() {
  1333. if r := recover(); r != nil {
  1334. // TODO: add method to alert on panic
  1335. return
  1336. }
  1337. }()
  1338. // listens for websocket closing handshake
  1339. defer wg.Done()
  1340. for {
  1341. if _, _, err := rw.ReadMessage(); err != nil {
  1342. errorchan <- nil
  1343. return
  1344. }
  1345. }
  1346. }()
  1347. go func() {
  1348. defer func() {
  1349. if r := recover(); r != nil {
  1350. // TODO: add method to alert on panic
  1351. return
  1352. }
  1353. }()
  1354. // listens for websocket closing handshake
  1355. defer wg.Done()
  1356. informer.SetWatchErrorHandler(func(r *cache.Reflector, err error) {
  1357. if strings.HasSuffix(err.Error(), ": Unauthorized") {
  1358. errorchan <- &AuthError{}
  1359. }
  1360. })
  1361. informer.AddEventHandler(cache.ResourceEventHandlerFuncs{
  1362. UpdateFunc: func(oldObj, newObj interface{}) {
  1363. secretObj, ok := newObj.(*v1.Secret)
  1364. if !ok {
  1365. errorchan <- fmt.Errorf("could not cast to secret")
  1366. return
  1367. }
  1368. helm_object, isNotHelmRelease, err := ParseSecretToHelmRelease(*secretObj, chartList)
  1369. if isNotHelmRelease && err == nil {
  1370. return
  1371. }
  1372. if err != nil {
  1373. errorchan <- err
  1374. return
  1375. }
  1376. msg := Message{
  1377. EventType: "UPDATE",
  1378. Object: helm_object,
  1379. }
  1380. rw.WriteJSON(msg)
  1381. },
  1382. AddFunc: func(obj interface{}) {
  1383. secretObj, ok := obj.(*v1.Secret)
  1384. if !ok {
  1385. errorchan <- fmt.Errorf("could not cast to secret")
  1386. return
  1387. }
  1388. helm_object, isNotHelmRelease, err := ParseSecretToHelmRelease(*secretObj, chartList)
  1389. if isNotHelmRelease && err == nil {
  1390. return
  1391. }
  1392. if err != nil {
  1393. errorchan <- err
  1394. return
  1395. }
  1396. msg := Message{
  1397. EventType: "ADD",
  1398. Object: helm_object,
  1399. }
  1400. rw.WriteJSON(msg)
  1401. },
  1402. DeleteFunc: func(obj interface{}) {
  1403. secretObj, ok := obj.(*v1.Secret)
  1404. if !ok {
  1405. errorchan <- fmt.Errorf("could not cast to secret")
  1406. return
  1407. }
  1408. helm_object, isNotHelmRelease, err := ParseSecretToHelmRelease(*secretObj, chartList)
  1409. if isNotHelmRelease && err == nil {
  1410. return
  1411. }
  1412. if err != nil {
  1413. errorchan <- err
  1414. return
  1415. }
  1416. msg := Message{
  1417. EventType: "DELETE",
  1418. Object: helm_object,
  1419. }
  1420. rw.WriteJSON(msg)
  1421. },
  1422. })
  1423. informer.Run(stopper)
  1424. }()
  1425. for err = range errorchan {
  1426. once.Do(func() {
  1427. close(stopper)
  1428. rw.Close()
  1429. })
  1430. }
  1431. return err
  1432. }
  1433. return a.RunWebsocketTask(run)
  1434. }
  1435. func (a *Agent) StreamPorterAgentLokiLog(
  1436. labels []string,
  1437. startTime string,
  1438. searchParam string,
  1439. limit uint32,
  1440. rw *websocket.WebsocketSafeReadWriter,
  1441. ) error {
  1442. run := func() error {
  1443. errorchan := make(chan error)
  1444. var wg sync.WaitGroup
  1445. var once sync.Once
  1446. var err error
  1447. wg.Add(2)
  1448. go func() {
  1449. wg.Wait()
  1450. close(errorchan)
  1451. }()
  1452. go func() {
  1453. defer func() {
  1454. if r := recover(); r != nil {
  1455. // TODO: add method to alert on panic
  1456. return
  1457. }
  1458. }()
  1459. defer wg.Done()
  1460. // listens for websocket closing handshake
  1461. for {
  1462. if _, _, err := rw.ReadMessage(); err != nil {
  1463. errorchan <- nil
  1464. return
  1465. }
  1466. }
  1467. }()
  1468. go func() {
  1469. defer func() {
  1470. if r := recover(); r != nil {
  1471. // TODO: add method to alert on panic
  1472. return
  1473. }
  1474. }()
  1475. defer wg.Done()
  1476. podList, err := a.Clientset.CoreV1().Pods("porter-agent-system").List(context.Background(), metav1.ListOptions{
  1477. LabelSelector: "control-plane=controller-manager",
  1478. })
  1479. if err != nil {
  1480. errorchan <- err
  1481. return
  1482. }
  1483. if len(podList.Items) == 0 {
  1484. errorchan <- fmt.Errorf("no porter agent pods found")
  1485. return
  1486. }
  1487. pod := podList.Items[0]
  1488. restConf, err := a.RESTClientGetter.ToRESTConfig()
  1489. if err != nil {
  1490. errorchan <- err
  1491. return
  1492. }
  1493. req := a.Clientset.CoreV1().RESTClient().
  1494. Post().
  1495. Resource("pods").
  1496. Name(pod.Name).
  1497. Namespace(pod.Namespace).
  1498. SubResource("exec")
  1499. cmd := []string{
  1500. "/porter/agent-cli",
  1501. "--start",
  1502. startTime,
  1503. }
  1504. for _, label := range labels {
  1505. cmd = append(cmd, "--label", label)
  1506. }
  1507. if searchParam != "" {
  1508. cmd = append(cmd, "--search", searchParam)
  1509. }
  1510. if limit > 0 {
  1511. cmd = append(cmd, "--limit", fmt.Sprintf("%d", limit))
  1512. }
  1513. opts := &v1.PodExecOptions{
  1514. Command: cmd,
  1515. Stdout: true,
  1516. Stderr: true,
  1517. }
  1518. req.VersionedParams(
  1519. opts,
  1520. scheme.ParameterCodec,
  1521. )
  1522. exec, err := remotecommand.NewSPDYExecutor(restConf, "POST", req.URL())
  1523. if err != nil {
  1524. errorchan <- err
  1525. return
  1526. }
  1527. err = exec.Stream(remotecommand.StreamOptions{
  1528. Stdin: nil,
  1529. Stdout: rw,
  1530. Stderr: rw,
  1531. })
  1532. if err != nil {
  1533. errorchan <- err
  1534. return
  1535. }
  1536. }()
  1537. for err = range errorchan {
  1538. once.Do(func() {
  1539. rw.Close()
  1540. })
  1541. }
  1542. return err
  1543. }
  1544. return a.RunWebsocketTask(run)
  1545. }
  1546. // CreateImagePullSecrets will create the required image pull secrets and
  1547. // return a map from the registry name to the name of the secret.
  1548. func (a *Agent) CreateImagePullSecrets(
  1549. repo repository.Repository,
  1550. namespace string,
  1551. linkedRegs map[string]*models.Registry,
  1552. doAuth *oauth2.Config,
  1553. ) (map[string]string, error) {
  1554. res := make(map[string]string)
  1555. for key, val := range linkedRegs {
  1556. _reg := registry.Registry(*val)
  1557. data, err := _reg.GetDockerConfigJSON(repo, doAuth)
  1558. if err != nil {
  1559. return nil, err
  1560. }
  1561. secretName := fmt.Sprintf("porter-%s-%d", val.ToRegistryType().Service, val.ID)
  1562. secret, err := a.Clientset.CoreV1().Secrets(namespace).Get(
  1563. context.TODO(),
  1564. secretName,
  1565. metav1.GetOptions{},
  1566. )
  1567. // if not found, create the secret
  1568. if err != nil && errors.IsNotFound(err) {
  1569. _, err = a.Clientset.CoreV1().Secrets(namespace).Create(
  1570. context.TODO(),
  1571. &v1.Secret{
  1572. ObjectMeta: metav1.ObjectMeta{
  1573. Name: secretName,
  1574. },
  1575. Data: map[string][]byte{
  1576. string(v1.DockerConfigJsonKey): data,
  1577. },
  1578. Type: v1.SecretTypeDockerConfigJson,
  1579. },
  1580. metav1.CreateOptions{},
  1581. )
  1582. if err != nil {
  1583. return nil, err
  1584. }
  1585. // add secret name to the map
  1586. res[key] = secretName
  1587. continue
  1588. } else if err != nil {
  1589. return nil, err
  1590. }
  1591. // otherwise, check that the secret contains the correct data: if
  1592. // if doesn't, update it
  1593. if !bytes.Equal(secret.Data[v1.DockerConfigJsonKey], data) {
  1594. _, err := a.Clientset.CoreV1().Secrets(namespace).Update(
  1595. context.TODO(),
  1596. &v1.Secret{
  1597. ObjectMeta: metav1.ObjectMeta{
  1598. Name: secretName,
  1599. },
  1600. Data: map[string][]byte{
  1601. string(v1.DockerConfigJsonKey): data,
  1602. },
  1603. Type: v1.SecretTypeDockerConfigJson,
  1604. },
  1605. metav1.UpdateOptions{},
  1606. )
  1607. if err != nil {
  1608. return nil, err
  1609. }
  1610. }
  1611. // add secret name to the map
  1612. res[key] = secretName
  1613. }
  1614. return res, nil
  1615. }
  1616. // helper that waits for pod to be ready
  1617. func (a *Agent) waitForPod(pod *v1.Pod) (error, bool) {
  1618. var (
  1619. w watch.Interface
  1620. err error
  1621. ok bool
  1622. )
  1623. // immediately after creating a pod, the API may return a 404. heuristically 1
  1624. // second seems to be plenty.
  1625. watchRetries := 3
  1626. for i := 0; i < watchRetries; i++ {
  1627. selector := fields.OneTermEqualSelector("metadata.name", pod.Name).String()
  1628. w, err = a.Clientset.CoreV1().
  1629. Pods(pod.Namespace).
  1630. Watch(context.Background(), metav1.ListOptions{FieldSelector: selector})
  1631. if err == nil {
  1632. break
  1633. }
  1634. time.Sleep(time.Second)
  1635. }
  1636. if err != nil {
  1637. return err, false
  1638. }
  1639. defer w.Stop()
  1640. expireTime := time.Now().Add(time.Second * 30)
  1641. for time.Now().Unix() <= expireTime.Unix() {
  1642. select {
  1643. case <-time.NewTicker(time.Second).C:
  1644. // poll every second in case we already missed the ready event while
  1645. // creating the listener.
  1646. pod, err = a.Clientset.CoreV1().
  1647. Pods(pod.Namespace).
  1648. Get(context.Background(), pod.Name, metav1.GetOptions{})
  1649. if err != nil && errors.IsNotFound(err) {
  1650. return IsNotFoundError, false
  1651. } else if err != nil {
  1652. return err, false
  1653. }
  1654. if isExited := isPodExited(pod); isExited || isPodReady(pod) {
  1655. return nil, isExited
  1656. }
  1657. case evt := <-w.ResultChan():
  1658. pod, ok = evt.Object.(*v1.Pod)
  1659. if !ok {
  1660. return fmt.Errorf("unexpected object type: %T", evt.Object), false
  1661. }
  1662. if isExited := isPodExited(pod); isExited || isPodReady(pod) {
  1663. return nil, isExited
  1664. }
  1665. }
  1666. }
  1667. return goerrors.New("timed out waiting for pod"), false
  1668. }
  1669. func isPodReady(pod *v1.Pod) bool {
  1670. ready := false
  1671. conditions := pod.Status.Conditions
  1672. for i := range conditions {
  1673. if conditions[i].Type == v1.PodReady {
  1674. ready = pod.Status.Conditions[i].Status == v1.ConditionTrue
  1675. }
  1676. }
  1677. return ready
  1678. }
  1679. func isPodExited(pod *v1.Pod) bool {
  1680. return pod.Status.Phase == v1.PodSucceeded || pod.Status.Phase == v1.PodFailed
  1681. }