aboutsummaryrefslogtreecommitdiffstats
path: root/SQLiteStudio3/sqlitestudiocli/commands/clicommandfactory.cpp
blob: 32f972fdf7f4fa8346d578f09e1b23d3d621f30b (plain) (blame)
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
#include "clicommandfactory.h"
#include "cli.h"
#include "clicommandadd.h"
#include "clicommandremove.h"
#include "clicommandexit.h"
#include "clicommanddblist.h"
#include "clicommanduse.h"
#include "clicommandopen.h"
#include "clicommandclose.h"
#include "clicommandsql.h"
#include "clicommandhelp.h"
#include "clicommandtables.h"
#include "clicommandmode.h"
#include "clicommandnullvalue.h"
#include "clicommandhistory.h"
#include "clicommanddir.h"
#include "clicommandpwd.h"
#include "clicommandcd.h"
#include "clicommandtree.h"
#include "clicommanddesc.h"
#include <QDebug>

QHash<QString,CliCommandFactory::CliCommandCreatorFunc> CliCommandFactory::mapping;

#define REGISTER_CMD(Cmd) registerCommand([]() -> CliCommand* {return new Cmd();})

void CliCommandFactory::init()
{
    REGISTER_CMD(CliCommandAdd);
    REGISTER_CMD(CliCommandRemove);
    REGISTER_CMD(CliCommandExit);
    REGISTER_CMD(CliCommandDbList);
    REGISTER_CMD(CliCommandUse);
    REGISTER_CMD(CliCommandOpen);
    REGISTER_CMD(CliCommandClose);
    REGISTER_CMD(CliCommandSql);
    REGISTER_CMD(CliCommandHelp);
    REGISTER_CMD(CliCommandTables);
    REGISTER_CMD(CliCommandMode);
    REGISTER_CMD(CliCommandNullValue);
    REGISTER_CMD(CliCommandHistory);
    REGISTER_CMD(CliCommandDir);
    REGISTER_CMD(CliCommandPwd);
    REGISTER_CMD(CliCommandCd);
    REGISTER_CMD(CliCommandTree);
    REGISTER_CMD(CliCommandDesc);
}

CliCommand *CliCommandFactory::getCommand(const QString &cmdName)
{
    if (!mapping.contains(cmdName))
        return nullptr;

    return mapping[cmdName]();
}

QHash<QString,CliCommand*> CliCommandFactory::getAllCommands()
{
    QHash<QString,CliCommand*> results;
    QHashIterator<QString,CliCommandFactory::CliCommandCreatorFunc> it(mapping);
    while (it.hasNext())
    {
        it.next();
        results[it.key()] = it.value()();
    }

    return results;
}

QStringList CliCommandFactory::getCommandNames()
{
    return mapping.keys();
}

void CliCommandFactory::registerCommand(CliCommandCreatorFunc func)
{
    CliCommand* cmd = func();
    cmd->defineSyntax();

    mapping[cmd->getName()] = func;
    for (const QString& alias : cmd->aliases())
        mapping[alias] = func;

    delete cmd;
}