skyhopper/deploy.py at main · artyom-beilis/skyhopper · 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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
import glob
import re
import os
import sys
import markdown
import shutil
import base64

from create_data import create_db
from po import tojson

def copyf(src,tgt):
    print("Copying %s -> %s" % (src,tgt))
    shutil.copyfile(src,tgt)

def get_ver():
    with open('Changelog.md') as f:
        first = list(f.readlines())[0]
    ver = first[first.find(': v')+3:].strip()
    return ver


def make_manual(lang = None):
    path = 'README.md' if lang is None else 'po/README_%s.md' % lang
    with open(path, "r",encoding="utf-8") as input_file:
        text = input_file.read()
        md  = markdown.Markdown(extensions=['toc'])
        html = md.convert(text)

    if lang is None:
        with open("manual.html", "w",encoding="utf-8") as output_file:
            with open('header.html','r') as f:
                output_file.write(f.read())
            output_file.write(html)
            with open('footer.html','r') as f:
                output_file.write(f.read())
    return html


def embed_service_worker(version):
    with open("sw.js","r") as f, open("sw_deploy.js","w") as out:
        content=f.read()
        content = content.replace('VERSION',version)
        out.write(content)

def png_encode(path):
    with open(path,'rb') as f:
        png = f.read()
        b64png = base64.b64encode(png).decode()
    return 'data:image/png;base64,%s' % b64png

def combine_manuals(mans):
    res = []
    for lang in mans:
        cls = 'l2r' if lang == 'en' else 'r2l'
        res.append("<div id='manual_tr_%s' class='%s manual_section'>\n%s\n</div>\n" % (lang,cls,mans[lang]))
    return '\n'.join(res)

def embed(manual,version):
    script=re.compile(r'^<script src="(.*)"></script>')
    ver=re.compile(r'.* \((version)\).*')
    urlpng=re.compile(r'^(.*)url\(([a-z0-9_\-]*\.png)\)(.*)$')
    urlpng2=re.compile(r'^(.*<img.*)src="([a-z0-9_\-\/]*\.png)"(.*)$')

    with open("astrohopper.html","r", encoding="utf-8") as f, open("astrohopper_deploy.html","w", encoding="utf-8") as out:
        for line in f.readlines():
            m = script.match(line)
            v = ver.match(line)
            u = urlpng.match(line)
            u2 = urlpng2.match(line)
            if m:
                with open(m.group(1),"r", encoding="utf-8") as inline:
                    line = inline.read()
                    out.write('<script>\n')
                    out.write(line)
                    out.write('</script>\n')
            elif v:
                out.write(line.replace('version',version))
            elif u or u2:
                if u:
                    pat = '%surl(%s)%s\n'
                else:
                    pat = '%ssrc="%s"%s\n'
                    u=u2
                b64png = png_encode(u.group(2))
                new_line = pat % (u.group(1),b64png,u.group(3))
                out.write(new_line)
            elif line.find('MANUAL') == 0:
                out.write(manual)
            else:
                out.write(line)

def deploy_files(target):
    copyf('astrohopper_deploy.html', os.path.join(target, "astrohopper.html"))
    copyf('sw_deploy.js', os.path.join(target, "sw.js"))

    for f in ['LICENSE', 'COPYING.md', 'manual.html', 'manifest.json']:
        copyf(f, os.path.join(target, f))


def add_ga():
    print("Adding Google Analytics to the file")
    with open('ga.html') as f:
        cnt=f.read()
    with open('astrohopper_deploy.html') as f:
        page = f.read()
    page = page.replace('</head>',cnt + '</head>')
    with open('astrohopper_deploy.html','w')  as f:
        f.write(page)

def main():
    create_db()
    ver = get_ver()
    manuals = dict()
    manuals["en"] = make_manual()

    for man in glob.glob(os.path.join("po", "README_*.md")):
        # Extract just the filename
        filename = os.path.basename(man)
        # Remove the prefix and extension
        lang = filename.replace('README_', '').replace('.md', '')
        if len(lang) < 2 or len(lang) > 3:
            continue
        print("  Manual for ",lang)
        manuals[lang] = make_manual(lang)

    embed(combine_manuals(manuals),ver)
    embed_service_worker(ver)
    if len(sys.argv) >= 2:
        if sys.argv[1] == '-g':
            add_ga()
            del sys.argv[1]
    if len(sys.argv) == 2:
        deploy_files(sys.argv[1])

if __name__ == "__main__":
    main()