storeimages.go RawHistoryFolder

1package store
2
3import (
4	"fmt"
5	"io"
6	"os"
7	"path/filepath"
8
9	"github.com/bnema/gordon/internal/common"
10	"github.com/docker/docker/api/types"
11)
12
13type StorageConfig struct {
14	StorageDir string
15	ImageID    string
16	Images     []types.ImageSummary
17}
18
19func NewStorageConfig(config *common.Config) *StorageConfig {
20	return &StorageConfig{
21		StorageDir: config.General.StorageDir,
22	}
23}
24
25func SaveImageToStorage(config *common.Config, filename string, buf io.Reader) (string, error) {
26	// Check if the folder exist if not create it
27	if _, err := os.Stat(config.General.StorageDir); os.IsNotExist(err) {
28		err := os.MkdirAll(config.General.StorageDir, 0755)
29		if err != nil {
30			return "", fmt.Errorf("failed to create storage directory: %v", err)
31		}
32	}
33
34	// Define the path where the image will be saved
35	saveInPath := filepath.Join(config.General.StorageDir, filename)
36
37	// Create or open a file for appending.
38	outFile, err := os.Create(saveInPath)
39	if err != nil {
40		return "", fmt.Errorf("failed to open file: %v", err)
41	}
42
43	// Write the uploaded file's content to the outFile
44	_, err = io.Copy(outFile, buf)
45	if err != nil {
46		return "", fmt.Errorf("failed to write file: %v", err)
47	}
48
49	// Check if file exists
50	if _, err := os.Stat(saveInPath); os.IsNotExist(err) {
51		return "", fmt.Errorf("file does not exist: %v", err)
52	}
53
54	return saveInPath, nil
55}
56
57func (sc *StorageConfig) DeleteImageFromStorage(imageId string) error {
58	// Delete the image from the storage directory
59	imagePath := filepath.Join(sc.StorageDir, imageId+".tar")
60	err := os.Remove(imagePath)
61	if err != nil {
62		return fmt.Errorf("failed to delete image from storage directory: %v", err)
63	}
64
65	return nil
66}