aboutsummaryrefslogtreecommitdiff
path: root/paludis/resolver/prefer_or_avoid_helper.cc
blob: 9b8405de1c6ce0c5677ad7421336d74b3465349c (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
/* vim: set sw=4 sts=4 et foldmethod=syntax : */

/*
 * Copyright (c) 2010, 2011, 2014 Ciaran McCreesh
 *
 * This file is part of the Paludis package manager. Paludis is free software;
 * you can redistribute it and/or modify it under the terms of the GNU General
 * Public License version 2, as published by the Free Software Foundation.
 *
 * Paludis is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
 * Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include <paludis/resolver/prefer_or_avoid_helper.hh>
#include <paludis/util/pimp-impl.hh>
#include <paludis/util/hashes.hh>
#include <paludis/util/tribool.hh>
#include <paludis/util/options.hh>
#include <paludis/name.hh>
#include <paludis/dep_spec.hh>
#include <paludis/match_package.hh>
#include <unordered_set>
#include <list>

using namespace paludis;
using namespace paludis::resolver;

namespace paludis
{
    template <>
    struct Imp<PreferOrAvoidHelper>
    {
        const Environment * const env;
        std::unordered_set<QualifiedPackageName, Hash<QualifiedPackageName> > prefer_names;
        std::unordered_set<QualifiedPackageName, Hash<QualifiedPackageName> > avoid_names;
        std::list<std::shared_ptr<const PackageIDSequence> > prefer_matching;
        std::list<std::shared_ptr<const PackageIDSequence> > avoid_matching;

        Imp(const Environment * const e) :
            env(e)
        {
        }
    };
}

PreferOrAvoidHelper::PreferOrAvoidHelper(const Environment * const e) :
    _imp(e)
{
}

PreferOrAvoidHelper::~PreferOrAvoidHelper() = default;

void
PreferOrAvoidHelper::add_prefer_name(const QualifiedPackageName & name)
{
    _imp->prefer_names.insert(name);
}

void
PreferOrAvoidHelper::add_avoid_name(const QualifiedPackageName & name)
{
    _imp->avoid_names.insert(name);
}

void
PreferOrAvoidHelper::add_prefer_matching(const std::shared_ptr<const PackageIDSequence> & s)
{
    _imp->prefer_matching.push_back(s);
}

void
PreferOrAvoidHelper::add_avoid_matching(const std::shared_ptr<const PackageIDSequence> & s)
{
    _imp->avoid_matching.push_back(s);
}

Tribool
PreferOrAvoidHelper::operator() (const PackageDepSpec & s, const std::shared_ptr<const PackageID> & id) const
{
    if (s.package_ptr())
    {
        if (_imp->prefer_names.end() != _imp->prefer_names.find(*s.package_ptr()))
            return true;

        if (_imp->avoid_names.end() != _imp->avoid_names.find(*s.package_ptr()))
            return false;
    }

    for (auto p(_imp->prefer_matching.begin()), p_end(_imp->prefer_matching.end()) ;
            p != p_end ; ++p)
    {
        bool all(true);
        for (auto q((*p)->begin()), q_end((*p)->end()) ; q != q_end ; ++q)
            if (! match_package(*_imp->env, s, *q, id, { }))
            {
                all = false;
                break;
            }

        if (all)
            return true;
    }

    for (auto p(_imp->avoid_matching.begin()), p_end(_imp->avoid_matching.end()) ;
            p != p_end ; ++p)
    {
        for (auto q((*p)->begin()), q_end((*p)->end()) ; q != q_end ; ++q)
            if (match_package(*_imp->env, s, *q, id, { }))
                return false;
    }

    return indeterminate;
}

namespace paludis
{
    template class Pimp<PreferOrAvoidHelper>;
}