90 lines
1.9 KiB
Go
90 lines
1.9 KiB
Go
package oweb
|
|
|
|
import (
|
|
"cmp"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/ollama/ollama/x/client/ollama"
|
|
)
|
|
|
|
func Missing(field string) error {
|
|
return &ollama.Error{
|
|
Status: 400,
|
|
Code: "missing",
|
|
Field: field,
|
|
Message: fmt.Sprintf("%s is required", field),
|
|
}
|
|
}
|
|
|
|
func Invalid(field, value, format string, args ...any) error {
|
|
return &ollama.Error{
|
|
Status: 400,
|
|
Code: "invalid",
|
|
Field: field,
|
|
Value: value,
|
|
Message: fmt.Sprintf(format, args...),
|
|
}
|
|
}
|
|
|
|
// Convenience errors
|
|
var (
|
|
ErrNotFound = &ollama.Error{Status: 404, Code: "not_found"}
|
|
ErrInternal = &ollama.Error{Status: 500, Code: "internal_error"}
|
|
ErrMethodNotAllowed = &ollama.Error{Status: 405, Code: "method_not_allowed"}
|
|
)
|
|
|
|
type HandlerFunc func(w http.ResponseWriter, r *http.Request) error
|
|
|
|
func Serve(h HandlerFunc, w http.ResponseWriter, r *http.Request) {
|
|
if err := h(w, r); err != nil {
|
|
// TODO: take a slog.Logger
|
|
log.Printf("error: %v", err)
|
|
var oe *ollama.Error
|
|
if !errors.As(err, &oe) {
|
|
oe = ErrInternal
|
|
}
|
|
oe.Status = cmp.Or(oe.Status, 400)
|
|
w.WriteHeader(oe.Status)
|
|
if err := EncodeJSON(w, oe); err != nil {
|
|
log.Printf("error encoding error: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func DecodeUserJSON[T any](field string, r io.Reader) (*T, error) {
|
|
v, err := DecodeJSON[T](r)
|
|
|
|
// Handle common JSON syntax errors
|
|
var e *json.SyntaxError
|
|
if errors.As(err, &e) {
|
|
return nil, Invalid(field, "", e.Error())
|
|
}
|
|
|
|
// Handle type errors
|
|
var se *json.UnmarshalTypeError
|
|
if errors.As(err, &se) {
|
|
return nil, Invalid(field, se.Value, "expected %s", se.Type)
|
|
}
|
|
|
|
// Return v and err as they were.
|
|
return v, err
|
|
}
|
|
|
|
func DecodeJSON[T any](r io.Reader) (*T, error) {
|
|
var v *T
|
|
if err := json.NewDecoder(r).Decode(&v); err != nil {
|
|
var zero T
|
|
return &zero, err
|
|
}
|
|
return v, nil
|
|
}
|
|
|
|
func EncodeJSON(w io.Writer, v any) error {
|
|
return json.NewEncoder(w).Encode(v)
|
|
}
|