/*
 * Copyright (c) 2017, 2024, Oracle and/or its affiliates.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License, version 2.0,
 * as published by the Free Software Foundation.
 *
 * This program is designed to work with certain software (including
 * but not limited to OpenSSL) that is licensed under separate terms,
 * as designated in a particular file or component or in included license
 * documentation.  The authors of MySQL hereby grant you an additional
 * permission to link the program and your derivative works with the
 * separately licensed software that they have either included with
 * the program or referenced in the documentation.
 *
 * 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, version 2.0, 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 St, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include "shellcore/completer.h"
#include <algorithm>
#include <cctype>
#include <iostream>
#include <iterator>
#include <set>

#include "mysqlshdk/libs/utils/logger.h"
#include "mysqlshdk/libs/utils/utils_file.h"
#include "mysqlshdk/libs/utils/utils_lexing.h"
#include "mysqlshdk/libs/utils/utils_path.h"
#include "mysqlshdk/libs/utils/utils_string.h"

namespace shcore {
namespace completer {
Completion_list Completer::complete(IShell_core::Mode mode,
                                    const std::string &buffer,
                                    const std::string &line,
                                    size_t *compl_offset) {
  Completion_list list;
  for (auto &iter : providers_) {
    if (iter.first.is_set(mode)) {
      size_t old_compl_offs = *compl_offset;
      auto tmp = iter.second->complete(buffer, line, compl_offset);
      if (tmp.empty()) {
        *compl_offset = old_compl_offs;
      } else {
        std::move(tmp.begin(), tmp.end(), std::back_inserter(list));
        if (old_compl_offs != *compl_offset) break;
      }
    }
  }
  return list;
}

namespace {

bool has_prefix(std::string_view path, std::string_view prefix) {
#if defined(_WIN32) || defined(__APPLE__)
  return shcore::str_ibeginswith(path, prefix);
#else
  return shcore::str_beginswith(path, prefix);
#endif
}
}  // namespace

Completion_list Completer::complete_path(const std::string &path) {
  Completion_list list;
  std::string dirname = shcore::path::dirname(path);
  std::string prefix = shcore::path::basename(path);

  // basename returns dir2 for /dir1/dir2/, but we want "" in this case
  if (!shcore::str_endswith(path, prefix)) {
    dirname = path;
    prefix = "";
  }

  try {
    shcore::iterdir(dirname.empty() ? "." : dirname, [&list, &prefix, &dirname](
                                                         const std::string &f) {
      if (f != "." && f != "..") {
        if (prefix.empty() || has_prefix(f, prefix)) {
          auto p = shcore::path::join_path("." == dirname ? "" : dirname, f);
          if (shcore::is_folder(p)) p = shcore::path::join_path(p, "");
          list.push_back(p);
        }
      }
      return true;
    });
  } catch (...) {
  }

  std::sort(std::begin(list), std::end(list));

  return list;
}

void Completer::add_provider(IShell_core::Mode_mask mode_mask,
                             std::shared_ptr<Provider> provider,
                             bool before_all) {
  if (before_all) {
    providers_.insert(providers_.begin(), {mode_mask, provider});
  } else {
    providers_.push_back({mode_mask, provider});
  }
}

void Completer::reset() { providers_.clear(); }

}  // namespace completer
}  // namespace shcore
