language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-27 13:59:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Add an AbilityResolver trait for abilities stores and views

Browse source
This commit is contained in:
Ayaz Hafiz 2022-07-27 12:54:13 -04:00
parent 6a4a3e2c60
commit b72c8705e5
No known key found for this signature in database
GPG key ID: 0E2A37416A25EF58
5 changed files with 150 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

23
crates/compiler/late_solve/src/lib.rs
View file

@ -10,6 +10,7 @@ use roc_collections::MutMap;
use roc_derive::SharedDerivedModule;
use roc_error_macros::internal_error;
use roc_module::symbol::ModuleId;
use roc_solve::ability::AbilityResolver;
use roc_solve::solve::Pools;
use roc_solve::specialize::{compact_lambda_sets_of_vars, DerivedEnv, Phase};
use roc_types::subs::{get_member_lambda_sets_at_region, Content, FlatType, LambdaSet};
@ -97,6 +98,28 @@ impl AbilitiesView<'_> {
}
}
impl<'a> AbilityResolver for AbilitiesView<'a> {
fn member_parent_and_signature_var(
&self,
ability_member: roc_module::symbol::Symbol,
) -> Option<(roc_module::symbol::Symbol, Variable)> {
self.with_module_abilities_store(ability_member.module_id(), |store| {
store
.member_def(ability_member)
.map(|def| (def.parent_ability, def.signature_var()))
})
}
fn get_implementation(
&self,
impl_key: roc_can::abilities::ImplKey,
) -> Option<roc_types::types::MemberImpl> {
self.with_module_abilities_store(impl_key.opaque.module_id(), |store| {
store.get_implementation(impl_key).copied()
})
}
}
pub struct LatePhase<'a> {
home: ModuleId,
abilities: &'a AbilitiesView<'a>,