| 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 | /*}}}*/ |