natpass/code/client/shell/shell.go

package shell

import (
	"fmt"
	"natpass/code/client/global"
	"natpass/code/client/pool"
	"net/http"

	"github.com/lwch/logging"
	"github.com/lwch/runtime"
)

type Shell struct {
	Name string
	cfg  global.Tunnel
}

// New new shell
func New(cfg global.Tunnel) *Shell {
	return &Shell{
		Name: cfg.Name,
		cfg:  cfg,
	}
}

// Handle handle shell
func (shell *Shell) Handle(pl *pool.Pool) {
	defer func() {
		if err := recover(); err != nil {
			logging.Error("close shell tunnel: %s, err=%v", shell.Name, err)
		}
	}()
	pf := func(cb func(*pool.Pool, http.ResponseWriter, *http.Request)) http.HandlerFunc {
		return func(w http.ResponseWriter, r *http.Request) {
			cb(pl, w, r)
		}
	}
	mux := http.NewServeMux()
	mux.HandleFunc("/new", pf(shell.New))
	svr := &http.Server{
		Addr:    fmt.Sprintf("%s:%d", shell.cfg.LocalAddr, shell.cfg.LocalPort),
		Handler: mux,
	}
	runtime.Assert(svr.ListenAndServe())
}