octopush/internal/services/storage/storage.go

package storage

import (
	"errors"
	"os"
	"path"

	"go.uber.org/zap"
	"golang.org/x/net/context"
)

// The idea behind storage is that we have file dir, with a git repo.
// This file repo can now take certain actions

type StorageConfig struct {
	Path string
}

func NewDefaultStorageConfig() (*StorageConfig, error) {
	tempDir, err := os.MkdirTemp(os.TempDir(), "")
	if err != nil {
		return nil, err
	}
	return &StorageConfig{
		Path: path.Join(tempDir, "kraken"),
	}, nil
}

type Service struct {
	logger *zap.Logger
	cfg    *StorageConfig
}

func NewService(logger *zap.Logger, cfg *StorageConfig) *Service {
	return &Service{logger: logger, cfg: cfg}
}

func (s *Service) getStoragePath(ctx context.Context) string {
	return path.Join(s.cfg.Path, "storage")
}

func (s *Service) InitializeStorage(ctx context.Context) error {
	return os.MkdirAll(s.getStoragePath(ctx), 0755)
}

func (s *Service) CleanupStorage(ctx context.Context) error {
	doneRemovingChan := make(chan struct{}, 1)
	go func(ctx context.Context) {
		s.logger.Debug("Removing all temp storage")
		os.RemoveAll(s.getStoragePath(ctx))
		doneRemovingChan <- struct{}{}
	}(ctx)

	select {
	case <-ctx.Done():
		return errors.New("could not cleanup storage aborting")
	case <-doneRemovingChan:
		return nil
	}

	return nil
}

func (s *Service) CreateArea(ctx context.Context) (*Area, error) {
	dir, err := os.MkdirTemp(s.getStoragePath(ctx), "*")
	if err != nil {
		return nil, err
	}

	return &Area{
		Path: dir,
	}, nil
}

func (s *Service) RemoveArea(ctx context.Context, area *Area) error {
	return os.RemoveAll(area.Path)
}