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
|
/*
* megatools - Mega.nz client library and tools
* Copyright (C) 2013 Ondřej Jirman <megi@xff.cz>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program 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.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include <string.h>
#include "shell.h"
#include "config.h"
#include "lib/alloc.h"
#ifdef G_OS_WIN32
#include <stdio.h>
#include <stdlib.h>
#include <windows.h>
#endif
#include <locale.h>
extern struct shell_tool shell_tool_df;
extern struct shell_tool shell_tool_dl;
extern struct shell_tool shell_tool_get;
extern struct shell_tool shell_tool_ls;
extern struct shell_tool shell_tool_mkdir;
extern struct shell_tool shell_tool_put;
extern struct shell_tool shell_tool_reg;
extern struct shell_tool shell_tool_rm;
extern struct shell_tool shell_tool_copy;
extern struct shell_tool shell_tool_test;
extern struct shell_tool shell_tool_export;
static struct shell_tool* tools[] = {
&shell_tool_dl,
&shell_tool_df,
&shell_tool_ls,
&shell_tool_test,
&shell_tool_export,
&shell_tool_get,
&shell_tool_put,
&shell_tool_copy,
&shell_tool_mkdir,
&shell_tool_rm,
&shell_tool_reg,
};
#ifdef G_OS_WIN32
static unsigned int initial_cp;
static void print_no_convert(const gchar *buf)
{
fputs(buf, stdout);
fflush(stdout);
}
static void printerr_no_convert(const gchar *buf)
{
fputs(buf, stderr);
fflush(stderr);
}
static void restore_console(void)
{
SetConsoleOutputCP(initial_cp);
}
#endif
int main(int ac, char *av[])
{
#ifdef G_OS_WIN32
setlocale(LC_ALL, "C");
av = g_win32_get_command_line();
ac = g_strv_length(av);
g_set_print_handler(print_no_convert);
g_set_printerr_handler(printerr_no_convert);
initial_cp = GetConsoleOutputCP();
SetConsoleOutputCP(CP_UTF8);
atexit(restore_console);
#else
setlocale(LC_ALL, "");
av = g_strdupv(av);
#endif
gc_free gchar* cmd_basename = g_path_get_basename(av[0]);
gchar* cmd_name = NULL;
if (g_str_has_suffix(cmd_basename, ".exe"))
cmd_basename[strlen(cmd_basename) - 4] = '\0';
if (g_str_has_prefix(cmd_basename, "mega"))
cmd_name = cmd_basename + 4;
if (cmd_name) {
// try to run a specifc <command> if we're run via mega<command>[.exe]
for (int i = 0; i < G_N_ELEMENTS(tools); i++) {
if (!strcmp(cmd_name, tools[i]->name))
return tools[i]->main(ac, av);
}
}
if (ac > 1) {
// otherwise try to find a command name based on the first argument
for (int i = 0; i < G_N_ELEMENTS(tools); i++) {
if (!strcmp(av[1], tools[i]->name)) {
av[1] = g_strdup_printf("megatools %s", av[1]);
return tools[i]->main(ac - 1, av + 1);
}
}
}
// show usage if we failed to run any specific command
g_print("Usage:\n");
for (int i = 0; i < G_N_ELEMENTS(tools); i++)
if (tools[i]->usages)
for (int j = 0; tools[i]->usages[j]; j++)
g_print(" megatools %s %s\n", tools[i]->name, tools[i]->usages[j]);
g_print("\n");
g_print("Run: megatools <command> --help for detailed options for each command.\n");
g_print("\n");
g_print("megatools " VERSION " - command line tools for Mega.nz\n");
g_print("Written by Ondrej Jirman <megi@xff.cz>, 2013-2025\n");
g_print("Go to https://xff.cz/megatools/ for more information\n");
return 1;
}
|
