mirror of
https://github.com/akyaiy/GoSally-mvp.git
synced 2026-01-03 02:52:25 +00:00
fmt
This commit is contained in:
@@ -1,4 +1,4 @@
|
|||||||
// The cmd package is the main package where all the main hooks and methods are called.
|
// The cmd package is the main package where all the main hooks and methods are called.
|
||||||
// GoSally uses spf13/cobra to organize all the calls.
|
// GoSally uses spf13/cobra to organize all the calls.
|
||||||
package cmd
|
package cmd
|
||||||
|
|
||||||
@@ -24,7 +24,7 @@ scripts in a given directory. For more information, visit: https://gosally.oblat
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute prepares global log, loads cmdline args
|
// Execute prepares global log, loads cmdline args
|
||||||
// and executes rootCmd.Execute()
|
// and executes rootCmd.Execute()
|
||||||
func Execute() {
|
func Execute() {
|
||||||
log.SetOutput(os.Stdout)
|
log.SetOutput(os.Stdout)
|
||||||
|
|||||||
@@ -85,7 +85,7 @@ func InitUUIDHook(_ context.Context, cs *corestate.CoreState, x *app.AppX) {
|
|||||||
corestate.NODE_UUID = uuid32
|
corestate.NODE_UUID = uuid32
|
||||||
}
|
}
|
||||||
|
|
||||||
// The hook is responsible for checking the initialization stage
|
// The hook is responsible for checking the initialization stage
|
||||||
// and restarting in some cases
|
// and restarting in some cases
|
||||||
func InitRuntimeHook(_ context.Context, cs *corestate.CoreState, x *app.AppX) {
|
func InitRuntimeHook(_ context.Context, cs *corestate.CoreState, x *app.AppX) {
|
||||||
if *x.Config.Env.ParentStagePID != os.Getpid() {
|
if *x.Config.Env.ParentStagePID != os.Getpid() {
|
||||||
@@ -138,7 +138,7 @@ func InitRuntimeHook(_ context.Context, cs *corestate.CoreState, x *app.AppX) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// post-init stage
|
// post-init stage
|
||||||
// The hook creates a run.lock file, which contains information
|
// The hook creates a run.lock file, which contains information
|
||||||
// about the process and the node, in the runtime directory.
|
// about the process and the node, in the runtime directory.
|
||||||
func InitRunlockHook(_ context.Context, cs *corestate.CoreState, x *app.AppX) {
|
func InitRunlockHook(_ context.Context, cs *corestate.CoreState, x *app.AppX) {
|
||||||
NodeApp.Fallback(func(ctx context.Context, cs *corestate.CoreState, x *app.AppX) {
|
NodeApp.Fallback(func(ctx context.Context, cs *corestate.CoreState, x *app.AppX) {
|
||||||
@@ -192,7 +192,7 @@ func InitRunlockHook(_ context.Context, cs *corestate.CoreState, x *app.AppX) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// The hook reads the configuration and replaces special expressions
|
// The hook reads the configuration and replaces special expressions
|
||||||
// (%tmp% and so on) in string fields with the required data.
|
// (%tmp% and so on) in string fields with the required data.
|
||||||
func InitConfigReplHook(_ context.Context, cs *corestate.CoreState, x *app.AppX) {
|
func InitConfigReplHook(_ context.Context, cs *corestate.CoreState, x *app.AppX) {
|
||||||
if !slices.Contains(*x.Config.Conf.DisableWarnings, "--WNonStdTmpDir") && os.TempDir() != "/tmp" {
|
if !slices.Contains(*x.Config.Conf.DisableWarnings, "--WNonStdTmpDir") && os.TempDir() != "/tmp" {
|
||||||
@@ -218,7 +218,7 @@ func InitConfigReplHook(_ context.Context, cs *corestate.CoreState, x *app.AppX)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// The hook is responsible for outputting the
|
// The hook is responsible for outputting the
|
||||||
// final config and asking for confirmation.
|
// final config and asking for confirmation.
|
||||||
func InitConfigPrintHook(ctx context.Context, cs *corestate.CoreState, x *app.AppX) {
|
func InitConfigPrintHook(ctx context.Context, cs *corestate.CoreState, x *app.AppX) {
|
||||||
if *x.Config.Conf.Node.ShowConfig {
|
if *x.Config.Conf.Node.ShowConfig {
|
||||||
@@ -254,7 +254,7 @@ func InitSLogHook(_ context.Context, cs *corestate.CoreState, x *app.AppX) {
|
|||||||
*x.SLog = *newSlog
|
*x.SLog = *newSlog
|
||||||
}
|
}
|
||||||
|
|
||||||
// The method goes through the entire config structure through
|
// The method goes through the entire config structure through
|
||||||
// reflection and replaces string fields with the required ones.
|
// reflection and replaces string fields with the required ones.
|
||||||
func processConfig(conf any, replacements map[string]any) error {
|
func processConfig(conf any, replacements map[string]any) error {
|
||||||
val := reflect.ValueOf(conf)
|
val := reflect.ValueOf(conf)
|
||||||
|
|||||||
@@ -11,8 +11,8 @@ import (
|
|||||||
"regexp"
|
"regexp"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/akyaiy/GoSally-mvp/src/internal/core/corestate"
|
|
||||||
"github.com/akyaiy/GoSally-mvp/src/internal/colors"
|
"github.com/akyaiy/GoSally-mvp/src/internal/colors"
|
||||||
|
"github.com/akyaiy/GoSally-mvp/src/internal/core/corestate"
|
||||||
"github.com/akyaiy/GoSally-mvp/src/internal/core/run_manager"
|
"github.com/akyaiy/GoSally-mvp/src/internal/core/run_manager"
|
||||||
"github.com/akyaiy/GoSally-mvp/src/internal/core/update"
|
"github.com/akyaiy/GoSally-mvp/src/internal/core/update"
|
||||||
"github.com/akyaiy/GoSally-mvp/src/internal/core/utils"
|
"github.com/akyaiy/GoSally-mvp/src/internal/core/utils"
|
||||||
@@ -22,6 +22,7 @@ import (
|
|||||||
"github.com/akyaiy/GoSally-mvp/src/internal/server/gateway"
|
"github.com/akyaiy/GoSally-mvp/src/internal/server/gateway"
|
||||||
"github.com/akyaiy/GoSally-mvp/src/internal/server/session"
|
"github.com/akyaiy/GoSally-mvp/src/internal/server/session"
|
||||||
"github.com/akyaiy/GoSally-mvp/src/internal/server/sv1"
|
"github.com/akyaiy/GoSally-mvp/src/internal/server/sv1"
|
||||||
|
"github.com/akyaiy/GoSally-mvp/src/internal/server/sv2"
|
||||||
"github.com/go-chi/chi/v5"
|
"github.com/go-chi/chi/v5"
|
||||||
"github.com/go-chi/cors"
|
"github.com/go-chi/cors"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
@@ -65,13 +66,20 @@ func RunHook(ctx context.Context, cs *corestate.CoreState, x *app.AppX) error {
|
|||||||
Ver: "v1",
|
Ver: "v1",
|
||||||
})
|
})
|
||||||
|
|
||||||
|
sv2 := sv2.InitServer(&sv2.HandlerInitStruct{
|
||||||
|
X: x,
|
||||||
|
CS: cs,
|
||||||
|
AllowedCmd: regexp.MustCompile(AllowedCmdPattern),
|
||||||
|
Ver: "v2",
|
||||||
|
})
|
||||||
|
|
||||||
session_manager := session.New(*x.Config.Conf.HTTPServer.SessionTTL)
|
session_manager := session.New(*x.Config.Conf.HTTPServer.SessionTTL)
|
||||||
|
|
||||||
s := gateway.InitGateway(&gateway.GatewayServerInit{
|
s := gateway.InitGateway(&gateway.GatewayServerInit{
|
||||||
SM: session_manager,
|
SM: session_manager,
|
||||||
CS: cs,
|
CS: cs,
|
||||||
X: x,
|
X: x,
|
||||||
}, serverv1)
|
}, serverv1, sv2)
|
||||||
|
|
||||||
r := chi.NewRouter()
|
r := chi.NewRouter()
|
||||||
r.Use(cors.Handler(cors.Options{
|
r.Use(cors.Handler(cors.Options{
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ func NewLuaPool() *LuaPool {
|
|||||||
pool: sync.Pool{
|
pool: sync.Pool{
|
||||||
New: func() any {
|
New: func() any {
|
||||||
L := lua.NewState()
|
L := lua.NewState()
|
||||||
|
|
||||||
return L
|
return L
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -30,6 +30,6 @@ func (lp *LuaPool) Put(L *lua.LState) {
|
|||||||
L.Close()
|
L.Close()
|
||||||
|
|
||||||
newL := lua.NewState()
|
newL := lua.NewState()
|
||||||
|
|
||||||
lp.pool.Put(newL)
|
lp.pool.Put(newL)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,13 +11,12 @@ import (
|
|||||||
type LuaEngineDeps struct {
|
type LuaEngineDeps struct {
|
||||||
HttpRequest *http.Request
|
HttpRequest *http.Request
|
||||||
JSONRPCRequest *rpc.RPCRequest
|
JSONRPCRequest *rpc.RPCRequest
|
||||||
SessionUUID string
|
SessionUUID string
|
||||||
ScriptPath string
|
ScriptPath string
|
||||||
}
|
}
|
||||||
|
|
||||||
type LuaEngineContract interface {
|
type LuaEngineContract interface {
|
||||||
Handle(deps *LuaEngineDeps) *rpc.RPCResponse
|
Handle(deps *LuaEngineDeps) *rpc.RPCResponse
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type LuaEngine struct {
|
type LuaEngine struct {
|
||||||
|
|||||||
@@ -80,10 +80,10 @@ func loadDBMod(llog *slog.Logger, sid string) func(*lua.LState) int {
|
|||||||
|
|
||||||
mt := L.NewTypeMetatable("gosally_db")
|
mt := L.NewTypeMetatable("gosally_db")
|
||||||
L.SetField(mt, "__index", L.SetFuncs(L.NewTable(), map[string]lua.LGFunction{
|
L.SetField(mt, "__index", L.SetFuncs(L.NewTable(), map[string]lua.LGFunction{
|
||||||
"exec": dbExec,
|
"exec": dbExec,
|
||||||
"query": dbQuery,
|
"query": dbQuery,
|
||||||
"query_row": dbQueryRow,
|
"query_row": dbQueryRow,
|
||||||
"close": dbClose,
|
"close": dbClose,
|
||||||
}))
|
}))
|
||||||
|
|
||||||
L.SetField(dbMod, "__seed", lua.LString(sid))
|
L.SetField(dbMod, "__seed", lua.LString(sid))
|
||||||
@@ -215,43 +215,43 @@ func dbExec(L *lua.LState) int {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func dbQueryRow(L *lua.LState) int {
|
func dbQueryRow(L *lua.LState) int {
|
||||||
ud := L.CheckUserData(1)
|
ud := L.CheckUserData(1)
|
||||||
conn, ok := ud.Value.(*DBConnection)
|
conn, ok := ud.Value.(*DBConnection)
|
||||||
if !ok {
|
if !ok {
|
||||||
L.Push(lua.LNil)
|
L.Push(lua.LNil)
|
||||||
L.Push(lua.LString("invalid database connection"))
|
L.Push(lua.LString("invalid database connection"))
|
||||||
return 2
|
return 2
|
||||||
}
|
}
|
||||||
|
|
||||||
query := L.CheckString(2)
|
query := L.CheckString(2)
|
||||||
|
|
||||||
var args []any
|
var args []any
|
||||||
if L.GetTop() >= 3 {
|
if L.GetTop() >= 3 {
|
||||||
params := L.CheckTable(3)
|
params := L.CheckTable(3)
|
||||||
params.ForEach(func(k lua.LValue, v lua.LValue) {
|
params.ForEach(func(k lua.LValue, v lua.LValue) {
|
||||||
args = append(args, ConvertLuaTypesToGolang(v))
|
args = append(args, ConvertLuaTypesToGolang(v))
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
if conn.log {
|
if conn.log {
|
||||||
conn.logger.Info("DB QueryRow",
|
conn.logger.Info("DB QueryRow",
|
||||||
slog.String("query", query),
|
slog.String("query", query),
|
||||||
slog.Any("params", args))
|
slog.Any("params", args))
|
||||||
}
|
}
|
||||||
|
|
||||||
mtx := getDBMutex(conn.dbPath)
|
mtx := getDBMutex(conn.dbPath)
|
||||||
mtx.RLock()
|
mtx.RLock()
|
||||||
defer mtx.RUnlock()
|
defer mtx.RUnlock()
|
||||||
|
|
||||||
db, err := sql.Open("sqlite", conn.dbPath+"?_busy_timeout=5000&_journal_mode=WAL&_sync=NORMAL&_cache_size=-10000")
|
db, err := sql.Open("sqlite", conn.dbPath+"?_busy_timeout=5000&_journal_mode=WAL&_sync=NORMAL&_cache_size=-10000")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
L.Push(lua.LNil)
|
L.Push(lua.LNil)
|
||||||
L.Push(lua.LString(err.Error()))
|
L.Push(lua.LString(err.Error()))
|
||||||
return 2
|
return 2
|
||||||
}
|
}
|
||||||
defer db.Close()
|
defer db.Close()
|
||||||
|
|
||||||
row := db.QueryRow(query, args...)
|
row := db.QueryRow(query, args...)
|
||||||
|
|
||||||
columns := []string{}
|
columns := []string{}
|
||||||
stmt, err := db.Prepare(query)
|
stmt, err := db.Prepare(query)
|
||||||
@@ -278,36 +278,36 @@ func dbQueryRow(L *lua.LState) int {
|
|||||||
columns = append(columns, c)
|
columns = append(columns, c)
|
||||||
}
|
}
|
||||||
|
|
||||||
colCount := len(columns)
|
colCount := len(columns)
|
||||||
values := make([]any, colCount)
|
values := make([]any, colCount)
|
||||||
valuePtrs := make([]any, colCount)
|
valuePtrs := make([]any, colCount)
|
||||||
for i := range columns {
|
for i := range columns {
|
||||||
valuePtrs[i] = &values[i]
|
valuePtrs[i] = &values[i]
|
||||||
}
|
}
|
||||||
|
|
||||||
err = row.Scan(valuePtrs...)
|
err = row.Scan(valuePtrs...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
L.Push(lua.LNil)
|
L.Push(lua.LNil)
|
||||||
return 1
|
return 1
|
||||||
}
|
}
|
||||||
L.Push(lua.LNil)
|
L.Push(lua.LNil)
|
||||||
L.Push(lua.LString(fmt.Sprintf("scan failed: %v", err)))
|
L.Push(lua.LString(fmt.Sprintf("scan failed: %v", err)))
|
||||||
return 2
|
return 2
|
||||||
}
|
}
|
||||||
|
|
||||||
rowTable := L.NewTable()
|
rowTable := L.NewTable()
|
||||||
for i, col := range columns {
|
for i, col := range columns {
|
||||||
val := values[i]
|
val := values[i]
|
||||||
if val == nil {
|
if val == nil {
|
||||||
L.SetField(rowTable, col, lua.LNil)
|
L.SetField(rowTable, col, lua.LNil)
|
||||||
} else {
|
} else {
|
||||||
L.SetField(rowTable, col, ConvertGolangTypesToLua(L, val))
|
L.SetField(rowTable, col, ConvertGolangTypesToLua(L, val))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
L.Push(rowTable)
|
L.Push(rowTable)
|
||||||
return 1
|
return 1
|
||||||
}
|
}
|
||||||
|
|
||||||
func dbQuery(L *lua.LState) int {
|
func dbQuery(L *lua.LState) int {
|
||||||
|
|||||||
12
src/internal/server/sv2/handle.go
Normal file
12
src/internal/server/sv2/handle.go
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
package sv2
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/akyaiy/GoSally-mvp/src/internal/server/rpc"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (h *Handler) Handle(_ context.Context, sid string, r *http.Request, req *rpc.RPCRequest) *rpc.RPCResponse {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
43
src/internal/server/sv2/server.go
Normal file
43
src/internal/server/sv2/server.go
Normal file
@@ -0,0 +1,43 @@
|
|||||||
|
// SV2 works with binaries, scripts, and anything else that has access to stdin/stdout.
|
||||||
|
// Modules run in a separate process and communicate via I/O.
|
||||||
|
package sv2
|
||||||
|
|
||||||
|
import (
|
||||||
|
"regexp"
|
||||||
|
|
||||||
|
"github.com/akyaiy/GoSally-mvp/src/internal/core/corestate"
|
||||||
|
"github.com/akyaiy/GoSally-mvp/src/internal/engine/app"
|
||||||
|
)
|
||||||
|
|
||||||
|
// HandlerV2InitStruct structure is only for initialization
|
||||||
|
type HandlerInitStruct struct {
|
||||||
|
Ver string
|
||||||
|
CS *corestate.CoreState
|
||||||
|
X *app.AppX
|
||||||
|
AllowedCmd *regexp.Regexp
|
||||||
|
}
|
||||||
|
|
||||||
|
type Handler struct {
|
||||||
|
cs *corestate.CoreState
|
||||||
|
x *app.AppX
|
||||||
|
|
||||||
|
// allowedCmd and listAllowedCmd are regular expressions used to validate command names.
|
||||||
|
allowedCmd *regexp.Regexp
|
||||||
|
|
||||||
|
ver string
|
||||||
|
}
|
||||||
|
|
||||||
|
func InitServer(o *HandlerInitStruct) *Handler {
|
||||||
|
return &Handler{
|
||||||
|
cs: o.CS,
|
||||||
|
x: o.X,
|
||||||
|
allowedCmd: o.AllowedCmd,
|
||||||
|
ver: o.Ver,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetVersion returns the API version of the HandlerV1, which is set during initialization.
|
||||||
|
// This version is used to identify the API version in the request routing.
|
||||||
|
func (h *Handler) GetVersion() string {
|
||||||
|
return h.ver
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user