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