homebox/backend/internal/core/services/service_items.go

package services

import (
	"context"
	"errors"

	"github.com/google/uuid"
	"github.com/hay-kot/homebox/backend/internal/data/repo"
	"github.com/rs/zerolog/log"
)

var (
	ErrNotFound     = errors.New("not found")
	ErrFileNotFound = errors.New("file not found")
)

type ItemService struct {
	repo *repo.AllRepos

	filepath string
	// at is a map of tokens to attachment IDs. This is used to store the attachment ID
	// for issued URLs
	at attachmentTokens
}

func (svc *ItemService) CsvImport(ctx context.Context, gid uuid.UUID, data [][]string) (int, error) {
	loaded := []csvRow{}

	// Skip first row
	for _, row := range data[1:] {
		// Skip empty rows
		if len(row) == 0 {
			continue
		}

		if len(row) != NumOfCols {
			return 0, ErrInvalidCsv
		}

		r := newCsvRow(row)
		loaded = append(loaded, r)
	}

	// validate rows
	var errMap = map[int][]error{}
	var hasErr bool
	for i, r := range loaded {

		errs := r.validate()

		if len(errs) > 0 {
			hasErr = true
			lineNum := i + 2

			errMap[lineNum] = errs
		}
	}

	if hasErr {
		for lineNum, errs := range errMap {
			for _, err := range errs {
				log.Error().Err(err).Int("line", lineNum).Msg("csv import error")
			}
		}
	}

	// Bootstrap the locations and labels so we can reuse the created IDs for the items
	locations := map[string]uuid.UUID{}
	existingLocation, err := svc.repo.Locations.GetAll(ctx, gid)
	if err != nil {
		return 0, err
	}
	for _, loc := range existingLocation {
		locations[loc.Name] = loc.ID
	}

	labels := map[string]uuid.UUID{}
	existingLabels, err := svc.repo.Labels.GetAll(ctx, gid)
	if err != nil {
		return 0, err
	}
	for _, label := range existingLabels {
		labels[label.Name] = label.ID
	}

	for _, row := range loaded {

		// Locations
		if _, exists := locations[row.Location]; !exists {
			result, err := svc.repo.Locations.Create(ctx, gid, repo.LocationCreate{
				Name:        row.Location,
				Description: "",
			})
			if err != nil {
				return 0, err
			}
			locations[row.Location] = result.ID
		}

		// Labels

		for _, label := range row.getLabels() {
			if _, exists := labels[label]; exists {
				continue
			}
			result, err := svc.repo.Labels.Create(ctx, gid, repo.LabelCreate{
				Name:        label,
				Description: "",
			})
			if err != nil {
				return 0, err
			}
			labels[label] = result.ID
		}
	}

	// Create the items
	var count int
	for _, row := range loaded {
		// Check Import Ref
		if row.Item.ImportRef != "" {
			exists, err := svc.repo.Items.CheckRef(ctx, gid, row.Item.ImportRef)
			if exists {
				continue
			}
			if err != nil {
				log.Err(err).Msg("error checking import ref")
			}
		}

		locationID := locations[row.Location]
		labelIDs := []uuid.UUID{}
		for _, label := range row.getLabels() {
			labelIDs = append(labelIDs, labels[label])
		}

		log.Info().
			Str("name", row.Item.Name).
			Str("location", row.Location).
			Msgf("Creating Item: %s", row.Item.Name)

		result, err := svc.repo.Items.Create(ctx, gid, repo.ItemCreate{
			ImportRef:   row.Item.ImportRef,
			Name:        row.Item.Name,
			Description: row.Item.Description,
			LabelIDs:    labelIDs,
			LocationID:  locationID,
		})

		if err != nil {
			return count, err
		}

		// Update the item with the rest of the data
		_, err = svc.repo.Items.UpdateByGroup(ctx, gid, repo.ItemUpdate{
			// Edges
			LocationID: locationID,
			LabelIDs:   labelIDs,

			// General Fields
			ID:          result.ID,
			Name:        result.Name,
			Description: result.Description,
			Insured:     row.Item.Insured,
			Notes:       row.Item.Notes,
			Quantity:    row.Item.Quantity,

			// Identifies the item as imported
			SerialNumber: row.Item.SerialNumber,
			ModelNumber:  row.Item.ModelNumber,
			Manufacturer: row.Item.Manufacturer,

			// Purchase
			PurchaseFrom:  row.Item.PurchaseFrom,
			PurchasePrice: row.Item.PurchasePrice,
			PurchaseTime:  row.Item.PurchaseTime,

			// Warranty
			LifetimeWarranty: row.Item.LifetimeWarranty,
			WarrantyExpires:  row.Item.WarrantyExpires,
			WarrantyDetails:  row.Item.WarrantyDetails,

			SoldTo:    row.Item.SoldTo,
			SoldPrice: row.Item.SoldPrice,
			SoldTime:  row.Item.SoldTime,
			SoldNotes: row.Item.SoldNotes,
		})

		if err != nil {
			return count, err
		}

		count++
	}
	return count, nil
}