toni/shrap
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

track installed packages

Browse Source
This commit is contained in:
Antoni Piasecki
2026-02-21 10:30:44 +01:00
parent 9f9c5b7c9c
commit 6559569986
4 changed files with 76 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
src/main.cc
View File

@@ -1,3 +1,4 @@
#include "db.h"
#include "parser.h"
#include "util.h"
#include <algorithm>
@@ -7,9 +8,9 @@
#include <thread>
#include <unordered_map>
static const std::string BUILD_PATH = "/tmp/shrap";
std::unordered_map<std::string, Expr> packages;
bool flag_raw = false;
bool flag_graph = false;
void load_packages() {
for (const auto &e : std::filesystem::directory_iterator("./packages/")) {
@@ -25,11 +26,15 @@ void load_packages() {
}
std::vector<std::string>
resolve_dependencies(const std::vector<std::string> &roots) {
resolve_dependencies(const std::vector<std::string> &roots, DB &db) {
std::unordered_map<std::string, int> state;
std::vector<std::string> order;
std::function<void(const std::string &)> dfs = [&](const std::string &name) {
if (db.is_installed(name)) {
return;
}
int s = state[name];
if (s == 2) {
return;
@@ -63,7 +68,7 @@ resolve_dependencies(const std::vector<std::string> &roots) {
return order;
}
void install_package(const std::string &name) {
void install_package(const std::string &name, DB &db) {
// TODO: track installed packages
if (!packages.contains(name)) {
@@ -79,10 +84,10 @@ void install_package(const std::string &name) {
Expr src = pkg.get_one("src");
std::string src_type = src.children[0].value;
std::string src_url = src.get_one("url").value;
std::string src_path = "/tmp/shrap/" + src.get_one("dir").value;
std::string src_path = BUILD_PATH + "/" + src.get_one("dir").value;
if (src_type == "tar") {
std::string archive_path = "/tmp/shrap/" + Util::basename(src_url);
std::string archive_path = BUILD_PATH + "/" + Util::basename(src_url);
std::string expected_hash = src.get_one("blake3").value;
bool needs_download = true;
@@ -110,7 +115,7 @@ void install_package(const std::string &name) {
}
std::cout << "\n\tExtracting...\n";
Util::shell_command("tar xf " + archive_path + " -C /tmp/shrap/");
Util::shell_command("tar xf " + archive_path + " -C " + BUILD_PATH);
} else {
throw std::runtime_error("unrecognized src type");
}
@@ -176,10 +181,12 @@ void install_package(const std::string &name) {
throw std::runtime_error("unrecognized step type");
}
}
db.mark_installed(pkg.get_one("name").value, pkg.get_one("version").value);
}
void generate_graph(const std::vector<std::string> &to_highlight) {
std::ofstream out("/tmp/shrap/graph.dot");
std::ofstream out(BUILD_PATH + "/graph.dot");
out << "digraph dependencies {\n";
if (to_highlight.empty()) {
@@ -202,17 +209,25 @@ void generate_graph(const std::vector<std::string> &to_highlight) {
out << "}\n";
out.close();
Util::shell_command("dot -Tpng graph.dot > graph.png", "/tmp/shrap");
std::cout << "Graph saved to /tmp/shrap/graph.png\n";
Util::shell_command("dot -Tpng graph.dot > graph.png", BUILD_PATH);
std::cout << "Graph saved to " << BUILD_PATH << "/graph.png\n";
}
int main(int argc, char **argv) {
auto args = std::span(argv, static_cast<size_t>(argc));
Util::shell_command("mkdir -p /tmp/shrap");
std::filesystem::create_directories(BUILD_PATH);
std::filesystem::create_directories(DB::DB_PATH);
DB db;
db.init_db();
load_packages();
bool flag_raw = false;
bool flag_graph = false;
std::vector<std::string> to_install;
for (size_t i = 1; i < args.size(); ++i) {
std::string arg = args[i];
if (arg == "-r") {
@@ -225,7 +240,7 @@ int main(int argc, char **argv) {
}
if (!flag_raw) {
to_install = resolve_dependencies(to_install);
to_install = resolve_dependencies(to_install, db);
}
if (flag_graph) {
@@ -256,7 +271,7 @@ int main(int argc, char **argv) {
for (const std::string &pkg : to_install) {
try {
install_package(pkg);
install_package(pkg, db);
} catch (std::exception &e) {
std::cerr << "ERROR: " << e.what() << std::endl;
return 1;