shithub: furgit

ref: 21e2f19ae2a3baa744b6bbed997f757f0af7e01f
dir: /receivepack/service/ingest_quarantine.go/

package service

import (
	"os"

	"codeberg.org/lindenii/furgit/internal/utils"
	"codeberg.org/lindenii/furgit/packfile/ingest"
)

func (service *Service) ingestQuarantine(
	result *Result,
	commands []Command,
	req *Request,
) (string, *os.Root, bool) {
	if !req.PackExpected {
		return "", nil, true
	}

	if req.Pack == nil {
		utils.BestEffortFprintf(service.opts.Progress, "unpack failed: missing pack stream.\n")

		result.UnpackError = "missing pack stream"
		fillCommandErrors(result, commands, "missing pack stream")

		return "", nil, false
	}

	if service.opts.ObjectsRoot == nil {
		utils.BestEffortFprintf(service.opts.Progress, "unpack failed: objects root not configured.\n")

		result.UnpackError = "objects root not configured"
		fillCommandErrors(result, commands, "objects root not configured")

		return "", nil, false
	}

	var err error

	err = service.opts.ExistingObjects.Refresh()
	if err != nil {
		utils.BestEffortFprintf(service.opts.Progress, "unpack failed: refresh existing objects: %v.\n", err)

		result.UnpackError = err.Error()
		fillCommandErrors(result, commands, err.Error())

		return "", nil, false
	}

	pending, err := ingest.Ingest(
		req.Pack,
		service.opts.Algorithm,
		ingest.Options{
			FixThin:       true,
			WriteRev:      true,
			Base:          service.opts.ExistingObjects,
			Progress:      service.opts.Progress,
			ProgressFlush: service.opts.ProgressFlush,
		},
	)
	if err != nil {
		utils.BestEffortFprintf(service.opts.Progress, "unpack failed: %v.\n", err)

		result.UnpackError = err.Error()
		fillCommandErrors(result, commands, err.Error())

		return "", nil, false
	}

	if pending.Header().ObjectCount == 0 {
		discarded, err := pending.Discard()
		if err != nil {
			utils.BestEffortFprintf(service.opts.Progress, "unpack failed: %v.\n", err)

			result.UnpackError = err.Error()
			fillCommandErrors(result, commands, err.Error())

			return "", nil, false
		}

		result.Ingest = &ingest.Result{
			PackHash:    discarded.PackHash,
			ObjectCount: discarded.ObjectCount,
		}

		utils.BestEffortFprintf(
			service.opts.Progress,
			"unpacking: done (%d objects, %s).\n",
			discarded.ObjectCount,
			discarded.PackHash,
		)

		return "", nil, true
	}

	utils.BestEffortFprintf(service.opts.Progress, "creating quarantine...\r")

	quarantineName, quarantineRoot, err := service.createQuarantineRoot()
	if err != nil {
		utils.BestEffortFprintf(service.opts.Progress, "unpack failed: %v.\n", err)

		result.UnpackError = err.Error()
		fillCommandErrors(result, commands, err.Error())

		return "", nil, false
	}

	quarantinePackRoot, err := service.openQuarantinePackRoot(quarantineRoot)
	if err != nil {
		utils.BestEffortFprintf(service.opts.Progress, "unpack failed: %v.\n", err)

		result.UnpackError = err.Error()
		fillCommandErrors(result, commands, err.Error())

		_ = quarantineRoot.Close()
		_ = service.opts.ObjectsRoot.RemoveAll(quarantineName)

		return "", nil, false
	}

	utils.BestEffortFprintf(service.opts.Progress, "creating quarantine: done.\n")
	utils.BestEffortFprintf(service.opts.Progress, "unpacking...\r")

	ingested, err := pending.Continue(quarantinePackRoot)

	_ = quarantinePackRoot.Close()

	if err != nil {
		utils.BestEffortFprintf(service.opts.Progress, "unpack failed: %v.\n", err)

		result.UnpackError = err.Error()
		fillCommandErrors(result, commands, err.Error())

		_ = quarantineRoot.Close()
		_ = service.opts.ObjectsRoot.RemoveAll(quarantineName)

		return "", nil, false
	}

	utils.BestEffortFprintf(service.opts.Progress, "unpacking: done (%d objects, %s).\n", ingested.ObjectCount, ingested.PackHash)

	result.Ingest = &ingested

	return quarantineName, quarantineRoot, true
}