Skip to content

Commit 943b01f

Browse files
zthzth
committed
start refactoring dot completion everywhere
1 parent 5ad0319 commit 943b01f

File tree

4 files changed

+439
-21
lines changed

4 files changed

+439
-21
lines changed

‎analysis/src/CompletionBackEnd.ml

Lines changed: 112 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -993,6 +993,105 @@ and getCompletionsForContextPath ~debug ~full ~opens ~rawOpens ~pos ~env ~exact
993993
path @ [fieldName]
994994
|> getCompletionsForPath ~debug ~opens ~full ~pos ~exact
995995
~completionContext:Field ~env ~scope
996+
| CPField {contextPath = cp; fieldName; fieldNameLoc} when Debug.verbose () ->
997+
(* TODO: this should only happen when the dot completion is at the end of the path *)
998+
if Debug.verbose () then print_endline "[ctx_path]--> dot completion!";
999+
let completionsForCtxPath =
1000+
cp
1001+
|> getCompletionsForContextPath ~debug ~full ~opens ~rawOpens ~pos ~env
1002+
~exact:true ~scope
1003+
|> completionsGetTypeEnv2 ~debug ~full ~opens ~rawOpens ~pos
1004+
in
1005+
(* These are the main completions for the dot. *)
1006+
let mainCompletions =
1007+
match completionsForCtxPath with
1008+
| Some (typ, env)
1009+
when typ |> TypeUtils.extractObjectType ~env ~package |> Option.is_some
1010+
->
1011+
(* Handle obj completion via dot *)
1012+
if Debug.verbose () then
1013+
Printf.printf "[dot_completion]--> Obj type found:\n";
1014+
let objEnv, obj =
1015+
typ |> TypeUtils.extractObjectType ~env ~package |> Option.get
1016+
in
1017+
obj |> TypeUtils.getObjFields
1018+
|> Utils.filterMap (fun (field, typ) ->
1019+
if Utils.checkName field ~prefix:fieldName ~exact then
1020+
let fullObjFieldName = Printf.sprintf "[\"%s\"]" field in
1021+
Some
1022+
(Completion.create fullObjFieldName ~range:fieldNameLoc
1023+
~insertText:fullObjFieldName ~env:objEnv
1024+
~kind:(Completion.ObjLabel typ))
1025+
else None)
1026+
| Some (typ, env)
1027+
when typ |> TypeUtils.extractRecordType ~env ~package |> Option.is_some
1028+
->
1029+
let env, fields, decl, _path, _attributes =
1030+
typ |> TypeUtils.extractRecordType ~env ~package |> Option.get
1031+
in
1032+
if Debug.verbose () then
1033+
Printf.printf "[dot_completion]--> Record type found\n";
1034+
let recordAsString =
1035+
decl.item.decl |> Shared.declToString decl.name.txt
1036+
in
1037+
fields
1038+
|> Utils.filterMap (fun field ->
1039+
if Utils.checkName field.fname.txt ~prefix:fieldName ~exact then
1040+
Some
1041+
(Completion.create field.fname.txt ~env
1042+
?deprecated:field.deprecated ~docstring:field.docstring
1043+
~kind:(Completion.Field (field, recordAsString)))
1044+
else None)
1045+
| Some (_typ, _env) ->
1046+
(* No more primary completions, for now. *)
1047+
[]
1048+
| None -> []
1049+
in
1050+
let pipeCompletions =
1051+
match completionsForCtxPath with
1052+
| None -> []
1053+
| Some (typ, envFromCompletionItem) -> (
1054+
let tPath = TypeUtils.pathFromTypeExpr typ in
1055+
match tPath with
1056+
| None -> []
1057+
| Some tPath ->
1058+
let completionPath =
1059+
(tPath |> Utils.expandPath |> List.tl |> List.rev)
1060+
@ (envFromCompletionItem.pathRev |> List.rev)
1061+
in
1062+
if List.length completionPath = 0 then []
1063+
else
1064+
let completions =
1065+
completionsForPipeFromCompletionPath ~envCompletionIsMadeFrom
1066+
~opens ~pos ~scope ~debug ~prefix:fieldName ~env ~rawOpens ~full
1067+
completionPath
1068+
in
1069+
completions
1070+
|> TypeUtils.filterPipeableFunctions ~env ~full
1071+
~lastPath:(Path.last tPath) ~replaceRange:fieldNameLoc)
1072+
in
1073+
(* Extra completions from configure extra module(s) *)
1074+
let extraCompletions =
1075+
match completionsForCtxPath with
1076+
| None -> []
1077+
| Some (typ, envFromCompletionItem) -> (
1078+
match
1079+
TypeUtils.getExtraModuleToCompleteFromForType typ
1080+
~env:envFromCompletionItem ~full
1081+
with
1082+
| None -> []
1083+
| Some completionPath ->
1084+
completionsForPipeFromCompletionPath ~envCompletionIsMadeFrom ~opens
1085+
~pos ~scope ~debug ~prefix:fieldName ~env ~rawOpens ~full
1086+
completionPath
1087+
|> TypeUtils.filterPipeableFunctions ~env ~full
1088+
~replaceRange:fieldNameLoc
1089+
?lastPath:
1090+
(match TypeUtils.pathFromTypeExpr typ with
1091+
| None -> None
1092+
| Some tPath -> Some (Path.last tPath)))
1093+
in
1094+
mainCompletions @ pipeCompletions @ extraCompletions
9961095
| CPField {contextPath = cp; fieldName; fieldNameLoc} -> (
9971096
if Debug.verbose () then print_endline "[ctx_path]--> CPField";
9981097
let completionsForCtxPath =
@@ -1056,8 +1155,8 @@ and getCompletionsForContextPath ~debug ~full ~opens ~rawOpens ~pos ~env ~exact
10561155
completionsForPipeFromCompletionPath ~opens ~pos ~scope ~debug
10571156
~prefix:fieldName ~envCompletionIsMadeFrom:env ~env:envFromExtracted
10581157
~rawOpens ~full completionPath
1059-
|> TypeUtils.filterPipeableFunctions ~env:envFromExtracted ~full ~path
1060-
~replaceRange:fieldNameLoc
1158+
|> TypeUtils.filterPipeableFunctions ~env:envFromExtracted ~full
1159+
~lastPath:(Path.last path) ~replaceRange:fieldNameLoc
10611160
| None -> []
10621161
in
10631162
pipeCompletionsForModule
@@ -1081,16 +1180,7 @@ and getCompletionsForContextPath ~debug ~full ~opens ~rawOpens ~pos ~env ~exact
10811180
| Some (typ, env) -> (
10821181
match typ |> TypeUtils.extractObjectType ~env ~package with
10831182
| Some (env, tObj) ->
1084-
let rec getFields (texp : Types.type_expr) =
1085-
match texp.desc with
1086-
| Tfield (name, _, t1, t2) ->
1087-
let fields = t2 |> getFields in
1088-
(name, t1) :: fields
1089-
| Tlink te | Tsubst te | Tpoly (te, []) -> te |> getFields
1090-
| Tvar None -> []
1091-
| _ -> []
1092-
in
1093-
tObj |> getFields
1183+
tObj |> TypeUtils.getObjFields
10941184
|> Utils.filterMap (fun (field, typ) ->
10951185
if Utils.checkName field ~prefix:label ~exact then
10961186
Some
@@ -1175,15 +1265,23 @@ and getCompletionsForContextPath ~debug ~full ~opens ~rawOpens ~pos ~env ~exact
11751265
completionsForPipeFromCompletionPath ~envCompletionIsMadeFrom ~opens
11761266
~pos ~scope ~debug ~prefix:funNamePrefix ~env ~rawOpens ~full
11771267
completionPath
1178-
|> TypeUtils.filterPipeableFunctions ~env ~full ?path:tPath
1268+
|> TypeUtils.filterPipeableFunctions ~env ~full
1269+
?lastPath:
1270+
(match tPath with
1271+
| None -> None
1272+
| Some tPath -> Some (Path.last tPath))
11791273
in
11801274
match completionPath with
11811275
| Some completionPath -> (
11821276
let completionsFromMainFn =
11831277
completionsForPipeFromCompletionPath ~envCompletionIsMadeFrom ~opens
11841278
~pos ~scope ~debug ~prefix:funNamePrefix ~env ~rawOpens ~full
11851279
completionPath
1186-
|> TypeUtils.filterPipeableFunctions ~env ~full ?path:tPath
1280+
|> TypeUtils.filterPipeableFunctions ~env ~full
1281+
?lastPath:
1282+
(match tPath with
1283+
| None -> None
1284+
| Some tPath -> Some (Path.last tPath))
11871285
in
11881286
let completions = completionsFromMainFn @ completionsFromExtraModule in
11891287
(* We add React element functions to the completion if we're in a JSX context *)

‎analysis/src/TypeUtils.ml

Lines changed: 16 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1159,13 +1159,13 @@ let getExtraModuleToCompleteFromForType ~env ~full (t : Types.type_expr) =
11591159

11601160
(** Checks whether the provided type represents a function that takes the provided path
11611161
as the first argument (meaning it's pipeable). *)
1162-
let rec fnTakesTypeAsFirstArg ~env ~full ~path t =
1162+
let rec fnTakesTypeAsFirstArg ~env ~full ~lastPath t =
11631163
match t.Types.desc with
11641164
| Tlink t1
11651165
| Tsubst t1
11661166
| Tpoly (t1, [])
11671167
| Tconstr (Pident {name = "function$"}, [t1; _], _) ->
1168-
fnTakesTypeAsFirstArg ~env ~full ~path t1
1168+
fnTakesTypeAsFirstArg ~env ~full ~lastPath t1
11691169
| Tarrow _ -> (
11701170
match extractFunctionType ~env ~package:full.package t with
11711171
| (Nolabel, t) :: _, _ -> (
@@ -1181,7 +1181,7 @@ let rec fnTakesTypeAsFirstArg ~env ~full ~path t =
11811181
Therefore, we can safely pluck out just the last part of the `path`, but need to use the entire name of the current type
11821182
we're comparing with.
11831183
*)
1184-
Path.name p = Path.last path || Path.name p = "t")
1184+
Path.name p = lastPath || Path.name p = "t")
11851185
| _ -> false)
11861186
| _ -> false
11871187

@@ -1201,14 +1201,14 @@ let transformCompletionToPipeCompletion ~env ~replaceRange
12011201
}
12021202

12031203
(** Filters out completions that are not pipeable from a list of completions. *)
1204-
let filterPipeableFunctions ~env ~full ?path ?replaceRange completions =
1205-
match path with
1204+
let filterPipeableFunctions ~env ~full ?lastPath ?replaceRange completions =
1205+
match lastPath with
12061206
| None -> completions
1207-
| Some path ->
1207+
| Some lastPath ->
12081208
completions
12091209
|> List.filter_map (fun (completion : Completion.t) ->
12101210
match completion.kind with
1211-
| Value t when fnTakesTypeAsFirstArg ~env ~full ~path t -> (
1211+
| Value t when fnTakesTypeAsFirstArg ~env ~full ~lastPath t -> (
12121212
match replaceRange with
12131213
| None -> Some completion
12141214
| Some replaceRange ->
@@ -1222,3 +1222,12 @@ let removeCurrentModuleIfNeeded ~envCompletionIsMadeFrom completionPath =
12221222
&& List.hd completionPath = envCompletionIsMadeFrom.QueryEnv.file.moduleName
12231223
then List.tl completionPath
12241224
else completionPath
1225+
1226+
let rec getObjFields (texp : Types.type_expr) =
1227+
match texp.desc with
1228+
| Tfield (name, _, t1, t2) ->
1229+
let fields = t2 |> getObjFields in
1230+
(name, t1) :: fields
1231+
| Tlink te | Tsubst te | Tpoly (te, []) -> te |> getObjFields
1232+
| Tvar None -> []
1233+
| _ -> []

‎analysis/tests/src/DotCompletionEverywhere.res

Lines changed: 65 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,65 @@
1+
let someObj = {
2+
"name": "hello",
3+
"age": 123,
4+
}
5+
// ^dv+
6+
// someObj.
7+
// ^com
8+
9+
// someObj.na
10+
// ^com
11+
12+
type rrr = {name: string}
13+
let rrr = {name: "hello"}
14+
15+
// rrr.n
16+
// ^com
17+
18+
module SomeMod = {
19+
module SomeOtherMod = {
20+
type x
21+
22+
@send external do: x => unit = "do"
23+
24+
external xx: x = "xx"
25+
}
26+
}
27+
28+
external x: SomeMod.SomeOtherMod.x = "x"
29+
30+
// x.
31+
// ^com
32+
33+
// SomeMod.SomeOtherMod.xx.
34+
// ^com
35+
36+
module Sss = {
37+
type rrr = {name: string}
38+
let rrr = {name: "hello"}
39+
let do = rrr => rrr.name
40+
}
41+
42+
// Sss.rrr.
43+
// ^com
44+
45+
@editor.completeFrom(DotCompletionEverywhere.X2)
46+
type x2x2 = {namee: string}
47+
let x2x2 = {namee: "hello"}
48+
49+
module X2 = {
50+
let stuff = x => x.namee
51+
}
52+
53+
// x2x2.
54+
// ^com
55+
56+
let obj = {
57+
"name": "ReScript",
58+
"number": 1,
59+
"nothing": true,
60+
}
61+
62+
// obj.
63+
// ^com
64+
65+
// ^dv-

0 commit comments

Comments
 (0)