translation_v2_detection.go

package googletranslatewrapper

import (
	"context"
	"fmt"

	"cloud.google.com/go/translate"
	"github.com/weiyuan-lane/google-translate-api/internal/utils/errorhandlers"
)

func (t TranslateV2Wrapper) DetectionsFromText(ctx context.Context, text string) ([]Detection, error) {
	googleDetections, err := t.translateClient.DetectLanguage(
		ctx,
		[]string{text},
	)

	if err != nil {
		return []Detection{}, errorhandlers.Wrap(
			errorhandlers.ErrGoogleTranslateV2DetectErrResponse,
			fmt.Sprintf("Google translate detection returned error %s", err.Error()),
		)
	}

	if len(googleDetections) == 0 {
		return []Detection{}, errorhandlers.Wrap(
			errorhandlers.ErrGoogleTranslateV2EmptyDetectionResponse,
			"Google translate not returning any results",
		)
	}

	detections := t.makeDetectionsResponse(googleDetections[0])

	return detections, nil
}

func (t TranslateV2Wrapper) makeDetectionsResponse(googleDetections []translate.Detection) []Detection {
	detections := make([]Detection, len(googleDetections))

	for i, googleDetection := range googleDetections {
		detections[i] = Detection{
			Confidence: googleDetection.Confidence,
			IsReliable: googleDetection.IsReliable,
			Language:   googleDetection.Language,
		}
	}

	return detections
}