306 lines
9.5 KiB
Go
306 lines
9.5 KiB
Go
/*
|
||
* Copyright (c) Mengning Software. 2025. All rights reserved.
|
||
* Authors: DevStar Team, panshuxiao
|
||
* Create: 2025-11-19
|
||
* Description: Coordinates Kubernetes operations for installs.
|
||
*/
|
||
|
||
package appstore
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
|
||
k8s "code.gitea.io/gitea/modules/k8s"
|
||
applicationv1 "code.gitea.io/gitea/modules/k8s/api/application/v1"
|
||
"code.gitea.io/gitea/modules/k8s/application"
|
||
"code.gitea.io/gitea/modules/log"
|
||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||
dynamicclient "k8s.io/client-go/dynamic"
|
||
)
|
||
|
||
// K8sManager handles Kubernetes-specific application operations
|
||
type K8sManager struct {
|
||
ctx context.Context
|
||
}
|
||
|
||
// K8sCredential describes how to reach a Kubernetes cluster.
|
||
type K8sCredential struct {
|
||
K8sURL string
|
||
Token string
|
||
}
|
||
|
||
// NewK8sManager creates a new Kubernetes manager
|
||
func NewK8sManager(ctx context.Context) *K8sManager {
|
||
return &K8sManager{
|
||
ctx: ctx,
|
||
}
|
||
}
|
||
|
||
// buildK8sClient creates a dynamic client from the provided credential.
|
||
func (km *K8sManager) buildK8sClient(cred *K8sCredential) (dynamicclient.Interface, error) {
|
||
if cred != nil && cred.K8sURL != "" && cred.Token != "" {
|
||
return k8s.GetKubernetesClientWithToken(km.ctx, cred.K8sURL, cred.Token)
|
||
}
|
||
return k8s.GetKubernetesClientWithToken(km.ctx, "", "")
|
||
}
|
||
|
||
// InstallAppToKubernetes installs an application to a Kubernetes cluster
|
||
func (km *K8sManager) InstallAppToKubernetes(app *App, cred *K8sCredential) error {
|
||
// Validate that the app supports Kubernetes deployment
|
||
// 优先检查实际部署类型，如果为空则检查支持的部署类型
|
||
if app.Deploy.Type != "" && app.Deploy.Type != "kubernetes" {
|
||
return &AppStoreError{
|
||
Code: "DEPLOYMENT_TYPE_ERROR",
|
||
Message: "Application is not configured for Kubernetes deployment",
|
||
}
|
||
}
|
||
if app.Deploy.Type == "" && app.DeploymentType != "kubernetes" && app.DeploymentType != "both" {
|
||
return &AppStoreError{
|
||
Code: "DEPLOYMENT_TYPE_ERROR",
|
||
Message: "Application does not support Kubernetes deployment",
|
||
}
|
||
}
|
||
|
||
// Get Kubernetes client
|
||
k8sClient, err := km.buildK8sClient(cred)
|
||
if err != nil {
|
||
return &AppStoreError{
|
||
Code: "KUBERNETES_CLIENT_ERROR",
|
||
Message: "Failed to create Kubernetes client",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
// Build Kubernetes create options
|
||
createOptions, err := BuildK8sCreateOptions(app)
|
||
if err != nil {
|
||
return &AppStoreError{
|
||
Code: "KUBERNETES_OPTIONS_ERROR",
|
||
Message: "Failed to build Kubernetes create options",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
// Create the application in Kubernetes
|
||
_, err = application.CreateApplication(km.ctx, k8sClient, createOptions)
|
||
if err != nil {
|
||
return &AppStoreError{
|
||
Code: "KUBERNETES_CREATE_ERROR",
|
||
Message: "Failed to create application in Kubernetes",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// UninstallAppFromKubernetes uninstalls an application from a Kubernetes cluster
|
||
func (km *K8sManager) UninstallAppFromKubernetes(app *App, cred *K8sCredential) error {
|
||
// Validate that the app supports Kubernetes deployment
|
||
// 优先检查实际部署类型，如果为空则检查支持的部署类型
|
||
if app.Deploy.Type != "" && app.Deploy.Type != "kubernetes" {
|
||
return &AppStoreError{
|
||
Code: "DEPLOYMENT_TYPE_ERROR",
|
||
Message: "Application is not configured for Kubernetes deployment",
|
||
}
|
||
}
|
||
if app.Deploy.Type == "" && app.DeploymentType != "kubernetes" && app.DeploymentType != "both" {
|
||
return &AppStoreError{
|
||
Code: "DEPLOYMENT_TYPE_ERROR",
|
||
Message: "Application does not support Kubernetes deployment",
|
||
}
|
||
}
|
||
|
||
// Get Kubernetes client
|
||
k8sClient, err := km.buildK8sClient(cred)
|
||
if err != nil {
|
||
return &AppStoreError{
|
||
Code: "KUBERNETES_CLIENT_ERROR",
|
||
Message: "Failed to create Kubernetes client",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
// Build Kubernetes delete options
|
||
deleteOptions, err := BuildK8sDeleteOptions(app, "")
|
||
if err != nil {
|
||
return &AppStoreError{
|
||
Code: "KUBERNETES_OPTIONS_ERROR",
|
||
Message: "Failed to build Kubernetes delete options",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
// Delete the application from Kubernetes
|
||
err = application.DeleteApplication(km.ctx, k8sClient, deleteOptions)
|
||
if err != nil {
|
||
return &AppStoreError{
|
||
Code: "KUBERNETES_DELETE_ERROR",
|
||
Message: "Failed to delete application from Kubernetes",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// GetAppFromKubernetes gets an application from a Kubernetes cluster
|
||
func (km *K8sManager) GetAppFromKubernetes(app *App, cred *K8sCredential) (interface{}, error) {
|
||
// Validate that the app supports Kubernetes deployment
|
||
// 优先检查实际部署类型，如果为空则检查支持的部署类型
|
||
if app.Deploy.Type != "" && app.Deploy.Type != "kubernetes" {
|
||
return nil, &AppStoreError{
|
||
Code: "DEPLOYMENT_TYPE_ERROR",
|
||
Message: "Application is not configured for Kubernetes deployment",
|
||
}
|
||
}
|
||
if app.Deploy.Type == "" && app.DeploymentType != "kubernetes" && app.DeploymentType != "both" {
|
||
return nil, &AppStoreError{
|
||
Code: "DEPLOYMENT_TYPE_ERROR",
|
||
Message: "Application does not support Kubernetes deployment",
|
||
}
|
||
}
|
||
|
||
// Get Kubernetes client
|
||
k8sClient, err := km.buildK8sClient(cred)
|
||
if err != nil {
|
||
return nil, &AppStoreError{
|
||
Code: "KUBERNETES_CLIENT_ERROR",
|
||
Message: "Failed to create Kubernetes client",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
// Build Kubernetes get options
|
||
getOptions, err := BuildK8sGetOptions(app, "", false)
|
||
if err != nil {
|
||
return nil, &AppStoreError{
|
||
Code: "KUBERNETES_OPTIONS_ERROR",
|
||
Message: "Failed to build Kubernetes get options",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
// Get the application from Kubernetes
|
||
return application.GetApplication(km.ctx, k8sClient, getOptions)
|
||
}
|
||
|
||
// ListAppsFromKubernetes lists applications from a Kubernetes cluster
|
||
func (km *K8sManager) ListAppsFromKubernetes(app *App, cred *K8sCredential) (*applicationv1.ApplicationList, error) {
|
||
// Validate that the app supports Kubernetes deployment
|
||
// 优先检查实际部署类型，如果为空则检查支持的部署类型
|
||
if app.Deploy.Type != "" && app.Deploy.Type != "kubernetes" {
|
||
return nil, &AppStoreError{
|
||
Code: "DEPLOYMENT_TYPE_ERROR",
|
||
Message: "Application is not configured for Kubernetes deployment",
|
||
}
|
||
}
|
||
if app.Deploy.Type == "" && app.DeploymentType != "kubernetes" && app.DeploymentType != "both" {
|
||
return nil, &AppStoreError{
|
||
Code: "DEPLOYMENT_TYPE_ERROR",
|
||
Message: "Application does not support Kubernetes deployment",
|
||
}
|
||
}
|
||
|
||
// Get Kubernetes client
|
||
k8sClient, err := km.buildK8sClient(cred)
|
||
if err != nil {
|
||
return nil, &AppStoreError{
|
||
Code: "KUBERNETES_CLIENT_ERROR",
|
||
Message: "Failed to create Kubernetes client",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
// Build Kubernetes list options
|
||
listOptions, err := BuildK8sListOptions("", metav1.ListOptions{})
|
||
if err != nil {
|
||
return nil, &AppStoreError{
|
||
Code: "KUBERNETES_OPTIONS_ERROR",
|
||
Message: "Failed to build Kubernetes list options",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
// List applications from Kubernetes
|
||
return application.ListApplications(km.ctx, k8sClient, listOptions)
|
||
}
|
||
|
||
// UpdateAppInKubernetes updates an application in a Kubernetes cluster
|
||
func (km *K8sManager) UpdateAppInKubernetes(app *App, cred *K8sCredential) error {
|
||
// Validate that the app supports Kubernetes deployment
|
||
// 优先检查实际部署类型，如果为空则检查支持的部署类型
|
||
if app.Deploy.Type != "" && app.Deploy.Type != "kubernetes" {
|
||
return &AppStoreError{
|
||
Code: "DEPLOYMENT_TYPE_ERROR",
|
||
Message: "Application is not configured for Kubernetes deployment",
|
||
}
|
||
}
|
||
if app.Deploy.Type == "" && app.DeploymentType != "kubernetes" && app.DeploymentType != "both" {
|
||
return &AppStoreError{
|
||
Code: "DEPLOYMENT_TYPE_ERROR",
|
||
Message: "Application does not support Kubernetes deployment",
|
||
}
|
||
}
|
||
|
||
// Get Kubernetes client
|
||
k8sClient, err := km.buildK8sClient(cred)
|
||
if err != nil {
|
||
return &AppStoreError{
|
||
Code: "KUBERNETES_CLIENT_ERROR",
|
||
Message: "Failed to create Kubernetes client",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
// 先获取现有的应用，以保留元数据（特别是 resourceVersion）
|
||
existingApp, err := km.GetAppFromKubernetes(app, cred)
|
||
if err != nil {
|
||
// 如果应用不存在，尝试创建新应用
|
||
log.Warn("Application not found in Kubernetes, attempting to create new application: %v", err)
|
||
return &AppStoreError{
|
||
Code: "APP_NOT_FOUND",
|
||
Message: "应用在 Kubernetes 中不存在，请先安装应用",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
// 确保获取到的是 Application 类型
|
||
appData, ok := existingApp.(*applicationv1.Application)
|
||
if !ok {
|
||
return &AppStoreError{
|
||
Code: "TYPE_ERROR",
|
||
Message: "无法获取现有应用数据",
|
||
Details: fmt.Sprintf("期望 *applicationv1.Application 类型，实际类型: %T", existingApp),
|
||
}
|
||
}
|
||
|
||
// Build Kubernetes update options，传入现有应用以保留元数据
|
||
updateOptions, err := BuildK8sUpdateOptions(app, "", appData)
|
||
if err != nil {
|
||
return &AppStoreError{
|
||
Code: "KUBERNETES_OPTIONS_ERROR",
|
||
Message: "Failed to build Kubernetes update options",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
// Update the application in Kubernetes
|
||
_, err = application.UpdateApplication(km.ctx, k8sClient, updateOptions)
|
||
if err != nil {
|
||
return &AppStoreError{
|
||
Code: "KUBERNETES_UPDATE_ERROR",
|
||
Message: "Failed to update application in Kubernetes",
|
||
Details: err.Error(),
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// IsApplicationReady 封装底层就绪判断，避免上层直接依赖 k8s/application 的实现
|
||
func (km *K8sManager) IsApplicationReady(status *applicationv1.ApplicationStatus) bool {
|
||
return application.IsApplicationStatusReady(status)
|
||
}
|