vieter/src/repo/add.v

152 lines
4.8 KiB
Coq
Raw Normal View History

2022-01-09 22:30:07 +01:00
module repo
import os
2022-01-19 18:54:33 +01:00
import package
import util
// Manages a group of repositories. Each repository contains one or more
// arch-repositories, each of which represent a specific architecture.
pub struct RepoGroupManager {
2022-01-11 16:37:49 +01:00
mut:
mutex shared util.Dummy
pub:
// Where to store repositories' files
2022-04-07 15:21:27 +02:00
repos_dir string [required]
2022-04-09 21:17:22 +02:00
// Where packages are stored; each arch-repository gets its own
// subdirectory
2022-01-13 21:47:14 +01:00
pkg_dir string [required]
// The default architecture to use for a repository. Whenever a package of
// arch "any" is added to a repo, it will also be added to this
// architecture.
default_arch string [required]
}
2022-02-02 13:24:31 +01:00
pub struct RepoAddResult {
pub:
name string
version string
archs []string
2022-02-02 13:24:31 +01:00
}
// new creates a new RepoGroupManager & creates the directories as needed
2022-11-01 21:59:18 +01:00
pub fn new(repos_dir string, pkg_dir string, default_arch string) !RepoGroupManager {
2022-04-07 15:21:27 +02:00
if !os.is_dir(repos_dir) {
2022-04-13 22:20:05 +02:00
os.mkdir_all(repos_dir) or { return error('Failed to create repos directory: $err.msg()') }
2022-01-19 18:54:33 +01:00
}
2022-01-13 21:47:14 +01:00
2022-01-19 18:54:33 +01:00
if !os.is_dir(pkg_dir) {
2022-04-13 22:20:05 +02:00
os.mkdir_all(pkg_dir) or { return error('Failed to create package directory: $err.msg()') }
2022-01-19 18:54:33 +01:00
}
2022-01-13 21:47:14 +01:00
return RepoGroupManager{
2022-04-07 15:21:27 +02:00
repos_dir: repos_dir
2022-01-19 18:54:33 +01:00
pkg_dir: pkg_dir
default_arch: default_arch
2022-01-19 18:54:33 +01:00
}
}
// add_pkg_from_path adds a package to a given repo, given the file path to the
// pkg archive. It's a wrapper around add_pkg_in_repo that parses the archive
// file, passes the result to add_pkg_in_repo, and hard links the archive to
// the right subdirectories in r.pkg_dir if it was successfully added.
2022-11-01 21:59:18 +01:00
pub fn (r &RepoGroupManager) add_pkg_from_path(repo string, pkg_path string) !RepoAddResult {
pkg := package.read_pkg_archive(pkg_path) or {
2022-04-13 22:20:05 +02:00
return error('Failed to read package file: $err.msg()')
}
2022-11-01 21:59:18 +01:00
archs := r.add_pkg_in_repo(repo, pkg)!
2022-01-19 18:54:33 +01:00
// If the add was successful, we move the file to the packages directory
for arch in archs {
repo_pkg_path := os.real_path(os.join_path(r.pkg_dir, repo, arch))
dest_path := os.join_path_single(repo_pkg_path, pkg.filename())
2022-11-01 21:59:18 +01:00
os.mkdir_all(repo_pkg_path)!
// We create hard links so that "any" arch packages aren't stored
// multiple times
2022-11-01 21:59:18 +01:00
os.link(pkg_path, dest_path)!
2022-01-19 18:54:33 +01:00
}
// After linking, we can remove the original file
2022-11-01 21:59:18 +01:00
os.rm(pkg_path)!
2022-02-02 13:24:31 +01:00
return RepoAddResult{
name: pkg.info.name
version: pkg.info.version
archs: archs
2022-02-02 13:24:31 +01:00
}
2022-01-11 14:40:25 +01:00
}
// add_pkg_in_repo adds a package to a given repo. This function is responsible
// for inspecting the package architecture. If said architecture is 'any', the
// package is added to each arch-repository within the given repo. A package of
// architecture 'any' is always added to the arch-repo defined by
// r.default_arch. If this arch-repo doesn't exist yet, it is created. If the
// architecture isn't 'any', the package is only added to the specific
// architecture.
2022-11-01 21:59:18 +01:00
fn (r &RepoGroupManager) add_pkg_in_repo(repo string, pkg &package.Pkg) ![]string {
// A package not of arch 'any' can be handled easily by adding it to the
// respective repo
if pkg.info.arch != 'any' {
2022-11-01 21:59:18 +01:00
r.add_pkg_in_arch_repo(repo, pkg.info.arch, pkg)!
return [pkg.info.arch]
}
mut arch_repos := []string{}
// If it is an "any" package, the package gets added to every currently
// present arch-repo. It will always get added to the r.default_arch repo,
// even if no or multiple others are present.
repo_dir := os.join_path_single(r.repos_dir, repo)
// If this is the first package that's added to the repo, the directory
// won't exist yet
if os.exists(repo_dir) {
2022-11-01 21:59:18 +01:00
arch_repos = os.ls(repo_dir)!
}
// The default_arch should always be updated when a package with arch 'any'
// is added.
if !arch_repos.contains(r.default_arch) {
arch_repos << r.default_arch
}
2022-03-27 18:22:30 +02:00
// Add the package to each found architecture
// NOTE: if any of these fail, the function fails. This means the user does
// not know which arch-repositories did succeed in adding the package, if
// any.
for arch in arch_repos {
2022-11-01 21:59:18 +01:00
r.add_pkg_in_arch_repo(repo, arch, pkg)!
}
return arch_repos
}
// add_pkg_in_arch_repo is the function that actually adds a package to a given
// arch-repo. It records the package's data in the arch-repo's desc & files
// files, and afterwards updates the db & files archives to reflect these
// changes.
2022-11-01 21:59:18 +01:00
fn (r &RepoGroupManager) add_pkg_in_arch_repo(repo string, arch string, pkg &package.Pkg) ! {
2022-04-07 15:21:27 +02:00
pkg_dir := os.join_path(r.repos_dir, repo, arch, '$pkg.info.name-$pkg.info.version')
// Remove the previous version of the package, if present
2022-11-01 21:59:18 +01:00
r.remove_pkg_from_arch_repo(repo, arch, pkg.info.name, false)!
os.mkdir_all(pkg_dir) or { return error('Failed to create package directory.') }
2022-01-19 18:54:33 +01:00
2022-11-01 21:59:18 +01:00
os.write_file(os.join_path_single(pkg_dir, 'desc'), pkg.to_desc()!) or {
os.rmdir_all(pkg_dir)!
2022-01-19 18:54:33 +01:00
return error('Failed to write desc file.')
2022-01-09 22:30:07 +01:00
}
2022-01-19 18:54:33 +01:00
os.write_file(os.join_path_single(pkg_dir, 'files'), pkg.to_files()) or {
2022-11-01 21:59:18 +01:00
os.rmdir_all(pkg_dir)!
2022-01-19 18:54:33 +01:00
return error('Failed to write files file.')
}
2022-11-01 21:59:18 +01:00
r.sync(repo, arch)!
2022-01-19 18:54:33 +01:00
}