function.go

package function

import (
	"context"
	"fmt"
	"os"
	"strings"

	"source.cloud.google.com/my-finance-6f9c3/bitbucket_pqadev_vm-cf/event"
	"source.cloud.google.com/my-finance-6f9c3/bitbucket_pqadev_vm-cf/internal"
)

var projectId = os.Getenv("PROJECT_ID")

func Image2Expense(ctx context.Context, e event.GCS) error {
	if !strings.Contains(e.ContentType, "image") {
		return fmt.Errorf("%s is not an image", e.SelfLink)
	}

	parts := strings.Split(e.Name, "/")
	if len(parts) <= 1 {
		return fmt.Errorf("Invalid gs path. Please upload the image below the user's folder.")
	}

	gs := fmt.Sprintf("gs://%s/%s", e.Bucket, e.Name)
	des, err := internal.Img2Text(ctx, gs)
	if err != nil {
		return err
	}
	fmt.Print(des)

	if len(des) == 0 {
		return fmt.Errorf("Could not detect text from the receipt image")
	}

	expense, err := internal.Text2ExpenseInfo(ctx, des)
	if err != nil {
		return err
	}

	id, err := internal.SaveExpense(ctx, projectId, parts[0], expense, des)
	if err != nil {
		return err
	}
	fmt.Printf("Document added: %s", id)
	
	return nil
}