feat: new stuff

pkgr/src/api/mod.rs

@@ -1,42 +1,60 @@
 use std::collections::HashMap;
+use serde;
+use serde::{Deserialize, Serialize};
 
-enum Query {
+pub mod client;
+
+#[derive(Serialize, Deserialize)]
+pub enum Query {
+    #[serde(rename = "pull")]
     Pull {
         name: String,
         version: Option<u128>,
         tags: Vec<String>
     },
+    #[serde(rename = "push")]
     Push {
+        // todo: review me pls
         _data: Vec<u8>
+    },
+    #[serde(rename = "index")]
+    Index {
+        request_update: bool,
+        fetch: bool
     }
 }
 
-struct Request {
+#[derive(Serialize, Deserialize)]
+pub struct Request {
     version: u32,
     id: String,
     token: Option<String>,
     query: HashMap<String, Query>
 }
 
-struct DataErrorDetails {
+#[derive(Serialize, Deserialize)]
+pub struct DataErrorDetails {
     actor: String,
     detailed_cause: String,
     recovery_options: Vec<String>,
 }
 
-struct DataError {
+#[derive(Serialize, Deserialize)]
+pub struct DataError {
     name: String,
     cause: Option<String>,
     details: Option<DataErrorDetails>
 }
 
-enum Data {
+#[derive(Serialize, Deserialize)]
+pub enum Data {
     Pull {
         _data: Option<Vec<u8>>,
     }
 }
 
-struct Response {
+#[derive(Serialize, Deserialize)]
+pub struct Response {
     version: u32,
     id: String,
     reply_to: String,

pkgr/src/config/repos.rs

@@ -22,4 +22,16 @@ impl Default for Repo {
 #[derive(Clone, Serialize, Deserialize)]
 pub struct RepoFile {
     pub repos: HashMap<String, Repo>
+}
+
+impl RepoFile {
+    pub fn from_path(path: &str) -> Result<RepoFile, String> {
+        match std::fs::read_to_string(path) {
+            Ok(s) => match toml::from_str(&s) {
+                Ok(c) => Ok(c),
+                Err(e) => Err(format!("failed to parse config: {}", e)),
+            },
+            Err(e) => Err(format!("failed to read config: {}", e)),
+        }
+    }
 }

pkgr/src/config/storage.rs

@@ -7,7 +7,10 @@ pub struct Storage {
     pub repo_file: String,
     /// Where to store pkgs data
     #[serde(default)]
-    pub data_dir: String
+    pub data_dir: String,
+    /// Where to store repo indexes
+    #[serde(default)]
+    pub index_dir: String,
 }
 
 impl Default for Storage {
@@ -15,6 +18,7 @@ impl Default for Storage {
         Storage {
             repo_file: String::from("/etc/pkgr.d/repos.toml"),
             data_dir: String::from("/var/lib/pkgr/packages"),
+            index_dir: String::from("/var/lib/pkgr/indexes"),
         }
     }
 }

pkgr/src/main.rs

@@ -9,6 +9,8 @@ mod commands;
 mod logging;
 /// Package and helpers.
 mod package;
+/// Repo and helpers
+mod repo;
 /// Process wrapper with logging wrapper.
 mod process;
 /// tmpfs wrapper.

pkgr/src/repo/index/index_package.rs

@@ -0,0 +1,18 @@
+use std::io;
+use std::path::Path;
+use serde::{Deserialize, Serialize};
+use url::Url;
+use pkgfile::PKGFile;
+use crate::api::Query;
+use crate::CONFIG;
+use crate::package::Package;
+use crate::repo::index::RepoIndex;
+
+/// This struct solely exists for indexing and has no real functionality.
+#[derive(Serialize, Deserialize)]
+pub struct IndexPackage {
+    name: String,
+    versions: Vec<u32>,
+    tags: Vec<String>,
+    uri: Url
+}

pkgr/src/repo/index/mod.rs

@@ -0,0 +1,58 @@
+use std::collections::HashMap;
+use std::io;
+use std::path::Path;
+use serde::{Deserialize, Serialize};
+use crate::api::Request;
+use crate::CONFIG;
+use crate::repo::index::index_package::IndexPackage;
+use crate::repo::Repo;
+use crate::types::fetch::{Fetch, TryFetch};
+use crate::util::create_uuid;
+
+pub mod index_package;
+
+#[derive(Serialize, Deserialize)]
+pub struct RepoIndex {
+    origin_repo: String,
+    packages: HashMap<String, IndexPackage>
+}
+
+impl RepoIndex {
+    // /// Fetch existing index or create a new one.
+    // pub fn from_repo(repo: Repo) -> Self {
+    //
+    // }
+    //
+    // /// Create new index.
+    // pub fn create_with_repo(repo: Repo) -> Self {
+    // }
+
+    /// Get repo
+    pub fn get_repo(&self) -> io::Result<Repo> {
+        Ok(Repo::from_name(&self.origin_repo)?)
+    }
+
+    pub fn from_path(path: &str) -> Result<RepoIndex, String> {
+        match std::fs::read_to_string(path) {
+            Ok(s) => match toml::from_str(&s) {
+                Ok(c) => Ok(c),
+                Err(e) => Err(format!("failed to parse config: {}", e)),
+            },
+            Err(e) => Err(format!("failed to read config: {}", e)),
+        }
+    }
+}
+
+impl TryFetch<Repo> for RepoIndex {
+    type Error = ();
+    /// Fetch
+    fn try_fetch(query: Repo) -> Result<Self, Self::Error> {
+        let path = CONFIG.with(|c| c.storage.index_dir.clone()) + query.uri.as_str() + ".toml";
+        if Path::new(path.as_str()).exists() {
+            RepoIndex::from_path(path.as_str())
+                .map_err(|_| ())
+        } else {
+            Err(())
+        }
+    }
+}

pkgr/src/repo/mod.rs

@@ -0,0 +1,42 @@
+use std::io;
+use std::io::{Error, ErrorKind};
+use std::path::Path;
+use url::Url;
+use crate::CONFIG;
+use crate::config::repos::RepoFile;
+
+/// Indexed repos
+pub mod index;
+
+pub struct Repo {
+    name: String,
+    uri: Url,
+}
+
+impl Repo {
+    pub fn from_name(name: &String) -> io::Result<Repo> {
+        let r = RepoFile::from_path(CONFIG.with(|c| c.storage.repo_file.clone()).as_str())
+            .map_err(|e| Error::new(ErrorKind::Other, e))?;
+        let r = r
+            .repos
+            .get(name)
+            .ok_or(Error::new(ErrorKind::InvalidData, "Could not get repo"))?;
+        Ok(Repo {
+            name: r.name.clone(),
+            uri: r.uri.clone()
+        })
+    }
+
+    pub fn get_name(&self) -> String {
+        self.name.clone()
+    }
+
+    pub fn get_uri(&self) -> Url {
+        self.uri.clone()
+    }
+
+    /// Fetch indexed repo
+    pub fn get_index(&self) {
+
+    }
+}

pkgr/src/util/mod.rs

@@ -1,2 +1,9 @@
 pub mod prompts;
-pub mod fs;
+/// Helpers for fs
+pub mod fs;
+
+/// Create a UUID
+pub fn create_uuid() -> String {
+    // TODO
+    String::from("rand")
+}