mirror of
https://github.com/VaalaCat/frp-panel.git
synced 2025-09-26 19:31:18 +08:00
96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/VaalaCat/frp-panel/conf"
|
|
"github.com/casbin/casbin/v2"
|
|
"github.com/gin-gonic/gin"
|
|
"google.golang.org/grpc/credentials"
|
|
)
|
|
|
|
type Application interface {
|
|
GetStreamLogHookMgr() StreamLogHookMgr
|
|
SetStreamLogHookMgr(StreamLogHookMgr)
|
|
GetShellPTYMgr() ShellPTYMgr
|
|
SetShellPTYMgr(ShellPTYMgr)
|
|
GetClientLogManager() ClientLogManager
|
|
SetClientLogManager(ClientLogManager)
|
|
GetDBManager() DBManager
|
|
SetDBManager(DBManager)
|
|
GetClientRecvMap() *sync.Map
|
|
SetClientRecvMap(*sync.Map)
|
|
GetClientsManager() ClientsManager
|
|
SetClientsManager(ClientsManager)
|
|
GetMasterCli() MasterClient
|
|
SetMasterCli(MasterClient)
|
|
GetClientRPCHandler() ClientRPCHandler
|
|
SetClientRPCHandler(ClientRPCHandler)
|
|
GetServerHandler() ServerHandler
|
|
SetServerHandler(ServerHandler)
|
|
GetClientController() ClientController
|
|
SetClientController(ClientController)
|
|
GetServerController() ServerController
|
|
SetServerController(ServerController)
|
|
GetConfig() conf.Config
|
|
SetConfig(conf.Config)
|
|
GetRPCCred() credentials.TransportCredentials
|
|
SetRPCCred(credentials.TransportCredentials)
|
|
GetCurrentRole() string
|
|
SetCurrentRole(string)
|
|
GetEnforcer() *casbin.Enforcer
|
|
SetEnforcer(*casbin.Enforcer)
|
|
GetPermManager() PermissionManager
|
|
SetPermManager(PermissionManager)
|
|
GetWorkerExecManager() WorkerExecManager
|
|
SetWorkerExecManager(WorkerExecManager)
|
|
GetWorkersManager() WorkersManager
|
|
SetWorkersManager(WorkersManager)
|
|
}
|
|
|
|
type Context struct {
|
|
context.Context
|
|
appInstance Application
|
|
}
|
|
|
|
func (c *Context) GetApp() Application {
|
|
return c.appInstance
|
|
}
|
|
|
|
func (c *Context) GetGinCtx() *gin.Context {
|
|
return c.Context.(*gin.Context)
|
|
}
|
|
|
|
func (c *Context) GetCtx() context.Context {
|
|
return c.Context
|
|
}
|
|
|
|
func (c *Context) Background() *Context {
|
|
return NewContext(context.Background(), c.appInstance)
|
|
}
|
|
|
|
func NewContext(c context.Context, appInstance Application) *Context {
|
|
return &Context{
|
|
Context: c,
|
|
appInstance: appInstance,
|
|
}
|
|
}
|
|
|
|
func NewApp() Application {
|
|
return &application{}
|
|
}
|
|
|
|
// var app *application
|
|
|
|
// func GetApp() Application {
|
|
// if app == nil {
|
|
// app = NewApp().(*application)
|
|
// }
|
|
// return app
|
|
// }
|
|
|
|
// func SetAppInstance(a Application) {
|
|
// app = a.(*application)
|
|
// }
|