cplusplus/Composite_FileDirEntry.h

Go to the documentation of this file.

 
 
#pragma once
#ifndef __COMPOSITE_FILEDIRENTRY_H__
#define __COMPOSITE_FILEDIRENTRY_H__
 
#include <string>
#include <vector>
#include "helpers/DateTime.h"
using Helpers::DateTime;
 
namespace DesignPatternExamples_cpp
{
 
    enum FileDirTypes
    {
        File,
 
        Directory
    };
 
    class FileDirEntry; // forward declaration
 
    typedef std::vector<std::shared_ptr<FileDirEntry>> FileDirEntryList;
 
    class FileDirEntry
    {
    protected:
        FileDirTypes fileDirType;
        std::string name;
        long length;
        DateTime whenModified;
 
    public:
        FileDirEntry()
            : fileDirType(FileDirTypes::File)
            , length(0)
        {
        }
 
        FileDirEntry(FileDirTypes type, std::string entryName, long size, DateTime modDate)
            : fileDirType(type)
            , name(entryName)
            , length(size)
            , whenModified(modDate)
        {
        }
 
        virtual FileDirTypes FileDirType() { return fileDirType; }
 
        virtual std::string Name() { return name; }
 
        virtual long Length() { return length; }
 
        virtual DateTime WhenModified() { return whenModified; }
 
        virtual FileDirEntryList Children() { return FileDirEntryList(); }
    };
 
 
    //########################################################################
    //########################################################################
 
    class FileEntry : public FileDirEntry
    {
    public:
        FileEntry(std::string entryName, long size, DateTime modDate)
            : FileDirEntry(FileDirTypes::File, entryName, size, modDate)
        {
        }
    };
 
 
 
    //########################################################################
    //########################################################################
 
 
    class DirEntry : public FileDirEntry
    {
    public:
        FileDirEntryList _children;
        bool _lengthSet;
 
        DirEntry(std::string entryName, DateTime modDate, FileDirEntryList children)
            : FileDirEntry(FileDirTypes::Directory, entryName, 0, modDate)
            , _children(children)
            , _lengthSet(false)
        {
        }
 
 
        long Length()
        {
            if (!_lengthSet)
            {
                // Recurse over all children, accumulating their size.
                for (FileDirEntryList::iterator entryIter = std::begin(_children);
                    entryIter != std::end(_children);
                    entryIter++)
                {
                    length += (*entryIter)->Length();
                }
                _lengthSet = true;
            }
            return length;
        }
 
 
        FileDirEntryList Children()
        {
            return _children;
        }
    };
 
} // end namespace
 
#endif //__COMPOSITE_FILEDIRENTRY_H__