aboutsummaryrefslogtreecommitdiff
blob: fd11ed14a60935071b786cb9b2f3999ece262eae (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
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
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
#!/usr/bin/env python -b
# Copyright 1999-2021 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2

import sys
from os import path as osp

if osp.isfile(
    osp.join(osp.dirname(osp.dirname(osp.realpath(__file__))), ".portage_not_installed")
):
    sys.path.insert(
        0, osp.join(osp.dirname(osp.dirname(osp.realpath(__file__))), "lib")
    )
import portage

portage._internal_caller = True
from portage import os
from portage._sets.files import StaticFileSet, WorldSelectedPackagesSet

import re
import tempfile
import textwrap

__candidatematcher__ = re.compile("^[0-9]+: \\*\\*\\* emerge ")
__noncandidatematcher__ = re.compile(
    " sync( |$)| clean( |$)| search( |$)|--oneshot|--fetchonly| unmerge( |$)"
)


def issyspkg(pkgline):
    return pkgline[0] == "*"


def iscandidate(logline):
    return __candidatematcher__.match(logline) and not __noncandidatematcher__.search(
        logline
    )


def getpkginfo(logline):
    logline = re.sub("^[0-9]+: \\*\\*\\* emerge ", "", logline)
    logline = logline.strip()
    logline = re.sub("(\\S+\\.(ebuild|tbz2))|(--\\S+)|inject ", "", logline)
    return logline.strip()


__uniqlist__ = []


def isunwanted(pkgline):
    if pkgline in ["world", "system", "depclean", "info", "regen", ""]:
        return False
    elif pkgline in __uniqlist__:
        return False
    elif not re.search("^[a-zA-Z<>=~]", pkgline):
        return False
    else:
        __uniqlist__.append(pkgline)
        return True


eroot = portage.settings["EROOT"]
world_file = os.path.join(eroot, portage.WORLD_FILE)

# show a little description if we have arguments
if len(sys.argv) >= 2 and sys.argv[1] in ["-h", "--help"]:
    print("This script regenerates the portage world file by checking the portage")
    print("logfile for all actions that you've done in the past. It ignores any")
    print("arguments except --help. It is recommended that you make a backup of")
    print("your existing world file (%s) before using this tool." % world_file)
    sys.exit(0)

worldlist = portage.grabfile(world_file)
syslist = [x for x in portage.settings.packages if issyspkg(x)]

if portage.settings.get("EMERGE_LOG_DIR"):
    logfile = portage.grabfile(
        os.path.join(portage.settings["EMERGE_LOG_DIR"], "emerge.log")
    )
else:
    logfile = portage.grabfile(os.path.join(eroot, "var/log/emerge.log"))
biglist = [getpkginfo(x) for x in logfile if iscandidate(x)]
tmplist = []
for l in biglist:
    tmplist += l.split()
biglist = [x for x in tmplist if isunwanted(x)]
# for p in biglist:
# 	print(p)
# sys.exit(0)

# resolving virtuals
realsyslist = []
for mykey in syslist:
    # drop the asterix
    mykey = mykey[1:]
    # print("candidate:",mykey)
    mylist = portage.db[eroot]["vartree"].dbapi.match(mykey)
    if mylist:
        mykey = portage.cpv_getkey(mylist[0])
        if mykey not in realsyslist:
            realsyslist.append(mykey)

for mykey in biglist:
    # print("checking:",mykey)
    try:
        mylist = portage.db[eroot]["vartree"].dbapi.match(mykey)
    except (portage.exception.InvalidAtom, KeyError):
        if "--debug" in sys.argv:
            print("* ignoring broken log entry for %s (likely injected)" % mykey)
    except ValueError as e:
        try:
            print("* %s is an ambiguous package name, candidates are:\n%s" % (mykey, e))
        except AttributeError:
            # FIXME: Find out what causes this (bug #344845).
            print("* %s is an ambiguous package name" % (mykey,))
        continue
    if mylist:
        # print "mylist:",mylist
        myfavkey = portage.cpv_getkey(mylist[0])
        if (myfavkey not in realsyslist) and (myfavkey not in worldlist):
            print("add to world:", myfavkey)
            worldlist.append(myfavkey)

if not worldlist:
    pass
else:
    existing_set = WorldSelectedPackagesSet(eroot)
    existing_set.load()

    if not existing_set:
        existing_set.replace(worldlist)
    else:
        old_world = existing_set._filename
        fd, tmp_filename = tempfile.mkstemp(
            suffix=".tmp",
            prefix=os.path.basename(old_world) + ".",
            dir=os.path.dirname(old_world),
        )
        os.close(fd)

        new_set = StaticFileSet(tmp_filename)
        new_set.update(worldlist)

        if existing_set.getAtoms() == new_set.getAtoms():
            os.unlink(tmp_filename)
        else:
            new_set.write()

            msg = (
                "Please review differences between old and new files, "
                + "and replace the old file if desired."
            )

            portage.util.writemsg_stdout("\n", noiselevel=-1)
            for line in textwrap.wrap(msg, 65):
                portage.util.writemsg_stdout("%s\n" % line, noiselevel=-1)
            portage.util.writemsg_stdout("\n", noiselevel=-1)
            portage.util.writemsg_stdout("  old: %s\n\n" % old_world, noiselevel=-1)
            portage.util.writemsg_stdout("  new: %s\n\n" % tmp_filename, noiselevel=-1)