summaryrefslogtreecommitdiff
path: root/paper/diffinator.py
blob: 5c26d351540fe51a381e50394ae3b8c99b058c12 (plain)
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
#!/usr/bin/env python3

import re
import subprocess

import click


@click.command()
@click.argument('texfile')
@click.argument('bibliography')
@click.argument('revision')
def generate_git_tex_diff(texfile, bibliography, revision):
    with open(texfile) as f:
        tex_lines = len(list(f))
    with open(bibliography) as f:
        bib_lines = len(list(f))

    tex_proc = subprocess.run(['git', 'diff', f'-U{tex_lines+1}', '--word-diff', '--color=always', revision, texfile],
            check=True, capture_output=True)

    bib_proc = subprocess.run(['git', 'diff', f'-U{bib_lines+1}', '--word-diff', '--color=always', revision, bibliography],
            check=True, capture_output=True)

    addition_re = re.compile('\033\\[32m\\{\\+(.*?)\\+\\}\033\\[m')
    deletion_re = re.compile('\033\\[31m\\[-(.*?)\\-]\033\\[m')
    csi_re = re.compile('\033\\[.*?m')
    bibtex_entry_def_re = re.compile('@.*?{(.*?),')

    bibliography_categories  = '\\DeclareBibliographyCategory{diff_new_entry}\n'
    bibliography_categories += '\\DeclareBibliographyCategory{diff_deleted_entry}\n'
    bibliography_categories += '\\AtEveryBibitem{\\ifcategory{diff_new_entry}{\\color{diffgreen}}{\\ifcategory{diff_deleted_entry}{\\color{diffred}}{\\color{black}}}}\n'

    added_entries, removed_entries = [], []
    for line in bib_proc.stdout.decode().splitlines():
        if (match := addition_re.fullmatch(line.strip())):
            if (entry_def := bibtex_entry_def_re.fullmatch(match.group(1))):
                added_entries.append(entry_def.group(1))

        if (match := deletion_re.fullmatch(line.strip())):
            if (entry_def := bibtex_entry_def_re.fullmatch(match.group(1))):
                removed_entries.append(entry_def.group(1))

    if added_entries:
        bibliography_categories += '\\addtocategory{diff_new_entry}{' + ','.join(added_entries) + '}\n'
    if removed_entries:
        bibliography_categories += '\\addtocategory{diff_deleted_entry}{' + ','.join(removed_entries) + '}\n'

    content_started = False
    document_started = False
    for line in tex_proc.stdout.decode().splitlines():

        if not content_started:
            if '@@' in line:
                content_started = True
            continue

        line = line.rstrip()
        if document_started: # diff results in preamble
            line = addition_re.sub(r' \\color{diffgreen}\1 \\color{black}', line)
            line = deletion_re.sub(r' \\color{diffred}\1 \\color{black}', line)

        else:
            if '\\begin{document}' in line:
                document_started = True
                print(bibliography_categories)
                print('\\definecolor{diffgreen}{HTML}{1e8449}')
                print('\\definecolor{diffred}{HTML}{cb4335}')


            line = addition_re.sub(r'\1', line)
            line = deletion_re.sub(r'\1', line)

        line = csi_re.sub('', line)
        print(line)

if __name__ == '__main__':
    generate_git_tex_diff()