forked from vieter-v/vieter
				
			Added extra comments; made linter happy
							parent
							
								
									cb2ba86200
								
							
						
					
					
						commit
						0dd4534e20
					
				| 
						 | 
					@ -99,8 +99,8 @@ fn parse_pkg_info_string(pkg_info_str &string) ?PkgInfo {
 | 
				
			||||||
	return pkg_info
 | 
						return pkg_info
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// read_pkg extracts the file list & .PKGINFO contents from an archive
 | 
					// read_pkg_archive extracts the file list & .PKGINFO contents from an archive
 | 
				
			||||||
// NOTE: this command currently only supports zstd-compressed tarballs
 | 
					// NOTE: this command only supports zstd- & gzip-compressed tarballs
 | 
				
			||||||
pub fn read_pkg_archive(pkg_path string) ?Pkg {
 | 
					pub fn read_pkg_archive(pkg_path string) ?Pkg {
 | 
				
			||||||
	if !os.is_file(pkg_path) {
 | 
						if !os.is_file(pkg_path) {
 | 
				
			||||||
		return error("'$pkg_path' doesn't exist or isn't a file.")
 | 
							return error("'$pkg_path' doesn't exist or isn't a file.")
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -16,7 +16,7 @@ pub:
 | 
				
			||||||
	pkg_dir string [required]
 | 
						pkg_dir string [required]
 | 
				
			||||||
	// The default architecture to use for a repository. In reality, this value
 | 
						// The default architecture to use for a repository. In reality, this value
 | 
				
			||||||
	// is only required when a package with architecture "any" is added as the
 | 
						// is only required when a package with architecture "any" is added as the
 | 
				
			||||||
	// first package to a repository.
 | 
						// first package of a repository.
 | 
				
			||||||
	default_arch string [required]
 | 
						default_arch string [required]
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
| 
						 | 
					@ -43,10 +43,14 @@ pub fn new(data_dir string, pkg_dir string, default_arch string) ?RepoGroupManag
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// add_from_path adds a package from an arbitrary path & moves it into the pkgs
 | 
					// add_pkg_from_path adds a package to a given repo, given the file path to the
 | 
				
			||||||
// directory if necessary.
 | 
					// 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 moves the archive to
 | 
				
			||||||
 | 
					// r.pkg_dir if it was successfully added.
 | 
				
			||||||
pub fn (r &RepoGroupManager) add_pkg_from_path(repo string, pkg_path string) ?RepoAddResult {
 | 
					pub fn (r &RepoGroupManager) add_pkg_from_path(repo string, pkg_path string) ?RepoAddResult {
 | 
				
			||||||
	pkg := package.read_pkg_archive(pkg_path) or { return error('Failed to read package file: $err.msg') }
 | 
						pkg := package.read_pkg_archive(pkg_path) or {
 | 
				
			||||||
 | 
							return error('Failed to read package file: $err.msg')
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	added := r.add_pkg_in_repo(repo, pkg) ?
 | 
						added := r.add_pkg_in_repo(repo, pkg) ?
 | 
				
			||||||
 | 
					
 | 
				
			||||||
| 
						 | 
					@ -66,36 +70,48 @@ pub fn (r &RepoGroupManager) add_pkg_from_path(repo string, pkg_path string) ?Re
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 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. If none
 | 
				
			||||||
 | 
					// exist, one is created for provided r.default_arch value. If the architecture
 | 
				
			||||||
 | 
					// isn't 'any', the package is only added to the specific architecture.
 | 
				
			||||||
fn (r &RepoGroupManager) add_pkg_in_repo(repo string, pkg &package.Pkg) ?bool {
 | 
					fn (r &RepoGroupManager) add_pkg_in_repo(repo string, pkg &package.Pkg) ?bool {
 | 
				
			||||||
	if pkg.info.arch == "any" {
 | 
						if pkg.info.arch != 'any' {
 | 
				
			||||||
		repo_dir := os.join_path_single(r.data_dir, repo)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		mut arch_repos := []string{}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if os.exists(repo_dir) {
 | 
					 | 
				
			||||||
			// We get a listing of all currently present arch-repos in the given repo
 | 
					 | 
				
			||||||
			arch_repos = os.ls(repo_dir) ?.filter(os.is_dir(os.join_path_single(repo_dir, it)))
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if arch_repos.len == 0 {
 | 
					 | 
				
			||||||
			arch_repos << r.default_arch
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		mut added := false
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		for arch in arch_repos {
 | 
					 | 
				
			||||||
			added = added || r.add_pkg_in_arch_repo(repo, arch, pkg) ?
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		return added
 | 
					 | 
				
			||||||
	}else{
 | 
					 | 
				
			||||||
		return r.add_pkg_in_arch_repo(repo, pkg.info.arch, pkg)
 | 
							return r.add_pkg_in_arch_repo(repo, pkg.info.arch, pkg)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						repo_dir := os.join_path_single(r.data_dir, repo)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						mut arch_repos := []string{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// If this is the first package to be added to the repository, it won't
 | 
				
			||||||
 | 
						// contain any arch-repos yet.
 | 
				
			||||||
 | 
						if os.exists(repo_dir) {
 | 
				
			||||||
 | 
							// We get a listing of all currently present arch-repos in the given repo
 | 
				
			||||||
 | 
							arch_repos = os.ls(repo_dir) ?.filter(os.is_dir(os.join_path_single(repo_dir,
 | 
				
			||||||
 | 
								it)))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if arch_repos.len == 0 {
 | 
				
			||||||
 | 
							arch_repos << r.default_arch
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						mut added := false
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// We add the package to each repository. If any of the repositories
 | 
				
			||||||
 | 
						// return true, the result of the function is also true.
 | 
				
			||||||
 | 
						for arch in arch_repos {
 | 
				
			||||||
 | 
							added = added || r.add_pkg_in_arch_repo(repo, arch, pkg) ?
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return added
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// add_pkg_in_repo adds the given package to the specified repo. A repo is an
 | 
					// add_pkg_in_arch_repo is the function that actually adds a package to a given
 | 
				
			||||||
// arbitrary subdirectory of r.repo_dir, but in practice, it will always be an
 | 
					// arch-repo. It records the package's data in the arch-repo's desc & files
 | 
				
			||||||
// architecture-specific version of some sub-repository.
 | 
					// files, and afterwards updates the db & files archives to reflect these
 | 
				
			||||||
 | 
					// changes. The function returns false if the package was already present in
 | 
				
			||||||
 | 
					// the repo, and true otherwise.
 | 
				
			||||||
fn (r &RepoGroupManager) add_pkg_in_arch_repo(repo string, arch string, pkg &package.Pkg) ?bool {
 | 
					fn (r &RepoGroupManager) add_pkg_in_arch_repo(repo string, arch string, pkg &package.Pkg) ?bool {
 | 
				
			||||||
	pkg_dir := os.join_path(r.data_dir, repo, arch, '$pkg.info.name-$pkg.info.version')
 | 
						pkg_dir := os.join_path(r.data_dir, repo, arch, '$pkg.info.name-$pkg.info.version')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
| 
						 | 
					@ -125,8 +141,9 @@ fn (r &RepoGroupManager) add_pkg_in_arch_repo(repo string, arch string, pkg &pac
 | 
				
			||||||
	return true
 | 
						return true
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// remove removes a package from the database. It returns false if the package
 | 
					// remove_pkg_from_arch_repo removes a package from an arch-repo's database. It
 | 
				
			||||||
// wasn't present in the database.
 | 
					// returns false if the package wasn't present in the database. It also
 | 
				
			||||||
 | 
					// optionally re-syncs the repo archives.
 | 
				
			||||||
fn (r &RepoGroupManager) remove_pkg_from_arch_repo(repo string, arch string, pkg &package.Pkg, sync bool) ?bool {
 | 
					fn (r &RepoGroupManager) remove_pkg_from_arch_repo(repo string, arch string, pkg &package.Pkg, sync bool) ?bool {
 | 
				
			||||||
	repo_dir := os.join_path(r.data_dir, repo, arch)
 | 
						repo_dir := os.join_path(r.data_dir, repo, arch)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
| 
						 | 
					@ -138,11 +155,14 @@ fn (r &RepoGroupManager) remove_pkg_from_arch_repo(repo string, arch string, pkg
 | 
				
			||||||
	// We iterate over every directory in the repo dir
 | 
						// We iterate over every directory in the repo dir
 | 
				
			||||||
	// TODO filter so we only check directories
 | 
						// TODO filter so we only check directories
 | 
				
			||||||
	for d in os.ls(repo_dir) ? {
 | 
						for d in os.ls(repo_dir) ? {
 | 
				
			||||||
 | 
							// Because a repository only allows a single version of each package,
 | 
				
			||||||
 | 
							// we need only compare whether the name of the package is the same,
 | 
				
			||||||
 | 
							// not the version.
 | 
				
			||||||
		name := d.split('-')#[..-2].join('-')
 | 
							name := d.split('-')#[..-2].join('-')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if name == pkg.info.name {
 | 
							if name == pkg.info.name {
 | 
				
			||||||
			// We lock the mutex here to prevent other routines from creating a
 | 
								// We lock the mutex here to prevent other routines from creating a
 | 
				
			||||||
			// new archive while we removed an entry
 | 
								// new archive while we remove an entry
 | 
				
			||||||
			lock r.mutex {
 | 
								lock r.mutex {
 | 
				
			||||||
				os.rmdir_all(os.join_path_single(repo_dir, d)) ?
 | 
									os.rmdir_all(os.join_path_single(repo_dir, d)) ?
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -42,7 +42,7 @@ pub fn server() ? {
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// This also creates the directories if needed
 | 
						// This also creates the directories if needed
 | 
				
			||||||
	repo := repo.new(conf.data_dir, conf.pkg_dir, "x86_64") or {
 | 
						repo := repo.new(conf.data_dir, conf.pkg_dir, 'x86_64') or {
 | 
				
			||||||
		logger.error(err.msg)
 | 
							logger.error(err.msg)
 | 
				
			||||||
		exit(1)
 | 
							exit(1)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
		Reference in New Issue