agent.go 44 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055
  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. errors2 "errors"
  23. "github.com/porter-dev/porter/internal/helm/grapher"
  24. appsv1 "k8s.io/api/apps/v1"
  25. batchv1 "k8s.io/api/batch/v1"
  26. batchv1beta1 "k8s.io/api/batch/v1beta1"
  27. v1 "k8s.io/api/core/v1"
  28. v1beta1 "k8s.io/api/extensions/v1beta1"
  29. netv1 "k8s.io/api/networking/v1"
  30. netv1beta1 "k8s.io/api/networking/v1beta1"
  31. "k8s.io/apimachinery/pkg/api/errors"
  32. metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  33. "k8s.io/apimachinery/pkg/fields"
  34. "k8s.io/apimachinery/pkg/runtime"
  35. "k8s.io/apimachinery/pkg/runtime/schema"
  36. "k8s.io/apimachinery/pkg/types"
  37. "k8s.io/apimachinery/pkg/watch"
  38. "k8s.io/cli-runtime/pkg/genericclioptions"
  39. "k8s.io/client-go/informers"
  40. "k8s.io/client-go/kubernetes"
  41. "k8s.io/client-go/rest"
  42. "k8s.io/client-go/tools/cache"
  43. "k8s.io/client-go/tools/remotecommand"
  44. "k8s.io/kubectl/pkg/scheme"
  45. rspb "helm.sh/helm/v3/pkg/release"
  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) (*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. return a.Clientset.CoreV1().Namespaces().Create(
  550. context.TODO(),
  551. &namespace,
  552. metav1.CreateOptions{},
  553. )
  554. }
  555. // GetNamespace gets the namespace given the name
  556. func (a *Agent) GetNamespace(name string) (*v1.Namespace, error) {
  557. ns, err := a.Clientset.CoreV1().Namespaces().Get(
  558. context.Background(),
  559. name,
  560. metav1.GetOptions{},
  561. )
  562. if err != nil {
  563. return nil, err
  564. }
  565. return ns, nil
  566. }
  567. // DeleteNamespace deletes the namespace given the name.
  568. func (a *Agent) DeleteNamespace(name string) error {
  569. // check if namespace exists
  570. checkNS, err := a.Clientset.CoreV1().Namespaces().Get(
  571. context.TODO(),
  572. name,
  573. metav1.GetOptions{},
  574. )
  575. // if the namespace is not found, don't return an error.
  576. if err != nil && errors.IsNotFound(err) {
  577. return nil
  578. }
  579. // if the namespace was found but is in the "Terminating" phase
  580. // we should ignore it and not return an error
  581. if checkNS != nil && checkNS.Status.Phase == v1.NamespaceTerminating {
  582. return nil
  583. }
  584. return a.Clientset.CoreV1().Namespaces().Delete(
  585. context.TODO(),
  586. name,
  587. metav1.DeleteOptions{},
  588. )
  589. }
  590. func (a *Agent) GetPorterAgent() (*appsv1.Deployment, error) {
  591. depl, err := a.Clientset.AppsV1().Deployments("porter-agent-system").Get(
  592. context.TODO(),
  593. "porter-agent-controller-manager",
  594. metav1.GetOptions{},
  595. )
  596. if err != nil && errors.IsNotFound(err) {
  597. return nil, IsNotFoundError
  598. }
  599. return depl, err
  600. }
  601. // ListJobsByLabel lists jobs in a namespace matching a label
  602. type Label struct {
  603. Key string
  604. Val string
  605. }
  606. func (a *Agent) ListJobsByLabel(namespace string, labels ...Label) ([]batchv1.Job, error) {
  607. selectors := make([]string, 0)
  608. for _, label := range labels {
  609. selectors = append(selectors, fmt.Sprintf("%s=%s", label.Key, label.Val))
  610. }
  611. resp, err := a.Clientset.BatchV1().Jobs(namespace).List(
  612. context.TODO(),
  613. metav1.ListOptions{
  614. LabelSelector: strings.Join(selectors, ","),
  615. },
  616. )
  617. if err != nil {
  618. return nil, err
  619. }
  620. return resp.Items, nil
  621. }
  622. // StreamJobs streams a list of jobs to the websocket writer, closing the connection once all jobs have been sent
  623. func (a *Agent) StreamJobs(namespace string, selectors string, rw *websocket.WebsocketSafeReadWriter) error {
  624. run := func() error {
  625. errorchan := make(chan error)
  626. ctx, cancel := context.WithCancel(context.Background())
  627. defer cancel()
  628. var wg sync.WaitGroup
  629. var once sync.Once
  630. var err error
  631. wg.Add(2)
  632. go func() {
  633. wg.Wait()
  634. close(errorchan)
  635. }()
  636. go func() {
  637. defer func() {
  638. if r := recover(); r != nil {
  639. // TODO: add method to alert on panic
  640. return
  641. }
  642. }()
  643. // listens for websocket closing handshake
  644. defer wg.Done()
  645. for {
  646. if _, _, err := rw.ReadMessage(); err != nil {
  647. errorchan <- nil
  648. return
  649. }
  650. }
  651. }()
  652. go func() {
  653. defer func() {
  654. if r := recover(); r != nil {
  655. // TODO: add method to alert on panic
  656. return
  657. }
  658. }()
  659. // listens for websocket closing handshake
  660. defer wg.Done()
  661. continueVal := ""
  662. for {
  663. if ctx.Err() != nil {
  664. errorchan <- nil
  665. return
  666. }
  667. labelSelector := "meta.helm.sh/release-name"
  668. if selectors != "" {
  669. labelSelector = selectors
  670. }
  671. jobs, err := a.Clientset.BatchV1().Jobs(namespace).List(
  672. ctx,
  673. metav1.ListOptions{
  674. Limit: 100,
  675. Continue: continueVal,
  676. LabelSelector: labelSelector,
  677. },
  678. )
  679. if err != nil {
  680. errorchan <- err
  681. return
  682. }
  683. for _, job := range jobs.Items {
  684. err := rw.WriteJSON(job)
  685. if err != nil {
  686. errorchan <- err
  687. return
  688. }
  689. }
  690. if jobs.Continue == "" {
  691. // we have reached the end of the list of jobs
  692. break
  693. } else {
  694. // start pagination
  695. continueVal = jobs.Continue
  696. }
  697. }
  698. // at this point, we can return the status finished
  699. err := rw.WriteJSON(map[string]interface{}{
  700. "streamStatus": "finished",
  701. })
  702. errorchan <- err
  703. }()
  704. for err = range errorchan {
  705. once.Do(func() {
  706. rw.Close()
  707. cancel()
  708. })
  709. }
  710. return err
  711. }
  712. return a.RunWebsocketTask(run)
  713. }
  714. // DeleteJob deletes the job in the given name and namespace.
  715. func (a *Agent) DeleteJob(name, namespace string) error {
  716. return a.Clientset.BatchV1().Jobs(namespace).Delete(
  717. context.TODO(),
  718. name,
  719. metav1.DeleteOptions{},
  720. )
  721. }
  722. // GetJobPods lists all pods belonging to a job in a namespace
  723. func (a *Agent) GetJobPods(namespace, jobName string) ([]v1.Pod, error) {
  724. resp, err := a.Clientset.CoreV1().Pods(namespace).List(
  725. context.TODO(),
  726. metav1.ListOptions{
  727. LabelSelector: fmt.Sprintf("%s=%s", "job-name", jobName),
  728. },
  729. )
  730. if err != nil {
  731. return nil, err
  732. }
  733. return resp.Items, nil
  734. }
  735. // GetIngress gets ingress given the name and namespace
  736. func (a *Agent) GetExtensionsV1Beta1Ingress(namespace string, name string) (*v1beta1.Ingress, error) {
  737. resp, err := a.Clientset.ExtensionsV1beta1().Ingresses(namespace).Get(
  738. context.TODO(),
  739. name,
  740. metav1.GetOptions{},
  741. )
  742. if err != nil && errors.IsNotFound(err) {
  743. return nil, IsNotFoundError
  744. } else if err != nil {
  745. return nil, err
  746. }
  747. return resp, nil
  748. }
  749. func (a *Agent) GetNetworkingV1Ingress(namespace string, name string) (*netv1.Ingress, error) {
  750. resp, err := a.Clientset.NetworkingV1().Ingresses(namespace).Get(
  751. context.TODO(),
  752. name,
  753. metav1.GetOptions{},
  754. )
  755. if err != nil && errors.IsNotFound(err) {
  756. return nil, IsNotFoundError
  757. } else if err != nil {
  758. return nil, err
  759. }
  760. return resp, nil
  761. }
  762. func (a *Agent) GetNetworkingV1Beta1Ingress(namespace string, name string) (*netv1beta1.Ingress, error) {
  763. resp, err := a.Clientset.NetworkingV1beta1().Ingresses(namespace).Get(
  764. context.TODO(),
  765. name,
  766. metav1.GetOptions{},
  767. )
  768. if err != nil && errors.IsNotFound(err) {
  769. return nil, IsNotFoundError
  770. } else if err != nil {
  771. return nil, err
  772. }
  773. return resp, nil
  774. }
  775. var IsNotFoundError = fmt.Errorf("not found")
  776. type BadRequestError struct {
  777. msg string
  778. }
  779. func (e *BadRequestError) Error() string {
  780. return e.msg
  781. }
  782. // GetDeployment gets the deployment given the name and namespace
  783. func (a *Agent) GetDeployment(c grapher.Object) (*appsv1.Deployment, error) {
  784. res, err := a.Clientset.AppsV1().Deployments(c.Namespace).Get(
  785. context.TODO(),
  786. c.Name,
  787. metav1.GetOptions{},
  788. )
  789. if err != nil && errors.IsNotFound(err) {
  790. return nil, IsNotFoundError
  791. } else if err != nil {
  792. return nil, err
  793. }
  794. res.Kind = c.Kind
  795. return res, nil
  796. }
  797. // GetStatefulSet gets the statefulset given the name and namespace
  798. func (a *Agent) GetStatefulSet(c grapher.Object) (*appsv1.StatefulSet, error) {
  799. res, err := a.Clientset.AppsV1().StatefulSets(c.Namespace).Get(
  800. context.TODO(),
  801. c.Name,
  802. metav1.GetOptions{},
  803. )
  804. if err != nil && errors.IsNotFound(err) {
  805. return nil, IsNotFoundError
  806. } else if err != nil {
  807. return nil, err
  808. }
  809. res.Kind = c.Kind
  810. return res, nil
  811. }
  812. // GetReplicaSet gets the replicaset given the name and namespace
  813. func (a *Agent) GetReplicaSet(c grapher.Object) (*appsv1.ReplicaSet, error) {
  814. res, err := a.Clientset.AppsV1().ReplicaSets(c.Namespace).Get(
  815. context.TODO(),
  816. c.Name,
  817. metav1.GetOptions{},
  818. )
  819. if err != nil && errors.IsNotFound(err) {
  820. return nil, IsNotFoundError
  821. } else if err != nil {
  822. return nil, err
  823. }
  824. res.Kind = c.Kind
  825. return res, nil
  826. }
  827. // GetDaemonSet gets the daemonset by name and namespace
  828. func (a *Agent) GetDaemonSet(c grapher.Object) (*appsv1.DaemonSet, error) {
  829. res, err := a.Clientset.AppsV1().DaemonSets(c.Namespace).Get(
  830. context.TODO(),
  831. c.Name,
  832. metav1.GetOptions{},
  833. )
  834. if err != nil && errors.IsNotFound(err) {
  835. return nil, IsNotFoundError
  836. } else if err != nil {
  837. return nil, err
  838. }
  839. res.Kind = c.Kind
  840. return res, nil
  841. }
  842. // GetJob gets the job by name and namespace
  843. func (a *Agent) GetJob(c grapher.Object) (*batchv1.Job, error) {
  844. res, err := a.Clientset.BatchV1().Jobs(c.Namespace).Get(
  845. context.TODO(),
  846. c.Name,
  847. metav1.GetOptions{},
  848. )
  849. if err != nil && errors.IsNotFound(err) {
  850. return nil, IsNotFoundError
  851. } else if err != nil {
  852. return nil, err
  853. }
  854. res.Kind = c.Kind
  855. return res, nil
  856. }
  857. // GetCronJob gets the CronJob by name and namespace
  858. func (a *Agent) GetCronJob(c grapher.Object) (*batchv1beta1.CronJob, error) {
  859. res, err := a.Clientset.BatchV1beta1().CronJobs(c.Namespace).Get(
  860. context.TODO(),
  861. c.Name,
  862. metav1.GetOptions{},
  863. )
  864. if err != nil && errors.IsNotFound(err) {
  865. return nil, IsNotFoundError
  866. } else if err != nil {
  867. return nil, err
  868. }
  869. res.Kind = c.Kind
  870. return res, nil
  871. }
  872. // GetPodsByLabel retrieves pods with matching labels
  873. func (a *Agent) GetPodsByLabel(selector string, namespace string) (*v1.PodList, error) {
  874. // Search in all namespaces for matching pods
  875. return a.Clientset.CoreV1().Pods(namespace).List(
  876. context.TODO(),
  877. metav1.ListOptions{
  878. LabelSelector: selector,
  879. },
  880. )
  881. }
  882. // GetPodByName retrieves a single instance of pod with given name
  883. func (a *Agent) GetPodByName(name string, namespace string) (*v1.Pod, error) {
  884. // Get pod by name
  885. pod, err := a.Clientset.CoreV1().Pods(namespace).Get(
  886. context.TODO(),
  887. name,
  888. metav1.GetOptions{},
  889. )
  890. if err != nil && errors.IsNotFound(err) {
  891. return nil, IsNotFoundError
  892. }
  893. if err != nil {
  894. return nil, err
  895. }
  896. return pod, nil
  897. }
  898. // DeletePod deletes a pod by name and namespace
  899. func (a *Agent) DeletePod(namespace string, name string) error {
  900. err := a.Clientset.CoreV1().Pods(namespace).Delete(
  901. context.TODO(),
  902. name,
  903. metav1.DeleteOptions{},
  904. )
  905. if err != nil && errors.IsNotFound(err) {
  906. return IsNotFoundError
  907. }
  908. return err
  909. }
  910. // GetPodLogs streams real-time logs from a given pod.
  911. func (a *Agent) GetPodLogs(namespace string, name string, selectedContainer string, rw *websocket.WebsocketSafeReadWriter) error {
  912. // get the pod to read in the list of contains
  913. pod, err := a.Clientset.CoreV1().Pods(namespace).Get(
  914. context.Background(),
  915. name,
  916. metav1.GetOptions{},
  917. )
  918. if err != nil && errors.IsNotFound(err) {
  919. return IsNotFoundError
  920. } else if err != nil {
  921. return fmt.Errorf("Cannot get logs from pod %s: %s", name, err.Error())
  922. }
  923. // see if container is ready and able to open a stream. If not, wait for container
  924. // to be ready.
  925. err, _ = a.waitForPod(pod)
  926. if err != nil && goerrors.Is(err, IsNotFoundError) {
  927. return IsNotFoundError
  928. } else if err != nil {
  929. return fmt.Errorf("Cannot get logs from pod %s: %s", name, err.Error())
  930. }
  931. container := pod.Spec.Containers[0].Name
  932. if len(selectedContainer) > 0 {
  933. container = selectedContainer
  934. }
  935. tails := int64(400)
  936. // follow logs
  937. podLogOpts := v1.PodLogOptions{
  938. Follow: true,
  939. TailLines: &tails,
  940. Container: container,
  941. }
  942. req := a.Clientset.CoreV1().Pods(namespace).GetLogs(name, &podLogOpts)
  943. podLogs, err := req.Stream(context.TODO())
  944. // in the case of bad request errors, such as if the pod is stuck in "ContainerCreating",
  945. // we'd like to pass this through to the client.
  946. if err != nil && errors.IsBadRequest(err) {
  947. return &BadRequestError{err.Error()}
  948. } else if err != nil {
  949. return fmt.Errorf("Cannot open log stream for pod %s: %s", name, err.Error())
  950. }
  951. r := bufio.NewReader(podLogs)
  952. errorchan := make(chan error)
  953. var wg sync.WaitGroup
  954. var once sync.Once
  955. wg.Add(2)
  956. go func() {
  957. wg.Wait()
  958. close(errorchan)
  959. }()
  960. go func() {
  961. defer func() {
  962. if r := recover(); r != nil {
  963. // TODO: add method to alert on panic
  964. return
  965. }
  966. }()
  967. // listens for websocket closing handshake
  968. defer wg.Done()
  969. for {
  970. if _, _, err := rw.ReadMessage(); err != nil {
  971. errorchan <- nil
  972. return
  973. }
  974. }
  975. }()
  976. go func() {
  977. defer func() {
  978. if r := recover(); r != nil {
  979. // TODO: add method to alert on panic
  980. return
  981. }
  982. }()
  983. defer wg.Done()
  984. for {
  985. bytes, err := r.ReadBytes('\n')
  986. if err != nil {
  987. errorchan <- err
  988. return
  989. }
  990. if _, writeErr := rw.Write(bytes); writeErr != nil {
  991. errorchan <- writeErr
  992. return
  993. }
  994. }
  995. }()
  996. for err = range errorchan {
  997. // only call these methods a single time
  998. once.Do(func() {
  999. rw.Close()
  1000. podLogs.Close()
  1001. })
  1002. }
  1003. return err
  1004. }
  1005. // GetPodLogs streams real-time logs from a given pod.
  1006. func (a *Agent) GetPreviousPodLogs(namespace string, name string, selectedContainer string) ([]string, error) {
  1007. // get the pod to read in the list of contains
  1008. pod, err := a.Clientset.CoreV1().Pods(namespace).Get(
  1009. context.Background(),
  1010. name,
  1011. metav1.GetOptions{},
  1012. )
  1013. if err != nil && errors.IsNotFound(err) {
  1014. return nil, IsNotFoundError
  1015. } else if err != nil {
  1016. return nil, fmt.Errorf("Cannot get logs from pod %s: %s", name, err.Error())
  1017. }
  1018. container := pod.Spec.Containers[0].Name
  1019. if len(selectedContainer) > 0 {
  1020. container = selectedContainer
  1021. }
  1022. tails := int64(400)
  1023. // follow logs
  1024. podLogOpts := v1.PodLogOptions{
  1025. Follow: true,
  1026. TailLines: &tails,
  1027. Container: container,
  1028. Previous: true,
  1029. }
  1030. req := a.Clientset.CoreV1().Pods(namespace).GetLogs(name, &podLogOpts)
  1031. podLogs, err := req.Stream(context.TODO())
  1032. // in the case of bad request errors, such as if the pod is stuck in "ContainerCreating",
  1033. // we'd like to pass this through to the client.
  1034. if err != nil && strings.Contains(err.Error(), "not found") {
  1035. return nil, IsNotFoundError
  1036. }
  1037. if err != nil && errors.IsBadRequest(err) {
  1038. return nil, &BadRequestError{err.Error()}
  1039. } else if err != nil {
  1040. return nil, fmt.Errorf("Cannot open log stream for pod %s: %s", name, err.Error())
  1041. }
  1042. defer podLogs.Close()
  1043. r := bufio.NewReader(podLogs)
  1044. var logs []string
  1045. for {
  1046. line, err := r.ReadString('\n')
  1047. logs = append(logs, line)
  1048. if err == io.EOF {
  1049. break
  1050. } else if err != nil {
  1051. return nil, err
  1052. }
  1053. }
  1054. return logs, nil
  1055. }
  1056. // StopJobWithJobSidecar sends a termination signal to a job running with a sidecar
  1057. func (a *Agent) StopJobWithJobSidecar(namespace, name string) error {
  1058. jobPods, err := a.GetJobPods(namespace, name)
  1059. if err != nil {
  1060. return err
  1061. }
  1062. podName := jobPods[0].ObjectMeta.Name
  1063. restConf, err := a.RESTClientGetter.ToRESTConfig()
  1064. restConf.GroupVersion = &schema.GroupVersion{
  1065. Group: "api",
  1066. Version: "v1",
  1067. }
  1068. restConf.NegotiatedSerializer = runtime.NewSimpleNegotiatedSerializer(runtime.SerializerInfo{})
  1069. restClient, err := rest.RESTClientFor(restConf)
  1070. if err != nil {
  1071. return err
  1072. }
  1073. req := restClient.Post().
  1074. Resource("pods").
  1075. Name(podName).
  1076. Namespace(namespace).
  1077. SubResource("exec")
  1078. req.Param("command", "./signal.sh")
  1079. req.Param("container", "sidecar")
  1080. req.Param("stdin", "true")
  1081. req.Param("stdout", "false")
  1082. req.Param("tty", "false")
  1083. exec, err := remotecommand.NewSPDYExecutor(restConf, "POST", req.URL())
  1084. if err != nil {
  1085. return err
  1086. }
  1087. return exec.Stream(remotecommand.StreamOptions{
  1088. Tty: false,
  1089. Stdin: strings.NewReader("./signal.sh"),
  1090. })
  1091. }
  1092. // RunWebsocketTask will run a websocket task. If the websocket returns an anauthorized error, it will restart
  1093. // the task some number of times until failing
  1094. func (a *Agent) RunWebsocketTask(task func() error) error {
  1095. lastTime := int64(0)
  1096. for {
  1097. if err := a.UpdateClientset(); err != nil {
  1098. return err
  1099. }
  1100. err := task()
  1101. if err == nil {
  1102. return nil
  1103. }
  1104. if !errors2.Is(err, &AuthError{}) {
  1105. return err
  1106. }
  1107. if time.Now().Unix()-lastTime < 60 { // don't regenerate connection if too many unauthorized errors
  1108. return err
  1109. }
  1110. lastTime = time.Now().Unix()
  1111. }
  1112. }
  1113. // StreamControllerStatus streams controller status. Supports Deployment, StatefulSet, ReplicaSet, and DaemonSet
  1114. // TODO: Support Jobs
  1115. func (a *Agent) StreamControllerStatus(kind string, selectors string, rw *websocket.WebsocketSafeReadWriter) error {
  1116. run := func() error {
  1117. // selectors is an array of max length 1. StreamControllerStatus accepts calls without the selectors argument.
  1118. // selectors argument is a single string with comma separated key=value pairs. (e.g. "app=porter,porter=true")
  1119. tweakListOptionsFunc := func(options *metav1.ListOptions) {
  1120. options.LabelSelector = selectors
  1121. }
  1122. factory := informers.NewSharedInformerFactoryWithOptions(
  1123. a.Clientset,
  1124. 0,
  1125. informers.WithTweakListOptions(tweakListOptionsFunc),
  1126. )
  1127. var informer cache.SharedInformer
  1128. // Spins up an informer depending on kind. Convert to lowercase for robustness
  1129. switch strings.ToLower(kind) {
  1130. case "deployment":
  1131. informer = factory.Apps().V1().Deployments().Informer()
  1132. case "statefulset":
  1133. informer = factory.Apps().V1().StatefulSets().Informer()
  1134. case "replicaset":
  1135. informer = factory.Apps().V1().ReplicaSets().Informer()
  1136. case "daemonset":
  1137. informer = factory.Apps().V1().DaemonSets().Informer()
  1138. case "job":
  1139. informer = factory.Batch().V1().Jobs().Informer()
  1140. case "cronjob":
  1141. informer = factory.Batch().V1beta1().CronJobs().Informer()
  1142. case "namespace":
  1143. informer = factory.Core().V1().Namespaces().Informer()
  1144. case "pod":
  1145. informer = factory.Core().V1().Pods().Informer()
  1146. }
  1147. stopper := make(chan struct{})
  1148. errorchan := make(chan error)
  1149. var wg sync.WaitGroup
  1150. var once sync.Once
  1151. var err error
  1152. wg.Add(2)
  1153. go func() {
  1154. wg.Wait()
  1155. close(errorchan)
  1156. }()
  1157. go func() {
  1158. defer func() {
  1159. if r := recover(); r != nil {
  1160. // TODO: add method to alert on panic
  1161. return
  1162. }
  1163. }()
  1164. // listens for websocket closing handshake
  1165. defer wg.Done()
  1166. for {
  1167. if _, _, err := rw.ReadMessage(); err != nil {
  1168. errorchan <- nil
  1169. return
  1170. }
  1171. }
  1172. }()
  1173. go func() {
  1174. defer func() {
  1175. if r := recover(); r != nil {
  1176. // TODO: add method to alert on panic
  1177. return
  1178. }
  1179. }()
  1180. // listens for websocket closing handshake
  1181. defer wg.Done()
  1182. informer.SetWatchErrorHandler(func(r *cache.Reflector, err error) {
  1183. if strings.HasSuffix(err.Error(), ": Unauthorized") {
  1184. errorchan <- &AuthError{}
  1185. }
  1186. })
  1187. informer.AddEventHandler(cache.ResourceEventHandlerFuncs{
  1188. UpdateFunc: func(oldObj, newObj interface{}) {
  1189. msg := Message{
  1190. EventType: "UPDATE",
  1191. Object: newObj,
  1192. Kind: strings.ToLower(kind),
  1193. }
  1194. err := rw.WriteJSON(msg)
  1195. if err != nil {
  1196. errorchan <- err
  1197. }
  1198. },
  1199. AddFunc: func(obj interface{}) {
  1200. msg := Message{
  1201. EventType: "ADD",
  1202. Object: obj,
  1203. Kind: strings.ToLower(kind),
  1204. }
  1205. err := rw.WriteJSON(msg)
  1206. if err != nil {
  1207. errorchan <- err
  1208. }
  1209. },
  1210. DeleteFunc: func(obj interface{}) {
  1211. msg := Message{
  1212. EventType: "DELETE",
  1213. Object: obj,
  1214. Kind: strings.ToLower(kind),
  1215. }
  1216. err := rw.WriteJSON(msg)
  1217. if err != nil {
  1218. errorchan <- err
  1219. }
  1220. },
  1221. })
  1222. informer.Run(stopper)
  1223. }()
  1224. for err = range errorchan {
  1225. once.Do(func() {
  1226. close(stopper)
  1227. rw.Close()
  1228. })
  1229. }
  1230. return err
  1231. }
  1232. return a.RunWebsocketTask(run)
  1233. }
  1234. var b64 = base64.StdEncoding
  1235. var magicGzip = []byte{0x1f, 0x8b, 0x08}
  1236. func decodeRelease(data string) (*rspb.Release, error) {
  1237. // base64 decode string
  1238. b, err := b64.DecodeString(data)
  1239. if err != nil {
  1240. return nil, err
  1241. }
  1242. // For backwards compatibility with releases that were stored before
  1243. // compression was introduced we skip decompression if the
  1244. // gzip magic header is not found
  1245. if bytes.Equal(b[0:3], magicGzip) {
  1246. r, err := gzip.NewReader(bytes.NewReader(b))
  1247. if err != nil {
  1248. return nil, err
  1249. }
  1250. defer r.Close()
  1251. b2, err := ioutil.ReadAll(r)
  1252. if err != nil {
  1253. return nil, err
  1254. }
  1255. b = b2
  1256. }
  1257. var rls rspb.Release
  1258. // unmarshal release object bytes
  1259. if err := json.Unmarshal(b, &rls); err != nil {
  1260. return nil, err
  1261. }
  1262. return &rls, nil
  1263. }
  1264. func contains(s []string, str string) bool {
  1265. for _, v := range s {
  1266. if v == str {
  1267. return true
  1268. }
  1269. }
  1270. return false
  1271. }
  1272. func ParseSecretToHelmRelease(secret v1.Secret, chartList []string) (*rspb.Release, bool, error) {
  1273. if secret.Type != "helm.sh/release.v1" {
  1274. return nil, true, nil
  1275. }
  1276. releaseData, ok := secret.Data["release"]
  1277. if !ok {
  1278. return nil, true, fmt.Errorf("release field not found")
  1279. }
  1280. helm_object, err := decodeRelease(string(releaseData))
  1281. if err != nil {
  1282. return nil, true, err
  1283. }
  1284. if len(chartList) > 0 && !contains(chartList, helm_object.Name) {
  1285. return nil, true, nil
  1286. }
  1287. return helm_object, false, nil
  1288. }
  1289. func (a *Agent) StreamHelmReleases(namespace string, chartList []string, selectors string, rw *websocket.WebsocketSafeReadWriter) error {
  1290. run := func() error {
  1291. tweakListOptionsFunc := func(options *metav1.ListOptions) {
  1292. options.LabelSelector = selectors
  1293. }
  1294. factory := informers.NewSharedInformerFactoryWithOptions(
  1295. a.Clientset,
  1296. 0,
  1297. informers.WithTweakListOptions(tweakListOptionsFunc),
  1298. informers.WithNamespace(namespace),
  1299. )
  1300. informer := factory.Core().V1().Secrets().Informer()
  1301. stopper := make(chan struct{})
  1302. errorchan := make(chan error)
  1303. var wg sync.WaitGroup
  1304. var once sync.Once
  1305. var err error
  1306. wg.Add(2)
  1307. go func() {
  1308. wg.Wait()
  1309. close(errorchan)
  1310. }()
  1311. go func() {
  1312. defer func() {
  1313. if r := recover(); r != nil {
  1314. // TODO: add method to alert on panic
  1315. return
  1316. }
  1317. }()
  1318. // listens for websocket closing handshake
  1319. defer wg.Done()
  1320. for {
  1321. if _, _, err := rw.ReadMessage(); err != nil {
  1322. errorchan <- nil
  1323. return
  1324. }
  1325. }
  1326. }()
  1327. go func() {
  1328. defer func() {
  1329. if r := recover(); r != nil {
  1330. // TODO: add method to alert on panic
  1331. return
  1332. }
  1333. }()
  1334. // listens for websocket closing handshake
  1335. defer wg.Done()
  1336. informer.SetWatchErrorHandler(func(r *cache.Reflector, err error) {
  1337. if strings.HasSuffix(err.Error(), ": Unauthorized") {
  1338. errorchan <- &AuthError{}
  1339. }
  1340. })
  1341. informer.AddEventHandler(cache.ResourceEventHandlerFuncs{
  1342. UpdateFunc: func(oldObj, newObj interface{}) {
  1343. secretObj, ok := newObj.(*v1.Secret)
  1344. if !ok {
  1345. errorchan <- fmt.Errorf("could not cast to secret")
  1346. return
  1347. }
  1348. helm_object, isNotHelmRelease, err := ParseSecretToHelmRelease(*secretObj, chartList)
  1349. if isNotHelmRelease && err == nil {
  1350. return
  1351. }
  1352. if err != nil {
  1353. errorchan <- err
  1354. return
  1355. }
  1356. msg := Message{
  1357. EventType: "UPDATE",
  1358. Object: helm_object,
  1359. }
  1360. rw.WriteJSON(msg)
  1361. },
  1362. AddFunc: func(obj interface{}) {
  1363. secretObj, ok := obj.(*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: "ADD",
  1378. Object: helm_object,
  1379. }
  1380. rw.WriteJSON(msg)
  1381. },
  1382. DeleteFunc: 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: "DELETE",
  1398. Object: helm_object,
  1399. }
  1400. rw.WriteJSON(msg)
  1401. },
  1402. })
  1403. informer.Run(stopper)
  1404. }()
  1405. for err = range errorchan {
  1406. once.Do(func() {
  1407. close(stopper)
  1408. rw.Close()
  1409. })
  1410. }
  1411. return err
  1412. }
  1413. return a.RunWebsocketTask(run)
  1414. }
  1415. func (a *Agent) StreamPorterAgentLokiLog(
  1416. labels []string,
  1417. startTime string,
  1418. searchParam string,
  1419. limit uint32,
  1420. rw *websocket.WebsocketSafeReadWriter,
  1421. ) error {
  1422. run := func() error {
  1423. errorchan := make(chan error)
  1424. var wg sync.WaitGroup
  1425. var once sync.Once
  1426. var err error
  1427. wg.Add(2)
  1428. go func() {
  1429. wg.Wait()
  1430. close(errorchan)
  1431. }()
  1432. go func() {
  1433. defer func() {
  1434. if r := recover(); r != nil {
  1435. // TODO: add method to alert on panic
  1436. return
  1437. }
  1438. }()
  1439. defer wg.Done()
  1440. // listens for websocket closing handshake
  1441. for {
  1442. if _, _, err := rw.ReadMessage(); err != nil {
  1443. errorchan <- nil
  1444. return
  1445. }
  1446. }
  1447. }()
  1448. go func() {
  1449. defer func() {
  1450. if r := recover(); r != nil {
  1451. // TODO: add method to alert on panic
  1452. return
  1453. }
  1454. }()
  1455. defer wg.Done()
  1456. podList, err := a.Clientset.CoreV1().Pods("porter-agent-system").List(context.Background(), metav1.ListOptions{
  1457. LabelSelector: "control-plane=controller-manager",
  1458. })
  1459. if err != nil {
  1460. errorchan <- err
  1461. return
  1462. }
  1463. if len(podList.Items) == 0 {
  1464. errorchan <- fmt.Errorf("no porter agent pods found")
  1465. return
  1466. }
  1467. pod := podList.Items[0]
  1468. restConf, err := a.RESTClientGetter.ToRESTConfig()
  1469. if err != nil {
  1470. errorchan <- err
  1471. return
  1472. }
  1473. req := a.Clientset.CoreV1().RESTClient().
  1474. Post().
  1475. Resource("pods").
  1476. Name(pod.Name).
  1477. Namespace(pod.Namespace).
  1478. SubResource("exec")
  1479. cmd := []string{
  1480. "/porter/agent-cli",
  1481. "--start",
  1482. startTime,
  1483. }
  1484. for _, label := range labels {
  1485. cmd = append(cmd, "--label", label)
  1486. }
  1487. if searchParam != "" {
  1488. cmd = append(cmd, "--search", searchParam)
  1489. }
  1490. if limit > 0 {
  1491. cmd = append(cmd, "--limit", fmt.Sprintf("%d", limit))
  1492. }
  1493. opts := &v1.PodExecOptions{
  1494. Command: cmd,
  1495. Stdout: true,
  1496. Stderr: true,
  1497. }
  1498. req.VersionedParams(
  1499. opts,
  1500. scheme.ParameterCodec,
  1501. )
  1502. exec, err := remotecommand.NewSPDYExecutor(restConf, "POST", req.URL())
  1503. if err != nil {
  1504. errorchan <- err
  1505. return
  1506. }
  1507. err = exec.Stream(remotecommand.StreamOptions{
  1508. Stdin: nil,
  1509. Stdout: rw,
  1510. Stderr: rw,
  1511. })
  1512. if err != nil {
  1513. errorchan <- err
  1514. return
  1515. }
  1516. }()
  1517. for err = range errorchan {
  1518. once.Do(func() {
  1519. rw.Close()
  1520. })
  1521. }
  1522. return err
  1523. }
  1524. return a.RunWebsocketTask(run)
  1525. }
  1526. // CreateImagePullSecrets will create the required image pull secrets and
  1527. // return a map from the registry name to the name of the secret.
  1528. func (a *Agent) CreateImagePullSecrets(
  1529. repo repository.Repository,
  1530. namespace string,
  1531. linkedRegs map[string]*models.Registry,
  1532. doAuth *oauth2.Config,
  1533. ) (map[string]string, error) {
  1534. res := make(map[string]string)
  1535. for key, val := range linkedRegs {
  1536. _reg := registry.Registry(*val)
  1537. data, err := _reg.GetDockerConfigJSON(repo, doAuth)
  1538. if err != nil {
  1539. return nil, err
  1540. }
  1541. secretName := fmt.Sprintf("porter-%s-%d", val.ToRegistryType().Service, val.ID)
  1542. secret, err := a.Clientset.CoreV1().Secrets(namespace).Get(
  1543. context.TODO(),
  1544. secretName,
  1545. metav1.GetOptions{},
  1546. )
  1547. // if not found, create the secret
  1548. if err != nil && errors.IsNotFound(err) {
  1549. _, err = a.Clientset.CoreV1().Secrets(namespace).Create(
  1550. context.TODO(),
  1551. &v1.Secret{
  1552. ObjectMeta: metav1.ObjectMeta{
  1553. Name: secretName,
  1554. },
  1555. Data: map[string][]byte{
  1556. string(v1.DockerConfigJsonKey): data,
  1557. },
  1558. Type: v1.SecretTypeDockerConfigJson,
  1559. },
  1560. metav1.CreateOptions{},
  1561. )
  1562. if err != nil {
  1563. return nil, err
  1564. }
  1565. // add secret name to the map
  1566. res[key] = secretName
  1567. continue
  1568. } else if err != nil {
  1569. return nil, err
  1570. }
  1571. // otherwise, check that the secret contains the correct data: if
  1572. // if doesn't, update it
  1573. if !bytes.Equal(secret.Data[v1.DockerConfigJsonKey], data) {
  1574. _, err := a.Clientset.CoreV1().Secrets(namespace).Update(
  1575. context.TODO(),
  1576. &v1.Secret{
  1577. ObjectMeta: metav1.ObjectMeta{
  1578. Name: secretName,
  1579. },
  1580. Data: map[string][]byte{
  1581. string(v1.DockerConfigJsonKey): data,
  1582. },
  1583. Type: v1.SecretTypeDockerConfigJson,
  1584. },
  1585. metav1.UpdateOptions{},
  1586. )
  1587. if err != nil {
  1588. return nil, err
  1589. }
  1590. }
  1591. // add secret name to the map
  1592. res[key] = secretName
  1593. }
  1594. return res, nil
  1595. }
  1596. // helper that waits for pod to be ready
  1597. func (a *Agent) waitForPod(pod *v1.Pod) (error, bool) {
  1598. var (
  1599. w watch.Interface
  1600. err error
  1601. ok bool
  1602. )
  1603. // immediately after creating a pod, the API may return a 404. heuristically 1
  1604. // second seems to be plenty.
  1605. watchRetries := 3
  1606. for i := 0; i < watchRetries; i++ {
  1607. selector := fields.OneTermEqualSelector("metadata.name", pod.Name).String()
  1608. w, err = a.Clientset.CoreV1().
  1609. Pods(pod.Namespace).
  1610. Watch(context.Background(), metav1.ListOptions{FieldSelector: selector})
  1611. if err == nil {
  1612. break
  1613. }
  1614. time.Sleep(time.Second)
  1615. }
  1616. if err != nil {
  1617. return err, false
  1618. }
  1619. defer w.Stop()
  1620. expireTime := time.Now().Add(time.Second * 30)
  1621. for time.Now().Unix() <= expireTime.Unix() {
  1622. select {
  1623. case <-time.NewTicker(time.Second).C:
  1624. // poll every second in case we already missed the ready event while
  1625. // creating the listener.
  1626. pod, err = a.Clientset.CoreV1().
  1627. Pods(pod.Namespace).
  1628. Get(context.Background(), pod.Name, metav1.GetOptions{})
  1629. if err != nil && errors.IsNotFound(err) {
  1630. return IsNotFoundError, false
  1631. } else if err != nil {
  1632. return err, false
  1633. }
  1634. if isExited := isPodExited(pod); isExited || isPodReady(pod) {
  1635. return nil, isExited
  1636. }
  1637. case evt := <-w.ResultChan():
  1638. pod, ok = evt.Object.(*v1.Pod)
  1639. if !ok {
  1640. return fmt.Errorf("unexpected object type: %T", evt.Object), false
  1641. }
  1642. if isExited := isPodExited(pod); isExited || isPodReady(pod) {
  1643. return nil, isExited
  1644. }
  1645. }
  1646. }
  1647. return goerrors.New("timed out waiting for pod"), false
  1648. }
  1649. func isPodReady(pod *v1.Pod) bool {
  1650. ready := false
  1651. conditions := pod.Status.Conditions
  1652. for i := range conditions {
  1653. if conditions[i].Type == v1.PodReady {
  1654. ready = pod.Status.Conditions[i].Status == v1.ConditionTrue
  1655. }
  1656. }
  1657. return ready
  1658. }
  1659. func isPodExited(pod *v1.Pod) bool {
  1660. return pod.Status.Phase == v1.PodSucceeded || pod.Status.Phase == v1.PodFailed
  1661. }