mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
63cd906cfc
- Merge all files inside openflare-server to the repository root directory. - Relocate agent, relay, and flared subprojects from internal/ to internal/apps/. - Combine docker-compose files and update build context paths to root. - Update GitHub workflows and Dockerfiles to refer to new directories and package names. - Rewrite Go package imports across all files. - Resolve database renew test race condition and clean up docs.
88 lines
2.3 KiB
Go
88 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log/slog"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/config"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/flared"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/frpc"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/heartbeat"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/httpclient"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/sync"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/wsclient"
|
|
)
|
|
|
|
func main() {
|
|
// Setup simple structured logging
|
|
slog.SetDefault(slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
|
|
Level: parseLevel(os.Getenv("LOG_LEVEL")),
|
|
})))
|
|
|
|
configPath := flag.String("config", "./flared.json", "flared config path")
|
|
flag.Parse()
|
|
|
|
cfg, err := config.Load(*configPath)
|
|
if err != nil {
|
|
slog.Error("load flared config failed", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
slog.Info("flared config loaded",
|
|
"server", cfg.ServerURL,
|
|
"frpc_path", cfg.FrpcPath,
|
|
"data_dir", cfg.DataDir,
|
|
"heartbeat_interval", cfg.HeartbeatInterval,
|
|
"sync_interval", cfg.SyncInterval,
|
|
)
|
|
|
|
frpcManager := frpc.NewManager(cfg)
|
|
_ = frpcManager.LoadState()
|
|
|
|
slog.Info("detected frpc version", "version", frpcManager.GetVersion())
|
|
|
|
httpClient := httpclient.New(cfg.ServerURL, cfg.InitialAuthToken(), cfg.RequestTimeout.Duration())
|
|
wsClient := wsclient.New(cfg.ServerURL, cfg.InitialAuthToken(), cfg.RequestTimeout.Duration())
|
|
|
|
syncService := sync.New(httpClient, frpcManager, cfg)
|
|
heartbeatService := heartbeat.New(httpClient, frpcManager, cfg)
|
|
|
|
runner := &flared.Runner{
|
|
Config: cfg,
|
|
FrpcManager: frpcManager,
|
|
HttpClient: httpClient,
|
|
WebSocketService: wsClient,
|
|
HeartbeatService: heartbeatService,
|
|
SyncService: syncService,
|
|
}
|
|
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
|
|
slog.Info("flared process started")
|
|
|
|
if err := runner.Run(ctx); err != nil && err != context.Canceled {
|
|
slog.Error("flared process exited with error", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
slog.Info("flared process stopped")
|
|
}
|
|
|
|
func parseLevel(value string) slog.Level {
|
|
switch strings.ToLower(strings.TrimSpace(value)) {
|
|
case "debug":
|
|
return slog.LevelDebug
|
|
case "warn", "warning":
|
|
return slog.LevelWarn
|
|
case "error":
|
|
return slog.LevelError
|
|
default:
|
|
return slog.LevelInfo
|
|
}
|
|
}
|