ibrcommon/ibrcommon/data/File.cpp

Go to the documentation of this file.

/*
 * File.cpp
 *
 *  Created on: 20.11.2009
 *      Author: morgenro
 */

#include "ibrcommon/config.h"
#include "ibrcommon/data/File.h"
#include <sstream>
#include <errno.h>
#include <sys/stat.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

namespace ibrcommon
{
        File::File()
         : _path(), _type(DT_UNKNOWN)
        {
        }

        File::File(const string path, const unsigned char t)
         : _path(path), _type(t)
        {
                removeSlash();
        }

        File::File(const string path)
         : _path(path), _type(DT_UNKNOWN)
        {
                removeSlash();
                update();
        }

        void File::removeSlash()
        {
                std::string::iterator iter = _path.end(); iter--;

                if ((*iter) == '/')
                {
                        _path.erase(iter);
                }
        }

        bool File::exists() const
        {
                struct stat st;
                if( stat(_path.c_str(), &st ) == 0)
                        return true;

                return false;
        }

        void File::update()
        {
                struct stat s;
                int type;

                if ( stat(_path.c_str(), &s) == 0 )
                {
                        type = s.st_mode & S_IFMT;

                        switch (type)
                        {
                                case S_IFREG:
                                        _type = DT_REG;
                                        break;

                                case S_IFLNK:
                                        _type = DT_LNK;
                                        break;

                                case S_IFDIR:
                                        _type = DT_DIR;
                                        break;

                                default:
                                        _type = DT_UNKNOWN;
                                        break;
                        }
                }
        }

        File::~File()
        {}

        unsigned char File::getType() const
        {
                return _type;
        }

        int File::getFiles(list<File> &files)
        {
                if (!isDirectory()) return -1;

                DIR *dp;
                struct dirent *dirp;
                if((dp = opendir(_path.c_str())) == NULL) {
                        return errno;
                }

                while ((dirp = readdir(dp)) != NULL)
                {
                        string name = string(dirp->d_name);
                        stringstream ss; ss << getPath() << "/" << name;
                        File file(ss.str(), dirp->d_type);
                        files.push_back(file);
                }
                closedir(dp);

                return 0;
        }

        bool File::isSystem() const
        {
                if ((_path.substr(_path.length() - 2, 2) == "..") || (_path.substr(_path.length() - 1, 1) == ".")) return true;
                return false;
        }

        bool File::isDirectory() const
        {
                if (_type == DT_DIR) return true;
                return false;
        }

        string File::getPath() const
        {
                return _path;
        }

        File File::get(string filename) const
        {
                stringstream ss; ss << getPath() << "/" << filename;
                File file(ss.str());

                return file;
        }

        int File::remove(bool recursive)
        {
                int ret;

                if (isSystem()) return -1;
                if (_type == DT_UNKNOWN) return -1;

                if (isDirectory())
                {
                        if (recursive)
                        {
                                // container for all files
                                list<File> files;

                                // get all files in this directory
                                if ((ret = getFiles(files)) < 0)
                                        return ret;

                                for (list<File>::iterator iter = files.begin(); iter != files.end(); iter++)
                                {
                                        if (!(*iter).isSystem())
                                        {
                                                if ((ret = (*iter).remove(recursive)) < 0)
                                                        return ret;
                                        }
                                }
                        }

                        ::rmdir(getPath().c_str());
                }
                else
                {
			::remove(getPath().c_str());
                }

                return 0;
        }

        File File::getParent() const
        {
                size_t pos = _path.find_last_of('/');
                return File(_path.substr(0, pos));
        }

        void File::createDirectory(File &path)
        {
                if (!path.exists())
                {
                        File parent = path.getParent();
                        File::createDirectory(parent);

                        // create path
                        ::mkdir(path.getPath().c_str(), 0700);

                        // update file information
                        path.update();
                }
        }

        size_t File::size() const
        {
                struct stat filestatus;
                stat( getPath().c_str(), &filestatus );
                return filestatus.st_size;
        }

        TemporaryFile::TemporaryFile(const File &path, const std::string prefix)
         : File(tmpname(path, prefix))
        {
        }

        TemporaryFile::~TemporaryFile()
        {
        }

        std::string TemporaryFile::tmpname(const File &path, const std::string prefix)
        {
                char *name;

                if ( (name = tempnam( path.getPath().c_str(), prefix.c_str() )) == NULL )
                        throw ibrcommon::IOException("Could not create a temporary name.");

                string filename(name);
                free(name);

                return filename;
        }
}

---