/*
 * Deskflow -- mouse and keyboard sharing utility
 * SPDX-FileCopyrightText: (C) 2024 Symless Ltd.
 * SPDX-License-Identifier: GPL-2.0-only WITH LicenseRef-OpenSSL-Exception
 */

#include "DotEnv.h"

#include "common/Settings.h"

#include <QCoreApplication>
#include <QDebug>
#include <QDir>
#include <QFile>
#include <QProcess>
#include <QProcessEnvironment>
#include <QStandardPaths>
#include <QString>
#include <QTextStream>

namespace deskflow::gui {

QPair<QString, QString> getPair(const QString &line);

bool open(QFile &file, const QString &filePath)
{
  file.setFileName(filePath);
  return file.open(QIODevice::ReadOnly | QIODevice::Text);
}

/**
 * @brief A _very_ basic Qt .env file parser.
 *
 * This function re-invents the wheel to save adding a library dependency.
 * It does not support many things that a proper .env parser would, such as
 * trailing comments, escaping characters, multiline values, etc.
 *
 * If this function is not sufficient, replace it with a library such as:
 * https://github.com/adeharo9/cpp-dotenv
 */
void dotenv(const QString &filename)
{
  QString filePath = filename;
  QFile file;
  if (!open(file, filePath)) {
    QFileInfo fileInfo(filePath);
    qInfo("no %s file in dir: %s", qPrintable(filename), qPrintable(fileInfo.absolutePath()));

    // if nothing in current dir, then try the config dir.
    // this makes it a bit easier for engineers in the field to have an easily
    // predictable location for the .env file.
    const auto orgDir = QDir(Settings::settingsPath());

    filePath = orgDir.filePath(filename);
    if (!open(file, filePath)) {
      qInfo("no %s file in app config dir: %s", qPrintable(filename), qPrintable(orgDir.absolutePath()));
      return;
    }
  }

  qInfo("loading env vars from: %s", qPrintable(filePath));

  QTextStream in(&file);
  while (!in.atEnd()) {
    QString line = in.readLine().trimmed();
    if (line.startsWith('#'))
      continue;

    auto [key, value] = getPair(line);
    if (key.isEmpty() || value.isEmpty())
      continue;

    const auto keyBytes = key.toUtf8();
    const auto valueBytes = value.toUtf8();

    const auto &key_c = keyBytes.constData();
    const auto &value_c = valueBytes.constData();

    qDebug("%s=%s", key_c, value_c);
    qputenv(keyBytes, valueBytes);
  }
}

QString stripQuotes(const QString &value)
{
  QString result = value;
  if (result.startsWith('"') && result.endsWith('"')) {
    result = result.mid(1, result.length() - 2);
  }
  return result;
}

QPair<QString, QString> getPair(const QString &line)
{
  auto pos = line.indexOf('=');
  if (pos == -1) {
    return QPair<QString, QString>("", "");
  }

  QString key = line.left(pos);
  QString value = line.mid(pos + 1);

  return QPair<QString, QString>(key.trimmed(), stripQuotes(value.trimmed()));
}

} // namespace deskflow::gui