oscarzhou
102 lines
3.3 KiB
Go
102 lines
3.3 KiB
Go
package customtemplates
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
|
|
httperror "github.com/portainer/libhttp/error"
|
|
"github.com/portainer/libhttp/request"
|
|
"github.com/portainer/libhttp/response"
|
|
portainer "github.com/portainer/portainer/api"
|
|
"github.com/portainer/portainer/api/stacks/stackutils"
|
|
)
|
|
|
|
// @id CustomTemplateGitFetch
|
|
// @summary Fetch the latest config file content based on custom template's git repository configuration
|
|
// @description Retrieve details about a template created from git repository method.
|
|
// @description **Access policy**: authenticated
|
|
// @tags custom_templates
|
|
// @security ApiKeyAuth
|
|
// @security jwt
|
|
// @produce json
|
|
// @param id path int true "Template identifier"
|
|
// @success 200 {object} portaineree.CustomTemplate "Success"
|
|
// @failure 400 "Invalid request"
|
|
// @failure 500 "Server error"
|
|
// @router /custom_templates/{id}/git_fetch [put]
|
|
func (handler *Handler) customTemplateGitFetch(w http.ResponseWriter, r *http.Request) *httperror.HandlerError {
|
|
customTemplateID, err := request.RetrieveNumericRouteVariableValue(r, "id")
|
|
if err != nil {
|
|
return httperror.BadRequest("Invalid Custom template identifier route variable", err)
|
|
}
|
|
|
|
customTemplate, err := handler.DataStore.CustomTemplate().CustomTemplate(portainer.CustomTemplateID(customTemplateID))
|
|
if handler.DataStore.IsErrObjectNotFound(err) {
|
|
return httperror.NotFound("Unable to find a custom template with the specified identifier inside the database", err)
|
|
} else if err != nil {
|
|
return httperror.InternalServerError("Unable to find a custom template with the specified identifier inside the database", err)
|
|
}
|
|
|
|
if customTemplate.GitConfig != nil {
|
|
// back up the current custom template folder
|
|
backupPath, err := backupCustomTemplate(customTemplate.ProjectPath)
|
|
if err != nil {
|
|
return httperror.InternalServerError("Failed to backup the custom template folder", err)
|
|
}
|
|
|
|
// remove backup custom template folder
|
|
defer cleanUpBackupCustomTemplate(backupPath)
|
|
|
|
commitHash, err := stackutils.DownloadGitRepository(*customTemplate.GitConfig, handler.GitService, func() string {
|
|
return customTemplate.ProjectPath
|
|
})
|
|
if err != nil {
|
|
err = rollbackCustomTemplate(backupPath, customTemplate.ProjectPath)
|
|
if err != nil {
|
|
return httperror.InternalServerError("Failed to rollback the custom template folder", err)
|
|
}
|
|
return httperror.InternalServerError(err.Error(), err)
|
|
}
|
|
|
|
if customTemplate.GitConfig.ConfigHash != commitHash {
|
|
customTemplate.GitConfig.ConfigHash = commitHash
|
|
}
|
|
|
|
err = handler.DataStore.CustomTemplate().UpdateCustomTemplate(customTemplate.ID, customTemplate)
|
|
if err != nil {
|
|
return httperror.InternalServerError("Unable to persist custom template changes inside the database", err)
|
|
}
|
|
}
|
|
|
|
return response.JSON(w, customTemplate)
|
|
}
|
|
|
|
func backupCustomTemplate(projectPath string) (string, error) {
|
|
stat, err := os.Stat(projectPath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
backupPath := fmt.Sprintf("%s-backup", projectPath)
|
|
err = os.Rename(projectPath, backupPath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
err = os.Mkdir(projectPath, stat.Mode())
|
|
if err != nil {
|
|
return backupPath, err
|
|
}
|
|
return backupPath, nil
|
|
}
|
|
|
|
func rollbackCustomTemplate(backupPath, projectPath string) error {
|
|
os.RemoveAll(projectPath)
|
|
return os.Rename(backupPath, projectPath)
|
|
}
|
|
|
|
func cleanUpBackupCustomTemplate(backupPath string) error {
|
|
return os.RemoveAll(backupPath)
|
|
}
|