Add DirectoryCollection class for creating directory-backed collections

[libs/datafile.git] / source / directorycollection.h

diff --git a/source/directorycollection.h b/source/directorycollection.h
new file mode 100644 (file)
index 0000000..93f9354
--- /dev/null
+++ b/source/directorycollection.h
@@ -0,0 +1,52 @@
+#ifndef MSP_DATAFILE_DIRECTORYCOLLECTION_H_
+#define MSP_DATAFILE_DIRECTORYCOLLECTION_H_
+
+#include <msp/fs/path.h>
+#include "collection.h"
+
+namespace Msp {
+namespace DataFile {
+
+/**
+A Collection that can automatically load items from files in a directory.
+*/
+class DirectoryCollection: public Collection
+{
+private:
+       std::list<FS::Path> dirs;
+
+public:
+       DirectoryCollection();
+
+protected:
+       void set_directory(const FS::Path &);
+       void add_directory(const FS::Path &);
+
+       template<typename T>
+       CollectionItemType<T> &add_type()
+       {
+               return Collection::add_type<T>().creator(&DirectoryCollection::create<T>);
+       }
+
+private:
+       template<typename T>
+       T *create(const std::string &name)
+       {
+               FS::Path file;
+               if(lookup_file(name, file))
+               {
+                       RefPtr<T> item = new T;
+                       load(*item, file.str());
+                       return item.release();
+               }
+               else
+                       return 0;
+       }
+
+       bool lookup_file(const std::string &, FS::Path &) const;
+};
+
+} // namespace DataFile
+} // namespace Msp
+
+#endif