desktop/src/tempdir.cc

   1 ///
   2 /// \file       tempdir.cc
   3 ///             Temp directory & file wrapper class
   4 ///
   5
   6 /*
   7     Copyright (C) 2009-2013, Chris Frey <cdfrey@foursquare.net>
   8     The idea to use glib's g_get_tmp_dir() came from opensync's osynctool.
   9
  10     This program is free software; you can redistribute it and/or modify
  11     it under the terms of the GNU General Public License as published by
  12     the Free Software Foundation; either version 2 of the License, or
  13     (at your option) any later version.
  14
  15     This program is distributed in the hope that it will be useful,
  16     but WITHOUT ANY WARRANTY; without even the implied warranty of
  17     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  18
  19     See the GNU General Public License in the COPYING file at the
  20     root directory of this project for more details.
  21 */
  22
  23 #include "tempdir.h"
  24 #include <errno.h>
  25 #include <stdlib.h>
  26 #include <unistd.h>
  27 #include <string.h>
  28 #include <glib.h>
  29 #include <sstream>
  30 #include <stdexcept>
  31 #include "i18n.h"
  32
  33 TempDir::TempDir(const char *basename)
  34         : m_template(0)
  35         , m_files(0)
  36 {
  37         m_template = g_strdup_printf("%s/%s-XXXXXX", g_get_tmp_dir(), basename);
  38         if( mkdtemp(m_template) == NULL ) {
  39                 g_free(m_template);
  40                 throw std::runtime_error(std::string(_C("Cannot create temp directory: ")) + strerror(errno));
  41         }
  42 }
  43
  44 TempDir::~TempDir()
  45 {
  46         // delete all files
  47         for( int i = 0; i < m_files; i++ ) {
  48                 unlink(MakeFilename(i).c_str());
  49         }
  50
  51         // delete directory
  52         rmdir(m_template);
  53
  54         // cleanup memory
  55         g_free(m_template);
  56 }
  57
  58 std::string TempDir::MakeFilename(int file_id) const
  59 {
  60         std::ostringstream oss;
  61         oss << m_template << "/" << file_id;
  62         return oss.str();
  63 }
  64
  65 std::string TempDir::GetNewFilename()
  66 {
  67         return MakeFilename(m_files++);
  68 }
  69