Initial open-source release (MIT): image2api AI gateway
Full Go backend + Vue 3 frontend, OpenAI-compatible API, multi-provider account pools, billing/admin, Docker one-command deploy with auto HTTPS. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,48 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"strings"
|
||||
|
||||
"backend/internal/repo"
|
||||
)
|
||||
|
||||
type ImageAccessService struct {
|
||||
generatedRoot string
|
||||
showcase *repo.ShowcaseRepository
|
||||
auth *AuthService
|
||||
}
|
||||
|
||||
func NewImageAccessService(generatedRoot string, showcase *repo.ShowcaseRepository, auth *AuthService) *ImageAccessService {
|
||||
return &ImageAccessService{
|
||||
generatedRoot: generatedRoot,
|
||||
showcase: showcase,
|
||||
auth: auth,
|
||||
}
|
||||
}
|
||||
|
||||
// Resolve validates the path params and returns the object key (user/name).
|
||||
// Existence isn't checked here — that's the storage GET's job (404 if missing).
|
||||
func (s *ImageAccessService) Resolve(user, name string) (string, error) {
|
||||
user = strings.TrimSpace(user)
|
||||
name = strings.TrimSpace(name)
|
||||
if user == "" || name == "" {
|
||||
return "", errors.New("missing path params")
|
||||
}
|
||||
// :user and :name are single path segments (gin won't match "/"); guard
|
||||
// against traversal tokens anyway.
|
||||
if strings.Contains(user, "..") || strings.Contains(name, "..") ||
|
||||
strings.ContainsAny(user, `/\`) || strings.ContainsAny(name, `/\`) {
|
||||
return "", errors.New("invalid image path")
|
||||
}
|
||||
return user + "/" + name, nil
|
||||
}
|
||||
|
||||
func (s *ImageAccessService) IsPublic(ctx context.Context, rel string) (bool, error) {
|
||||
return s.showcase.IsPublicFile(ctx, rel)
|
||||
}
|
||||
|
||||
func (s *ImageAccessService) IsAuthorized(ctx context.Context, sessionCookie, owner string) (bool, error) {
|
||||
return s.auth.IsAuthorizedForPrivateImage(ctx, sessionCookie, owner)
|
||||
}
|
||||
Reference in New Issue
Block a user