/*
 *    Copyright (C) 2013  Hong Jen Yee (PCMan) <pcman.tw@gmail.com>
 *
 *    This program is free software; you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation; either version 2 of the License, or
 *    (at your option) any later version.
 *
 *    This program is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License along
 *    with this program; if not, write to the Free Software Foundation, Inc.,
 *    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include "xdgdir.h"
#include <QStandardPaths>
#include <QFile>
#include <QDir>
#include <QSaveFile>

QString XdgDir::readUserDirsFile() {
    QFile file(QStandardPaths::writableLocation(QStandardPaths::ConfigLocation) + QStringLiteral("/user-dirs.dirs"));
    if(file.open(QIODevice::ReadOnly | QIODevice::Text)) {
        QByteArray data = file.readAll();
        file.close();
        return QString::fromLocal8Bit(data);
    }
    return QString();
}

QString XdgDir::readDesktopDir() {
    QString str = readUserDirsFile();
    if(str.isEmpty())
        return QStandardPaths::writableLocation(QStandardPaths::HomeLocation) + QStringLiteral("/Desktop");
    QRegExp reg(QStringLiteral("XDG_DESKTOP_DIR=\"([^\n]*)\""));
    if(reg.lastIndexIn(str) != -1) {
        str = reg.cap(1);
        if(str.startsWith(QStringLiteral("$HOME")))
            str = QStandardPaths::writableLocation(QStandardPaths::HomeLocation) + str.mid(5);
        return str;
    }
    return QString();
}

void XdgDir::setDesktopDir(QString path) {
    QString home = QStandardPaths::writableLocation(QStandardPaths::HomeLocation);
    if(path.startsWith(home))
        path = QStringLiteral("$HOME") + path.mid(home.length());
    QString str = readUserDirsFile();
    QRegExp reg(QStringLiteral("XDG_DESKTOP_DIR=\"([^\n]*)\""));
    QString line = QStringLiteral("XDG_DESKTOP_DIR=\"") + path + '\"';
    if(reg.indexIn(str) != -1)
        str.replace(reg, line);
    else {
        if(!str.endsWith('\n'))
            str += '\n';
        str += line + '\n';
    }
    QString dir = QStandardPaths::writableLocation(QStandardPaths::ConfigLocation);
    if(QDir().mkpath(dir)) { // write the file
        QSaveFile file(dir + QStringLiteral("/user-dirs.dirs"));
        if(file.open(QIODevice::WriteOnly | QIODevice::Text)) {
            file.write(str.toLocal8Bit());
            file.commit();
        }
    }
}