]>
Commit | Line | Data |
---|---|---|
e43a426e MV |
1 | // -*- mode: cpp; mode: fold -*- |
2 | // Description /*{{{*/ | |
3 | /* ##################################################################### | |
4 | apt-helper - cmdline helpers | |
5 | ##################################################################### */ | |
6 | /*}}}*/ | |
7 | // Include Files /*{{{*/ | |
8 | #include <config.h> | |
9 | ||
453b82a3 | 10 | #include <apt-pkg/configuration.h> |
e43a426e MV |
11 | #include <apt-pkg/cmndline.h> |
12 | #include <apt-pkg/error.h> | |
13 | #include <apt-pkg/init.h> | |
14 | #include <apt-pkg/strutl.h> | |
15 | #include <apt-pkg/pkgsystem.h> | |
16 | #include <apt-pkg/fileutl.h> | |
17 | #include <apt-pkg/acquire.h> | |
18 | #include <apt-pkg/acquire-item.h> | |
c6ee61ea | 19 | #include <apt-pkg/proxy.h> |
e43a426e MV |
20 | |
21 | #include <apt-private/acqprogress.h> | |
22 | #include <apt-private/private-output.h> | |
0d58c26a | 23 | #include <apt-private/private-download.h> |
e43a426e MV |
24 | #include <apt-private/private-cmndline.h> |
25 | ||
453b82a3 DK |
26 | #include <iostream> |
27 | #include <string> | |
28 | #include <vector> | |
e43a426e MV |
29 | |
30 | #include <apti18n.h> | |
31 | /*}}}*/ | |
e43a426e | 32 | |
c6ee61ea MV |
33 | static bool DoAutoDetectProxy(CommandLine &CmdL) |
34 | { | |
35 | if (CmdL.FileSize() != 2) | |
36 | return _error->Error(_("Need one URL as argument")); | |
37 | URI ServerURL(CmdL.FileList[1]); | |
38 | AutoDetectProxy(ServerURL); | |
39 | std::string SpecificProxy = _config->Find("Acquire::"+ServerURL.Access+"::Proxy::" + ServerURL.Host); | |
40 | ioprintf(std::cout, "Using proxy '%s' for URL '%s'\n", | |
41 | SpecificProxy.c_str(), std::string(ServerURL).c_str()); | |
42 | ||
43 | return true; | |
44 | } | |
45 | ||
c3ccac92 | 46 | static bool DoDownloadFile(CommandLine &CmdL) |
e43a426e MV |
47 | { |
48 | if (CmdL.FileSize() <= 2) | |
49 | return _error->Error(_("Must specify at least one pair url/filename")); | |
50 | ||
dfad5bee | 51 | AcqTextStatus Stat(std::cout, ScreenWidth,_config->FindI("quiet",0)); |
04a54261 DK |
52 | pkgAcquire Fetcher(&Stat); |
53 | ||
ed793a19 DK |
54 | size_t fileind = 0; |
55 | std::vector<std::string> targetfiles; | |
56 | while (fileind + 2 <= CmdL.FileSize()) | |
57 | { | |
58 | std::string download_uri = CmdL.FileList[fileind + 1]; | |
59 | std::string targetfile = CmdL.FileList[fileind + 2]; | |
60 | std::string hash; | |
61 | if (CmdL.FileSize() > fileind + 3) | |
62 | hash = CmdL.FileList[fileind + 3]; | |
63 | // we use download_uri as descr and targetfile as short-descr | |
64 | new pkgAcqFile(&Fetcher, download_uri, hash, 0, download_uri, targetfile, | |
65 | "dest-dir-ignored", targetfile); | |
66 | targetfiles.push_back(targetfile); | |
67 | fileind += 3; | |
68 | } | |
460601d5 DK |
69 | |
70 | // Disable drop-privs if "_apt" can not write to the target dir | |
71 | CheckDropPrivsMustBeDisabled(Fetcher); | |
72 | ||
0d58c26a | 73 | bool Failed = false; |
ed793a19 | 74 | if (AcquireRun(Fetcher, 0, &Failed, NULL) == false || Failed == true) |
0d58c26a | 75 | return _error->Error(_("Download Failed")); |
ed793a19 DK |
76 | if (targetfiles.empty() == false) |
77 | for (std::vector<std::string>::const_iterator f = targetfiles.begin(); f != targetfiles.end(); ++f) | |
78 | if (FileExists(*f) == false) | |
79 | return _error->Error(_("Download Failed")); | |
80 | ||
e43a426e MV |
81 | return true; |
82 | } | |
83 | ||
65512241 | 84 | static bool ShowHelp(CommandLine &) |
e43a426e | 85 | { |
249aec3b | 86 | ioprintf(std::cout, "%s %s (%s)\n", PACKAGE, PACKAGE_VERSION, COMMON_ARCH); |
e43a426e MV |
87 | |
88 | if (_config->FindB("version") == true) | |
89 | return true; | |
90 | ||
0d58c26a | 91 | std::cout << |
e43a426e MV |
92 | _("Usage: apt-helper [options] command\n" |
93 | " apt-helper [options] download-file uri target-path\n" | |
94 | "\n" | |
95 | "apt-helper is a internal helper for apt\n" | |
96 | "\n" | |
97 | "Commands:\n" | |
98 | " download-file - download the given uri to the target-path\n" | |
c6ee61ea | 99 | " auto-detect-proxy - detect proxy using apt.conf\n" |
e43a426e MV |
100 | "\n" |
101 | " This APT helper has Super Meep Powers.\n"); | |
102 | return true; | |
103 | } | |
104 | ||
105 | ||
106 | int main(int argc,const char *argv[]) /*{{{*/ | |
107 | { | |
108 | CommandLine::Dispatch Cmds[] = {{"help",&ShowHelp}, | |
109 | {"download-file", &DoDownloadFile}, | |
c6ee61ea | 110 | {"auto-detect-proxy", &DoAutoDetectProxy}, |
e43a426e MV |
111 | {0,0}}; |
112 | ||
113 | std::vector<CommandLine::Args> Args = getCommandArgs( | |
114 | "apt-download", CommandLine::GetCommand(Cmds, argc, argv)); | |
115 | ||
116 | // Set up gettext support | |
117 | setlocale(LC_ALL,""); | |
118 | textdomain(PACKAGE); | |
119 | ||
120 | // Parse the command line and initialize the package library | |
ad7e0941 DK |
121 | CommandLine CmdL; |
122 | ParseCommandLine(CmdL, Cmds, Args.data(), &_config, &_system, argc, argv, ShowHelp); | |
e43a426e MV |
123 | |
124 | InitOutput(); | |
125 | ||
126 | // Match the operation | |
127 | CmdL.DispatchArg(Cmds); | |
128 | ||
129 | // Print any errors or warnings found during parsing | |
130 | bool const Errors = _error->PendingError(); | |
131 | if (_config->FindI("quiet",0) > 0) | |
132 | _error->DumpErrors(); | |
133 | else | |
134 | _error->DumpErrors(GlobalError::DEBUG); | |
135 | return Errors == true ? 100 : 0; | |
136 | } | |
137 | /*}}}*/ |