67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package middlewares
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/metrue/fx/config"
|
|
"github.com/metrue/fx/constants"
|
|
dockerHTTP "github.com/metrue/fx/container_runtimes/docker/http"
|
|
"github.com/metrue/fx/context"
|
|
"github.com/metrue/fx/infra"
|
|
dockerInfra "github.com/metrue/fx/infra/docker"
|
|
k8sInfra "github.com/metrue/fx/infra/k8s"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// Provision make sure infrastructure is healthy
|
|
func Provision(ctx context.Contexter) (err error) {
|
|
fxConfig := ctx.Get("config").(*config.Config)
|
|
cloud := fxConfig.Clouds[fxConfig.CurrentCloud]
|
|
|
|
var deployer infra.Deployer
|
|
if os.Getenv("KUBECONFIG") != "" {
|
|
deployer, err = k8sInfra.CreateDeployer(os.Getenv("KUBECONFIG"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ctx.Set("cloud_type", config.CloudTypeK8S)
|
|
} else if cloud["type"] == config.CloudTypeDocker {
|
|
provisioner := dockerInfra.CreateProvisioner(cloud["host"], cloud["user"])
|
|
ok, err := provisioner.HealthCheck()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !ok {
|
|
if _, err := provisioner.Provision(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
docker, err := dockerHTTP.Create(cloud["host"], constants.AgentPort)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "please make sure docker is installed and running on your host")
|
|
}
|
|
|
|
// TODO should clean up, but it needed in middlewares.Build
|
|
ctx.Set("docker", docker)
|
|
deployer, err = dockerInfra.CreateDeployer(docker)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ctx.Set("cloud_type", config.CloudTypeDocker)
|
|
} else if cloud["type"] == config.CloudTypeK8S {
|
|
deployer, err = k8sInfra.CreateDeployer(cloud["kubeconfig"])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ctx.Set("cloud_type", config.CloudTypeK8S)
|
|
} else {
|
|
return fmt.Errorf("unsupport cloud type %s, please make sure you config is correct", cloud["type"])
|
|
}
|
|
|
|
ctx.Set("deployer", deployer)
|
|
|
|
return nil
|
|
}
|