download_helper.h 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
  1. /* <!-- copyright */
  2. /*
  3. * aria2 - The high speed download utility
  4. *
  5. * Copyright (C) 2006 Tatsuhiro Tsujikawa
  6. *
  7. * This program is free software; you can redistribute it and/or modify
  8. * it under the terms of the GNU General Public License as published by
  9. * the Free Software Foundation; either version 2 of the License, or
  10. * (at your option) any later version.
  11. *
  12. * This program is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU General Public License
  18. * along with this program; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. *
  21. * In addition, as a special exception, the copyright holders give
  22. * permission to link the code of portions of this program with the
  23. * OpenSSL library under certain conditions as described in each
  24. * individual source file, and distribute linked combinations
  25. * including the two.
  26. * You must obey the GNU General Public License in all respects
  27. * for all of the code used other than OpenSSL. If you modify
  28. * file(s) with this exception, you may extend this exception to your
  29. * version of the file(s), but you are not obligated to do so. If you
  30. * do not wish to do so, delete this exception statement from your
  31. * version. If you delete this exception statement from all source
  32. * files in the program, then also delete it here.
  33. */
  34. /* copyright --> */
  35. #ifndef D_DOWNLOAD_HELPER_H
  36. #define D_DOWNLOAD_HELPER_H
  37. #include "common.h"
  38. #include <string>
  39. #include <vector>
  40. #include <set>
  41. #include "SharedHandle.h"
  42. namespace aria2 {
  43. class RequestGroup;
  44. class Option;
  45. class MetadataInfo;
  46. class DownloadContext;
  47. class UriListParser;
  48. #ifdef ENABLE_BITTORRENT
  49. // Create RequestGroup object using torrent file specified by
  50. // torrent-file option. If non-empty torrentData is specified, then
  51. // it is used as a content of torrent file instead. If
  52. // adjustAnnounceUri is true, announce URIs are adjusted using
  53. // bittorrent::adjustAnnounceUri(). In this function,
  54. // force-sequential is ignored.
  55. void createRequestGroupForBitTorrent
  56. (std::vector<SharedHandle<RequestGroup> >& result,
  57. const SharedHandle<Option>& option,
  58. const std::vector<std::string>& uris,
  59. const std::string& torrentData = "",
  60. bool adjustAnnounceUri = true);
  61. #endif // ENABLE_BITTORRENT
  62. #ifdef ENABLE_METALINK
  63. // Create RequestGroup objects using Metalink file specified by
  64. // metalink-file option. If non-empty metalinkData is specified, it is
  65. // used as a content of metalink file instead.
  66. void createRequestGroupForMetalink
  67. (std::vector<SharedHandle<RequestGroup> >& result,
  68. const SharedHandle<Option>& option,
  69. const std::string& metalinkData = "");
  70. #endif // ENABLE_METALINK
  71. // Reads one entry from uriListParser and creates RequestGroups from
  72. // it and store them in result. If the bad entry is found, this
  73. // function just skips it and reads next entry. If at least one
  74. // RequestGroup is created successfully, this function returns true
  75. // and created RequestGroups are stored in result. If no RequestGroup
  76. // is created and uriListParser reads all input, this function returns
  77. // false. The option is used as a option template.
  78. bool createRequestGroupFromUriListParser
  79. (std::vector<SharedHandle<RequestGroup> >& result,
  80. const Option* option,
  81. UriListParser* uriListParser);
  82. // Creates UriListParser using given filename. If filename is "-",
  83. // then UriListParser is configured to read from standard input.
  84. // Otherwise, this function first checks file denoted by filename
  85. // exists. If it does not exist, this function throws exception.
  86. // This function returns SharedHandle<UriListParser> object if it
  87. // succeeds.
  88. SharedHandle<UriListParser> openUriListParser(const std::string& filename);
  89. // Create RequestGroup objects from reading file specified by input-file option.
  90. // If the value of input-file option is "-", stdin is used as a input source.
  91. // Each line is treated as if it is provided in command-line argument.
  92. // The additional out and dir options can be specified after each line of URIs.
  93. // This optional line must start with white space(s).
  94. void createRequestGroupForUriList
  95. (std::vector<SharedHandle<RequestGroup> >& result,
  96. const SharedHandle<Option>& option);
  97. // Create RequestGroup object using provided uris. If ignoreLocalPath
  98. // is true, a path to torrent file abd metalink file are ignored. If
  99. // throwOnError is true, exception will be thrown when Metalink
  100. // Document or .torrent file cannot be parsed or unrecognized URI is
  101. // given. If throwOnError is false, these errors are just logged as
  102. // error.
  103. void createRequestGroupForUri
  104. (std::vector<SharedHandle<RequestGroup> >& result,
  105. const SharedHandle<Option>& option,
  106. const std::vector<std::string>& uris,
  107. bool ignoreForceSequential = false,
  108. bool ignoreLocalPath = false,
  109. bool throwOnError = false);
  110. template<typename InputIterator>
  111. void setMetadataInfo
  112. (InputIterator first, InputIterator last, const SharedHandle<MetadataInfo>& mi)
  113. {
  114. for(; first != last; ++first) {
  115. (*first)->setMetadataInfo(mi);
  116. }
  117. }
  118. SharedHandle<MetadataInfo>
  119. createMetadataInfoFromFirstFileEntry(const SharedHandle<DownloadContext>& dctx);
  120. // Removes option value which is only effective at the first
  121. // construction time.
  122. void removeOneshotOption(const SharedHandle<Option>& option);
  123. } // namespace aria2
  124. #endif // D_DOWNLOAD_HELPER_H