text/x-rust
•
5.03 KB
•
179 lines
use std::sync::Arc;
use askama::Template;
use axum::{body::Body, extract::State, http::Response};
use tower_cookies::Cookies;
use crate::{
AccessType, GlobalState, Project, User, UserModel,
controllers::not_found,
extractors::{PathUserProjectRef, PathUserProjectRefPath},
models::ContentPage,
services::{
markdown::{self, Frontmatter},
repository::{EntryKind, GitRefInfo, GitSummary, GitTreeEntry},
session,
},
views::ThemedRender,
};
#[derive(Template)]
#[template(path = "project/tree.html")]
struct ProjectTreeTemplate<'a> {
owner: Arc<User>,
project: &'a Project,
access_level: AccessType,
ssh_clone_url: String,
http_clone_url: String,
selected_branch: String,
summary: GitSummary,
info: Option<GitRefInfo>,
tree_entries: Vec<GitTreeEntry>,
current_path: String,
path_parts: Vec<String>,
readme_html: Option<String>,
readme_frontmatter: Frontmatter,
parent_path: Option<String>,
logged_in_user: Option<Arc<User>>,
sidebar_projects: Vec<Project>,
content_pages: Vec<ContentPage>,
active_tab: &'static str,
}
/// Helper to calculate parent path
fn get_parent_path(path: &str) -> Option<String> {
if path.is_empty() {
return None;
}
let parts: Vec<&str> = path.split('/').collect();
if parts.len() <= 1 {
return Some(String::new()); // Return to root
}
Some(parts[..parts.len() - 1].join("/"))
}
pub async fn project_tree_root_get(
State(state): State<GlobalState>,
cookies: Cookies,
PathUserProjectRef(owner, project, git_ref): PathUserProjectRef,
) -> Response<Body> {
render_tree_page(&state, cookies, owner, project, git_ref, String::new()).await
}
pub async fn project_tree_get(
State(state): State<GlobalState>,
cookies: Cookies,
PathUserProjectRefPath(owner, project, git_ref, path): PathUserProjectRefPath,
) -> Response<Body> {
render_tree_page(&state, cookies, owner, project, git_ref, path).await
}
async fn render_tree_page(
state: &GlobalState,
cookies: Cookies,
owner: Arc<User>,
project: Project,
git_ref: String,
path: String,
) -> Response<Body> {
let logged_in_user = session::current_user(state, &cookies).await.ok();
let content_pages = state.config.content_pages.clone();
let sidebar_projects = if let Some(ref user) = logged_in_user {
user.sidebar_projects(state).await
} else {
vec![]
};
let access_level = project
.access_level(logged_in_user.as_ref().map(|user| user.slug.clone()))
.await;
if access_level == AccessType::None {
return not_found(logged_in_user, sidebar_projects, content_pages);
}
let Some(summary) = state.repo.get_git_summary(&owner.slug, &project.slug).await else {
return not_found(logged_in_user, sidebar_projects, content_pages);
};
// Get tree entries
let tree_result = state
.repo
.get_git_tree(&owner.slug, &project.slug, &git_ref, &path)
.await;
let tree_entries = match tree_result {
Ok(entries) => entries,
Err(_) => return not_found(logged_in_user, sidebar_projects, content_pages),
};
let ssh_clone_url = project.ssh_clone_url(&state.config.ssh_public_host);
let http_clone_url = project.http_clone_url(&state.config.base_url);
let info = state
.repo
.get_git_info(&owner.slug, &project.slug, &git_ref, 1, 0)
.await;
let selected_branch = info
.as_ref()
.map(|i| i.branch_name.to_string())
.unwrap_or_else(|| git_ref.clone());
// Try to load README.md from current directory
let readme_path = if path.is_empty() {
"README.md".to_string()
} else {
format!("{}/README.md", path)
};
let readme_result = state
.repo
.get_git_file(&owner.slug, &project.slug, &git_ref, &readme_path)
.await;
let (readme_html, readme_frontmatter) = readme_result
.ok()
.filter(|_| {
// Check if README exists in current directory tree
tree_entries
.iter()
.any(|e| e.filename == "README.md" && e.kind.is_blob())
})
.map(|b| {
let content = String::from_utf8_lossy(&b);
let (frontmatter, html) = markdown::parse_and_render(&content);
(Some(html), frontmatter)
})
.unwrap_or((None, vec![]));
let parent_path = get_parent_path(&path);
let path_parts: Vec<String> = if path.is_empty() {
vec![]
} else {
path.split('/').map(|s| s.to_string()).collect()
};
let template = ProjectTreeTemplate {
owner,
project: &project,
access_level,
ssh_clone_url,
http_clone_url,
summary,
info,
selected_branch,
tree_entries,
current_path: path,
path_parts,
readme_html,
readme_frontmatter,
parent_path,
logged_in_user,
sidebar_projects,
content_pages,
active_tab: "code",
};
template.response()
}
