aboutsummaryrefslogtreecommitdiff
path: root/src/clients/cave/owner_common.cc
blob: 156c4a1c3bb180c3e4a4b3d6eac92a79e4114d3d (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
/* vim: set sw=4 sts=4 et foldmethod=syntax : */

/*
 * Copyright (c) 2008 Alexander Færøy
 * Copyright (c) 2010 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 "owner_common.hh"
#include <paludis/action.hh>
#include <paludis/contents.hh>
#include <paludis/environment.hh>
#include <paludis/filter.hh>
#include <paludis/filtered_generator.hh>
#include <paludis/generator.hh>
#include <paludis/metadata_key.hh>
#include <paludis/name.hh>
#include <paludis/package_database.hh>
#include <paludis/package_id.hh>
#include <paludis/repository.hh>
#include <paludis/selection.hh>
#include <paludis/util/indirect_iterator-impl.hh>
#include <paludis/util/wrapped_forward_iterator.hh>
#include <paludis/util/fs_stat.hh>
#include <algorithm>
#include <functional>

using namespace paludis;

namespace
{
    bool handle_full(const std::string & q, const std::shared_ptr<const ContentsEntry> & e)
    {
        return q == stringify(e->location_key()->value());
    }

    bool handle_basename(const std::string & q, const std::shared_ptr<const ContentsEntry> & e)
    {
        return q == e->location_key()->value().basename();
    }

    bool handle_partial(const std::string & q, const std::shared_ptr<const ContentsEntry> & e)
    {
        return std::string::npos != stringify(e->location_key()->value()).find(q);
    }
}

int
paludis::cave::owner_common(
        const std::shared_ptr<Environment> & env,
        const std::string & match,
        const std::string & q,
        const bool dereference,
        const std::function<void (const std::shared_ptr<const PackageID> &)> & callback)
{
    bool found(false);
    std::function<bool (const std::string &, const std::shared_ptr<const ContentsEntry> &)> handler;
    std::string query(q);

    if (dereference)
    {
        FSPath query_path(query);
        FSStat query_path_stat(query_path);
        if (query_path_stat.exists())
        {
            if (query_path_stat.is_symlink())
                query = stringify(query_path.realpath_if_exists());
        }
    }

    if (query.length() >= 2 && '/' == query.at(query.length() - 1))
        query.erase(query.length() - 1);

    if ("full" == match)
        handler = handle_full;
    else if ("basename" == match)
        handler = handle_basename;
    else if ("partial" == match)
        handler = handle_partial;
    else
    {
        if (! query.empty() && '/' == query.at(0))
            handler = handle_full;
        else if (std::string::npos != query.find("/"))
            handler = handle_partial;
        else
            handler = handle_basename;
    }

    std::shared_ptr<const PackageIDSequence> ids((*env)[selection::AllVersionsSorted(generator::All() |
                filter::InstalledAtRoot(env->preferred_root_key()->value()))]);

    for (PackageIDSequence::ConstIterator p(ids->begin()), p_end(ids->end()); p != p_end; ++p)
    {
        if (! (*p)->contents_key())
            continue;

        std::shared_ptr<const Contents> contents((*p)->contents_key()->value());
        if (contents->end() != std::find_if(contents->begin(), contents->end(), std::bind(handler, query,
                        std::placeholders::_1)))
        {
            callback(*p);
            found = true;
        }
    }

    return found ? EXIT_SUCCESS : EXIT_FAILURE;
}