mockforge-core 0.3.115

Shared logic for MockForge - routing, validation, latency, proxy
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

//! PR generator for creating pull requests
//!
//! This module provides a unified interface for generating PRs across different providers.

use crate::pr_generation::templates::{PRTemplate, PRTemplateContext};
use crate::pr_generation::types::{PRProvider, PRRequest, PRResult};
use crate::pr_generation::{GitHubPRClient, GitLabPRClient};

/// PR generator that works with multiple providers
#[derive(Debug, Clone)]
pub struct PRGenerator {
    provider: PRProvider,
    github_client: Option<GitHubPRClient>,
    gitlab_client: Option<GitLabPRClient>,
}

impl PRGenerator {
    /// Create a new PR generator for GitHub
    pub fn new_github(owner: String, repo: String, token: String, base_branch: String) -> Self {
        Self {
            provider: PRProvider::GitHub,
            github_client: Some(GitHubPRClient::new(owner, repo, token, base_branch)),
            gitlab_client: None,
        }
    }

    /// Create a new PR generator for GitLab
    pub fn new_gitlab(owner: String, repo: String, token: String, base_branch: String) -> Self {
        Self {
            provider: PRProvider::GitLab,
            github_client: None,
            gitlab_client: Some(GitLabPRClient::new(owner, repo, token, base_branch)),
        }
    }

    /// Generate and create a PR from template context
    pub async fn create_pr_from_context(
        &self,
        context: PRTemplateContext,
        files: Vec<crate::pr_generation::types::PRFileChange>,
        labels: Vec<String>,
        reviewers: Vec<String>,
    ) -> crate::Result<PRResult> {
        let title = PRTemplate::generate_title(&context);
        let body = PRTemplate::generate_body(&context);

        let branch = format!(
            "{}-{}-{}",
            "mockforge/contract-update",
            context.method.to_lowercase(),
            &uuid::Uuid::new_v4().to_string()[..8]
        );

        let request = PRRequest {
            title,
            body,
            branch,
            files,
            labels,
            reviewers,
        };

        self.create_pr(request).await
    }

    /// Create a PR
    pub async fn create_pr(&self, request: PRRequest) -> crate::Result<PRResult> {
        match self.provider {
            PRProvider::GitHub => {
                let client = self
                    .github_client
                    .as_ref()
                    .ok_or_else(|| crate::Error::internal("GitHub client not configured"))?;
                client.create_pr(request).await
            }
            PRProvider::GitLab => {
                let client = self
                    .gitlab_client
                    .as_ref()
                    .ok_or_else(|| crate::Error::internal("GitLab client not configured"))?;
                client.create_pr(request).await
            }
        }
    }
}