Skip to main content

Mountain/RPC/CocoonService/Provider/
ProvideCompletionItems.rs

1//! Forward a completion request to the registered provider and project
2//! the suggestions into the gRPC `CompletionItem` shape.
3
4use tonic::{Response, Status};
5use url::Url;
6use CommonLibrary::LanguageFeature::{
7	DTO::{
8		CompletionContextDTO::{CompletionContextDTO, CompletionTriggerKindDTO},
9		PositionDTO::PositionDTO,
10	},
11	LanguageFeatureProviderRegistry::LanguageFeatureProviderRegistry,
12};
13use ::Vine::Generated::{CompletionItem, ProvideCompletionItemsRequest, ProvideCompletionItemsResponse};
14
15use crate::{RPC::CocoonService::CocoonServiceImpl, dev_log};
16
17pub async fn Fn(
18	Service:&CocoonServiceImpl,
19
20	Request:ProvideCompletionItemsRequest,
21) -> Result<Response<ProvideCompletionItemsResponse>, Status> {
22	dev_log!(
23		"cocoon",
24		"[CocoonService] Providing completions for provider {}",
25		Request.provider_handle
26	);
27
28	let URI = Request.uri.as_ref().map(|U| U.value.as_str()).unwrap_or("");
29
30	let DocumentURI = Url::parse(URI).map_err(|E| Status::invalid_argument(format!("Invalid URI: {}", E)))?;
31
32	let Position_ = Request.position.as_ref();
33
34	let PositionDTO_ = PositionDTO {
35		LineNumber:Position_.map(|P| P.line).unwrap_or(0),
36
37		Column:Position_.map(|P| P.character).unwrap_or(0),
38	};
39
40	let ContextDTO = CompletionContextDTO {
41		TriggerKind:CompletionTriggerKindDTO::Invoke,
42
43		TriggerCharacter:if Request.trigger_character.is_empty() {
44			None
45		} else {
46			Some(Request.trigger_character.clone())
47		},
48	};
49
50	match Service
51		.environment
52		.ProvideCompletions(DocumentURI, PositionDTO_, ContextDTO, None)
53		.await
54	{
55		Ok(Some(List)) => {
56			let Items = List
57				.Suggestions
58				.iter()
59				.map(|S| {
60					CompletionItem {
61						label:S.Label.as_str().map(|L| L.to_string()).unwrap_or_default(),
62						kind:format!("{}", S.Kind),
63						detail:S.Detail.clone().unwrap_or_default(),
64						documentation:Vec::new(),
65						insert_text:S.InsertText.as_ref().and_then(|V| V.as_str()).unwrap_or("").to_string(),
66					}
67				})
68				.collect();
69
70			Ok(Response::new(ProvideCompletionItemsResponse { items:Items }))
71		},
72
73		Ok(None) => Ok(Response::new(ProvideCompletionItemsResponse { items:Vec::new() })),
74
75		Err(Error) => Err(Status::internal(format!("Completions failed: {}", Error))),
76	}
77}