#include "workspace.h"

#include <regex>

#include "common/log/log.h"
#include "common/util/FileUtil.h"
#include "common/util/ast_util.h"
#include "common/util/string_util.h"

#include "lsp/lsp_util.h"
#include "lsp/protocol/common_types.h"
#include "tree_sitter/api.h"

// Declare the `tree_sitter_opengoal` function, which is
// implemented by the `tree-sitter-opengoal` library.
extern "C" {
extern const TSLanguage* tree_sitter_opengoal();
}

const TSLanguage* g_opengoalLang = tree_sitter_opengoal();

Workspace::Workspace(){};
Workspace::~Workspace(){};

bool Workspace::is_initialized() {
  return m_initialized;
};

void Workspace::set_initialized(bool new_value) {
  m_initialized = new_value;
}

Workspace::FileType Workspace::determine_filetype_from_languageid(const std::string& language_id) {
  if (language_id == "opengoal") {
    return FileType::OpenGOAL;
  } else if (language_id == "opengoal-ir") {
    return FileType::OpenGOALIR;
  }
  return FileType::Unsupported;
}

Workspace::FileType Workspace::determine_filetype_from_uri(const LSPSpec::DocumentUri& file_uri) {
  if (str_util::ends_with(file_uri, ".gc")) {
    return FileType::OpenGOAL;
  } else if (str_util::ends_with(file_uri, "ir2.asm")) {
    return FileType::OpenGOALIR;
  }
  return FileType::Unsupported;
}

std::optional<std::reference_wrapper<WorkspaceOGFile>> Workspace::get_tracked_og_file(
    const LSPSpec::URI& file_uri) {
  auto it = m_tracked_og_files.find(file_uri);
  if (it == m_tracked_og_files.end()) {
    return std::nullopt;
  }
  return std::ref(it->second);
}

std::optional<std::reference_wrapper<WorkspaceIRFile>> Workspace::get_tracked_ir_file(
    const LSPSpec::URI& file_uri) {
  auto it = m_tracked_ir_files.find(file_uri);
  if (it == m_tracked_ir_files.end()) {
    return std::nullopt;
  }
  return std::ref(it->second);
}

std::optional<DefinitionMetadata> Workspace::get_definition_info_from_all_types(
    const std::string& symbol_name,
    const LSPSpec::DocumentUri& all_types_uri) {
  if (m_tracked_all_types_files.count(all_types_uri) == 0) {
    return {};
  }
  const auto& dts = m_tracked_all_types_files[all_types_uri]->m_dts;
  if (dts->symbol_metadata_map.count(symbol_name) == 0) {
    return {};
  }
  return dts->symbol_metadata_map.at(symbol_name);
}

// TODO - a gross hack that should go away when the language isn't so tightly coupled to the jak
// games
//
// This is bad because jak 2 now uses some code from the jak1 folder, and also wouldn't be able to
// be determined (jak1 or jak2?) if we had a proper 'common' folder(s).
std::optional<GameVersion> Workspace::determine_game_version_from_uri(
    const LSPSpec::DocumentUri& uri) {
  const auto path = lsp_util::uri_to_path(uri);
  if (str_util::contains(path, "goal_src/jak1")) {
    return GameVersion::Jak1;
  } else if (str_util::contains(path, "goal_src/jak2")) {
    return GameVersion::Jak2;
  } else if (str_util::contains(path, "goal_src/jak3")) {
    return GameVersion::Jak3;
  }
  return {};
}

std::vector<symbol_info::SymbolInfo*> Workspace::get_symbols_starting_with(
    const GameVersion game_version,
    const std::string& symbol_prefix) {
  if (m_compiler_instances.find(game_version) == m_compiler_instances.end()) {
    lg::debug("Compiler not instantiated for game version - {}",
              version_to_game_name(game_version));
    return {};
  }
  const auto& compiler = m_compiler_instances[game_version].get();
  return compiler->lookup_symbol_info_by_prefix(symbol_prefix);
}

std::optional<symbol_info::SymbolInfo*> Workspace::get_global_symbol_info(
    const WorkspaceOGFile& file,
    const std::string& symbol_name) {
  if (m_compiler_instances.find(file.m_game_version) == m_compiler_instances.end()) {
    lg::debug("Compiler not instantiated for game version - {}",
              version_to_game_name(file.m_game_version));
    return {};
  }
  const auto& compiler = m_compiler_instances[file.m_game_version].get();
  const auto symbol_infos = compiler->lookup_exact_name_info(symbol_name);
  if (symbol_infos.empty()) {
    return {};
  } else if (symbol_infos.size() > 1) {
    // TODO - handle this (overriden methods is the main issue here)
    lg::debug("Found symbol info, but found multiple infos - {}", symbol_infos.size());
    return {};
  }
  const auto& symbol = symbol_infos.at(0);
  return symbol;
}

// TODO - consolidate what is needed into `SymbolInfo`
std::optional<std::pair<TypeSpec, Type*>> Workspace::get_symbol_typeinfo(
    const WorkspaceOGFile& file,
    const std::string& symbol_name) {
  if (m_compiler_instances.find(file.m_game_version) == m_compiler_instances.end()) {
    lg::debug("Compiler not instantiated for game version - {}",
              version_to_game_name(file.m_game_version));
    return {};
  }
  const auto& compiler = m_compiler_instances[file.m_game_version].get();
  const auto typespec = compiler->lookup_typespec(symbol_name);
  if (typespec) {
    // NOTE - for some reason calling with the symbol's typespec and the symbol itself produces
    // different results!
    const auto full_type_info = compiler->type_system().lookup_type_no_throw(symbol_name);
    if (full_type_info != nullptr) {
      return std::make_pair(typespec.value(), full_type_info);
    }
  }
  return {};
}

std::optional<symbol_info::DefinitionLocation> Workspace::get_symbol_def_location(
    const WorkspaceOGFile& /*file*/,
    const symbol_info::SymbolInfo* symbol_info) {
  const auto& def_loc = symbol_info->m_def_location;
  if (!def_loc) {
    return {};
  }
  return def_loc;
}

std::vector<std::tuple<std::string, std::string, Docs::DefinitionLocation>>
Workspace::get_symbols_parent_type_path(const std::string& symbol_name,
                                        const GameVersion game_version) {
  if (m_compiler_instances.find(game_version) == m_compiler_instances.end()) {
    lg::debug("Compiler not instantiated for game version - {}",
              version_to_game_name(game_version));
    return {};
  }

  // name, docstring, def_loc
  std::vector<std::tuple<std::string, std::string, Docs::DefinitionLocation>> parents = {};

  const auto& compiler = m_compiler_instances[game_version].get();
  const auto parent_path = compiler->type_system().get_path_up_tree(symbol_name);
  for (const auto& parent : parent_path) {
    const auto symbol_infos = compiler->lookup_exact_name_info(parent);
    if (symbol_infos.empty()) {
      continue;
    }
    symbol_info::SymbolInfo* symbol_info;
    if (symbol_infos.size() > 1) {
      for (const auto& info : symbol_infos) {
        if (info->m_kind == symbol_info::Kind::TYPE) {
          symbol_info = info;
        }
      }
    } else {
      symbol_info = symbol_infos.at(0);
    }
    if (!symbol_info) {
      continue;
    }
    const auto& def_loc = symbol_info->m_def_location;
    if (!def_loc) {
      continue;
    }
    Docs::DefinitionLocation new_def_loc;
    new_def_loc.filename = lsp_util::uri_from_path(def_loc->file_path);
    new_def_loc.line_idx = def_loc->line_idx;
    new_def_loc.char_idx = def_loc->char_idx;
    parents.push_back({parent, symbol_info->m_docstring, new_def_loc});
  }
  return parents;
}

std::vector<std::tuple<std::string, std::string, Docs::DefinitionLocation>>
Workspace::get_types_subtypes(const std::string& symbol_name, const GameVersion game_version) {
  if (m_compiler_instances.find(game_version) == m_compiler_instances.end()) {
    lg::debug("Compiler not instantiated for game version - {}",
              version_to_game_name(game_version));
    return {};
  }

  // name, docstring, def_loc
  std::vector<std::tuple<std::string, std::string, Docs::DefinitionLocation>> subtypes = {};

  const auto& compiler = m_compiler_instances[game_version].get();
  const auto subtype_names =
      compiler->type_system().search_types_by_parent_type_strict(symbol_name);
  for (const auto& subtype_name : subtype_names) {
    const auto symbol_infos = compiler->lookup_exact_name_info(subtype_name);
    if (symbol_infos.empty()) {
      continue;
    } else if (symbol_infos.size() > 1) {
      continue;
    }
    const auto& symbol_info = symbol_infos.at(0);
    const auto& def_loc = symbol_info->m_def_location;
    if (!def_loc) {
      continue;
    }
    Docs::DefinitionLocation new_def_loc;
    new_def_loc.filename = lsp_util::uri_from_path(def_loc->file_path);
    new_def_loc.line_idx = def_loc->line_idx;
    new_def_loc.char_idx = def_loc->char_idx;
    subtypes.push_back({subtype_name, symbol_info->m_docstring, new_def_loc});
  }
  return subtypes;
}

std::unordered_map<std::string, s64> Workspace::get_enum_entries(const std::string& enum_name,
                                                                 const GameVersion game_version) {
  if (m_compiler_instances.find(game_version) == m_compiler_instances.end()) {
    lg::debug("Compiler not instantiated for game version - {}",
              version_to_game_name(game_version));
    return {};
  }

  const auto& compiler = m_compiler_instances[game_version].get();
  const auto enum_info = compiler->type_system().try_enum_lookup(enum_name);
  if (!enum_info) {
    return {};
  }
  return enum_info->entries();
}

void Workspace::start_tracking_file(const LSPSpec::DocumentUri& file_uri,
                                    const std::string& language_id,
                                    const std::string& content) {
  if (language_id == "opengoal-ir") {
    lg::debug("new ir file - {}", file_uri);
    WorkspaceIRFile file(content);
    m_tracked_ir_files[file_uri] = file;
    if (!file.m_all_types_uri.empty()) {
      if (m_tracked_all_types_files.count(file.m_all_types_uri) == 0) {
        lg::debug("new all-types file - {}", file.m_all_types_uri);
        m_tracked_all_types_files[file.m_all_types_uri] = std::make_unique<WorkspaceAllTypesFile>(
            file.m_all_types_uri, file.m_game_version, file.m_all_types_file_path);
        m_tracked_all_types_files[file.m_all_types_uri]->parse_type_system();
      }
    }
  } else if (language_id == "opengoal") {
    if (m_tracked_og_files.find(file_uri) != m_tracked_og_files.end()) {
      lg::debug("Already tracking - {}", file_uri);
      return;
    }
    auto game_version = determine_game_version_from_uri(file_uri);
    if (!game_version) {
      lg::debug("Could not determine game version from path - {}", file_uri);
      return;
    }

    if (m_compiler_instances.find(*game_version) == m_compiler_instances.end()) {
      lg::debug(
          "first time encountering a OpenGOAL file for game version - {}, initializing a compiler",
          version_to_game_name(*game_version));
      const auto project_path =
          file_util::try_get_project_path_from_path(lsp_util::uri_to_path(file_uri));
      lg::debug("Detected project path - {}", project_path.value());
      if (!file_util::setup_project_path(project_path)) {
        lg::debug("unable to setup project path, not initializing a compiler");
        return;
      }
      const std::string progress_title =
          fmt::format("Compiling {}", version_to_game_name_external(game_version.value()));
      m_requester.send_progress_create_request(progress_title, "compiling project", -1);
      m_compiler_instances.emplace(game_version.value(),
                                   std::make_unique<Compiler>(game_version.value()));
      try {
        // TODO - this should happen on a separate thread so the LSP is not blocking during this
        // lengthy step
        // TODO - make this a setting (disable indexing)
        // TODO - ask water if there is a fancy way to reduce memory usage (disabling coloring,
        // etc?)
        m_compiler_instances.at(*game_version)
            ->run_front_end_on_string("(make-group \"all-code\")");
        m_requester.send_progress_finish_request(progress_title, "indexed");
      } catch (std::exception& e) {
        // TODO - If it fails, annotate errors (DIAGNOSTIC TODO)
        m_requester.send_progress_finish_request(progress_title, "failed");
        lg::debug("error when {}", progress_title);
      }
    }
    m_tracked_og_files.emplace(file_uri, WorkspaceOGFile(file_uri, content, *game_version));
    m_tracked_og_files[file_uri].update_symbols(
        m_compiler_instances.at(*game_version)
            ->lookup_symbol_info_by_file(lsp_util::uri_to_path(file_uri)));
  }
}

void Workspace::update_tracked_file(const LSPSpec::DocumentUri& file_uri,
                                    const std::string& content) {
  lg::debug("potentially updating - {}", file_uri);
  // Check if the file is already tracked or not, this is done because change events don't give
  // language details it's assumed you are keeping track of that!
  if (m_tracked_ir_files.find(file_uri) != m_tracked_ir_files.end()) {
    lg::debug("updating tracked IR file - {}", file_uri);
    WorkspaceIRFile file(content);
    m_tracked_ir_files[file_uri] = file;
    // There is the potential for the all-types to have changed, albeit this is probably never going
    // to happen
    if (!file.m_all_types_uri.empty() &&
        m_tracked_all_types_files.count(file.m_all_types_uri) == 0) {
      auto& all_types_file = m_tracked_all_types_files[file.m_all_types_uri];
      all_types_file->m_file_path = file.m_all_types_file_path;
      all_types_file->m_uri = file.m_all_types_uri;
      all_types_file->m_game_version = file.m_game_version;
      all_types_file->update_type_system();
    }
  } else if (m_tracked_all_types_files.find(file_uri) != m_tracked_all_types_files.end()) {
    lg::debug("updating tracked all types file - {}", file_uri);
    // If the all-types file has changed, re-parse it
    // NOTE - this assumes its still for the same game version!
    m_tracked_all_types_files[file_uri]->update_type_system();
  } else if (m_tracked_og_files.find(file_uri) != m_tracked_og_files.end()) {
    lg::debug("updating tracked OG file - {}", file_uri);
    m_tracked_og_files[file_uri].parse_content(content);
    // re-`ml` the file
    const auto game_version = m_tracked_og_files[file_uri].m_game_version;
    if (m_compiler_instances.find(game_version) == m_compiler_instances.end()) {
      lg::debug("No compiler initialized for - {}", version_to_game_name(game_version));
      return;
    }
  }
}

void Workspace::tracked_file_will_save(const LSPSpec::DocumentUri& file_uri) {
  lg::debug("file will be saved - {}", file_uri);
  if (m_tracked_og_files.find(file_uri) != m_tracked_og_files.end()) {
    // goalc is not an incremental compiler (yet) so I believe it will be a better UX
    // to re-compile on the file save, rather than as the user is typing
    const auto game_version = m_tracked_og_files[file_uri].m_game_version;
    if (m_compiler_instances.find(game_version) == m_compiler_instances.end()) {
      lg::debug("No compiler initialized for - {}", version_to_game_name(game_version));
      return;
    }
    CompilationOptions options;
    options.filename = lsp_util::uri_to_path(file_uri);
    // re-compile the file
    m_compiler_instances.at(game_version)->asm_file(options);
    // Update symbols for this specific file
    const auto symbol_infos =
        m_compiler_instances.at(game_version)->lookup_symbol_info_by_file(options.filename);
    m_tracked_og_files[file_uri].update_symbols(symbol_infos);
  }
}

// clang-format off
void Workspace::update_global_index(const GameVersion game_version) {
  // TODO - project wide indexing potentially (ie. finding references)
};
// clang-format on

void Workspace::stop_tracking_file(const LSPSpec::DocumentUri& file_uri) {
  m_tracked_ir_files.erase(file_uri);
  m_tracked_all_types_files.erase(file_uri);
  m_tracked_og_files.erase(file_uri);
}

WorkspaceOGFile::WorkspaceOGFile(const LSPSpec::DocumentUri& uri,
                                 const std::string& content,
                                 const GameVersion& game_version)
    : m_uri(uri), m_game_version(game_version), version(0) {
  const auto [line_count, line_ending] =
      file_util::get_majority_file_line_endings_and_count(content);
  m_line_count = line_count;
  m_line_ending = line_ending;
  lg::info("Added new OG file. {} symbols and {} diagnostics", m_symbols.size(),
           m_diagnostics.size());
  parse_content(content);
}

void WorkspaceOGFile::parse_content(const std::string& content) {
  m_content = content;
  auto parser = ts_parser_new();
  if (ts_parser_set_language(parser, g_opengoalLang)) {
    // Get the AST for the current state of the file
    // TODO - eventually, we should consider doing partial updates of the AST
    // but right now the LSP just receives the entire document so that's a larger change.
    m_ast.reset(ts_parser_parse_string(parser, NULL, m_content.c_str(), m_content.length()),
                TreeSitterTreeDeleter());
  }
  ts_parser_delete(parser);
}

void WorkspaceOGFile::update_symbols(const std::vector<symbol_info::SymbolInfo*>& symbol_infos) {
  m_symbols.clear();
  // TODO - sorting by definition location would be nice (maybe VSCode already does this?)
  for (const auto& symbol_info : symbol_infos) {
    LSPSpec::DocumentSymbol lsp_sym;
    lsp_sym.m_name = symbol_info->m_name;
    lsp_sym.m_detail = symbol_info->m_docstring;
    switch (symbol_info->m_kind) {
      case symbol_info::Kind::CONSTANT:
        lsp_sym.m_kind = LSPSpec::SymbolKind::Constant;
        break;
      case symbol_info::Kind::FUNCTION:
        lsp_sym.m_kind = LSPSpec::SymbolKind::Function;
        break;
      case symbol_info::Kind::GLOBAL_VAR:
        lsp_sym.m_kind = LSPSpec::SymbolKind::Variable;
        break;
      case symbol_info::Kind::MACRO:
        lsp_sym.m_kind = LSPSpec::SymbolKind::Operator;
        break;
      case symbol_info::Kind::METHOD:
        lsp_sym.m_name = fmt::format("{}::{}", symbol_info->m_type, symbol_info->m_name);
        lsp_sym.m_kind = LSPSpec::SymbolKind::Method;
        break;
      case symbol_info::Kind::TYPE:
        lsp_sym.m_kind = LSPSpec::SymbolKind::Class;
        break;
      default:
        lsp_sym.m_kind = LSPSpec::SymbolKind::Object;
        break;
    }
    if (symbol_info->m_def_location) {
      lsp_sym.m_range = LSPSpec::Range(symbol_info->m_def_location->line_idx,
                                       symbol_info->m_def_location->char_idx);
    } else {
      lsp_sym.m_range = LSPSpec::Range(0, 0);
    }
    // TODO - would be nice to make this accurate but we don't store that info yet
    lsp_sym.m_selectionRange = lsp_sym.m_range;
    if (symbol_info->m_kind == symbol_info::Kind::TYPE) {
      std::vector<LSPSpec::DocumentSymbol> type_symbols = {};
      for (const auto& field : symbol_info->m_type_fields) {
        LSPSpec::DocumentSymbol field_sym;
        field_sym.m_name = field.name;
        field_sym.m_detail = field.description;
        if (field.is_array) {
          field_sym.m_kind = LSPSpec::SymbolKind::Array;
        } else {
          field_sym.m_kind = LSPSpec::SymbolKind::Field;
        }
        // TODO - we don't store the line number for fields
        field_sym.m_range = lsp_sym.m_range;
        field_sym.m_selectionRange = lsp_sym.m_selectionRange;
        type_symbols.push_back(field_sym);
      }
      for (const auto& method : symbol_info->m_type_methods) {
        LSPSpec::DocumentSymbol method_sym;
        method_sym.m_name = method.name;
        method_sym.m_kind = LSPSpec::SymbolKind::Method;
        // TODO - we don't store the line number for fields
        method_sym.m_range = lsp_sym.m_range;
        method_sym.m_selectionRange = lsp_sym.m_selectionRange;
        type_symbols.push_back(method_sym);
      }
      for (const auto& state : symbol_info->m_type_states) {
        LSPSpec::DocumentSymbol state_sym;
        state_sym.m_name = state.name;
        state_sym.m_kind = LSPSpec::SymbolKind::Event;
        // TODO - we don't store the line number for fields
        state_sym.m_range = lsp_sym.m_range;
        state_sym.m_selectionRange = lsp_sym.m_selectionRange;
        type_symbols.push_back(state_sym);
      }
      lsp_sym.m_children = type_symbols;
    }
    m_symbols.push_back(lsp_sym);
  }
}

std::optional<std::string> WorkspaceOGFile::get_symbol_at_position(
    const LSPSpec::Position position) const {
  if (m_ast) {
    TSNode root_node = ts_tree_root_node(m_ast.get());
    TSNode found_node =
        ts_node_descendant_for_point_range(root_node, {position.m_line, position.m_character},
                                           {position.m_line, position.m_character});
    if (!ts_node_has_error(found_node)) {
      uint32_t start = ts_node_start_byte(found_node);
      uint32_t end = ts_node_end_byte(found_node);
      const std::string node_str = m_content.substr(start, end - start);
      lg::debug("AST SAP - {}", node_str);
      const std::string node_name = ts_node_type(found_node);
      if (node_name == "sym_name") {
        return node_str;
      }
    } else {
      // found_node = ts_node_child(found_node, 0);
      // TODO - maybe get this one (but check if has an error)
      return {};
    }
  }
  return {};
}

std::vector<OpenGOALFormResult> WorkspaceOGFile::search_for_forms_that_begin_with(
    std::vector<std::string> prefix) const {
  std::vector<OpenGOALFormResult> results = {};
  if (!m_ast) {
    return results;
  }

  TSNode root_node = ts_tree_root_node(m_ast.get());
  std::vector<TSNode> found_nodes = {};
  ast_util::search_for_forms_that_begin_with(m_content, root_node, prefix, found_nodes);

  for (const auto& node : found_nodes) {
    std::vector<std::string> tokens = {};
    for (size_t i = 0; i < ts_node_child_count(node); i++) {
      const auto child_node = ts_node_child(node, i);
      const auto contents = ast_util::get_source_code(m_content, child_node);
      tokens.push_back(contents);
    }
    const auto start_point = ts_node_start_point(node);
    const auto end_point = ts_node_end_point(node);
    results.push_back(
        {tokens, {start_point.row, start_point.column}, {end_point.row, end_point.column}});
  }

  return results;
}

WorkspaceIRFile::WorkspaceIRFile(const std::string& content) {
  const auto line_ending = file_util::get_majority_file_line_endings(content);
  m_lines = str_util::split_string(content, line_ending);

  bool in_opengoal_block = false;
  for (size_t i = 0; i < m_lines.size(); i++) {
    const auto& line = m_lines.at(i);
    if (m_all_types_uri == "") {
      find_all_types_path(line);
    }
    if (str_util::contains(line, ";;-*-OpenGOAL-Start-*-")) {
      in_opengoal_block = true;
    } else if (str_util::contains(line, ";;-*-OpenGOAL-End-*-")) {
      in_opengoal_block = false;
    }
    find_function_symbol(i, line);
    identify_diagnostics(i, line, in_opengoal_block);
  }

  lg::info("Added new IR file. {} lines with {} symbols and {} diagnostics", m_lines.size(),
           m_symbols.size(), m_diagnostics.size());
}

// This is kind of a hack, but to ensure consistency.  The file will reference the all-types.gc
// file it was generated with, this lets us accurately jump to the definition properly!
void WorkspaceIRFile::find_all_types_path(const std::string& line) {
  std::regex regex("; ALL_TYPES=(.*)=(.*)");
  std::smatch matches;

  if (std::regex_search(line, matches, regex)) {
    if (matches.size() == 3) {
      const auto& game_version = matches[1];
      const auto& all_types_path = matches[2];
      lg::debug("Found DTS Path - {} : {}", game_version.str(), all_types_path.str());
      auto all_types_uri = lsp_util::uri_from_path(fs::path(all_types_path.str()));
      lg::debug("DTS URI - {}", all_types_uri);
      if (valid_game_version(game_version.str())) {
        m_game_version = game_name_to_version(game_version.str());
        m_all_types_uri = all_types_uri;
        m_all_types_file_path = fs::path(all_types_path.str());
      } else {
        lg::error("Invalid game version, ignoring - {}", game_version.str());
      }
    }
  }
}

void WorkspaceIRFile::find_function_symbol(const uint32_t line_num_zero_based,
                                           const std::string& line) {
  std::regex regex("; \\.function (.*)");
  std::smatch matches;

  if (std::regex_search(line, matches, regex)) {
    // NOTE - assumes we can only find 1 function per line
    if (matches.size() == 2) {
      const auto& match = matches[1];
      lg::info("Adding Symbol - {}", match.str());
      LSPSpec::DocumentSymbol new_symbol;
      new_symbol.m_name = match.str();
      new_symbol.m_kind = LSPSpec::SymbolKind::Function;
      LSPSpec::Range symbol_range;
      symbol_range.m_start = {line_num_zero_based, 0};
      symbol_range.m_end = {line_num_zero_based, 0};  // NOTE - set on the next function
      new_symbol.m_range = symbol_range;
      LSPSpec::Range symbol_selection_range;
      symbol_selection_range.m_start = {line_num_zero_based, 0};
      symbol_selection_range.m_end = {line_num_zero_based, (uint32_t)line.length() - 1};
      new_symbol.m_selectionRange = symbol_selection_range;
      m_symbols.push_back(new_symbol);
    }
  }

  std::regex end_function("^;; \\.endfunction\\s*$");
  if (std::regex_match(line, end_function)) {
    // Set the previous symbols end-line
    if (!m_symbols.empty()) {
      m_symbols[m_symbols.size() - 1].m_range.m_end.m_line = line_num_zero_based - 1;
    }
  }
}

void WorkspaceIRFile::identify_diagnostics(const uint32_t line_num_zero_based,
                                           const std::string& line,
                                           const bool /*in_opengoal_block*/) {
  std::regex info_regex(";; INFO: (.*)");
  std::regex warn_regex(";; WARN: (.*)");
  std::regex error_regex(";; ERROR: (.*)");
  std::smatch info_matches;
  std::smatch warn_matches;
  std::smatch error_matches;

  LSPSpec::Range diag_range;
  diag_range.m_start = {line_num_zero_based, 0};
  diag_range.m_end = {line_num_zero_based, (uint32_t)line.length() - 1};

  // Check for an info-level warnings
  if (std::regex_search(line, info_matches, info_regex)) {
    // NOTE - assumes we can only find 1 function per line
    if (info_matches.size() == 2) {
      const auto& match = info_matches[1];
      LSPSpec::Diagnostic new_diag;
      new_diag.m_severity = LSPSpec::DiagnosticSeverity::Information;
      new_diag.m_message = match.str();
      new_diag.m_range = diag_range;
      new_diag.m_source = "OpenGOAL LSP";
      m_diagnostics.push_back(new_diag);
      return;
    }
  }
  // Check for a warn level warnings
  if (std::regex_search(line, warn_matches, warn_regex)) {
    // NOTE - assumes we can only find 1 function per line
    if (warn_matches.size() == 2) {
      const auto& match = warn_matches[1];
      LSPSpec::Diagnostic new_diag;
      new_diag.m_severity = LSPSpec::DiagnosticSeverity::Warning;
      new_diag.m_message = match.str();
      new_diag.m_range = diag_range;
      new_diag.m_source = "OpenGOAL LSP";
      m_diagnostics.push_back(new_diag);
      return;
    }
  }

  // Check for a error level warnings
  if (std::regex_search(line, error_matches, error_regex)) {
    // NOTE - assumes we can only find 1 function per line
    if (error_matches.size() == 2) {
      const auto& match = error_matches[1];
      LSPSpec::Diagnostic new_diag;
      new_diag.m_severity = LSPSpec::DiagnosticSeverity::Error;
      new_diag.m_message = match.str();
      new_diag.m_range = diag_range;
      new_diag.m_source = "OpenGOAL LSP";
      m_diagnostics.push_back(new_diag);
      return;
    }
  }
}

std::optional<std::string> WorkspaceIRFile::get_mips_instruction_at_position(
    const LSPSpec::Position position) const {
  // Split the line on typical word boundaries
  std::string line = m_lines.at(position.m_line);
  std::smatch matches;
  std::regex regex("[\\w\\.]+");

  if (std::regex_search(line, matches, regex)) {
    const auto& match = matches[0];
    auto match_start = matches.position(0);
    auto match_end = match_start + match.length();
    if (position.m_character >= match_start && position.m_character <= match_end) {
      return match;
    }
  }

  return {};
}

std::optional<std::string> WorkspaceIRFile::get_symbol_at_position(
    const LSPSpec::Position position) const {
  // Split the line on typical word boundaries
  std::string line = m_lines.at(position.m_line);
  std::smatch matches;
  std::regex regex("[\\w\\.\\-_!<>*]+");
  std::regex_token_iterator<std::string::iterator> rend;

  std::regex_token_iterator<std::string::iterator> match(line.begin(), line.end(), regex);
  while (match != rend) {
    auto match_start = std::distance(line.begin(), match->first);
    auto match_end = match_start + match->length();
    if (position.m_character >= match_start && position.m_character <= match_end) {
      return match->str();
    }
    match++;
  }

  return {};
}

void WorkspaceAllTypesFile::parse_type_system() {
  lg::debug("DTS Loading - '{}'", m_file_path.string());
  m_dts->parse_type_defs({m_file_path.string()});
  lg::debug("DTS Loaded At - '{}'", m_file_path.string());
}

void WorkspaceAllTypesFile::update_type_system() {
  m_dts = std::make_unique<decompiler::DecompilerTypeSystem>(m_game_version);
  parse_type_system();
}