download_helper.h 6.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158
  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 <memory>
  42. namespace aria2 {
  43. class RequestGroup;
  44. class Option;
  45. class MetadataInfo;
  46. class DownloadContext;
  47. class UriListParser;
  48. class ValueBase;
  49. class GroupId;
  50. #ifdef ENABLE_BITTORRENT
  51. // Create RequestGroup object using torrent file specified by
  52. // metaInfoUri, which is treated as local file path. If non-empty
  53. // torrentData is specified, then it is used as a content of torrent
  54. // file instead. If adjustAnnounceUri is true, announce URIs are
  55. // adjusted using bittorrent::adjustAnnounceUri(). In this function,
  56. // force-sequential is ignored.
  57. void createRequestGroupForBitTorrent
  58. (std::vector<std::shared_ptr<RequestGroup>>& result,
  59. const std::shared_ptr<Option>& option,
  60. const std::vector<std::string>& uris,
  61. const std::string& metaInfoUri,
  62. const std::string& torrentData = "",
  63. bool adjustAnnounceUri = true);
  64. // Create RequestGroup object using already decoded torrent metainfo
  65. // structure. If adjustAnnounceUri is true, announce URIs are
  66. // adjusted using bittorrent::adjustAnnounceUri(). In this function,
  67. // force-sequential is ignored.
  68. void createRequestGroupForBitTorrent
  69. (std::vector<std::shared_ptr<RequestGroup>>& result,
  70. const std::shared_ptr<Option>& option,
  71. const std::vector<std::string>& uris,
  72. const std::string& metaInfoUri,
  73. const ValueBase* torrent,
  74. bool adjustAnnounceUri = true);
  75. #endif // ENABLE_BITTORRENT
  76. #ifdef ENABLE_METALINK
  77. // Create RequestGroup objects using Metalink file specified by
  78. // metalink-file option. If non-empty metalinkData is specified, it is
  79. // used as a content of metalink file instead.
  80. void createRequestGroupForMetalink
  81. (std::vector<std::shared_ptr<RequestGroup>>& result,
  82. const std::shared_ptr<Option>& option,
  83. const std::string& metalinkData = "");
  84. #endif // ENABLE_METALINK
  85. // Reads one entry from uriListParser and creates RequestGroups from
  86. // it and store them in result. If the bad entry is found, this
  87. // function just skips it and reads next entry. If at least one
  88. // RequestGroup is created successfully, this function returns true
  89. // and created RequestGroups are stored in result. If no RequestGroup
  90. // is created and uriListParser reads all input, this function returns
  91. // false. The option is used as a option template.
  92. bool createRequestGroupFromUriListParser
  93. (std::vector<std::shared_ptr<RequestGroup>>& result,
  94. const Option* option,
  95. UriListParser* uriListParser);
  96. // Creates UriListParser using given filename. If filename is "-",
  97. // then UriListParser is configured to read from standard input.
  98. // Otherwise, this function first checks file denoted by filename
  99. // exists. If it does not exist, this function throws exception.
  100. // This function returns std::shared_ptr<UriListParser> object if it
  101. // succeeds.
  102. std::shared_ptr<UriListParser> openUriListParser(const std::string& filename);
  103. // Create RequestGroup objects from reading file specified by input-file option.
  104. // If the value of input-file option is "-", stdin is used as a input source.
  105. // Each line is treated as if it is provided in command-line argument.
  106. // The additional out and dir options can be specified after each line of URIs.
  107. // This optional line must start with white space(s).
  108. void createRequestGroupForUriList
  109. (std::vector<std::shared_ptr<RequestGroup>>& result,
  110. const std::shared_ptr<Option>& option);
  111. // Create RequestGroup object using provided uris. If ignoreLocalPath
  112. // is true, a path to torrent file abd metalink file are ignored. If
  113. // throwOnError is true, exception will be thrown when Metalink
  114. // Document or .torrent file cannot be parsed or unrecognized URI is
  115. // given. If throwOnError is false, these errors are just logged as
  116. // error.
  117. void createRequestGroupForUri
  118. (std::vector<std::shared_ptr<RequestGroup>>& result,
  119. const std::shared_ptr<Option>& option,
  120. const std::vector<std::string>& uris,
  121. bool ignoreForceSequential = false,
  122. bool ignoreLocalPath = false,
  123. bool throwOnError = false);
  124. template<typename InputIterator>
  125. void setMetadataInfo
  126. (InputIterator first, InputIterator last,
  127. const std::shared_ptr<MetadataInfo>& mi)
  128. {
  129. for(; first != last; ++first) {
  130. (*first)->setMetadataInfo(mi);
  131. }
  132. }
  133. std::shared_ptr<MetadataInfo>
  134. createMetadataInfoFromFirstFileEntry
  135. (const std::shared_ptr<GroupId>& gid,
  136. const std::shared_ptr<DownloadContext>& dctx);
  137. // Removes option value which is only effective at the first
  138. // construction time.
  139. void removeOneshotOption(const std::shared_ptr<Option>& option);
  140. } // namespace aria2
  141. #endif // D_DOWNLOAD_HELPER_H