aboutsummaryrefslogtreecommitdiff
path: root/src/main.cpp
blob: ca85b6573e7c415f818c66731da3c115438f155e (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
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
/*
 * This file is part of smolbote. It's copyrighted by the contributors recorded
 * in the version control history of the file, available from its original
 * location: https://neueland.iserlohn-fortress.net/gitea/aqua/smolbote
 *
 * SPDX-License-Identifier: GPL-3.0
 */

#include "browser.h"
#include "builtins.h"
#include "configuration.h"
#include "crashhandler.h"
#include "session/sessiondialog.h"
#include "session_json.hpp"
#include "settings.h"
#include "util.h"
#include "version.h"
#include <QFile>
#include <QPluginLoader>
#include <QStandardPaths>
#include <iostream>
#include <memory>
#include <pluginloader.h>
#include <spdlog/spdlog.h>

// a helper function to join the keys of a command_map into a string
inline std::string join_keys(const command_map &map, const std::string sep = ", ")
{
    std::vector<std::string> keys(map.size());
    std::transform(map.begin(), map.end(), keys.begin(), [](auto pair) { return pair.first; });
    std::sort(keys.begin(), keys.end());

    std::string k;
    std::for_each(keys.begin(), keys.end() - 1, [&k, &sep](const std::string &piece) { k += piece + sep; });
    k += keys.back();

    return k;
}

int main(int argc, char **argv)
{
    // change log pattern
    spdlog::set_pattern("[%^%l%$] [%P:%t] %v");
#ifdef QT_DEBUG
    spdlog::set_level(spdlog::level::debug); // Set global log level to debug
#endif

    const command_map commands{
        { "configuration", builtins::configuration },
        { "bookmarks", builtins::bookmarks },
    };

    const std::vector<std::string> args(argv + 1, argv + argc);
    args::ArgumentParser parser("smolbote: yet another no-frills browser", "Subcommands: " + join_keys(commands));
    parser.Prog(argv[0]);

    args::HelpFlag cmd_help(parser, "help", "Display this help message.", { 'h', "help" });
    args::Flag cmd_version(parser, "version", "Display version information.", { 'v', "version" });
    args::Flag cmd_build(parser, "build", "Display build commit.", { 'b', "build" });

    args::ValueFlag<std::string> cmd_config(parser, "config", "Set the configuration file.", { 'c', "config" });

    args::Flag cmd_noRemote(parser, "no-remote", "Do not accept or send remote commands.", { "no-remote" });

    args::Flag cmd_pickSession(parser, "pick-session", "Show all available sessions and select which one to open", { "pick-session" });
    args::ValueFlag<std::string> cmd_session(parser, "session", "Open the specified session.", { 's', "session" });

    args::PositionalList<std::string> cmd_args(parser, "URL(s)", "List of URLs to open");
    cmd_args.KickOut(true);

    try {
        auto next = parser.ParseArgs(args);

        if(cmd_version)
            return builtins::version();
        if(cmd_build)
            return builtins::build();

        // create and load configuration
        spdlog::debug("Loading configuration {}", init_conf(args::get(cmd_config)));

        if(cmd_args) {
            const auto front = args::get(cmd_args).front();
            const auto cmd = commands.find(front);
            if(cmd != commands.end()) {
                return cmd->second(argv[0], next, std::end(args));
            }
        }

    } catch(args::Help &) {
        std::cout << parser;
        return 0;

    } catch(args::Error &e) {
        std::cerr << e.what() << std::endl;
        std::cerr << parser;
        return -1;
    }

    // argc, argv, allowSecondary
    Browser app(argc, argv);
    // set this, otherwise the webview becomes black when using a stylesheet
    app.setAttribute(Qt::AA_DontCreateNativeWidgetSiblings, true);

    {
        Configuration conf;

        if(conf.value<bool>("usebreakpad").value()) {
            CrashHandler::Context ctx(
                conf.value<std::string>("path.crashdump").value(),
                conf.value<std::string>("path.crashhandler").value());

            if(CrashHandler::install_handler(ctx)) {
                spdlog::info("Installed breakpad crash handler: {}", ctx.dumppath);
            } else {
                spdlog::warn("Failed to install breakpad crash handler: {}", ctx.dumppath);
            }
        }

        // load plugins
        app.loadPlugins(Util::files(conf.value<QString>("plugins.path").value(), { "*.so", "*.dll" }),
            [](const auto *loader) {
                if(loader->isLoaded())
                    spdlog::info("Loaded plugin [{}]", qUtf8Printable(loader->fileName()));
                else
                    spdlog::warn("Loading plugin [{}] failed: {}", qUtf8Printable(loader->fileName()), qUtf8Printable(loader->errorString()));
            });
    }

    const auto profile = []() {
        Configuration c;
        return c.value<QString>("profile.default").value();
    }();

    app.setup();

    QStringList urls;

    for(const auto &u : args::get(cmd_args)) {
        urls.append(QString::fromStdString(u));
    }
    if(urls.isEmpty())
        urls.append(QString());

    // if app is primary, create new sessions from received messages
    if(app.isPrimary() && !cmd_noRemote) {
        QObject::connect(&app, &Browser::receivedMessage, &app, [&app](quint32 instanceId, QByteArray message) {
            Q_UNUSED(instanceId);
            JsonSession session(message);
            app.open(session.get());
        });
    }

    {
        const auto session = [&]() {
            if(cmd_session) {
                QFile sessionJson(QString::fromStdString(args::get(cmd_session)));
                if(sessionJson.open(QIODevice::ReadOnly | QIODevice::Text)) {
                    return JsonSession(sessionJson.readAll());
                }
            }
            if(cmd_pickSession) {
                SessionDialog dlg;
                if(const auto pick = dlg.pickSession()) {
                    return JsonSession(pick.value());
                }
            }
            return JsonSession(profile, urls);
        }();

        if(app.isPrimary() || cmd_noRemote) {
            app.open(session.get());
        } else {
            // app is secondary and not standalone
            return app.sendMessage(session.serialize());
        }
    }

    return app.exec();
}