despiegk
27a536ab9a
* 'development' of github.com:Incubaid/herolib: refactor: Simplify handler signatures and add server runner fix: improve Redis response parsing and error handling fix: Correct AGEClient method receivers and error syntax
97 lines
2.6 KiB
V
97 lines
2.6 KiB
V
module heroserver
|
|
|
|
import veb
|
|
import freeflowuniverse.herolib.schemas.jsonrpc
|
|
import freeflowuniverse.herolib.crypt
|
|
import freeflowuniverse.herolib.heroserver.auth
|
|
import freeflowuniverse.herolib.heroserver.handlers
|
|
|
|
pub struct ServerConfig {
|
|
pub mut:
|
|
port int = 8080
|
|
auth_config auth.AuthConfig
|
|
}
|
|
|
|
pub struct HeroServer {
|
|
pub mut:
|
|
config ServerConfig
|
|
auth_manager &auth.AuthManager
|
|
handler_registry &handlers.HandlerRegistry
|
|
age_client &crypt.AGEClient
|
|
}
|
|
|
|
pub struct Context {
|
|
veb.Context
|
|
}
|
|
|
|
// Start the server
|
|
pub fn (mut s HeroServer) start() ! {
|
|
veb.run[HeroServer, Context](mut s, s.config.port)
|
|
}
|
|
|
|
// Authentication endpoints
|
|
@['/register'; post]
|
|
pub fn (mut s HeroServer) register(mut ctx Context) veb.Result {
|
|
// Implementation for pubkey registration
|
|
return ctx.text('not implemented')
|
|
}
|
|
|
|
@['/authreq'; post]
|
|
pub fn (mut s HeroServer) authreq(mut ctx Context) veb.Result {
|
|
// Implementation for authentication request
|
|
return ctx.text('not implemented')
|
|
}
|
|
|
|
@['/auth'; post]
|
|
pub fn (mut s HeroServer) auth(mut ctx Context) veb.Result {
|
|
// Implementation for authentication verification
|
|
return ctx.text('not implemented')
|
|
}
|
|
|
|
// API endpoints
|
|
@['/api/:handler_type'; post]
|
|
pub fn (mut s HeroServer) api(mut ctx Context, handler_type string) veb.Result {
|
|
// Validate session
|
|
session_key := ctx.get_custom_header('Authorization') or { '' }
|
|
if !s.auth_manager.validate_session(session_key) {
|
|
return ctx.request_error('Invalid session')
|
|
}
|
|
|
|
// Get handler and process request
|
|
mut handler := s.handler_registry.get(handler_type) or { return ctx.not_found() }
|
|
|
|
request := jsonrpc.decode_request(ctx.req.data) or {
|
|
return ctx.request_error('Invalid JSON-RPC request')
|
|
}
|
|
|
|
response := handler.handle(request) or { return ctx.server_error('Handler error') }
|
|
|
|
return ctx.json(response)
|
|
}
|
|
|
|
// Documentation endpoints
|
|
@['/doc/:handler_type'; get]
|
|
pub fn (mut s HeroServer) doc(mut ctx Context, handler_type string) veb.Result {
|
|
handler := s.handler_registry.get(handler_type) or { return ctx.not_found() }
|
|
|
|
doc_html := s.generate_documentation(handler_type, handler) or {
|
|
return ctx.server_error('Documentation generation failed')
|
|
}
|
|
|
|
return ctx.html(doc_html)
|
|
}
|
|
|
|
// new_server creates a new HeroServer instance
|
|
pub fn new_server(config ServerConfig) !&HeroServer {
|
|
mut auth_manager := auth.new_auth_manager(config.auth_config)!
|
|
mut handler_registry := handlers.new_handler_registry()!
|
|
mut age_client := crypt.new_age_client(crypt.AGEClientConfig{})!
|
|
|
|
return &HeroServer{
|
|
config: config
|
|
auth_manager: auth_manager
|
|
handler_registry: handler_registry
|
|
age_client: age_client
|
|
}
|
|
}
|