]>
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 | ||
52 | pkgAcquire Fetcher; | |
53 | AcqTextStatus Stat(ScreenWidth, _config->FindI("quiet",0)); | |
54 | Fetcher.Setup(&Stat); | |
55 | std::string download_uri = CmdL.FileList[1]; | |
56 | std::string targetfile = CmdL.FileList[2]; | |
57 | std::string hash; | |
58 | if (CmdL.FileSize() > 3) | |
59 | hash = CmdL.FileList[3]; | |
60 | // we use download_uri as descr and targetfile as short-descr | |
61 | new pkgAcqFile(&Fetcher, download_uri, hash, 0, download_uri, targetfile, | |
62 | "dest-dir-ignored", targetfile); | |
63 | Fetcher.Run(); | |
64 | bool Failed = false; | |
65 | if (AcquireRun(Fetcher, 0, &Failed, NULL) == false || Failed == true || | |
66 | FileExists(targetfile) == false) | |
67 | return _error->Error(_("Download Failed")); | |
68 | return true; | |
69 | } | |
70 | ||
71 | static bool ShowHelp(CommandLine &) | |
72 | { | |
73 | ioprintf(std::cout,_("%s %s for %s compiled on %s %s\n"),PACKAGE,PACKAGE_VERSION, | |
74 | COMMON_ARCH,__DATE__,__TIME__); | |
75 | ||
76 | if (_config->FindB("version") == true) | |
77 | return true; | |
78 | ||
79 | std::cout << | |
80 | _("Usage: apt-helper [options] command\n" | |
81 | " apt-helper [options] download-file uri target-path\n" | |
82 | "\n" | |
83 | "apt-helper is a internal helper for apt\n" | |
84 | "\n" | |
85 | "Commands:\n" | |
86 | " download-file - download the given uri to the target-path\n" | |
87 | " auto-detect-proxy - detect proxy using apt.conf\n" | |
88 | "\n" | |
89 | " This APT helper has Super Meep Powers.\n"); | |
90 | return true; | |
91 | } | |
92 | ||
93 | ||
94 | int main(int argc,const char *argv[]) /*{{{*/ | |
95 | { | |
96 | CommandLine::Dispatch Cmds[] = {{"help",&ShowHelp}, | |
97 | {"download-file", &DoDownloadFile}, | |
98 | {"auto-detect-proxy", &DoAutoDetectProxy}, | |
99 | {0,0}}; | |
100 | ||
101 | std::vector<CommandLine::Args> Args = getCommandArgs( | |
102 | "apt-download", CommandLine::GetCommand(Cmds, argc, argv)); | |
103 | ||
104 | // Set up gettext support | |
105 | setlocale(LC_ALL,""); | |
106 | textdomain(PACKAGE); | |
107 | ||
108 | // Parse the command line and initialize the package library | |
109 | CommandLine CmdL(Args.data(),_config); | |
110 | if (pkgInitConfig(*_config) == false || | |
111 | CmdL.Parse(argc,argv) == false || | |
112 | pkgInitSystem(*_config,_system) == false) | |
113 | { | |
114 | if (_config->FindB("version") == true) | |
115 | ShowHelp(CmdL); | |
116 | _error->DumpErrors(); | |
117 | return 100; | |
118 | } | |
119 | ||
120 | // See if the help should be shown | |
121 | if (_config->FindB("help") == true || | |
122 | _config->FindB("version") == true || | |
123 | CmdL.FileSize() == 0) | |
124 | { | |
125 | ShowHelp(CmdL); | |
126 | return 0; | |
127 | } | |
128 | ||
129 | InitOutput(); | |
130 | ||
131 | // Match the operation | |
132 | CmdL.DispatchArg(Cmds); | |
133 | ||
134 | // Print any errors or warnings found during parsing | |
135 | bool const Errors = _error->PendingError(); | |
136 | if (_config->FindI("quiet",0) > 0) | |
137 | _error->DumpErrors(); | |
138 | else | |
139 | _error->DumpErrors(GlobalError::DEBUG); | |
140 | return Errors == true ? 100 : 0; | |
141 | } | |
142 | /*}}}*/ |