mirror of
https://github.com/kubenetworks/kubevpn.git
synced 2025-12-24 11:51:13 +08:00
72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
package action
|
|
|
|
import (
|
|
"io"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/spf13/pflag"
|
|
|
|
"github.com/wencaiwulue/kubevpn/v2/pkg/config"
|
|
"github.com/wencaiwulue/kubevpn/v2/pkg/daemon/rpc"
|
|
"github.com/wencaiwulue/kubevpn/v2/pkg/handler"
|
|
"github.com/wencaiwulue/kubevpn/v2/pkg/ssh"
|
|
"github.com/wencaiwulue/kubevpn/v2/pkg/util"
|
|
)
|
|
|
|
func (svr *Server) Reset(req *rpc.ResetRequest, resp rpc.Daemon_ResetServer) error {
|
|
defer func() {
|
|
util.InitLoggerForServer(true)
|
|
log.SetOutput(svr.LogFile)
|
|
config.Debug = false
|
|
}()
|
|
out := io.MultiWriter(newResetWarp(resp), svr.LogFile)
|
|
util.InitLoggerForClient(config.Debug)
|
|
log.SetOutput(out)
|
|
|
|
connect := &handler.ConnectOptions{
|
|
Namespace: req.Namespace,
|
|
Workloads: req.Workloads,
|
|
}
|
|
|
|
file, err := util.ConvertToTempKubeconfigFile([]byte(req.KubeconfigBytes))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
flags := pflag.NewFlagSet("", pflag.ContinueOnError)
|
|
flags.AddFlag(&pflag.Flag{
|
|
Name: "kubeconfig",
|
|
DefValue: file,
|
|
})
|
|
var sshConf = ssh.ParseSshFromRPC(req.SshJump)
|
|
var ctx = resp.Context()
|
|
var path string
|
|
path, err = ssh.SshJump(ctx, sshConf, flags, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = connect.InitClient(util.InitFactoryByPath(path, req.Namespace))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = connect.Reset(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type resetWarp struct {
|
|
server rpc.Daemon_ResetServer
|
|
}
|
|
|
|
func (r *resetWarp) Write(p []byte) (n int, err error) {
|
|
_ = r.server.Send(&rpc.ResetResponse{
|
|
Message: string(p),
|
|
})
|
|
return len(p), nil
|
|
}
|
|
|
|
func newResetWarp(server rpc.Daemon_ResetServer) io.Writer {
|
|
return &resetWarp{server: server}
|
|
}
|