server

Documentation

Index

• func IsImageError(err error) bool
• func IsInvalidRequest(err error) bool
• func IsNoPortAvailable(err error) bool
• func IsSSHNotAvailable(err error) bool
• func IsTunnelError(err error) bool
• type Error
  • func NewError(code ErrorCode, message string, err error) *Error
  • func (e *Error) Error() string
  • func (e *Error) Unwrap() error
• type ErrorCode
• type Server
  • func New(cfg *config.Config, opts ...ServerOption) *Server
  • func (s *Server) Done() <-chan struct{}
  • func (s *Server) MessageChan() <-chan types.Message
  • func (s *Server) Ready() <-chan struct{}
  • func (s *Server) SendPostRequest(ctx context.Context, url, key, model string, useStream bool)
  • func (s *Server) Shutdown() error
  • func (s *Server) Start(ctx context.Context) error
  • func (s *Server) TunnelURL() string
• type ServerOption
  • func WithHTTPServer(server interfaces.HTTPServer) ServerOption
  • func WithImageGenerator(ig interfaces.ImageGenerator) ServerOption
  • func WithRouter(router interfaces.Router) ServerOption
  • func WithTunnel(t interfaces.Tunnel) ServerOption

Functions

func IsImageError

func IsImageError(err error) bool

IsImageError checks if the error is an image generation error

func IsInvalidRequest

func IsInvalidRequest(err error) bool

IsInvalidRequest checks if the error is an invalid request error

func IsNoPortAvailable

func IsNoPortAvailable(err error) bool

IsNoPortAvailable checks if the error is a no port available error

func IsSSHNotAvailable

func IsSSHNotAvailable(err error) bool

IsSSHNotAvailable checks if the error is an SSH not available error

func IsTunnelError

func IsTunnelError(err error) bool

IsTunnelError checks if the error is a tunnel error

Types

type Error

type Error struct {
	Code    ErrorCode
	Message string
	Err     error
}

Error represents a server error

func NewError

func NewError(code ErrorCode, message string, err error) *Error

NewError creates a new server error

func (*Error) Error

func (e *Error) Error() string

Error implements the error interface

func (*Error) Unwrap

func (e *Error) Unwrap() error

Unwrap returns the underlying error

type ErrorCode

type ErrorCode int

ErrorCode represents different types of server errors

const (
	ErrSSHNotAvailable ErrorCode = iota + 1
	ErrNoPortAvailable
	ErrTunnelStart
	ErrImageGeneration
	ErrInvalidRequest
	ErrServerStart
	ErrServerShutdown
)

type Server

type Server struct {
	// contains filtered or unexported fields
}

Server represents the main application server

func New

func New(cfg *config.Config, opts ...ServerOption) *Server

New creates a new server instance

func (*Server) Done

func (s *Server) Done() <-chan struct{}

Done returns the done channel

func (*Server) MessageChan

func (s *Server) MessageChan() <-chan types.Message

MessageChan returns the message channel

func (*Server) Ready

func (s *Server) Ready() <-chan struct{}

Ready returns the ready channel

func (*Server) SendPostRequest

func (s *Server) SendPostRequest(ctx context.Context, url, key, model string, useStream bool)

SendPostRequest sends a POST request to test the API

func (*Server) Shutdown

func (s *Server) Shutdown() error

Shutdown gracefully shuts down the server

func (*Server) Start

func (s *Server) Start(ctx context.Context) error

Start starts the server

func (*Server) TunnelURL

func (s *Server) TunnelURL() string

TunnelURL returns the tunnel URL

type ServerOption

type ServerOption func(*Server)

ServerOption represents a server configuration option

func WithHTTPServer

func WithHTTPServer(server interfaces.HTTPServer) ServerOption

WithHTTPServer sets a custom HTTP server

func WithImageGenerator

func WithImageGenerator(ig interfaces.ImageGenerator) ServerOption

WithImageGenerator sets a custom image generator

func WithRouter

func WithRouter(router interfaces.Router) ServerOption

WithRouter sets a custom router

func WithTunnel

func WithTunnel(t interfaces.Tunnel) ServerOption

WithTunnel sets a custom tunnel