mirror of
https://github.com/blacktop/ipsw.git
synced 2026-05-08 12:22:26 +00:00
blacktop
122 lines
2.5 KiB
Go
122 lines
2.5 KiB
Go
// Package server The ipsw API.
|
|
//
|
|
// Schemes: http
|
|
// Host: localhost:8080
|
|
// BasePath: /v1
|
|
// Version: v1.0
|
|
//
|
|
// Consumes:
|
|
// - application/json
|
|
//
|
|
// Produces:
|
|
// - application/json
|
|
//
|
|
// swagger:meta
|
|
package server
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"os/signal"
|
|
"path/filepath"
|
|
"runtime"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/apex/log"
|
|
"github.com/blacktop/ipsw/api"
|
|
"github.com/blacktop/ipsw/api/server/routes"
|
|
"github.com/blacktop/ipsw/api/types"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// Config is the server config
|
|
type Config struct {
|
|
Host string
|
|
Port int
|
|
Socket string
|
|
Debug bool
|
|
}
|
|
|
|
// Server is the main server struct
|
|
type Server struct {
|
|
router *gin.Engine
|
|
server *http.Server
|
|
conf *Config
|
|
}
|
|
|
|
// NewServer creates a new server
|
|
func NewServer(conf *Config) *Server {
|
|
return &Server{
|
|
router: gin.Default(),
|
|
conf: conf,
|
|
}
|
|
}
|
|
|
|
// Start starts the server
|
|
func (s *Server) Start() error {
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
|
|
s.router.GET("/version", func(c *gin.Context) {
|
|
c.JSON(http.StatusOK, types.Version{
|
|
ApiVersion: api.DefaultVersion,
|
|
OSType: runtime.GOOS,
|
|
BuilderVersion: types.BuildVersion,
|
|
})
|
|
})
|
|
|
|
rg := s.router.Group("/v" + api.DefaultVersion)
|
|
|
|
routes.Add(rg)
|
|
|
|
s.server = &http.Server{
|
|
Addr: fmt.Sprintf(":%d", s.conf.Port),
|
|
Handler: s.router,
|
|
}
|
|
|
|
go func() {
|
|
if len(s.conf.Socket) > 0 {
|
|
l, err := net.Listen("unix", filepath.Clean(s.conf.Socket))
|
|
if err != nil {
|
|
log.Fatalf("server: failed to listen: %v\n", err)
|
|
}
|
|
if err := s.server.Serve(l); err != nil && err != http.ErrServerClosed {
|
|
log.Fatalf("server: failed to serve: %v\n", err)
|
|
}
|
|
} else {
|
|
if err := s.server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
log.Fatalf("server: failed to listen and serve: %v\n", err)
|
|
}
|
|
}
|
|
}()
|
|
|
|
// Listen for the interrupt signal.
|
|
<-ctx.Done()
|
|
|
|
// Restore default behavior on the interrupt signal and notify user of shutdown.
|
|
stop()
|
|
|
|
log.Warn("Shutting down gracefully: Press Ctrl+C again to force")
|
|
|
|
return s.Stop()
|
|
}
|
|
|
|
// Stop stops the server
|
|
func (s *Server) Stop() error {
|
|
// The context is used to inform the server it has 5 seconds to finish
|
|
// the request it is currently handling
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
if err := s.server.Shutdown(ctx); err != nil {
|
|
return fmt.Errorf("server forced to shutdown: %v", err)
|
|
}
|
|
|
|
log.Info("Server Exiting")
|
|
|
|
return nil
|
|
}
|