From cbaf0619d912d80aab067a4cc7ca8d92f079ff72 Mon Sep 17 00:00:00 2001
From: Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>
Date: Wed, 13 May 2009 14:13:36 +0000
Subject: [PATCH] 2009-05-13  Tatsuhiro Tsujikawa 
 <t-tujikawa@users.sourceforge.net>

	Added addMetalink XML-RPC command.
	* src/XmlRpcMethodFactory.cc
	* src/XmlRpcMethodImpl.cc
	* src/XmlRpcMethodImpl.h
	* src/download_helper.cc
	* src/download_helper.h
---
 ChangeLog                  |  9 +++++++++
 src/XmlRpcMethodFactory.cc |  2 ++
 src/XmlRpcMethodImpl.cc    | 31 +++++++++++++++++++++++++++++++
 src/XmlRpcMethodImpl.h     |  5 +++++
 src/download_helper.cc     | 16 ++++++++++++----
 src/download_helper.h      | 12 +++++++-----
 6 files changed, 66 insertions(+), 9 deletions(-)

diff --git a/ChangeLog b/ChangeLog
index ffe0cf6b..0cc03d24 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,12 @@
+2009-05-13  Tatsuhiro Tsujikawa  <t-tujikawa@users.sourceforge.net>
+
+	Added addMetalink XML-RPC command.
+	* src/XmlRpcMethodFactory.cc
+	* src/XmlRpcMethodImpl.cc
+	* src/XmlRpcMethodImpl.h
+	* src/download_helper.cc
+	* src/download_helper.h
+
 2009-05-13  Tatsuhiro Tsujikawa  <t-tujikawa@users.sourceforge.net>
 
 	Fixed the bug that POU peers are never choked in seeding.
diff --git a/src/XmlRpcMethodFactory.cc b/src/XmlRpcMethodFactory.cc
index 9995734b..f54c14c3 100644
--- a/src/XmlRpcMethodFactory.cc
+++ b/src/XmlRpcMethodFactory.cc
@@ -48,6 +48,8 @@ XmlRpcMethodFactory::create(const std::string& methodName)
     return SharedHandle<XmlRpcMethod>(new AddUriXmlRpcMethod());
   } else if(methodName == "aria2.addTorrent") {
     return SharedHandle<XmlRpcMethod>(new AddTorrentXmlRpcMethod());
+  } else if(methodName == "aria2.addMetalink") {
+    return SharedHandle<XmlRpcMethod>(new AddMetalinkXmlRpcMethod());
   } else if(methodName == "aria2.remove") {
     return SharedHandle<XmlRpcMethod>(new RemoveXmlRpcMethod());
   } else if(methodName == "aria2.tellStatus") {
diff --git a/src/XmlRpcMethodImpl.cc b/src/XmlRpcMethodImpl.cc
index 3c478efe..5d815ac2 100644
--- a/src/XmlRpcMethodImpl.cc
+++ b/src/XmlRpcMethodImpl.cc
@@ -140,6 +140,37 @@ BDE AddTorrentXmlRpcMethod::process
   }
 } 
 
+BDE AddMetalinkXmlRpcMethod::process
+(const XmlRpcRequest& req, DownloadEngine* e)
+{
+  const BDE& params = req._params;
+  assert(params.isList());
+  if(params.empty() || !params[0].isString()) {
+    throw DlAbortEx("Metalink data is not provided.");
+  }
+  
+  SharedHandle<Option> requestOption(new Option(*e->option));
+  if(params.size() > 1 && params[1].isDict()) {
+    gatherRequestOption(requestOption, params[1]);
+  }
+  std::deque<SharedHandle<RequestGroup> > result;
+  createRequestGroupForMetalink(result, requestOption, params[0].s());
+  if(!result.empty()) {
+    e->_requestGroupMan->addReservedGroup(result);
+    BDE resParams = BDE::list();
+    BDE gids = BDE::list();
+    for(std::deque<SharedHandle<RequestGroup> >::const_iterator i =
+	  result.begin(); i != result.end(); ++i) {
+      gids << BDE(Util::itos((*i)->getGID()));
+    }
+    resParams << gids;
+    return resParams;
+  } else {
+    throw DlAbortEx("No files to download.");
+  }
+} 
+
+
 BDE RemoveXmlRpcMethod::process(const XmlRpcRequest& req, DownloadEngine* e)
 {
   const BDE& params = req._params;
diff --git a/src/XmlRpcMethodImpl.h b/src/XmlRpcMethodImpl.h
index ca7fa54e..ed377df6 100644
--- a/src/XmlRpcMethodImpl.h
+++ b/src/XmlRpcMethodImpl.h
@@ -56,6 +56,11 @@ protected:
   virtual BDE process(const XmlRpcRequest& req, DownloadEngine* e);
 };
 
+class AddMetalinkXmlRpcMethod:public XmlRpcMethod {
+protected:
+  virtual BDE process(const XmlRpcRequest& req, DownloadEngine* e);
+};
+
 class GetUrisXmlRpcMethod:public XmlRpcMethod {
 protected:
   virtual BDE process(const XmlRpcRequest& req, DownloadEngine* e);
diff --git a/src/download_helper.cc b/src/download_helper.cc
index 52026994..042e103f 100644
--- a/src/download_helper.cc
+++ b/src/download_helper.cc
@@ -60,6 +60,7 @@
 #include "Util.h"
 #include "array_fun.h"
 #include "OptionHandler.h"
+#include "ByteArrayDiskWriter.h"
 
 namespace aria2 {
 
@@ -252,11 +253,18 @@ void createRequestGroupForBitTorrent
 #ifdef ENABLE_METALINK
 void createRequestGroupForMetalink
 (std::deque<SharedHandle<RequestGroup> >& result,
- const SharedHandle<Option>& option)
+ const SharedHandle<Option>& option,
+ const std::string& metalinkData)
 {
-  Metalink2RequestGroup().generate(result,
-				   option->get(PREF_METALINK_FILE),
-				   option);
+  if(metalinkData.empty()) {
+    Metalink2RequestGroup().generate(result,
+				     option->get(PREF_METALINK_FILE),
+				     option);
+  } else {
+    SharedHandle<ByteArrayDiskWriter> dw(new ByteArrayDiskWriter());
+    dw->setString(metalinkData);
+    Metalink2RequestGroup().generate(result, dw, option);
+  }
   if(result.empty()) {
     throw DlAbortEx(MSG_NO_FILES_TO_DOWNLOAD);
   }
diff --git a/src/download_helper.h b/src/download_helper.h
index 8bc1597b..44fe3bc2 100644
--- a/src/download_helper.h
+++ b/src/download_helper.h
@@ -52,8 +52,8 @@ const std::vector<std::string>& listRequestOptions();
 
 #ifdef ENABLE_BITTORRENT
 // Create RequestGroup object using torrent file specified by
-// torrent-file option.  If torrentData is specified, then it is used
-// as a content of torrent file in stead. In this function,
+// torrent-file option.  If non-empty torrentData is specified, then
+// it is used as a content of torrent file instead. In this function,
 // force-sequential is ignored.
 void createRequestGroupForBitTorrent
 (std::deque<SharedHandle<RequestGroup> >& result,
@@ -63,11 +63,13 @@ void createRequestGroupForBitTorrent
 #endif // ENABLE_BITTORRENT
 
 #ifdef ENABLE_METALINK
-// Create RequestGroup objects using Metalink file specified by metalink-file
-// option.
+// Create RequestGroup objects using Metalink file specified by
+// metalink-file option. If non-empty metalinkData is specified, it is
+// used as a content of metalink file instead.
 void createRequestGroupForMetalink
 (std::deque<SharedHandle<RequestGroup> >& result,
- const SharedHandle<Option>& option);
+ const SharedHandle<Option>& option,
+ const std::string& metalinkData = "");
 #endif // ENABLE_METALINK
 
 // Create RequestGroup objects from reading file specified by input-file option.