git » autoupdaters.git » main » tree

[main] / autoupdater_helpers / __init__.py 

  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
#!/usr/bin/env python3

def pkgbuild_ver():
    for line in open("PKGBUILD", "r"):
        if line.startswith("pkgver="):
            return line[7:].strip()

def pkgbuild_new_ver(ver, sha256sum=None, rel="1", sumtype="sha256", sum=None):
    import os
    sumpfx = sumtype + "sums=("
    if sha256sum:
        sum = sha256sum

    # Skip one line with a checksum
    skipsumln = False
    # Skip the whole rest of the checksum section
    skipsumlines = False
    with open("PKGBUILD.new", "w") as outf:
        for line in open("PKGBUILD"):
            if skipsumln:
                if ')' in line:
                    skipsumln = False
                elif "'" not in line:
                    continue
                else:
                    skipsumln = False
                    continue
            if skipsumlines:
                if ')' in line:
                    skipsumlines = False
                continue
            if line.startswith("pkgver="):
                outf.write("pkgver=" + ver + "\n")
            elif line.startswith("pkgrel="):
                outf.write(f"pkgrel={rel}\n")
            elif line.startswith(sumpfx):
                if sum:
                    outf.write(sumpfx + "'" + sum + "'\n")
                    if "'" not in line:
                        skipsumln = True
                else:
                    if ")" not in line:
                        skipsumlines = True
            else:
                outf.write(line)

    os.rename("PKGBUILD", "PKGBUILD.old")
    os.rename("PKGBUILD.new", "PKGBUILD")
    return

def ssl_ctx():
    import ssl
    cabundle = "/etc/pki/tls/certs/ca-bundle.crt"
    ctx = ssl.create_default_context(cafile=cabundle)
    return ctx

def _log(desc):
    with open("/sources/autoupdaters/results.log", "a") as f:
        f.seek(0,2)
        f.write(desc + "\n")


def mpkg(pkg, ver=None, carch=None):
    statepath = "/tmp/armlfs-evt"
    activepath = statepath + "/active"
    donepath = statepath + "/done"
    failpath = statepath + "/failed"
    from os import makedirs, chdir, getcwd, rename, environ
    from os.path import join
    from subprocess import run, CalledProcessError
    desc = f"{pkg} {ver}" if ver else pkg
    fn = desc.replace(" ","-")
    makedirs(activepath, exist_ok=True)
    fpath = join(activepath,fn)
    with open(fpath,"w") as f:
        f.write(desc + "\n")
    prev_path = getcwd()
    chdir("/sources/base-pkgbuilds")
    myenv = None
    if carch:
        myenv = dict(environ, CARCH=carch)
    try:
        run(["./mpkg.sh", pkg], env=myenv, check=True)
        chdir(prev_path)
        makedirs(donepath, exist_ok=True)
        rename(fpath, join(donepath,fn))
        _log(desc + " DONE")
    except CalledProcessError:
        chdir(prev_path)
        makedirs(failpath, exist_ok=True)
        rename(fpath, join(failpath,fn))
        _log(desc + " FAIL")
        raise

def parse_pkgbuild_ver_sum(fn):
    meta = {}
    sumnextln = False
    for line in open(fn):
        if sumnextln:
            if "'" in line:
                _,meta['sum'],_ = line.strip().split("'")
                sumnextln = False
            if ")" in line:
                sumnextln = False
            continue
        if line.startswith("pkgver="):
            meta['ver'] = line[7:].strip()
        if "sums=(" in line:
            type, sum = line.strip().split("sums=(",maxsplit=1)
            if type not in ("b2", "sha512", "sha384", "sha256", "sha224", "sha1", "md5", "ck"):
                continue
            if "'" not in sum:
                sumnextln = True
            else:
                _,meta['sum'],_ = sum.strip().split("'")
            meta['sumtype'] = type
        if 'ver' in meta and 'sum' in meta:
            break
    return meta

def get_arch_pkgbuild_ver_sum(pkg):
    from tempfile import TemporaryDirectory
    from subprocess import run
    url = f"https://gitlab.archlinux.org/archlinux/packaging/packages/{pkg}.git"
    with TemporaryDirectory(prefix=f"tmp-{pkg}-") as dir:
        cmd = ["git", "clone", "--depth=1", url, dir ]
        run(cmd, check=True)
        return parse_pkgbuild_ver_sum(dir + '/PKGBUILD')