]>
Commit | Line | Data |
---|---|---|
1 | // -*- mode: cpp; mode: fold -*- | |
2 | // Description /*{{{*/ | |
3 | /* ##################################################################### | |
4 | apt-helper - cmdline helpers | |
5 | ##################################################################### */ | |
6 | /*}}}*/ | |
7 | // Include Files /*{{{*/ | |
8 | #include <config.h> | |
9 | ||
10 | #include <apt-pkg/configuration.h> | |
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> | |
19 | #include <apt-pkg/proxy.h> | |
20 | ||
21 | #include <apt-private/acqprogress.h> | |
22 | #include <apt-private/private-output.h> | |
23 | #include <apt-private/private-download.h> | |
24 | #include <apt-private/private-cmndline.h> | |
25 | ||
26 | #include <iostream> | |
27 | #include <string> | |
28 | #include <vector> | |
29 | ||
30 | #include <apti18n.h> | |
31 | /*}}}*/ | |
32 | ||
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 | ||
46 | static bool DoDownloadFile(CommandLine &CmdL) | |
47 | { | |
48 | if (CmdL.FileSize() <= 2) | |
49 | return _error->Error(_("Must specify at least one pair url/filename")); | |
50 | ||
51 | pkgAcquire Fetcher; | |
52 | AcqTextStatus Stat(ScreenWidth, _config->FindI("quiet",0)); | |
53 | Fetcher.Setup(&Stat); | |
54 | ||
55 | size_t fileind = 0; | |
56 | std::vector<std::string> targetfiles; | |
57 | while (fileind + 2 <= CmdL.FileSize()) | |
58 | { | |
59 | std::string download_uri = CmdL.FileList[fileind + 1]; | |
60 | std::string targetfile = CmdL.FileList[fileind + 2]; | |
61 | std::string hash; | |
62 | if (CmdL.FileSize() > fileind + 3) | |
63 | hash = CmdL.FileList[fileind + 3]; | |
64 | // we use download_uri as descr and targetfile as short-descr | |
65 | new pkgAcqFile(&Fetcher, download_uri, hash, 0, download_uri, targetfile, | |
66 | "dest-dir-ignored", targetfile); | |
67 | targetfiles.push_back(targetfile); | |
68 | fileind += 3; | |
69 | } | |
70 | ||
71 | bool Failed = false; | |
72 | if (AcquireRun(Fetcher, 0, &Failed, NULL) == false || Failed == true) | |
73 | return _error->Error(_("Download Failed")); | |
74 | if (targetfiles.empty() == false) | |
75 | for (std::vector<std::string>::const_iterator f = targetfiles.begin(); f != targetfiles.end(); ++f) | |
76 | if (FileExists(*f) == false) | |
77 | return _error->Error(_("Download Failed")); | |
78 | ||
79 | return true; | |
80 | } | |
81 | ||
82 | static bool ShowHelp(CommandLine &) | |
83 | { | |
84 | ioprintf(std::cout,_("%s %s for %s compiled on %s %s\n"),PACKAGE,PACKAGE_VERSION, | |
85 | COMMON_ARCH,__DATE__,__TIME__); | |
86 | ||
87 | if (_config->FindB("version") == true) | |
88 | return true; | |
89 | ||
90 | std::cout << | |
91 | _("Usage: apt-helper [options] command\n" | |
92 | " apt-helper [options] download-file uri target-path\n" | |
93 | "\n" | |
94 | "apt-helper is a internal helper for apt\n" | |
95 | "\n" | |
96 | "Commands:\n" | |
97 | " download-file - download the given uri to the target-path\n" | |
98 | " auto-detect-proxy - detect proxy using apt.conf\n" | |
99 | "\n" | |
100 | " This APT helper has Super Meep Powers.\n"); | |
101 | return true; | |
102 | } | |
103 | ||
104 | ||
105 | int main(int argc,const char *argv[]) /*{{{*/ | |
106 | { | |
107 | CommandLine::Dispatch Cmds[] = {{"help",&ShowHelp}, | |
108 | {"download-file", &DoDownloadFile}, | |
109 | {"auto-detect-proxy", &DoAutoDetectProxy}, | |
110 | {0,0}}; | |
111 | ||
112 | std::vector<CommandLine::Args> Args = getCommandArgs( | |
113 | "apt-download", CommandLine::GetCommand(Cmds, argc, argv)); | |
114 | ||
115 | // Set up gettext support | |
116 | setlocale(LC_ALL,""); | |
117 | textdomain(PACKAGE); | |
118 | ||
119 | // Parse the command line and initialize the package library | |
120 | CommandLine CmdL(Args.data(),_config); | |
121 | if (pkgInitConfig(*_config) == false || | |
122 | CmdL.Parse(argc,argv) == false || | |
123 | pkgInitSystem(*_config,_system) == false) | |
124 | { | |
125 | if (_config->FindB("version") == true) | |
126 | ShowHelp(CmdL); | |
127 | _error->DumpErrors(); | |
128 | return 100; | |
129 | } | |
130 | ||
131 | // See if the help should be shown | |
132 | if (_config->FindB("help") == true || | |
133 | _config->FindB("version") == true || | |
134 | CmdL.FileSize() == 0) | |
135 | { | |
136 | ShowHelp(CmdL); | |
137 | return 0; | |
138 | } | |
139 | ||
140 | InitOutput(); | |
141 | ||
142 | // Match the operation | |
143 | CmdL.DispatchArg(Cmds); | |
144 | ||
145 | // Print any errors or warnings found during parsing | |
146 | bool const Errors = _error->PendingError(); | |
147 | if (_config->FindI("quiet",0) > 0) | |
148 | _error->DumpErrors(); | |
149 | else | |
150 | _error->DumpErrors(GlobalError::DEBUG); | |
151 | return Errors == true ? 100 : 0; | |
152 | } | |
153 | /*}}}*/ |