You've already forked img-proxy-url-generator
50 lines
1.0 KiB
Go
50 lines
1.0 KiB
Go
package grpc
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/siteworxpro/img-proxy-url-generator/config"
|
|
"github.com/siteworxpro/img-proxy-url-generator/generator"
|
|
"log"
|
|
"strings"
|
|
)
|
|
|
|
type GeneratorService struct {
|
|
UnimplementedGeneratorServer
|
|
imgGenerator *generator.Generator
|
|
}
|
|
|
|
func NewService(config *config.Config) (*GeneratorService, error) {
|
|
g, err := generator.NewGenerator(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &GeneratorService{imgGenerator: g}, nil
|
|
}
|
|
|
|
func (s *GeneratorService) Generate(c context.Context, r *UrlRequest) (*UrlResponse, error) {
|
|
defer c.Done()
|
|
|
|
var err error
|
|
format := generator.DEF
|
|
|
|
if r.Format != nil {
|
|
format, err = s.imgGenerator.StringToFormat(r.Format.String())
|
|
if err != nil {
|
|
println(err.Error())
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
url, err := s.imgGenerator.GenerateUrl(r.Image, r.Params, format)
|
|
if err != nil {
|
|
println(err.Error())
|
|
return nil, err
|
|
}
|
|
|
|
log.Println(fmt.Sprintf("%s - [%s] - (%s)", r.Image, strings.Join(r.Params, ","), url))
|
|
|
|
return &UrlResponse{Url: url}, nil
|
|
}
|