agent.go 50 KB

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