Skip to main content

Mountain/RPC/CocoonService/Provider/
ProvideDefinition.rs

1//! Resolve "go to definition" via the registered provider, mapping each
2//! result location into the gRPC `Location` shape.
3
4use tonic::{Response, Status};
5use url::Url;
6use CommonLibrary::LanguageFeature::{
7	DTO::PositionDTO::PositionDTO,
8	LanguageFeatureProviderRegistry::LanguageFeatureProviderRegistry,
9};
10use ::Vine::Generated::{Location, Position, ProvideDefinitionRequest, ProvideDefinitionResponse, Range, Uri};
11
12use crate::{RPC::CocoonService::CocoonServiceImpl, dev_log};
13
14pub async fn Fn(
15	Service:&CocoonServiceImpl,
16
17	Request:ProvideDefinitionRequest,
18) -> Result<Response<ProvideDefinitionResponse>, Status> {
19	dev_log!(
20		"cocoon",
21		"[CocoonService] Providing definition for provider {}",
22		Request.provider_handle
23	);
24
25	let URI = Request.uri.as_ref().map(|U| U.value.as_str()).unwrap_or("");
26
27	let DocumentURI = Url::parse(URI).map_err(|E| Status::invalid_argument(format!("Invalid URI: {}", E)))?;
28
29	let Position_ = Request.position.as_ref();
30
31	let PositionDTO_ = PositionDTO {
32		LineNumber:Position_.map(|P| P.line).unwrap_or(0),
33
34		Column:Position_.map(|P| P.character).unwrap_or(0),
35	};
36
37	match Service.environment.ProvideDefinition(DocumentURI, PositionDTO_).await {
38		Ok(Some(Locations)) => {
39			let Mapped = Locations
40				.iter()
41				.map(|Loc| {
42					Location {
43						uri:Some(Uri { value:Loc.Uri.to_string() }),
44						range:Some(Range {
45							start:Some(Position { line:Loc.Range.StartLineNumber, character:Loc.Range.StartColumn }),
46							end:Some(Position { line:Loc.Range.EndLineNumber, character:Loc.Range.EndColumn }),
47						}),
48					}
49				})
50				.collect();
51
52			Ok(Response::new(ProvideDefinitionResponse { locations:Mapped }))
53		},
54
55		Ok(None) => Ok(Response::new(ProvideDefinitionResponse { locations:Vec::new() })),
56
57		Err(Error) => Err(Status::internal(format!("Definition failed: {}", Error))),
58	}
59}