use crate::models::common::DEFAULT_REVISION;
use crate::error::AppError;
use crate::service::RepoService;
use crate::session::Session;

impl RepoService {
    pub async fn git_commit_stats(
        &self,
        ctx: &Session,
        wk_name: &str,
        repo_name: &str,
        revision: &str,
    ) -> Result<crate::pb::repo::CommitStats, AppError> {
        let user_uid = ctx.user().ok_or(AppError::Unauthorized)?;
        let repo = self.resolve_repo(wk_name, repo_name).await?;
        self.ensure_repo_readable(user_uid, &repo).await?;
        let ws = self.resolve_workspace(wk_name).await?;
        let header = self.repo_header(&repo, &ws);
        let mut svc = self.git_client(&repo)?.commit;
        let resp = svc
            .get_commit_stats(tonic::Request::new(
                crate::pb::repo::GetCommitStatsRequest {
                    repository: Some(header),
                    revision: Some(crate::pb::repo::ObjectSelector {
                        selector: Some(crate::pb::repo::object_selector::Selector::Revision(
                            crate::pb::repo::ObjectName {
                                revision: revision.to_string(),
                            },
                        )),
                    }),
                },
            ))
            .await
            .map_err(|e| AppError::InternalServerError(e.to_string()))?;
        Ok(resp.into_inner())
    }

    pub async fn git_count_commits(
        &self,
        ctx: &Session,
        wk_name: &str,
        repo_name: &str,
        revision: Option<&str>,
        path: Option<&str>,
        since: Option<&str>,
        until: Option<&str>,
    ) -> Result<crate::pb::repo::CountCommitsResponse, AppError> {
        let user_uid = ctx.user().ok_or(AppError::Unauthorized)?;
        let repo = self.resolve_repo(wk_name, repo_name).await?;
        self.ensure_repo_readable(user_uid, &repo).await?;
        let ws = self.resolve_workspace(wk_name).await?;
        let header = self.repo_header(&repo, &ws);
        let mut svc = self.git_client(&repo)?.commit;
        let resp = svc
            .count_commits(tonic::Request::new(crate::pb::repo::CountCommitsRequest {
                repository: Some(header),
                revision: revision.unwrap_or(DEFAULT_REVISION).to_string(),
                path: path.unwrap_or_default().to_string(),
                since: since.unwrap_or_default().to_string(),
                until: until.unwrap_or_default().to_string(),
            }))
            .await
            .map_err(|e| AppError::InternalServerError(e.to_string()))?;
        Ok(resp.into_inner())
    }

    pub async fn git_count_diverging_commits(
        &self,
        ctx: &Session,
        wk_name: &str,
        repo_name: &str,
        left: &str,
        right: &str,
    ) -> Result<crate::pb::repo::CountDivergingCommitsResponse, AppError> {
        let user_uid = ctx.user().ok_or(AppError::Unauthorized)?;
        let repo = self.resolve_repo(wk_name, repo_name).await?;
        self.ensure_repo_readable(user_uid, &repo).await?;
        let ws = self.resolve_workspace(wk_name).await?;
        let header = self.repo_header(&repo, &ws);
        let mut svc = self.git_client(&repo)?.commit;
        let resp = svc
            .count_diverging_commits(tonic::Request::new(
                crate::pb::repo::CountDivergingCommitsRequest {
                    repository: Some(header),
                    left: left.to_string(),
                    right: right.to_string(),
                },
            ))
            .await
            .map_err(|e| AppError::InternalServerError(e.to_string()))?;
        Ok(resp.into_inner())
    }
}