Mountain/RPC/CocoonService/Provider/
ProvideCompletionItems.rs1use 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}