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
}