diff UnitTestsSources/RestApiTests.cpp @ 969:3dce528b0cc2

RestApiHierarchy
author Sebastien Jodogne <s.jodogne@gmail.com>
date Mon, 30 Jun 2014 13:17:49 +0200
parents 8ed284e79850
children 1a3817d84f39
line wrap: on
line diff
--- a/UnitTestsSources/RestApiTests.cpp	Fri Jun 27 17:48:55 2014 +0200
+++ b/UnitTestsSources/RestApiTests.cpp	Mon Jun 30 13:17:49 2014 +0200
@@ -42,6 +42,7 @@
 #include "../Core/Uuid.h"
 #include "../Core/OrthancException.h"
 #include "../Core/Compression/ZlibCompressor.h"
+#include "../Core/RestApi/RestApiHierarchy.h"
 
 using namespace Orthanc;
 
@@ -192,460 +193,6 @@
 
 
 
-namespace Orthanc
-{
-  class RestApiHierarchy
-  {
-  private:
-    struct Handlers
-    {
-      typedef std::list<RestApi::GetHandler>  GetHandlers;
-      typedef std::list<RestApi::PostHandler>  PostHandlers;
-      typedef std::list<RestApi::PutHandler>  PutHandlers;
-      typedef std::list<RestApi::DeleteHandler>  DeleteHandlers;
-
-      GetHandlers  getHandlers_;
-      PostHandlers  postHandlers_;
-      PutHandlers  putHandlers_;
-      DeleteHandlers  deleteHandlers_;
-
-      bool HasGet() const
-      {
-        return getHandlers_.size() > 0;
-      }
-
-      void Register(RestApi::GetHandler handler)
-      {
-        getHandlers_.push_back(handler);
-      }
-
-      void Register(RestApi::PutHandler handler)
-      {
-        putHandlers_.push_back(handler);
-      }
-
-      void Register(RestApi::PostHandler handler)
-      {
-        postHandlers_.push_back(handler);
-      }
-
-      void Register(RestApi::DeleteHandler handler)
-      {
-        deleteHandlers_.push_back(handler);
-      }
-
-      bool IsEmpty() const
-      {
-        return (getHandlers_.empty() &&
-                postHandlers_.empty() &&
-                putHandlers_.empty() &&
-                deleteHandlers_.empty());
-      }
-    };
-
-
-    typedef std::map<std::string, RestApiHierarchy*>  Children;
-    typedef bool (*ResourceCallback) (Handlers&,
-                                      const UriComponents& uri,
-                                      const RestApiPath::Components& components,
-                                      const UriComponents& trailing,
-                                      void* call);
-
-    Handlers  handlers_;
-    Children  children_;
-    Children  wildcardChildren_;
-    Handlers  universalHandlers_;
-
-
-    static RestApiHierarchy& AddChild(Children& children,
-                                      const std::string& name)
-    {
-      Children::iterator it = children.find(name);
-
-      if (it == children.end())
-      {
-        // Create new child
-        RestApiHierarchy *child = new RestApiHierarchy;
-        children[name] = child;
-        return *child;
-      }
-      else
-      {
-        return *it->second;
-      }
-    }
-
-
-    static void DeleteChildren(Children& children)
-    {
-      for (Children::iterator it = children.begin();
-           it != children.end(); it++)
-      {
-        delete it->second;
-      }
-    }
-
-
-    template <typename Handler>
-    void RegisterInternal(const RestApiPath& path,
-                          Handler handler,
-                          size_t level)
-    {
-      if (path.GetLevelCount() == level)
-      {
-        if (path.IsUniversalTrailing())
-        {
-          universalHandlers_.Register(handler);
-        }
-        else
-        {
-          handlers_.Register(handler);
-        }
-      }
-      else
-      {
-        RestApiHierarchy* child;
-        if (path.IsWildcardLevel(level))
-        {
-          child = &AddChild(wildcardChildren_, path.GetWildcardName(level));
-        }
-        else
-        {
-          child = &AddChild(children_, path.GetLevelName(level));
-        }
-
-        child->RegisterInternal(path, handler, level + 1);
-      }
-    }
-
-
-    bool LookupHandler(RestApiPath::Components& components,
-                       const UriComponents& uri,
-                       ResourceCallback callback,
-                       size_t level,
-                       void* call)
-    {
-      assert(uri.size() >= level);
-      UriComponents trailing;
-
-      // Look for an exact match on the resource of interest
-      if (uri.size() == level)
-      {
-        if (!handlers_.IsEmpty() &&
-            callback(handlers_, uri, components, trailing, call))
-        {
-          return true;
-        }
-      }
-
-
-      // Try and go down in the hierarchy, using an exact match for the child
-      Children::const_iterator child = children_.find(uri[level]);
-      if (child != children_.end())
-      {
-        if (child->second->LookupHandler(components, uri, callback, level + 1, call))
-        {
-          return true;
-        }
-      }
-
-
-      // Try and go down in the hierarchy, using wildcard rules for children
-      for (child = wildcardChildren_.begin();
-           child != wildcardChildren_.end(); child++)
-      {
-        RestApiPath::Components subComponents = components;
-        subComponents[child->first] = uri[level];
-
-        if (child->second->LookupHandler(components, uri, callback, level + 1, call))
-        {
-          return true;
-        }        
-      }
-
-
-      // As a last resort, call the universal handlers, if any
-      if (!universalHandlers_.IsEmpty())
-      {
-        trailing.resize(uri.size() - level);
-        size_t pos = 0;
-        for (size_t i = level; i < uri.size(); i++, pos++)
-        {
-          trailing[pos] = uri[i];
-        }
-
-        assert(pos == trailing.size());
-
-        if (callback(universalHandlers_, uri, components, trailing, call))
-        {
-          return true;
-        }
-      }
-
-      return false;
-    }
-
-
-    bool GetDirectory(Json::Value& result,
-                      const UriComponents& uri,
-                      size_t level)
-    {
-      if (uri.size() == level)
-      {
-        if (!handlers_.HasGet() && 
-            universalHandlers_.IsEmpty() &&
-            wildcardChildren_.size() == 0)
-        {
-          result = Json::arrayValue;
-
-          for (Children::const_iterator it = children_.begin();
-               it != children_.end(); it++)
-          {
-            result.append(it->first);
-          }
-
-          return true;
-        }
-        else
-        {
-          return false;
-        }
-      }
-
-      Children::const_iterator child = children_.find(uri[level]);
-      if (child != children_.end())
-      {
-        if (child->second->GetDirectory(result, uri, level + 1))
-        {
-          return true;
-        }
-      }
-
-      for (child = wildcardChildren_.begin(); 
-           child != wildcardChildren_.end(); child++)
-      {
-        if (child->second->GetDirectory(result, uri, level + 1))
-        {
-          return true;
-        }
-      }
-
-      return false;
-    }
-                       
-
-    static bool GetCallback(Handlers& handlers,
-                            const UriComponents& uri,
-                            const RestApiPath::Components& components,
-                            const UriComponents& trailing,
-                            void* call)
-    {
-      for (Handlers::GetHandlers::iterator
-             it = handlers.getHandlers_.begin(); 
-           it != handlers.getHandlers_.end(); it++)
-      {
-        // TODO RETURN BOOL
-
-        (*it) (*reinterpret_cast<RestApi::GetCall*>(call));
-        return true;
-      }
-
-      return false;
-    }
-
-
-    static bool PostCallback(Handlers& handlers,
-                             const UriComponents& uri,
-                             const RestApiPath::Components& components,
-                             const UriComponents& trailing,
-                             void* call)
-    {
-      for (Handlers::PostHandlers::iterator
-             it = handlers.postHandlers_.begin(); 
-           it != handlers.postHandlers_.end(); it++)
-      {
-        // TODO RETURN BOOL
-
-        (*it) (*reinterpret_cast<RestApi::PostCall*>(call));
-        return true;
-      }
-
-      return false;
-    }
-
-
-    static bool PutCallback(Handlers& handlers,
-                            const UriComponents& uri,
-                            const RestApiPath::Components& components,
-                            const UriComponents& trailing,
-                            void* call)
-    {
-      for (Handlers::PutHandlers::iterator
-             it = handlers.putHandlers_.begin(); 
-           it != handlers.putHandlers_.end(); it++)
-      {
-        // TODO RETURN BOOL
-
-        (*it) (*reinterpret_cast<RestApi::PutCall*>(call));
-        return true;
-      }
-
-      return false;
-    }
-
-
-    static bool DeleteCallback(Handlers& handlers,
-                               const UriComponents& uri,
-                               const RestApiPath::Components& components,
-                               const UriComponents& trailing,
-                               void* call)
-    {
-      for (Handlers::DeleteHandlers::iterator
-             it = handlers.deleteHandlers_.begin(); 
-           it != handlers.deleteHandlers_.end(); it++)
-      {
-        // TODO RETURN BOOL
-
-        (*it) (*reinterpret_cast<RestApi::DeleteCall*>(call));
-        return true;
-      }
-
-      return false;
-    }
-
-
-  public:
-    ~RestApiHierarchy()
-    {
-      DeleteChildren(children_);
-      DeleteChildren(wildcardChildren_);
-    }
-
-    void Register(const RestApiPath& path,
-                  RestApi::GetHandler handler)
-    {
-      RegisterInternal(path, handler, 0);
-    }
-
-    void Register(const RestApiPath& path,
-                  RestApi::PutHandler handler)
-    {
-      RegisterInternal(path, handler, 0);
-    }
-
-    void Register(const RestApiPath& path,
-                  RestApi::PostHandler handler)
-    {
-      RegisterInternal(path, handler, 0);
-    }
-
-    void Register(const RestApiPath& path,
-                  RestApi::DeleteHandler handler)
-    {
-      RegisterInternal(path, handler, 0);
-    }
-
-    void CreateSiteMap(Json::Value& target) const
-    {
-      if (children_.size() == 0)
-      {
-        std::string s = " ";
-        if (handlers_.getHandlers_.size() != 0)
-        {
-          s += "GET ";
-        }
-
-        if (handlers_.postHandlers_.size() != 0)
-        {
-          s += "POST ";
-        }
-
-        if (handlers_.putHandlers_.size() != 0)
-        {
-          s += "PUT ";
-        }
-
-        if (handlers_.deleteHandlers_.size() != 0)
-        {
-          s += "DELETE ";
-        }
-
-        target = s;
-      }
-      else
-      {
-        target = Json::objectValue;
-      
-        for (Children::const_iterator it = children_.begin();
-             it != children_.end(); it++)
-        {
-          it->second->CreateSiteMap(target[it->first]);
-        }
-      }
-      
-      /*for (Children::const_iterator it = wildcardChildren_.begin();
-           it != wildcardChildren_.end(); it++)
-      {
-        it->second->CreateSiteMap(target["* (" + it->first + ")"]);
-        }*/
-    }
-
-    bool GetDirectory(Json::Value& result,
-                      const UriComponents& uri)
-    {
-      return GetDirectory(result, uri, 0);
-    }
-
-    bool GetDirectory(Json::Value& result,
-                      const std::string& uri)
-    {
-      UriComponents c;
-      Toolbox::SplitUriComponents(c, uri);
-      return GetDirectory(result, c, 0);
-    }
-
-    bool Handle(RestApi::GetCall& call,
-                const UriComponents& uri)
-    {
-      RestApiPath::Components components;
-      return LookupHandler(components, uri, GetCallback, 0, &call);
-    }    
-
-    bool Handle(RestApi::PutCall& call,
-                const UriComponents& uri)
-    {
-      RestApiPath::Components components;
-      return LookupHandler(components, uri, PutCallback, 0, &call);
-    }    
-
-    bool Handle(RestApi::PostCall& call,
-                const UriComponents& uri)
-    {
-      RestApiPath::Components components;
-      return LookupHandler(components, uri, PostCallback, 0, &call);
-    }    
-
-    bool Handle(RestApi::DeleteCall& call,
-                const UriComponents& uri)
-    {
-      RestApiPath::Components components;
-      return LookupHandler(components, uri, DeleteCallback, 0, &call);
-    }    
-
-    bool Handle(RestApi::GetCall& call,
-                const std::string& uri)
-    {
-      UriComponents c;
-      Toolbox::SplitUriComponents(c, uri);
-      return Handle(call, c);
-    }    
-  };
-
-}
-
-
-
-
 static int testValue;
 
 template <int value>
@@ -655,6 +202,25 @@
 }
 
 
+static bool GetDirectory(Json::Value& target,
+                         RestApiHierarchy& hierarchy, 
+                         const std::string& uri)
+{
+  UriComponents p;
+  Toolbox::SplitUriComponents(p, uri);
+  return hierarchy.GetDirectory(target, p);
+}
+
+
+static bool HandleGet(RestApiHierarchy& hierarchy, 
+                      const std::string& uri)
+{
+  UriComponents p;
+  Toolbox::SplitUriComponents(p, uri);
+  return hierarchy.Handle(*reinterpret_cast<RestApi::GetCall*>(NULL), p);
+}
+
+
 TEST(RestApi, RestApiHierarchy)
 {
   RestApiHierarchy root;
@@ -668,32 +234,32 @@
   std::cout << m;
 
   Json::Value d;
-  ASSERT_FALSE(root.GetDirectory(d, "/hello"));
+  ASSERT_FALSE(GetDirectory(d, root, "/hello"));
 
-  ASSERT_TRUE(root.GetDirectory(d, "/hello/a")); 
+  ASSERT_TRUE(GetDirectory(d, root, "/hello/a")); 
   ASSERT_EQ(1u, d.size());
   ASSERT_EQ("test3", d[0].asString());
 
-  ASSERT_TRUE(root.GetDirectory(d, "/hello/world"));
+  ASSERT_TRUE(GetDirectory(d, root, "/hello/world"));
   ASSERT_EQ(2u, d.size());
 
-  ASSERT_TRUE(root.GetDirectory(d, "/hello/a/test3"));
+  ASSERT_TRUE(GetDirectory(d, root, "/hello/a/test3"));
   ASSERT_EQ(1u, d.size());
   ASSERT_EQ("test4", d[0].asString());
 
-  ASSERT_FALSE(root.GetDirectory(d, "/hello/world/test"));
-  ASSERT_FALSE(root.GetDirectory(d, "/hello/world/test2"));
-  ASSERT_FALSE(root.GetDirectory(d, "/hello2"));
+  ASSERT_FALSE(GetDirectory(d, root, "/hello/world/test"));
+  ASSERT_FALSE(GetDirectory(d, root, "/hello/world/test2"));
+  ASSERT_FALSE(GetDirectory(d, root, "/hello2"));
 
   testValue = 0;
-  ASSERT_TRUE(root.Handle(*reinterpret_cast<RestApi::GetCall*>(NULL), "/hello/world/test"));
+  ASSERT_TRUE(HandleGet(root, "/hello/world/test"));
   ASSERT_EQ(testValue, 1);
-  ASSERT_TRUE(root.Handle(*reinterpret_cast<RestApi::GetCall*>(NULL), "/hello/world/test2"));
+  ASSERT_TRUE(HandleGet(root, "/hello/world/test2"));
   ASSERT_EQ(testValue, 2);
-  ASSERT_TRUE(root.Handle(*reinterpret_cast<RestApi::GetCall*>(NULL), "/hello/b/test3/test4"));
+  ASSERT_TRUE(HandleGet(root, "/hello/b/test3/test4"));
   ASSERT_EQ(testValue, 3);
-  ASSERT_FALSE(root.Handle(*reinterpret_cast<RestApi::GetCall*>(NULL), "/hello/b/test3/test"));
+  ASSERT_FALSE(HandleGet(root, "/hello/b/test3/test"));
   ASSERT_EQ(testValue, 3);
-  ASSERT_TRUE(root.Handle(*reinterpret_cast<RestApi::GetCall*>(NULL), "/hello2/a/b"));
+  ASSERT_TRUE(HandleGet(root, "/hello2/a/b"));
   ASSERT_EQ(testValue, 4);
 }