feat: define core forge interface to implement starchart
This commit is contained in:
parent
ec9bb69f98
commit
72a6ce9564
2 changed files with 157 additions and 0 deletions
18
forge/forge-core/Cargo.toml
Normal file
18
forge/forge-core/Cargo.toml
Normal file
|
@ -0,0 +1,18 @@
|
|||
[package]
|
||||
name = "forge-core"
|
||||
repository = "https://github.com/forgeflux-org/starchart"
|
||||
version = "0.1.0"
|
||||
authors = ["realaravinth <realaravinth@batsense.net>"]
|
||||
description = "ForgeFlux StarChart - Federated forge spider"
|
||||
documentation = "https://forgeflux.org/"
|
||||
edition = "2021"
|
||||
license = "AGPLv3 or later version"
|
||||
|
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
||||
async-trait = "0.1.51"
|
||||
|
||||
[dependencies.db-core]
|
||||
path = "../../db/db-core"
|
139
forge/forge-core/src/lib.rs
Normal file
139
forge/forge-core/src/lib.rs
Normal file
|
@ -0,0 +1,139 @@
|
|||
/*
|
||||
* ForgeFlux StarChart - A federated software forge spider
|
||||
* Copyright © 2022 Aravinth Manivannan <realaravinth@batsense.net>
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero 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 Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
use std::collections::{HashMap, HashSet};
|
||||
use std::sync::Arc;
|
||||
|
||||
use async_trait::async_trait;
|
||||
use db_core::prelude::*;
|
||||
|
||||
pub mod prelude {
|
||||
pub use super::*;
|
||||
pub use async_trait::async_trait;
|
||||
}
|
||||
|
||||
pub mod dev {
|
||||
pub use super::*;
|
||||
pub use async_trait::async_trait;
|
||||
pub use db_core;
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct User<'a> {
|
||||
/// hostname of the forge instance: with scheme but remove trailing slash
|
||||
/// hostname can be derived from html_link also, but used to link to user's forge instance
|
||||
pub hostname: &'a str,
|
||||
/// username of the user
|
||||
pub username: Arc<String>,
|
||||
/// html link to the user profile
|
||||
pub html_link: String,
|
||||
/// OPTIONAL: html link to the user's profile photo
|
||||
pub profile_photo: Option<String>,
|
||||
}
|
||||
|
||||
impl<'a> From<&'a User<'a>> for AddUser<'a> {
|
||||
fn from(u: &'a User) -> Self {
|
||||
Self {
|
||||
hostname: u.hostname,
|
||||
username: u.username.as_str(),
|
||||
html_link: &u.html_link,
|
||||
profile_photo: u.profile_photo.as_deref(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
/// add new repository to database
|
||||
pub struct Repository<'a> {
|
||||
/// html link to the repository
|
||||
pub html_link: String,
|
||||
/// repository topic tags
|
||||
pub tags: Option<Vec<Arc<String>>>,
|
||||
/// hostname of the forge instance: with scheme but remove trailing slash
|
||||
/// hostname can be deras_ref().map(|p| p.as_str()),ived from html_link also, but used to link to user's forge instance
|
||||
pub hostname: &'a str,
|
||||
/// repository name
|
||||
pub name: String,
|
||||
/// repository owner
|
||||
pub owner: Arc<User<'a>>,
|
||||
/// repository description, if any
|
||||
pub description: Option<String>,
|
||||
/// repository website, if any
|
||||
pub website: Option<String>,
|
||||
}
|
||||
|
||||
impl<'a> From<&'a Repository<'a>> for AddRepository<'a> {
|
||||
fn from(r: &'a Repository) -> Self {
|
||||
let tags = if let Some(rtags) = &r.tags {
|
||||
let mut tags = Vec::with_capacity(rtags.len());
|
||||
for t in rtags.iter() {
|
||||
tags.push(t.as_str());
|
||||
}
|
||||
Some(tags)
|
||||
} else {
|
||||
None
|
||||
};
|
||||
Self {
|
||||
hostname: r.hostname,
|
||||
name: &r.name,
|
||||
description: r.description.as_deref(),
|
||||
owner: r.owner.username.as_str(),
|
||||
tags,
|
||||
html_link: &r.html_link,
|
||||
website: r.website.as_deref(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub type UserMap<'a> = HashMap<Arc<String>, Arc<User<'a>>>;
|
||||
pub type Tags = HashSet<Arc<String>>;
|
||||
pub type Repositories<'a> = Vec<Repository<'a>>;
|
||||
|
||||
pub struct CrawlResp<'a> {
|
||||
pub repos: Repositories<'a>,
|
||||
pub tags: Tags,
|
||||
pub users: UserMap<'a>,
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
pub trait SCForge: std::marker::Send + std::marker::Sync + CloneSPForge {
|
||||
async fn is_forge(&self) -> bool;
|
||||
async fn crawl(&self, limit: u64, page: u64) -> CrawlResp;
|
||||
fn get_hostname(&self) -> &str;
|
||||
fn forge_type(&self) -> ForgeImplementation;
|
||||
}
|
||||
|
||||
/// Trait to clone SCForge
|
||||
pub trait CloneSPForge {
|
||||
/// clone Forge
|
||||
fn clone_forge(&self) -> Box<dyn SCForge>;
|
||||
}
|
||||
|
||||
impl<T> CloneSPForge for T
|
||||
where
|
||||
T: SCForge + Clone + 'static,
|
||||
{
|
||||
fn clone_forge(&self) -> Box<dyn SCForge> {
|
||||
Box::new(self.clone())
|
||||
}
|
||||
}
|
||||
|
||||
impl Clone for Box<dyn SCForge> {
|
||||
fn clone(&self) -> Self {
|
||||
(**self).clone_forge()
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue