text/x-rust
•
7.98 KB
•
264 lines
use std::sync::Arc;
use askama::Template;
use axum::{
body::Body,
extract::{Query, State},
http::{Response, StatusCode, header},
};
use serde::Deserialize;
use tower_cookies::Cookies;
use crate::{
AccessType, GlobalState, Project, User,
controllers::not_found,
extractors::PathUserProjectRefPath,
models::{ContentPage, user::UserModel},
services::{
markdown::{self, Frontmatter},
repository::{GitRefInfo, GitSummary, get_git_file, get_git_info, get_git_summary},
session,
},
views::ThemedRender,
};
#[derive(Template)]
#[template(path = "project/blob.html")]
struct ProjectBlobTemplate<'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>,
file_path: String,
path_parts: Vec<String>,
file_content: String,
is_binary: bool,
is_image: bool,
is_source: bool,
is_rendered_markdown: bool,
markdown_frontmatter: Frontmatter,
raw_url: String,
source_url: Option<String>,
logged_in_user: Option<Arc<User>>,
sidebar_projects: Vec<Project>,
content_pages: Vec<ContentPage>,
active_tab: &'static str,
mime_type: String,
line_count: Option<usize>,
human_readable_size: String,
}
#[derive(Deserialize)]
pub struct BlobParams {
/// Return raw file bytes with appropriate content-type
pub raw: Option<String>,
/// For markdown files, show raw markdown source in <pre><code>
pub source: Option<String>,
}
fn human_readable_size(bytes: usize) -> String {
const UNITS: &[&str] = &["B", "KB", "MB", "GB", "TB"];
let mut size = bytes as f64;
let mut unit_index = 0;
while size >= 1024.0 && unit_index < UNITS.len() - 1 {
size /= 1024.0;
unit_index += 1;
}
if unit_index == 0 {
format!("{} {}", bytes, UNITS[0])
} else {
format!("{:.2} {}", size, UNITS[unit_index])
}
}
pub async fn project_blob_get(
State(state): State<GlobalState>,
cookies: Cookies,
Query(params): Query<BlobParams>,
PathUserProjectRefPath(owner, project, git_ref, path): PathUserProjectRefPath,
) -> 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) = get_git_summary(&state, &owner.slug, &project.slug).await else {
return not_found(logged_in_user, sidebar_projects, content_pages);
};
// Get file content
let file_result = get_git_file(&state, &owner.slug, &project.slug, &git_ref, &path).await;
let file_obj = match file_result {
Ok(obj) => obj,
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 = get_git_info(&state, &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());
// Detect file type
let is_markdown = path.to_lowercase().ends_with(".md");
// Build URLs for action links
let raw_url = format!("{}?raw", project.uri_blob(&git_ref, &path));
let source_url = if is_markdown {
Some(format!("{}?source", project.uri_blob(&git_ref, &path)))
} else {
None
};
// Handle ?raw mode - return raw bytes with appropriate content-type
if params.raw.is_some() {
let mime_type = mime_guess::from_path(&path).first_or_octet_stream();
return Response::builder()
.status(StatusCode::OK)
.header(header::CONTENT_TYPE, mime_type.as_ref())
.body(Body::from(file_obj.clone()))
.unwrap();
}
// Check if it's an image file (based on mime type)
let mime_type = mime_guess::from_path(&path).first_or_octet_stream();
let is_image = mime_type.type_() == mime_guess::mime::IMAGE;
// Check if binary
let is_binary = file_obj.contains(&0u8);
// Handle image files (binary or text-based like SVG)
if is_image {
let path_parts: Vec<String> = path.split('/').map(|s| s.to_string()).collect();
let file_size = file_obj.len();
let template = ProjectBlobTemplate {
owner,
project: &project,
access_level,
ssh_clone_url,
http_clone_url,
summary,
info,
selected_branch,
file_path: path.clone(),
path_parts,
file_content: String::new(),
is_binary,
is_image: true,
is_rendered_markdown: false,
is_source: false,
markdown_frontmatter: vec![],
raw_url,
source_url: None,
logged_in_user,
sidebar_projects,
content_pages,
active_tab: "code",
mime_type: mime_type.to_string(),
line_count: None,
human_readable_size: human_readable_size(file_size),
};
return template.response();
}
// Handle other binary files
if is_binary {
let path_parts: Vec<String> = path.split('/').map(|s| s.to_string()).collect();
let file_size = file_obj.len();
let template = ProjectBlobTemplate {
owner,
project: &project,
access_level,
ssh_clone_url,
http_clone_url,
summary,
info,
selected_branch,
file_path: path.clone(),
path_parts,
file_content: String::from("Binary file"),
is_binary: true,
is_image: false,
is_source: false,
is_rendered_markdown: false,
markdown_frontmatter: vec![],
raw_url,
source_url: None,
logged_in_user,
sidebar_projects,
content_pages,
active_tab: "code",
mime_type: mime_type.to_string(),
line_count: None,
human_readable_size: human_readable_size(file_size),
};
return template.response();
}
// Handle text files
let text_content = String::from_utf8_lossy(&file_obj).to_string();
let file_size = file_obj.len();
let line_count = text_content.lines().count();
// Render markdown if it's a .md file and ?source is not specified
let (file_content, is_rendered_markdown, markdown_frontmatter) =
if is_markdown && params.source.is_none() {
let (frontmatter, html) = markdown::parse_and_render(&text_content);
(html, true, frontmatter)
} else {
(text_content, false, vec![])
};
let path_parts: Vec<String> = path.split('/').map(|s| s.to_string()).collect();
let template = ProjectBlobTemplate {
owner,
project: &project,
access_level,
ssh_clone_url,
http_clone_url,
summary,
info,
selected_branch,
file_path: path.clone(),
path_parts,
file_content,
is_binary: false,
is_image: false,
is_source: params.source.is_some(),
is_rendered_markdown,
markdown_frontmatter,
raw_url,
source_url,
logged_in_user,
sidebar_projects,
content_pages,
active_tab: "code",
mime_type: mime_type.to_string(),
line_count: Some(line_count),
human_readable_size: human_readable_size(file_size),
};
template.response()
}
