symbolique 0.1.0

Symbol table pipeline for language servers — parse, link, merge, and resolve symbols across files, built on the laburnum LSP framework.
// Copyright Two Neutron Stars Incorporated and contributors
// SPDX-License-Identifier: BlueOak-1.0.0

//! Linked-level symbol reader extension trait.
//!
//! This module provides [`LinkedSymboliqueReadExt`], for querying linked-level
//! symbols from `QueryClient`.

use {
  crate::{
    core::{Ident, SymbolPath, Value},
    partitions::{linked::LinkedSymbols, records::SymbolEntry, symbols::Symbols},
  },
  laburnum::database::{
    HasPartition, RecordRef,
    query::QueryClient,
    storage::Partitions,
  },
};

/// Extension trait for reading linked-level symbols from partitions.
///
/// Provides typed methods for querying symbols written during the linking
/// stage. Methods query the `LinkedSymbols` index and optionally follow handles
/// to the shared `Symbols` partition.
pub trait LinkedSymboliqueReadExt<P: Partitions> {
  /// Look up a linked symbol by exact sort key.
  fn linked_symbol_at_path<V, I, Path>(
    &self,
    sort_key: &str,
  ) -> Option<SymbolEntry<V, I, Path>>
  where
    V: Value<I>,
    I: Ident,
    Path: SymbolPath,
    P::Stores: HasPartition<LinkedSymbols<V, I, Path>>;

  /// Query linked symbols with a sort key prefix.
  fn linked_symbols_with_prefix<V, I, Path>(
    &self,
    prefix: &str,
  ) -> Vec<(String, SymbolEntry<V, I, Path>)>
  where
    V: Value<I>,
    I: Ident,
    Path: SymbolPath,
    P::Stores: HasPartition<LinkedSymbols<V, I, Path>>;

  /// Follow a symbol entry's handle to retrieve the full symbol shape.
  fn resolve_linked_symbol<V, I, Path>(
    &self,
    entry: &SymbolEntry<V, I, Path>,
  ) -> Option<RecordRef<'_, Symbols<V, I, Path>>>
  where
    V: Value<I> + 'static,
    I: Ident,
    Path: SymbolPath + 'static,
    P::Stores: HasPartition<Symbols<V, I, Path>>;
}

impl<P: Partitions> LinkedSymboliqueReadExt<P> for QueryClient<P> {
  fn linked_symbol_at_path<V, I, Path>(
    &self,
    sort_key: &str,
  ) -> Option<SymbolEntry<V, I, Path>>
  where
    V: Value<I>,
    I: Ident,
    Path: SymbolPath,
    P::Stores: HasPartition<LinkedSymbols<V, I, Path>>,
  {
    self.index_get::<LinkedSymbols<V, I, Path>>(sort_key)
  }

  fn linked_symbols_with_prefix<V, I, Path>(
    &self,
    prefix: &str,
  ) -> Vec<(String, SymbolEntry<V, I, Path>)>
  where
    V: Value<I>,
    I: Ident,
    Path: SymbolPath,
    P::Stores: HasPartition<LinkedSymbols<V, I, Path>>,
  {
    self.index_range::<LinkedSymbols<V, I, Path>>(prefix)
  }

  fn resolve_linked_symbol<V, I, Path>(
    &self,
    entry: &SymbolEntry<V, I, Path>,
  ) -> Option<RecordRef<'_, Symbols<V, I, Path>>>
  where
    V: Value<I> + 'static,
    I: Ident,
    Path: SymbolPath + 'static,
    P::Stores: HasPartition<Symbols<V, I, Path>>,
  {
    self.get::<Symbols<V, I, Path>>(entry.symbol.content_hash())
  }
}