chlog/main.py at main · nathanielheitsch/chlog · GitHub

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
import asyncio
import logging
import sys
from typing import Optional
import inquirer
import typer
from typing_extensions import Annotated
from rich.progress import Progress
from common.ai_service import AIService
from common.git import getBranches, getCommitMessages, getDiff
from lib.writer import Writer

logging.basicConfig(stream=sys.stdout, level=logging.INFO,
                    format='[%(asctime)s] %(levelname)s - %(message)s')

loader_total = 0

def selectBranch(base=True) -> str:
    branches = getBranches()
    questions = [
        inquirer.List('branch',
                      message="Which branch/commit would you like to compare {} ({})".format(
                          "with" if not base else "against", "newer" if not base else "older"),
                      choices=branches,
                      ),
    ]
    answers = inquirer.prompt(questions)
    return answers["branch"]


def gen(new_commit: Annotated[Optional[str], typer.Argument()] = None, old_commit: Annotated[Optional[str], typer.Argument()] = None, token: Annotated[str, typer.Option(help="API Token for AI Services")] = None, title: Annotated[str, typer.Option(help="Title of the changelog")] = None):
    if new_commit == None:
        new_commit = selectBranch(False)
    if old_commit == None:
        old_commit = selectBranch()
    diffs = getDiff(new_commit, old_commit)
    if diffs == None:
        logging.info(
            f'There is no difference between {old_commit} and {new_commit}')
        return
    commit_messages = getCommitMessages(new_commit, old_commit)
    logging.info("Processing {} diffs over {} commits".format(len(diffs), len(commit_messages.splitlines())))
    asyncio.run(pursue(diffs, commit_messages, token, title if title else f"{old_commit} to {new_commit}"))


def writeDiff(result):
    logging.debug(result)
    return Writer.writeDiff(result)


async def pursue(diffs, commit_messages, token: str, title: str):
    ai_service = AIService(token)
    Writer.init()
    with Progress(
        transient=True,
    ) as progress:
        diff_len = len(diffs)
        task = progress.add_task(description="Processing...", total=diff_len)
        chlog = ""
        i=0
        def inc(s: str):
            nonlocal chlog
            nonlocal i
            i += 1
            progress.update(task, advance=1, description="Processing {}/{}".format(str(i).rjust(len(str(diff_len)) - len(str(i)), " "), diff_len))
            chlog += s
        await ai_service.ai.chlog(diffs, commit_messages, inc)
        progress.update(task, description="Finalizing Summary...")
        summary = await ai_service.ai.summarize(chlog, commit_messages)
        while summary.startswith("\n"):
            summary = summary[1:]
        while summary.endswith("\n"):
            summary = summary[:-1]
        progress.update(task, advance=1)
        writeDiff(f"{title}\n{summary}")
        logging.info("Success! See the CHANGELOG")


if __name__ == "__main__":
    typer.run(gen)