forked from Plemya-x/ALR
		
	feat: add checksum for git downloader
This commit is contained in:
		
							
								
								
									
										94
									
								
								pkg/dlcache/dlcache.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										94
									
								
								pkg/dlcache/dlcache.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,94 @@ | ||||
| // This file was originally part of the project "LURE - Linux User REpository", created by Elara Musayelyan. | ||||
| // It has been modified as part of "ALR - Any Linux Repository" by the ALR Authors. | ||||
| // | ||||
| // ALR - Any Linux Repository | ||||
| // Copyright (C) 2025 The ALR Authors | ||||
| // | ||||
| // This program is free software: you can redistribute it and/or modify | ||||
| // it under the terms of the GNU General Public License as published by | ||||
| // the Free Software Foundation, either version 3 of the License, or | ||||
| // (at your option) any later version. | ||||
| // | ||||
| // This program is distributed in the hope that it will be useful, | ||||
| // but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
| // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
| // GNU General Public License for more details. | ||||
| // | ||||
| // You should have received a copy of the GNU General Public License | ||||
| // along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  | ||||
| package dlcache | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"os" | ||||
| 	"path/filepath" | ||||
|  | ||||
| 	"gitea.plemya-x.ru/Plemya-x/ALR/internal/config" | ||||
| ) | ||||
|  | ||||
| type Config interface { | ||||
| 	GetPaths() *config.Paths | ||||
| } | ||||
|  | ||||
| type DownloadCache struct { | ||||
| 	cfg Config | ||||
| } | ||||
|  | ||||
| func New(cfg Config) *DownloadCache { | ||||
| 	return &DownloadCache{ | ||||
| 		cfg, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (dc *DownloadCache) BasePath(ctx context.Context) string { | ||||
| 	return filepath.Join( | ||||
| 		dc.cfg.GetPaths().CacheDir, "dl", | ||||
| 	) | ||||
| } | ||||
|  | ||||
| // New creates a new directory with the given ID in the cache. | ||||
| // If a directory with the same ID already exists, | ||||
| // it will be deleted before creating a new one. | ||||
| func (dc *DownloadCache) New(ctx context.Context, id string) (string, error) { | ||||
| 	h, err := hashID(id) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	itemPath := filepath.Join(dc.BasePath(ctx), h) | ||||
|  | ||||
| 	fi, err := os.Stat(itemPath) | ||||
| 	if err == nil || (fi != nil && !fi.IsDir()) { | ||||
| 		err = os.RemoveAll(itemPath) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	err = os.MkdirAll(itemPath, 0o755) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	return itemPath, nil | ||||
| } | ||||
|  | ||||
| // Get checks if an entry with the given ID | ||||
| // already exists in the cache, and if so, | ||||
| // returns the directory and true. If it | ||||
| // does not exist, it returns an empty string | ||||
| // and false. | ||||
| func (dc *DownloadCache) Get(ctx context.Context, id string) (string, bool) { | ||||
| 	h, err := hashID(id) | ||||
| 	if err != nil { | ||||
| 		return "", false | ||||
| 	} | ||||
| 	itemPath := filepath.Join(dc.BasePath(ctx), h) | ||||
|  | ||||
| 	_, err = os.Stat(itemPath) | ||||
| 	if err != nil { | ||||
| 		return "", false | ||||
| 	} | ||||
|  | ||||
| 	return itemPath, true | ||||
| } | ||||
							
								
								
									
										104
									
								
								pkg/dlcache/dlcache_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										104
									
								
								pkg/dlcache/dlcache_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,104 @@ | ||||
| // This file was originally part of the project "LURE - Linux User REpository", created by Elara Musayelyan. | ||||
| // It has been modified as part of "ALR - Any Linux Repository" by the ALR Authors. | ||||
| // | ||||
| // ALR - Any Linux Repository | ||||
| // Copyright (C) 2025 The ALR Authors | ||||
| // | ||||
| // This program is free software: you can redistribute it and/or modify | ||||
| // it under the terms of the GNU General Public License as published by | ||||
| // the Free Software Foundation, either version 3 of the License, or | ||||
| // (at your option) any later version. | ||||
| // | ||||
| // This program is distributed in the hope that it will be useful, | ||||
| // but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
| // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
| // GNU General Public License for more details. | ||||
| // | ||||
| // You should have received a copy of the GNU General Public License | ||||
| // along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  | ||||
| package dlcache_test | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"crypto/sha1" | ||||
| 	"encoding/hex" | ||||
| 	"io" | ||||
| 	"os" | ||||
| 	"path/filepath" | ||||
| 	"testing" | ||||
|  | ||||
| 	"gitea.plemya-x.ru/Plemya-x/ALR/internal/config" | ||||
| 	"gitea.plemya-x.ru/Plemya-x/ALR/pkg/dlcache" | ||||
| ) | ||||
|  | ||||
| type TestALRConfig struct { | ||||
| 	CacheDir string | ||||
| } | ||||
|  | ||||
| func (c *TestALRConfig) GetPaths() *config.Paths { | ||||
| 	return &config.Paths{ | ||||
| 		CacheDir: c.CacheDir, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func prepare(t *testing.T) *TestALRConfig { | ||||
| 	t.Helper() | ||||
|  | ||||
| 	dir, err := os.MkdirTemp("/tmp", "alr-dlcache-test.*") | ||||
| 	if err != nil { | ||||
| 		panic(err) | ||||
| 	} | ||||
|  | ||||
| 	return &TestALRConfig{ | ||||
| 		CacheDir: dir, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func cleanup(t *testing.T, cfg *TestALRConfig) { | ||||
| 	t.Helper() | ||||
| 	os.Remove(cfg.CacheDir) | ||||
| } | ||||
|  | ||||
| func TestNew(t *testing.T) { | ||||
| 	cfg := prepare(t) | ||||
| 	defer cleanup(t, cfg) | ||||
|  | ||||
| 	dc := dlcache.New(cfg) | ||||
|  | ||||
| 	ctx := context.Background() | ||||
|  | ||||
| 	const id = "https://example.com" | ||||
| 	dir, err := dc.New(ctx, id) | ||||
| 	if err != nil { | ||||
| 		t.Errorf("Expected no error, got %s", err) | ||||
| 	} | ||||
|  | ||||
| 	exp := filepath.Join(dc.BasePath(ctx), sha1sum(id)) | ||||
| 	if dir != exp { | ||||
| 		t.Errorf("Expected %s, got %s", exp, dir) | ||||
| 	} | ||||
|  | ||||
| 	fi, err := os.Stat(dir) | ||||
| 	if err != nil { | ||||
| 		t.Errorf("stat: expected no error, got %s", err) | ||||
| 	} | ||||
|  | ||||
| 	if !fi.IsDir() { | ||||
| 		t.Errorf("Expected cache item to be a directory") | ||||
| 	} | ||||
|  | ||||
| 	dir2, ok := dc.Get(ctx, id) | ||||
| 	if !ok { | ||||
| 		t.Errorf("Expected Get() to return valid value") | ||||
| 	} | ||||
| 	if dir2 != dir { | ||||
| 		t.Errorf("Expected %s from Get(), got %s", dir, dir2) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func sha1sum(id string) string { | ||||
| 	h := sha1.New() | ||||
| 	_, _ = io.WriteString(h, id) | ||||
| 	return hex.EncodeToString(h.Sum(nil)) | ||||
| } | ||||
							
								
								
									
										32
									
								
								pkg/dlcache/utils.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										32
									
								
								pkg/dlcache/utils.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,32 @@ | ||||
| // ALR - Any Linux Repository | ||||
| // Copyright (C) 2025 The ALR Authors | ||||
| // | ||||
| // This program is free software: you can redistribute it and/or modify | ||||
| // it under the terms of the GNU General Public License as published by | ||||
| // the Free Software Foundation, either version 3 of the License, or | ||||
| // (at your option) any later version. | ||||
| // | ||||
| // This program is distributed in the hope that it will be useful, | ||||
| // but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
| // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
| // GNU General Public License for more details. | ||||
| // | ||||
| // You should have received a copy of the GNU General Public License | ||||
| // along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  | ||||
| package dlcache | ||||
|  | ||||
| import ( | ||||
| 	"crypto/sha1" | ||||
| 	"encoding/hex" | ||||
| 	"io" | ||||
| ) | ||||
|  | ||||
| func hashID(id string) (string, error) { | ||||
| 	h := sha1.New() | ||||
| 	_, err := io.WriteString(h, id) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return hex.EncodeToString(h.Sum(nil)), nil | ||||
| } | ||||
		Reference in New Issue
	
	Block a user