content-generator.ts

import { format } from "date-fns";
import { GitHubTrackerSettings, RepositoryTracking, ProjectData } from "./types";
import { escapeBody, escapeYamlString } from "./util/escapeUtils";
import {
	createIssueTemplateData,
	createPullRequestTemplateData,
	processContentTemplate
} from "./util/templateUtils";
import { FileHelpers } from "./util/file-helpers";

export class ContentGenerator {
	constructor(
		private fileHelpers: FileHelpers,
	) {}

	/**
	 * Create issue content using template or default format
	 */
	public async createIssueContent(
		issue: any,
		repo: RepositoryTracking,
		comments: any[],
		settings: GitHubTrackerSettings,
		projectData?: ProjectData[],
		subIssues?: any[],
		parentIssue?: any,
	): Promise<string> {
		// Determine whether to escape hash tags (repo setting takes precedence if ignoreGlobalSettings is true)
		const shouldEscapeHashTags = repo.ignoreGlobalSettings ? repo.escapeHashTags : settings.escapeHashTags;

		// Check if custom template is enabled and load template content
		if (repo.useCustomIssueContentTemplate && repo.issueContentTemplate) {
			const templateContent = await this.fileHelpers.loadTemplateContent(repo.issueContentTemplate);
			if (templateContent) {
				const templateData = createIssueTemplateData(
					issue,
					repo.repository,
					comments,
					settings.dateFormat,
					settings.escapeMode,
					shouldEscapeHashTags,
					projectData,
					subIssues,
					parentIssue
				);
				return processContentTemplate(templateContent, templateData, settings.dateFormat);
			}
		}

		// Fallback to default template
		let frontmatter = `---
title: "${escapeYamlString(issue.title)}"
number: ${issue.number}
state: "${issue.state}"
type: "issue"
created: "${
			settings.dateFormat !== ""
				? format(new Date(issue.created_at), settings.dateFormat)
				: new Date(issue.created_at).toLocaleString()
		}"
updated: "${
			settings.dateFormat !== ""
				? format(new Date(issue.updated_at), settings.dateFormat)
				: new Date(issue.updated_at).toLocaleString()
		}"
url: "${issue.html_url}"
opened_by: "${issue.user?.login}"
assignees: [${(
			issue.assignees?.map(
				(assignee: { login: string }) => '"' + assignee.login + '"',
			) || []
		).join(", ")}]
labels: [${(
			issue.labels?.map(
				(label: { name: string }) => '"' + label.name + '"',
			) || []
		).join(", ")}]
updateMode: "${repo.issueUpdateMode}"
allowDelete: ${repo.allowDeleteIssue ? true : false}`;

		// Add projectData if available
		if (projectData && projectData.length > 0) {
			frontmatter += `
projectData:`;
			for (const project of projectData) {
				frontmatter += `
  - projectId: "${project.projectId}"`;
			}
		}

		frontmatter += `
---

# ${escapeBody(issue.title, settings.escapeMode, false)}
${
	issue.body
		? escapeBody(issue.body, settings.escapeMode, shouldEscapeHashTags)
		: "No description found"
}

${this.fileHelpers.formatComments(comments, settings.escapeMode, settings.dateFormat, shouldEscapeHashTags)}`;

		return frontmatter;
	}

	/**
	 * Create pull request content using template or default format
	 */
	public async createPullRequestContent(
		pr: any,
		repo: RepositoryTracking,
		comments: any[],
		settings: GitHubTrackerSettings,
		projectData?: ProjectData[],
	): Promise<string> {
		// Determine whether to escape hash tags (repo setting takes precedence if ignoreGlobalSettings is true)
		const shouldEscapeHashTags = repo.ignoreGlobalSettings ? repo.escapeHashTags : settings.escapeHashTags;

		// Check if custom template is enabled and load template content
		if (repo.useCustomPullRequestContentTemplate && repo.pullRequestContentTemplate) {
			const templateContent = await this.fileHelpers.loadTemplateContent(repo.pullRequestContentTemplate);
			if (templateContent) {
				const templateData = createPullRequestTemplateData(
					pr,
					repo.repository,
					comments,
					settings.dateFormat,
					settings.escapeMode,
					shouldEscapeHashTags,
					projectData
				);
				return processContentTemplate(templateContent, templateData, settings.dateFormat);
			}
		}

		// Fallback to default template
		let frontmatter = `---
title: "${escapeYamlString(pr.title)}"
number: ${pr.number}
state: "${pr.state}"
type: "pr"
created: "${
			settings.dateFormat !== ""
				? format(new Date(pr.created_at), settings.dateFormat)
				: new Date(pr.created_at).toLocaleString()
		}"
updated: "${
			settings.dateFormat !== ""
				? format(new Date(pr.updated_at), settings.dateFormat)
				: new Date(pr.updated_at).toLocaleString()
		}"
url: "${pr.html_url}"
opened_by: "${pr.user?.login}"
assignees: [${(
			pr.assignees?.map(
				(assignee: { login: string }) => '"' + assignee.login + '"',
			) || []
		).join(", ")}]
requested_reviewers: [${(
			pr.requested_reviewers?.map(
				(reviewer: { login: string }) => '"' + reviewer.login + '"',
			) || []
		).join(", ")}]
labels: [${(
			pr.labels?.map(
				(label: { name: string }) => '"' + label.name + '"',
			) || []
		).join(", ")}]
updateMode: "${repo.pullRequestUpdateMode}"
allowDelete: ${repo.allowDeletePullRequest ? true : false}`;

		// Add projectData if available
		if (projectData && projectData.length > 0) {
			frontmatter += `
projectData:`;
			for (const project of projectData) {
				frontmatter += `
  - projectId: "${project.projectId}"`;
			}
		}

		frontmatter += `
---

# ${escapeBody(pr.title, settings.escapeMode, false)}
${
	pr.body
		? escapeBody(pr.body, settings.escapeMode, shouldEscapeHashTags)
		: "No description found"
}

${this.fileHelpers.formatComments(comments, settings.escapeMode, settings.dateFormat, shouldEscapeHashTags)}`;

		return frontmatter;
	}
}