deploy_handler.go 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. package api
  2. import (
  3. "archive/tar"
  4. "bytes"
  5. "compress/gzip"
  6. "errors"
  7. "fmt"
  8. "io"
  9. "io/ioutil"
  10. "net/http"
  11. "strings"
  12. "github.com/porter-dev/porter/internal/models"
  13. "gopkg.in/yaml.v2"
  14. )
  15. // HandleDeployTemplate triggers a chart deployment from a template
  16. func (app *App) HandleDeployTemplate(w http.ResponseWriter, r *http.Request) {
  17. tgt := "hello-porter"
  18. baseURL := "https://porter-dev.github.io/chart-repo/"
  19. resp, err := http.Get(baseURL + "index.yaml")
  20. if err != nil {
  21. fmt.Println(err)
  22. return
  23. }
  24. defer resp.Body.Close()
  25. body, _ := ioutil.ReadAll(resp.Body)
  26. form := models.IndexYAML{}
  27. if err := yaml.Unmarshal([]byte(body), &form); err != nil {
  28. fmt.Println(err)
  29. return
  30. }
  31. // Loop over charts in index.yaml
  32. for k := range form.Entries {
  33. indexChart := form.Entries[k][0]
  34. tarURL := indexChart.Urls[0]
  35. splits := strings.Split(tarURL, "-")
  36. strAcc := splits[0]
  37. for i := 1; i < len(splits)-1; i++ {
  38. strAcc += "-" + splits[i]
  39. }
  40. // Unpack the target chart and retrieve values.yaml
  41. if strAcc == tgt {
  42. tgtURL := baseURL + tarURL
  43. values, err := processValues(tgtURL)
  44. if err != nil {
  45. fmt.Println(err)
  46. return
  47. }
  48. defaultValues := *values
  49. defaultValues["replicaCount"] = 87
  50. fmt.Println(defaultValues["replicaCount"])
  51. for k := range *values {
  52. fmt.Println(k)
  53. }
  54. }
  55. }
  56. }
  57. func processValues(tgtURL string) (*map[string]interface{}, error) {
  58. resp, err := http.Get(tgtURL)
  59. if err != nil {
  60. fmt.Println(err)
  61. return nil, err
  62. }
  63. defer resp.Body.Close()
  64. body, _ := ioutil.ReadAll(resp.Body)
  65. buf := bytes.NewBuffer(body)
  66. gzf, err := gzip.NewReader(buf)
  67. if err != nil {
  68. fmt.Println(err)
  69. return nil, err
  70. }
  71. // Process tarball to generate FormYAML and retrieve markdown
  72. tarReader := tar.NewReader(gzf)
  73. for {
  74. header, err := tarReader.Next()
  75. if err == io.EOF {
  76. break
  77. } else if err != nil {
  78. fmt.Println(err)
  79. return nil, err
  80. }
  81. name := header.Name
  82. switch header.Typeflag {
  83. case tar.TypeDir:
  84. continue
  85. case tar.TypeReg:
  86. // Handle values.yaml located in archive
  87. if strings.Contains(name, "values.yaml") {
  88. bufForm := new(bytes.Buffer)
  89. _, err := io.Copy(bufForm, tarReader)
  90. if err != nil {
  91. fmt.Println(err)
  92. return nil, err
  93. }
  94. // Unmarshal yaml byte buffer
  95. form := make(map[string]interface{})
  96. if err := yaml.Unmarshal(bufForm.Bytes(), &form); err != nil {
  97. fmt.Println(err)
  98. return nil, err
  99. }
  100. return &form, nil
  101. }
  102. default:
  103. fmt.Printf("%s : %c %s %s\n",
  104. "Unknown type",
  105. header.Typeflag,
  106. "in file",
  107. name,
  108. )
  109. }
  110. }
  111. return nil, errors.New("no values.yaml found")
  112. }