errorpages
This commit is contained in:
parent
474ed89dda
commit
be57476a8e
135
core/runner/http.go
Normal file
135
core/runner/http.go
Normal file
|
|
@ -0,0 +1,135 @@
|
|||
package runner
|
||||
|
||||
import (
|
||||
luajit "git.sharkk.net/Sky/LuaJIT-to-Go"
|
||||
)
|
||||
|
||||
// HTTPResponse represents an HTTP response from Lua
|
||||
type HTTPResponse struct {
|
||||
Status int `json:"status"`
|
||||
Headers map[string]string `json:"headers"`
|
||||
Body any `json:"body"`
|
||||
}
|
||||
|
||||
// NewHTTPResponse creates a default HTTP response
|
||||
func NewHTTPResponse() *HTTPResponse {
|
||||
return &HTTPResponse{
|
||||
Status: 200,
|
||||
Headers: make(map[string]string),
|
||||
}
|
||||
}
|
||||
|
||||
// LuaHTTPModule is the pure Lua implementation of the HTTP module
|
||||
const LuaHTTPModule = `
|
||||
-- Table to store response data
|
||||
__http_responses = {}
|
||||
|
||||
-- HTTP module implementation
|
||||
local http = {
|
||||
-- Set HTTP status code
|
||||
set_status = function(code)
|
||||
if type(code) ~= "number" then
|
||||
error("http.set_status: status code must be a number", 2)
|
||||
end
|
||||
|
||||
local resp = __http_responses[1] or {}
|
||||
resp.status = code
|
||||
__http_responses[1] = resp
|
||||
end,
|
||||
|
||||
-- Set HTTP header
|
||||
set_header = function(name, value)
|
||||
if type(name) ~= "string" or type(value) ~= "string" then
|
||||
error("http.set_header: name and value must be strings", 2)
|
||||
end
|
||||
|
||||
local resp = __http_responses[1] or {}
|
||||
resp.headers = resp.headers or {}
|
||||
resp.headers[name] = value
|
||||
__http_responses[1] = resp
|
||||
end
|
||||
}
|
||||
|
||||
-- Set content type; set_header helper
|
||||
http.set_content_type = function(content_type)
|
||||
http.set_header("Content-Type", content_type)
|
||||
end
|
||||
|
||||
-- Install HTTP module
|
||||
_G.http = http
|
||||
|
||||
-- Override sandbox executor to clear HTTP responses
|
||||
local old_execute_sandbox = __execute_sandbox
|
||||
__execute_sandbox = function(bytecode, ctx)
|
||||
-- Clear previous response for this thread
|
||||
__http_responses[1] = nil
|
||||
|
||||
-- Execute the original function
|
||||
local result = old_execute_sandbox(bytecode, ctx)
|
||||
|
||||
-- Return the result unchanged
|
||||
return result
|
||||
end
|
||||
|
||||
-- Make sure the HTTP module is accessible in sandbox
|
||||
if __env_system and __env_system.base_env then
|
||||
__env_system.base_env.http = http
|
||||
end
|
||||
`
|
||||
|
||||
// HTTPModuleInitFunc returns an initializer function for the HTTP module
|
||||
func HTTPModuleInitFunc() StateInitFunc {
|
||||
return func(state *luajit.State) error {
|
||||
// Initialize pure Lua HTTP module
|
||||
return state.DoString(LuaHTTPModule)
|
||||
}
|
||||
}
|
||||
|
||||
// GetHTTPResponse extracts the HTTP response from Lua state
|
||||
func GetHTTPResponse(state *luajit.State) (*HTTPResponse, bool) {
|
||||
response := NewHTTPResponse()
|
||||
|
||||
// Get response table
|
||||
state.GetGlobal("__http_responses")
|
||||
if state.IsNil(-1) {
|
||||
state.Pop(1)
|
||||
return response, false
|
||||
}
|
||||
|
||||
// Check for response at thread index
|
||||
state.PushNumber(1)
|
||||
state.GetTable(-2)
|
||||
if state.IsNil(-1) {
|
||||
state.Pop(2)
|
||||
return response, false
|
||||
}
|
||||
|
||||
// Get status
|
||||
state.GetField(-1, "status")
|
||||
if state.IsNumber(-1) {
|
||||
response.Status = int(state.ToNumber(-1))
|
||||
}
|
||||
state.Pop(1)
|
||||
|
||||
// Get headers
|
||||
state.GetField(-1, "headers")
|
||||
if state.IsTable(-1) {
|
||||
// Iterate through headers table
|
||||
state.PushNil() // Start iteration
|
||||
for state.Next(-2) {
|
||||
// Stack has key at -2 and value at -1
|
||||
if state.IsString(-2) && state.IsString(-1) {
|
||||
key := state.ToString(-2)
|
||||
value := state.ToString(-1)
|
||||
response.Headers[key] = value
|
||||
}
|
||||
state.Pop(1) // Pop value, leave key for next iteration
|
||||
}
|
||||
}
|
||||
state.Pop(1)
|
||||
|
||||
// Clean up
|
||||
state.Pop(2) // Pop response table and __http_responses
|
||||
|
||||
return response, true
|
||||
}
|
||||
227
core/utils/errorpages.go
Normal file
227
core/utils/errorpages.go
Normal file
|
|
@ -0,0 +1,227 @@
|
|||
package utils
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
// ErrorPageConfig holds configuration for generating error pages
|
||||
type ErrorPageConfig struct {
|
||||
OverrideDir string // Directory where override templates are stored
|
||||
DebugMode bool // Whether to show debug information
|
||||
}
|
||||
|
||||
// ErrorType represents HTTP error types
|
||||
type ErrorType int
|
||||
|
||||
const (
|
||||
ErrorTypeNotFound ErrorType = 404
|
||||
ErrorTypeMethodNotAllowed ErrorType = 405
|
||||
ErrorTypeInternalError ErrorType = 500
|
||||
)
|
||||
|
||||
// ErrorPage generates an HTML error page based on the error type
|
||||
// It first checks for an override file, and if not found, generates a default page
|
||||
func ErrorPage(config ErrorPageConfig, errorType ErrorType, url string, errMsg string) string {
|
||||
// Check for override file
|
||||
if config.OverrideDir != "" {
|
||||
var filename string
|
||||
switch errorType {
|
||||
case ErrorTypeNotFound:
|
||||
filename = "404.html"
|
||||
case ErrorTypeMethodNotAllowed:
|
||||
filename = "405.html"
|
||||
case ErrorTypeInternalError:
|
||||
filename = "500.html"
|
||||
}
|
||||
|
||||
if filename != "" {
|
||||
overridePath := filepath.Join(config.OverrideDir, filename)
|
||||
if content, err := os.ReadFile(overridePath); err == nil {
|
||||
return string(content)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// No override found, generate default page
|
||||
switch errorType {
|
||||
case ErrorTypeNotFound:
|
||||
return generateNotFoundHTML(url)
|
||||
case ErrorTypeMethodNotAllowed:
|
||||
return generateMethodNotAllowedHTML(url)
|
||||
case ErrorTypeInternalError:
|
||||
return generateInternalErrorHTML(config.DebugMode, url, errMsg)
|
||||
default:
|
||||
// Fallback to internal error
|
||||
return generateInternalErrorHTML(config.DebugMode, url, errMsg)
|
||||
}
|
||||
}
|
||||
|
||||
// NotFoundPage generates a 404 Not Found error page
|
||||
func NotFoundPage(config ErrorPageConfig, url string) string {
|
||||
return ErrorPage(config, ErrorTypeNotFound, url, "")
|
||||
}
|
||||
|
||||
// MethodNotAllowedPage generates a 405 Method Not Allowed error page
|
||||
func MethodNotAllowedPage(config ErrorPageConfig, url string) string {
|
||||
return ErrorPage(config, ErrorTypeMethodNotAllowed, url, "")
|
||||
}
|
||||
|
||||
// InternalErrorPage generates a 500 Internal Server Error page
|
||||
func InternalErrorPage(config ErrorPageConfig, url string, errMsg string) string {
|
||||
return ErrorPage(config, ErrorTypeInternalError, url, errMsg)
|
||||
}
|
||||
|
||||
// generateInternalErrorHTML creates a 500 Internal Server Error page
|
||||
func generateInternalErrorHTML(debugMode bool, url string, errMsg string) string {
|
||||
errorMessages := []string{
|
||||
"Oops! Something went wrong",
|
||||
"Oh no! The server choked",
|
||||
"Well, this is embarrassing...",
|
||||
"Houston, we have a problem",
|
||||
"Gremlins in the system",
|
||||
"The server is taking a coffee break",
|
||||
"Moonshark encountered a lunar eclipse",
|
||||
"Our code monkeys are working on it",
|
||||
"The server is feeling under the weather",
|
||||
"500 Brain Not Found",
|
||||
}
|
||||
|
||||
randomMessage := errorMessages[rand.Intn(len(errorMessages))]
|
||||
codeContent := url
|
||||
if errMsg != "" {
|
||||
codeContent = url + " → " + errMsg
|
||||
}
|
||||
return generateErrorHTML("500", randomMessage, "Internal Server Error", debugMode, codeContent)
|
||||
}
|
||||
|
||||
// generateNotFoundHTML creates a 404 Not Found error page
|
||||
func generateNotFoundHTML(url string) string {
|
||||
errorMessages := []string{
|
||||
"Nothing to see here",
|
||||
"This page is on vacation",
|
||||
"The page is missing in action",
|
||||
"This page has left the building",
|
||||
"This page is in another castle",
|
||||
"Sorry, we can't find that",
|
||||
"The page you're looking for doesn't exist",
|
||||
"Lost in space",
|
||||
"That's a 404",
|
||||
"Page not found",
|
||||
}
|
||||
|
||||
randomMessage := errorMessages[rand.Intn(len(errorMessages))]
|
||||
return generateErrorHTML("404", randomMessage, "Page Not Found", false, url)
|
||||
}
|
||||
|
||||
// generateMethodNotAllowedHTML creates a 405 Method Not Allowed error page
|
||||
func generateMethodNotAllowedHTML(url string) string {
|
||||
errorMessages := []string{
|
||||
"That's not how this works",
|
||||
"Method not allowed",
|
||||
"Wrong way!",
|
||||
"This method is not supported",
|
||||
"You can't do that here",
|
||||
"Sorry, wrong door",
|
||||
"That method won't work here",
|
||||
"Try a different approach",
|
||||
"Access denied for this method",
|
||||
"Method mismatch",
|
||||
}
|
||||
|
||||
randomMessage := errorMessages[rand.Intn(len(errorMessages))]
|
||||
return generateErrorHTML("405", randomMessage, "Method Not Allowed", false, url)
|
||||
}
|
||||
|
||||
// generateErrorHTML creates the common HTML structure for error pages
|
||||
func generateErrorHTML(errorCode, mainMessage, subMessage string, showDebugInfo bool, codeContent string) string {
|
||||
errorHTML := `<!doctype html>
|
||||
<html>
|
||||
<head>
|
||||
<title>` + errorCode + `</title>
|
||||
<style>
|
||||
:root {
|
||||
--bg-color: #2d2e2d;
|
||||
--bg-gradient: linear-gradient(to bottom, #2d2e2d 0%, #000 100%);
|
||||
--text-color: white;
|
||||
--code-bg: rgba(255, 255, 255, 0.1);
|
||||
}
|
||||
|
||||
@media (prefers-color-scheme: light) {
|
||||
:root {
|
||||
--bg-color: #f5f5f5;
|
||||
--bg-gradient: linear-gradient(to bottom, #f5f5f5 0%, #ddd 100%);
|
||||
--text-color: #333;
|
||||
--code-bg: rgba(0, 0, 0, 0.1);
|
||||
}
|
||||
}
|
||||
|
||||
body {
|
||||
font-family: sans-serif;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
align-items: center;
|
||||
justify-content: center;
|
||||
height: 100vh;
|
||||
margin: 0;
|
||||
background-color: var(--bg-color);
|
||||
color: var(--text-color);
|
||||
background: var(--bg-gradient);
|
||||
}
|
||||
h1 {
|
||||
font-size: 4rem;
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
}
|
||||
p {
|
||||
font-size: 1.5rem;
|
||||
margin: 0.5rem 0;
|
||||
padding: 0;
|
||||
}
|
||||
.sub-message {
|
||||
font-size: 1.2rem;
|
||||
margin-bottom: 1rem;
|
||||
opacity: 0.8;
|
||||
}
|
||||
code {
|
||||
display: inline-block;
|
||||
font-size: 1rem;
|
||||
font-family: monospace;
|
||||
background-color: var(--code-bg);
|
||||
padding: 0.25em 0.5em;
|
||||
border-radius: 0.25em;
|
||||
margin-top: 1rem;
|
||||
max-width: 90vw;
|
||||
overflow-wrap: break-word;
|
||||
word-break: break-all;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<div>
|
||||
<h1>` + errorCode + `</h1>
|
||||
<p>` + mainMessage + `</p>
|
||||
<div class="sub-message">` + subMessage + `</div>`
|
||||
|
||||
if codeContent != "" {
|
||||
errorHTML += `
|
||||
<code>` + codeContent + `</code>`
|
||||
}
|
||||
|
||||
// Add a note for debug mode
|
||||
if showDebugInfo {
|
||||
errorHTML += `
|
||||
<p style="font-size: 0.9rem; margin-top: 1rem;">
|
||||
An error occurred while processing your request.<br>
|
||||
Please check the server logs for details.
|
||||
</p>`
|
||||
}
|
||||
|
||||
errorHTML += `
|
||||
</div>
|
||||
</body>
|
||||
</html>`
|
||||
|
||||
return errorHTML
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user