aboutsummaryrefslogtreecommitdiff
blob: 93c05c0c6275bfe1a0b8cc1dc8e817f8bd77681d (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
"""
Package source manager is responsible for selecting the correct source
for a requested package, by a provided string.

Package sources can provide several handlers with different priorities.
A handler is a String -> Result function, which tries to parse the passed
value and return Ok(value).

The package would be handled by the handler with the lowest priority, which
was added the first.

Example:
    @dispatcher.source
    class BgoSource():
        @dispatcher.handler(priority=5)
        def parse_int(uri):
            if uri.is_decimal():
                return Result.Ok(int(uri))
            elif uri[0] == '#' and uri[1:].is_decimal():
                return Result.Ok(int(uri[1:]))
            return Result.Err('NaN')

        @dispatcher.handler(priority=1)
        def parse_url(uri):
            if uri.startswith('http://bugs.gentoo.org/'):
            ...
"""
#TODO: efficient sorted insertion
#import bisect
import inspect

class PackageDispatcher():
    def __init__(self):
        self.handlers = []

    def source(self, cls):
        for m, obj in inspect.getmembers(cls):
            if isinstance(obj, self.handler._handler):
                self.register_package_handler(cls, obj.handler, obj.priority)
        return cls

    class handler():
        class _handler():
            def __init__(self, handler):
                self.handler = handler
            def __call__(self, *args):
                return self.handler(*args)

        def __init__(self, priority=1000, *args, **kwargs):
            self.priority = priority

        def __call__(self, func, *args, **kwargs):
            x = self._handler(func)
            x.priority = self.priority
            return x

    def register_package_handler(self, source, handler, priority):
        i = 0
        for i in range(len(self.handlers)):
            if self.handlers[0] > priority:
                break
        self.handlers.insert(i, (priority, source, handler))

    def get_package_source(self, uri):
        for priority, source, handler in self.handlers:
            if handler(uri).is_ok():
                return source
        return None