blob: 394589b02b18f18a13c03159cc45d679e8fe33d2 [file] [log] [blame]
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001/* vi:set ts=8 sts=4 sw=4 noet:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * vim9class.c: Vim9 script class support
12 */
13
14#define USING_FLOAT_STUFF
15#include "vim.h"
16
17#if defined(FEAT_EVAL) || defined(PROTO)
18
19// When not generating protos this is included in proto.h
20#ifdef PROTO
21# include "vim9.h"
22#endif
23
Yegappan Lakshmanane651e112023-09-04 07:51:01 +020024static class_T *first_class = NULL;
25static class_T *next_nonref_class = NULL;
26
27/*
28 * Call this function when a class has been created. It will be added to the
29 * list headed by "first_class".
30 */
31 static void
32class_created(class_T *cl)
33{
34 if (first_class != NULL)
35 {
36 cl->class_next_used = first_class;
37 first_class->class_prev_used = cl;
38 }
39 first_class = cl;
40}
41
42/*
43 * Call this function when a class has been cleared and is about to be freed.
44 * It is removed from the list headed by "first_class".
45 */
46 static void
47class_cleared(class_T *cl)
48{
49 if (cl->class_next_used != NULL)
50 cl->class_next_used->class_prev_used = cl->class_prev_used;
51 if (cl->class_prev_used != NULL)
52 cl->class_prev_used->class_next_used = cl->class_next_used;
53 else if (first_class == cl)
54 first_class = cl->class_next_used;
55
56 // update the next class to check if needed
57 if (cl == next_nonref_class)
58 next_nonref_class = cl->class_next_used;
59}
60
Bram Moolenaarc1c365c2022-12-04 20:13:24 +000061/*
Bram Moolenaard505d172022-12-18 21:42:55 +000062 * Parse a member declaration, both object and class member.
63 * Returns OK or FAIL. When OK then "varname_end" is set to just after the
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +020064 * variable name and "type_ret" is set to the declared or detected type.
Bram Moolenaard505d172022-12-18 21:42:55 +000065 * "init_expr" is set to the initialisation expression (allocated), if there is
Bram Moolenaar554d0312023-01-05 19:59:18 +000066 * one. For an interface "init_expr" is NULL.
Bram Moolenaard505d172022-12-18 21:42:55 +000067 */
68 static int
69parse_member(
Yegappan Lakshmananeb91e242023-08-31 18:10:46 +020070 exarg_T *eap,
71 char_u *line,
72 char_u *varname,
Yegappan Lakshmanan3775f772023-09-01 22:05:45 +020073 int has_public, // TRUE if "public" seen before "varname"
Yegappan Lakshmananeb91e242023-08-31 18:10:46 +020074 char_u **varname_end,
Yegappan Lakshmanan3775f772023-09-01 22:05:45 +020075 garray_T *type_list,
Yegappan Lakshmananeb91e242023-08-31 18:10:46 +020076 type_T **type_ret,
77 char_u **init_expr)
Bram Moolenaard505d172022-12-18 21:42:55 +000078{
79 *varname_end = to_name_end(varname, FALSE);
80 if (*varname == '_' && has_public)
81 {
82 semsg(_(e_public_member_name_cannot_start_with_underscore_str), line);
83 return FAIL;
84 }
85
86 char_u *colon = skipwhite(*varname_end);
87 char_u *type_arg = colon;
88 type_T *type = NULL;
89 if (*colon == ':')
90 {
91 if (VIM_ISWHITE(**varname_end))
92 {
93 semsg(_(e_no_white_space_allowed_before_colon_str), varname);
94 return FAIL;
95 }
96 if (!VIM_ISWHITE(colon[1]))
97 {
98 semsg(_(e_white_space_required_after_str_str), ":", varname);
99 return FAIL;
100 }
101 type_arg = skipwhite(colon + 1);
102 type = parse_type(&type_arg, type_list, TRUE);
103 if (type == NULL)
104 return FAIL;
105 }
106
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +0200107 char_u *init_arg = skipwhite(type_arg);
108 if (type == NULL && *init_arg != '=')
Bram Moolenaard505d172022-12-18 21:42:55 +0000109 {
110 emsg(_(e_type_or_initialization_required));
111 return FAIL;
112 }
113
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +0200114 if (init_expr == NULL && *init_arg == '=')
Bram Moolenaard505d172022-12-18 21:42:55 +0000115 {
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +0200116 emsg(_(e_cannot_initialize_member_in_interface));
117 return FAIL;
118 }
119
120 if (*init_arg == '=')
121 {
122 evalarg_T evalarg;
123 char_u *expr_start, *expr_end;
124
125 if (!VIM_ISWHITE(init_arg[-1]) || !VIM_ISWHITE(init_arg[1]))
Bram Moolenaard505d172022-12-18 21:42:55 +0000126 {
127 semsg(_(e_white_space_required_before_and_after_str_at_str),
128 "=", type_arg);
129 return FAIL;
130 }
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +0200131 init_arg = skipwhite(init_arg + 1);
Bram Moolenaard505d172022-12-18 21:42:55 +0000132
Bram Moolenaard505d172022-12-18 21:42:55 +0000133 fill_evalarg_from_eap(&evalarg, eap, FALSE);
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +0200134 (void)skip_expr_concatenate(&init_arg, &expr_start, &expr_end, &evalarg);
Bram Moolenaard505d172022-12-18 21:42:55 +0000135
Yegappan Lakshmanand4e4ecb2023-08-27 18:35:45 +0200136 // No type specified for the member. Set it to "any" and the correct
137 // type will be set when the object is instantiated.
Bram Moolenaard505d172022-12-18 21:42:55 +0000138 if (type == NULL)
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +0200139 type = &t_any;
Bram Moolenaard505d172022-12-18 21:42:55 +0000140
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +0200141 *init_expr = vim_strnsave(expr_start, expr_end - expr_start);
142 // Free the memory pointed by expr_start.
Bram Moolenaard505d172022-12-18 21:42:55 +0000143 clear_evalarg(&evalarg, NULL);
144 }
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +0200145 else if (!valid_declaration_type(type))
Bram Moolenaard505d172022-12-18 21:42:55 +0000146 return FAIL;
147
148 *type_ret = type;
Bram Moolenaard505d172022-12-18 21:42:55 +0000149 return OK;
150}
151
152/*
153 * Add a member to an object or a class.
154 * Returns OK when successful, "init_expr" will be consumed then.
155 * Returns FAIL otherwise, caller might need to free "init_expr".
156 */
157 static int
158add_member(
Yegappan Lakshmanan3775f772023-09-01 22:05:45 +0200159 garray_T *gap,
160 char_u *varname,
161 char_u *varname_end,
162 int has_public,
163 type_T *type,
164 char_u *init_expr)
Bram Moolenaard505d172022-12-18 21:42:55 +0000165{
166 if (ga_grow(gap, 1) == FAIL)
167 return FAIL;
168 ocmember_T *m = ((ocmember_T *)gap->ga_data) + gap->ga_len;
169 m->ocm_name = vim_strnsave(varname, varname_end - varname);
=?UTF-8?q?Ola=20S=C3=B6der?=d8742472023-03-05 13:12:32 +0000170 m->ocm_access = has_public ? VIM_ACCESS_ALL
171 : *varname == '_' ? VIM_ACCESS_PRIVATE : VIM_ACCESS_READ;
Bram Moolenaard505d172022-12-18 21:42:55 +0000172 m->ocm_type = type;
173 if (init_expr != NULL)
174 m->ocm_init = init_expr;
175 ++gap->ga_len;
176 return OK;
177}
178
179/*
180 * Move the class or object members found while parsing a class into the class.
181 * "gap" contains the found members.
Bram Moolenaar83677162023-01-08 19:54:10 +0000182 * "parent_members" points to the members in the parent class (if any)
183 * "parent_count" is the number of members in the parent class
Bram Moolenaard505d172022-12-18 21:42:55 +0000184 * "members" will be set to the newly allocated array of members and
185 * "member_count" set to the number of members.
186 * Returns OK or FAIL.
187 */
188 static int
189add_members_to_class(
190 garray_T *gap,
Bram Moolenaar83677162023-01-08 19:54:10 +0000191 ocmember_T *parent_members,
192 int parent_count,
Bram Moolenaard505d172022-12-18 21:42:55 +0000193 ocmember_T **members,
194 int *member_count)
195{
Bram Moolenaar83677162023-01-08 19:54:10 +0000196 *member_count = parent_count + gap->ga_len;
197 *members = *member_count == 0 ? NULL
198 : ALLOC_MULT(ocmember_T, *member_count);
199 if (*member_count > 0 && *members == NULL)
Bram Moolenaard505d172022-12-18 21:42:55 +0000200 return FAIL;
Bram Moolenaar83677162023-01-08 19:54:10 +0000201 for (int i = 0; i < parent_count; ++i)
202 {
203 // parent members need to be copied
Bram Moolenaarae3205a2023-01-15 20:49:00 +0000204 ocmember_T *m = *members + i;
205 *m = parent_members[i];
206 m->ocm_name = vim_strsave(m->ocm_name);
207 if (m->ocm_init != NULL)
208 m->ocm_init = vim_strsave(m->ocm_init);
Bram Moolenaar83677162023-01-08 19:54:10 +0000209 }
Bram Moolenaar8efdcee2022-12-19 12:18:09 +0000210 if (gap->ga_len > 0)
Bram Moolenaar83677162023-01-08 19:54:10 +0000211 // new members are moved
212 mch_memmove(*members + parent_count,
213 gap->ga_data, sizeof(ocmember_T) * gap->ga_len);
Bram Moolenaard505d172022-12-18 21:42:55 +0000214 VIM_CLEAR(gap->ga_data);
215 return OK;
216}
217
218/*
Bram Moolenaar29ac5df2023-01-16 19:43:47 +0000219 * Convert a member index "idx" of interface "itf" to the member index of class
220 * "cl" implementing that interface.
221 */
222 int
Ernie Rael18143d32023-09-04 22:30:41 +0200223object_index_from_itf_index(class_T *itf, int is_method, int idx, class_T *cl,
224 int is_static)
Bram Moolenaar29ac5df2023-01-16 19:43:47 +0000225{
Ernie Rael18143d32023-09-04 22:30:41 +0200226 if (idx >= (is_method ? itf->class_obj_method_count
227 : is_static ? itf->class_class_member_count
Bram Moolenaard0200c82023-01-28 15:19:40 +0000228 : itf->class_obj_member_count))
Bram Moolenaar29ac5df2023-01-16 19:43:47 +0000229 {
230 siemsg("index %d out of range for interface %s", idx, itf->class_name);
231 return 0;
232 }
Yegappan Lakshmanan74cc13c2023-08-13 17:41:26 +0200233
234 // If "cl" is the interface or the class that is extended, then the method
235 // index can be used directly and there is no need to search for the method
236 // index in one of the child classes.
237 if (cl == itf)
238 return idx;
239
Yegappan Lakshmanancc0bcf42023-09-08 19:12:03 +0200240 itf2class_T *i2c = NULL;
241 int searching = TRUE;
242 int method_offset = 0;
243
Ernie Raelcf138d42023-09-06 20:45:03 +0200244 for (class_T *super = cl; super != NULL && searching;
245 super = super->class_extends)
Yegappan Lakshmanancc0bcf42023-09-08 19:12:03 +0200246 {
Ernie Raelcf138d42023-09-06 20:45:03 +0200247 for (i2c = itf->class_itf2class; i2c != NULL; i2c = i2c->i2c_next)
Yegappan Lakshmanancc0bcf42023-09-08 19:12:03 +0200248 {
Ernie Raelcf138d42023-09-06 20:45:03 +0200249 if (i2c->i2c_class == super && i2c->i2c_is_method == is_method)
250 {
251 searching = FALSE;
252 break;
253 }
Yegappan Lakshmanancc0bcf42023-09-08 19:12:03 +0200254 }
255 if (searching && is_method)
256 // The parent class methods are stored after the current class
257 // methods.
258 method_offset += is_static
259 ? super->class_class_function_count_child
260 : super->class_obj_method_count_child;
261 }
Bram Moolenaar29ac5df2023-01-16 19:43:47 +0000262 if (i2c == NULL)
263 {
264 siemsg("class %s not found on interface %s",
265 cl->class_name, itf->class_name);
266 return 0;
267 }
Ernie Rael18143d32023-09-04 22:30:41 +0200268 if (is_static)
269 {
270 // TODO: Need a table for fast lookup?
271 char_u *name = itf->class_class_members[idx].ocm_name;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +0200272 int m_idx = class_member_idx(i2c->i2c_class, name, 0);
273 if (m_idx >= 0)
274 return m_idx;
275
Ernie Rael18143d32023-09-04 22:30:41 +0200276 siemsg("class %s, interface %s, static %s not found",
277 cl->class_name, itf->class_name, name);
278 return 0;
279 }
280 else
281 {
282 // A table follows the i2c for the class
283 int *table = (int *)(i2c + 1);
Yegappan Lakshmanancc0bcf42023-09-08 19:12:03 +0200284 // "method_offset" is 0, if method is in the current class. If method
285 // is in a parent class, then it is non-zero.
286 return table[idx] + method_offset;
Ernie Rael18143d32023-09-04 22:30:41 +0200287 }
Bram Moolenaar29ac5df2023-01-16 19:43:47 +0000288}
289
290/*
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200291 * Check whether a class named "extends_name" is present. If the class is
292 * valid, then "extends_clp" is set with the class pointer.
293 * Returns TRUE if the class name "extends_names" is a valid class.
294 */
295 static int
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200296validate_extends_class(
297 char_u *extends_name,
298 class_T **extends_clp,
299 int is_class)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200300{
301 typval_T tv;
302 int success = FALSE;
303
304 tv.v_type = VAR_UNKNOWN;
305 if (eval_variable_import(extends_name, &tv) == FAIL)
306 {
307 semsg(_(e_class_name_not_found_str), extends_name);
308 return success;
309 }
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +0200310
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200311 if (tv.v_type != VAR_CLASS || tv.vval.v_class == NULL
312 || (is_class
313 && (tv.vval.v_class->class_flags & CLASS_INTERFACE) != 0)
314 || (!is_class
315 && (tv.vval.v_class->class_flags & CLASS_INTERFACE) == 0))
316 // a interface cannot extend a class and a class cannot extend an
317 // interface.
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +0200318 semsg(_(e_cannot_extend_str), extends_name);
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200319 else
320 {
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +0200321 class_T *extends_cl = tv.vval.v_class;
322 ++extends_cl->class_refcount;
323 *extends_clp = extends_cl;
324 success = TRUE;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200325 }
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +0200326 clear_tv(&tv);
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200327
328 return success;
329}
330
331/*
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200332 * Check method names in the parent class lineage to make sure the access is
333 * the same for overridden methods.
334 */
335 static int
336validate_extends_methods(
337 garray_T *objmethods_gap,
338 class_T *extends_cl)
339{
340 class_T *super = extends_cl;
341 int method_count = objmethods_gap->ga_len;
342 ufunc_T **cl_fp = (ufunc_T **)(objmethods_gap->ga_data);
343
344 while (super != NULL)
345 {
346 int extends_method_count = super->class_obj_method_count_child;
347 if (extends_method_count == 0)
348 {
349 super = super->class_extends;
350 continue;
351 }
352
353 ufunc_T **extends_methods = super->class_obj_methods;
354
355 for (int i = 0; i < extends_method_count; i++)
356 {
357 char_u *pstr = extends_methods[i]->uf_name;
358 int extends_private = (*pstr == '_');
359 if (extends_private)
360 pstr++;
361
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200362 // When comparing the method names, ignore the access type (public
363 // and private methods are considered the same).
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200364 for (int j = 0; j < method_count; j++)
365 {
366 char_u *qstr = cl_fp[j]->uf_name;
367 int priv_method = (*qstr == '_');
368 if (priv_method)
369 qstr++;
370 if (STRCMP(pstr, qstr) == 0 && priv_method != extends_private)
371 {
372 // Method access is different between the super class and
373 // the subclass
374 semsg(_(e_method_str_of_class_str_has_different_access),
375 cl_fp[j]->uf_name, super->class_name);
376 return FALSE;
377 }
378 }
379 }
380 super = super->class_extends;
381 }
382
383 return TRUE;
384}
385
386/*
387 * Check whether a object member variable in "objmembers_gap" is a duplicate of
388 * a member in any of the extended parent class lineage. Returns TRUE if there
389 * are no duplicates.
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200390 */
391 static int
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200392extends_check_dup_members(
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200393 garray_T *objmembers_gap,
394 class_T *extends_cl)
395{
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200396 int member_count = objmembers_gap->ga_len;
397 if (member_count == 0)
398 return TRUE;
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200399
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200400 ocmember_T *members = (ocmember_T *)(objmembers_gap->ga_data);
401
402 // Validate each member variable
403 for (int c_i = 0; c_i < member_count; c_i++)
404 {
405 class_T *p_cl = extends_cl;
406 ocmember_T *c_m = members + c_i;
407 char_u *pstr = (*c_m->ocm_name == '_')
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200408 ? c_m->ocm_name + 1 : c_m->ocm_name;
409
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200410 // Check in all the parent classes in the lineage
411 while (p_cl != NULL)
412 {
413 int p_member_count = p_cl->class_obj_member_count;
414 if (p_member_count == 0)
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200415 {
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200416 p_cl = p_cl->class_extends;
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200417 continue;
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200418 }
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200419 ocmember_T *p_members = p_cl->class_obj_members;
420
421 // Compare against all the members in the parent class
422 for (int p_i = 0; p_i < p_member_count; p_i++)
423 {
424 ocmember_T *p_m = p_members + p_i;
425 char_u *qstr = (*p_m->ocm_name == '_')
426 ? p_m->ocm_name + 1 : p_m->ocm_name;
427 if (STRCMP(pstr, qstr) == 0)
428 {
429 semsg(_(e_duplicate_member_str), c_m->ocm_name);
430 return FALSE;
431 }
432 }
433
434 p_cl = p_cl->class_extends;
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200435 }
436 }
437
438 return TRUE;
439}
440
441/*
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200442 * Compare the variable type of interface variables in "objmembers_gap" against
443 * the variable in any of the extended super interface lineage. Used to
444 * compare the variable types when extending interfaces. Returns TRUE if the
445 * variable types are the same.
446 */
447 static int
448extends_check_intf_var_type(
449 garray_T *objmembers_gap,
450 class_T *extends_cl)
451{
452 int member_count = objmembers_gap->ga_len;
453 if (member_count == 0)
454 return TRUE;
455
456 ocmember_T *members = (ocmember_T *)(objmembers_gap->ga_data);
457
458 // Validate each member variable
459 for (int c_i = 0; c_i < member_count; c_i++)
460 {
461 class_T *p_cl = extends_cl;
462 ocmember_T *c_m = members + c_i;
463 int var_found = FALSE;
464
465 // Check in all the parent classes in the lineage
466 while (p_cl != NULL && !var_found)
467 {
468 int p_member_count = p_cl->class_obj_member_count;
469 if (p_member_count == 0)
470 {
471 p_cl = p_cl->class_extends;
472 continue;
473 }
474 ocmember_T *p_members = p_cl->class_obj_members;
475
476 // Compare against all the members in the parent class
477 for (int p_i = 0; p_i < p_member_count; p_i++)
478 {
479 where_T where = WHERE_INIT;
480 ocmember_T *p_m = p_members + p_i;
481
482 if (STRCMP(p_m->ocm_name, c_m->ocm_name) != 0)
483 continue;
484
485 // Ensure the type is matching.
486 where.wt_func_name = (char *)c_m->ocm_name;
487 where.wt_kind = WT_MEMBER;
488
489 if (check_type(p_m->ocm_type, c_m->ocm_type, TRUE,
490 where) == FAIL)
491 return FALSE;
492
493 var_found = TRUE;
494 }
495
496 p_cl = p_cl->class_extends;
497 }
498 }
499
500 return TRUE;
501}
502
503/*
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +0200504 * When extending an abstract class, check whether all the abstract methods in
505 * the parent class are implemented. Returns TRUE if all the methods are
506 * implemented.
507 */
508 static int
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200509validate_abstract_class_methods(
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +0200510 garray_T *classmethods_gap,
511 garray_T *objmethods_gap,
512 class_T *extends_cl)
513{
514 for (int loop = 1; loop <= 2; ++loop)
515 {
516 // loop == 1: check class methods
517 // loop == 2: check object methods
518 int extends_method_count = loop == 1
519 ? extends_cl->class_class_function_count
520 : extends_cl->class_obj_method_count;
521 if (extends_method_count == 0)
522 continue;
523
524 ufunc_T **extends_methods = loop == 1
525 ? extends_cl->class_class_functions
526 : extends_cl->class_obj_methods;
527
528 int method_count = loop == 1 ? classmethods_gap->ga_len
529 : objmethods_gap->ga_len;
530 ufunc_T **cl_fp = (ufunc_T **)(loop == 1
531 ? classmethods_gap->ga_data
532 : objmethods_gap->ga_data);
533
534 for (int i = 0; i < extends_method_count; i++)
535 {
536 ufunc_T *uf = extends_methods[i];
537 if ((uf->uf_flags & FC_ABSTRACT) == 0)
538 continue;
539
540 int method_found = FALSE;
541
542 for (int j = 0; j < method_count; j++)
543 {
544 if (STRCMP(uf->uf_name, cl_fp[j]->uf_name) == 0)
545 {
546 method_found = TRUE;
547 break;
548 }
549 }
550
551 if (!method_found)
552 {
553 semsg(_(e_abstract_method_str_not_found), uf->uf_name);
554 return FALSE;
555 }
556 }
557 }
558
559 return TRUE;
560}
561
562/*
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200563 * Returns TRUE if the interface variable "if_var" is present in the list of
564 * variables in "cl_mt" or in the parent lineage of one of the extended classes
565 * in "extends_cl". For a class variable, 'is_class_var' is TRUE.
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200566 */
567 static int
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200568intf_variable_present(
569 char_u *intf_class_name,
570 ocmember_T *if_var,
571 int is_class_var,
572 ocmember_T *cl_mt,
573 int cl_member_count,
574 class_T *extends_cl)
575{
576 int variable_present = FALSE;
577
578 for (int cl_i = 0; cl_i < cl_member_count; ++cl_i)
579 {
580 ocmember_T *m = &cl_mt[cl_i];
581 where_T where = WHERE_INIT;
582
583 if (STRCMP(if_var->ocm_name, m->ocm_name) != 0)
584 continue;
585
586 // Ensure the access type is same
587 if (if_var->ocm_access != m->ocm_access)
588 {
589 semsg(_(e_member_str_of_interface_str_has_different_access),
590 if_var->ocm_name, intf_class_name);
591 return FALSE;
592 }
593
594 // Ensure the type is matching.
595 if (m->ocm_type == &t_any)
596 {
597 // variable type is not specified. Use the variable type in the
598 // interface.
599 m->ocm_type = if_var->ocm_type;
600 }
601 else
602 {
603 where.wt_func_name = (char *)m->ocm_name;
604 where.wt_kind = WT_MEMBER;
605 if (check_type(if_var->ocm_type, m->ocm_type, TRUE,
606 where) == FAIL)
607 return FALSE;
608 }
609
610 variable_present = TRUE;
611 break;
612 }
613
614 if (!variable_present && extends_cl != NULL)
615 {
616 int ext_cl_count = is_class_var
617 ? extends_cl->class_class_member_count
618 : extends_cl->class_obj_member_count;
619 ocmember_T *ext_cl_mt = is_class_var
620 ? extends_cl->class_class_members
621 : extends_cl->class_obj_members;
622 return intf_variable_present(intf_class_name, if_var,
623 is_class_var, ext_cl_mt,
624 ext_cl_count,
625 extends_cl->class_extends);
626 }
627
628 return variable_present;
629}
630
631/*
632 * Check the variables of the interface class "ifcl" match the class variables
633 * ("classmembers_gap") and object variables ("objmembers_gap") of a class.
634 * Returns TRUE if the class and object variables names are valid.
635 */
636 static int
637validate_interface_variables(
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200638 char_u *intf_class_name,
639 class_T *ifcl,
640 garray_T *classmembers_gap,
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200641 garray_T *objmembers_gap,
642 class_T *extends_cl)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200643{
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200644 for (int loop = 1; loop <= 2; ++loop)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200645 {
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200646 // loop == 1: check class variables
647 // loop == 2: check object variables
648 int is_class_var = (loop == 1);
649 int if_count = is_class_var ? ifcl->class_class_member_count
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200650 : ifcl->class_obj_member_count;
651 if (if_count == 0)
652 continue;
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200653 ocmember_T *if_ms = is_class_var ? ifcl->class_class_members
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200654 : ifcl->class_obj_members;
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200655 ocmember_T *cl_ms = (ocmember_T *)(is_class_var
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200656 ? classmembers_gap->ga_data
657 : objmembers_gap->ga_data);
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200658 int cl_count = is_class_var ? classmembers_gap->ga_len
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200659 : objmembers_gap->ga_len;
660 for (int if_i = 0; if_i < if_count; ++if_i)
661 {
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200662 if (!intf_variable_present(intf_class_name, &if_ms[if_i],
663 is_class_var, cl_ms, cl_count, extends_cl))
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200664 {
665 semsg(_(e_member_str_of_interface_str_not_implemented),
666 if_ms[if_i].ocm_name, intf_class_name);
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200667 return FALSE;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200668 }
669 }
670 }
671
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200672 return TRUE;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200673}
674
675/*
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200676 * Returns TRUE if the method signature of "if_method" and "cl_method" matches.
677 */
678 static int
679intf_method_type_matches(ufunc_T *if_method, ufunc_T *cl_method)
680{
681 where_T where = WHERE_INIT;
682
683 // Ensure the type is matching.
684 where.wt_func_name = (char *)if_method->uf_name;
685 where.wt_kind = WT_METHOD;
686 if (check_type(if_method->uf_func_type, cl_method->uf_func_type, TRUE,
687 where) == FAIL)
688 return FALSE;
689
690 return TRUE;
691}
692
693/*
694 * Returns TRUE if the interface method "if_ufunc" is present in the list of
695 * methods in "cl_fp" or in the parent lineage of one of the extended classes
696 * in "extends_cl". For a class method, 'is_class_method' is TRUE.
697 */
698 static int
699intf_method_present(
700 ufunc_T *if_ufunc,
701 int is_class_method,
702 ufunc_T **cl_fp,
703 int cl_count,
704 class_T *extends_cl)
705{
706 int method_present = FALSE;
707
708 for (int cl_i = 0; cl_i < cl_count; ++cl_i)
709 {
710 char_u *cl_name = cl_fp[cl_i]->uf_name;
711 if (STRCMP(if_ufunc->uf_name, cl_name) == 0)
712 {
713 // Ensure the type is matching.
714 if (!intf_method_type_matches(if_ufunc, cl_fp[cl_i]))
715 return FALSE;
716 method_present = TRUE;
717 break;
718 }
719 }
720
721 if (!method_present && extends_cl != NULL)
722 {
723 ufunc_T **ext_cl_fp = (ufunc_T **)(is_class_method
724 ? extends_cl->class_class_functions
725 : extends_cl->class_obj_methods);
726 int ext_cl_count = is_class_method
727 ? extends_cl->class_class_function_count
728 : extends_cl->class_obj_method_count;
729 return intf_method_present(if_ufunc, is_class_method, ext_cl_fp,
730 ext_cl_count,
731 extends_cl->class_extends);
732 }
733
734 return method_present;
735}
736
737/*
738 * Validate that a new class implements all the class/instance methods in the
739 * interface "ifcl". The new class methods are in "classfunctions_gap" and the
740 * new object methods are in "objmemthods_gap". Also validates the method
741 * types.
742 * Returns TRUE if all the interface class/object methods are implemented in
743 * the new class.
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200744 */
745 static int
746validate_interface_methods(
747 char_u *intf_class_name,
748 class_T *ifcl,
749 garray_T *classfunctions_gap,
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200750 garray_T *objmethods_gap,
751 class_T *extends_cl)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200752{
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200753 for (int loop = 1; loop <= 2; ++loop)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200754 {
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200755 // loop == 1: check class methods
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200756 // loop == 2: check object methods
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200757 int is_class_method = (loop == 1);
758 int if_count = is_class_method ? ifcl->class_class_function_count
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200759 : ifcl->class_obj_method_count;
760 if (if_count == 0)
761 continue;
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200762 ufunc_T **if_fp = is_class_method ? ifcl->class_class_functions
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200763 : ifcl->class_obj_methods;
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200764 ufunc_T **cl_fp = (ufunc_T **)(is_class_method
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200765 ? classfunctions_gap->ga_data
766 : objmethods_gap->ga_data);
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200767 int cl_count = is_class_method ? classfunctions_gap->ga_len
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200768 : objmethods_gap->ga_len;
769 for (int if_i = 0; if_i < if_count; ++if_i)
770 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +0200771 char_u *if_name = if_fp[if_i]->uf_name;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200772
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200773 if (!intf_method_present(if_fp[if_i], is_class_method, cl_fp,
774 cl_count, extends_cl))
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200775 {
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200776 semsg(_(e_method_str_of_interface_str_not_implemented),
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200777 if_name, intf_class_name);
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200778 return FALSE;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200779 }
780 }
781 }
782
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200783 return TRUE;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200784}
785
786/*
787 * Validate all the "implements" classes when creating a new class. The
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200788 * classes are returned in "intf_classes". The class functions, class members,
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200789 * object methods and object members in the new class are in
790 * "classfunctions_gap", "classmembers_gap", "objmethods_gap", and
791 * "objmembers_gap" respectively.
792 */
793 static int
794validate_implements_classes(
795 garray_T *impl_gap,
796 class_T **intf_classes,
797 garray_T *classfunctions_gap,
798 garray_T *classmembers_gap,
799 garray_T *objmethods_gap,
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200800 garray_T *objmembers_gap,
801 class_T *extends_cl)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200802{
803 int success = TRUE;
804
805 for (int i = 0; i < impl_gap->ga_len && success; ++i)
806 {
807 char_u *impl = ((char_u **)impl_gap->ga_data)[i];
808 typval_T tv;
809 tv.v_type = VAR_UNKNOWN;
810 if (eval_variable_import(impl, &tv) == FAIL)
811 {
812 semsg(_(e_interface_name_not_found_str), impl);
813 success = FALSE;
814 break;
815 }
816
817 if (tv.v_type != VAR_CLASS
818 || tv.vval.v_class == NULL
819 || (tv.vval.v_class->class_flags & CLASS_INTERFACE) == 0)
820 {
821 semsg(_(e_not_valid_interface_str), impl);
822 success = FALSE;
823 clear_tv(&tv);
824 break;
825 }
826
827 class_T *ifcl = tv.vval.v_class;
828 intf_classes[i] = ifcl;
829 ++ifcl->class_refcount;
830
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200831 // check the variables of the interface match the members of the class
832 success = validate_interface_variables(impl, ifcl, classmembers_gap,
833 objmembers_gap, extends_cl);
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200834
835 // check the functions/methods of the interface match the
836 // functions/methods of the class
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200837 if (success)
838 success = validate_interface_methods(impl, ifcl,
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200839 classfunctions_gap, objmethods_gap,
840 extends_cl);
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200841 clear_tv(&tv);
842 }
843
844 return success;
845}
846
847/*
848 * Check no function argument name is used as a class member.
849 * (Object members are always accessed with "this." prefix, so no need
850 * to check them.)
851 */
852 static int
853check_func_arg_names(
854 garray_T *classfunctions_gap,
855 garray_T *objmethods_gap,
856 garray_T *classmembers_gap)
857{
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200858 // loop 1: class functions, loop 2: object methods
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200859 for (int loop = 1; loop <= 2; ++loop)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200860 {
861 garray_T *gap = loop == 1 ? classfunctions_gap : objmethods_gap;
862
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200863 for (int fi = 0; fi < gap->ga_len; ++fi)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200864 {
865 ufunc_T *uf = ((ufunc_T **)gap->ga_data)[fi];
866
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200867 for (int i = 0; i < uf->uf_args.ga_len; ++i)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200868 {
869 char_u *aname = ((char_u **)uf->uf_args.ga_data)[i];
870 garray_T *mgap = classmembers_gap;
871
872 // Check all the class member names
873 for (int mi = 0; mi < mgap->ga_len; ++mi)
874 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +0200875 char_u *mname =
876 ((ocmember_T *)mgap->ga_data + mi)->ocm_name;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200877 if (STRCMP(aname, mname) == 0)
878 {
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200879 if (uf->uf_script_ctx.sc_sid > 0)
880 SOURCING_LNUM = uf->uf_script_ctx.sc_lnum;
881
882 semsg(_(e_argument_already_declared_in_class_str),
883 aname);
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200884
885 return FALSE;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200886 }
887 }
888 }
889 }
890 }
891
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +0200892 return TRUE;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200893}
894
895/*
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +0200896 * Returns TRUE if the member "varname" is already defined.
897 */
898 static int
899is_duplicate_member(garray_T *mgap, char_u *varname, char_u *varname_end)
900{
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +0200901 char_u *name = vim_strnsave(varname, varname_end - varname);
902 char_u *pstr = (*name == '_') ? name + 1 : name;
903 int dup = FALSE;
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +0200904
905 for (int i = 0; i < mgap->ga_len; ++i)
906 {
907 ocmember_T *m = ((ocmember_T *)mgap->ga_data) + i;
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +0200908 char_u *qstr = *m->ocm_name == '_' ? m->ocm_name + 1 : m->ocm_name;
909 if (STRCMP(pstr, qstr) == 0)
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +0200910 {
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +0200911 semsg(_(e_duplicate_member_str), name);
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +0200912 dup = TRUE;
913 break;
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +0200914 }
915 }
916
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +0200917 vim_free(name);
918 return dup;
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +0200919}
920
921/*
922 * Returns TRUE if the method "name" is already defined.
923 */
924 static int
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200925is_duplicate_method(
926 garray_T *classmethods_gap,
927 garray_T *objmethods_gap,
928 char_u *name)
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +0200929{
930 char_u *pstr = (*name == '_') ? name + 1 : name;
931
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200932 // loop 1: class methods, loop 2: object methods
933 for (int loop = 1; loop <= 2; loop++)
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +0200934 {
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200935 garray_T *fgap = (loop == 1) ? classmethods_gap : objmethods_gap;
936 for (int i = 0; i < fgap->ga_len; ++i)
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +0200937 {
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200938 char_u *n = ((ufunc_T **)fgap->ga_data)[i]->uf_name;
939 char_u *qstr = *n == '_' ? n + 1 : n;
940 if (STRCMP(pstr, qstr) == 0)
941 {
942 semsg(_(e_duplicate_function_str), name);
943 return TRUE;
944 }
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +0200945 }
946 }
947
948 return FALSE;
949}
950
951/*
Gianmaria Bajo4b9777a2023-08-29 22:26:30 +0200952 * Returns TRUE if the constructor is valid.
953 */
954 static int
955is_valid_constructor(ufunc_T *uf, int is_abstract, int has_static)
956{
957 // Constructors are not allowed in abstract classes.
958 if (is_abstract)
959 {
960 emsg(_(e_cannot_define_new_function_in_abstract_class));
961 return FALSE;
962 }
963 // A constructor is always static, no need to define it so.
964 if (has_static)
965 {
966 emsg(_(e_cannot_define_new_function_as_static));
967 return FALSE;
968 }
969 // A return type should not be specified for the new()
970 // constructor method.
971 if (uf->uf_ret_type->tt_type != VAR_VOID)
972 {
973 emsg(_(e_cannot_use_a_return_type_with_new));
974 return FALSE;
975 }
976 return TRUE;
977}
978
979/*
Yegappan Lakshmananb1027282023-08-19 11:26:42 +0200980 * Update the interface class lookup table for the member index on the
981 * interface to the member index in the class implementing the interface.
982 * And a lookup table for the object method index on the interface
983 * to the object method index in the class implementing the interface.
984 * This is also used for updating the lookup table for the extended class
985 * hierarchy.
986 */
987 static int
988update_member_method_lookup_table(
Yegappan Lakshmanan3775f772023-09-01 22:05:45 +0200989 class_T *ifcl,
990 class_T *cl,
Yegappan Lakshmananeb91e242023-08-31 18:10:46 +0200991 garray_T *objmethods,
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200992 int pobj_method_offset)
Yegappan Lakshmananb1027282023-08-19 11:26:42 +0200993{
994 if (ifcl == NULL)
995 return OK;
996
997 // Table for members.
998 itf2class_T *if2cl = alloc_clear(sizeof(itf2class_T)
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +0200999 + ifcl->class_obj_member_count * sizeof(int));
Yegappan Lakshmananb1027282023-08-19 11:26:42 +02001000 if (if2cl == NULL)
1001 return FAIL;
1002 if2cl->i2c_next = ifcl->class_itf2class;
1003 ifcl->class_itf2class = if2cl;
1004 if2cl->i2c_class = cl;
1005 if2cl->i2c_is_method = FALSE;
1006
1007 for (int if_i = 0; if_i < ifcl->class_obj_member_count; ++if_i)
1008 for (int cl_i = 0; cl_i < cl->class_obj_member_count; ++cl_i)
1009 {
1010 if (STRCMP(ifcl->class_obj_members[if_i].ocm_name,
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001011 cl->class_obj_members[cl_i].ocm_name) == 0)
Yegappan Lakshmananb1027282023-08-19 11:26:42 +02001012 {
1013 int *table = (int *)(if2cl + 1);
1014 table[if_i] = cl_i;
1015 break;
1016 }
1017 }
1018
1019 // Table for methods.
1020 if2cl = alloc_clear(sizeof(itf2class_T)
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001021 + ifcl->class_obj_method_count * sizeof(int));
Yegappan Lakshmananb1027282023-08-19 11:26:42 +02001022 if (if2cl == NULL)
1023 return FAIL;
1024 if2cl->i2c_next = ifcl->class_itf2class;
1025 ifcl->class_itf2class = if2cl;
1026 if2cl->i2c_class = cl;
1027 if2cl->i2c_is_method = TRUE;
1028
1029 for (int if_i = 0; if_i < ifcl->class_obj_method_count; ++if_i)
1030 {
1031 int done = FALSE;
1032 for (int cl_i = 0; cl_i < objmethods->ga_len; ++cl_i)
1033 {
1034 if (STRCMP(ifcl->class_obj_methods[if_i]->uf_name,
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001035 ((ufunc_T **)objmethods->ga_data)[cl_i]->uf_name) == 0)
Yegappan Lakshmananb1027282023-08-19 11:26:42 +02001036 {
1037 int *table = (int *)(if2cl + 1);
1038 table[if_i] = cl_i;
1039 done = TRUE;
1040 break;
1041 }
1042 }
1043
1044 // extended class object method is not overridden by the child class.
1045 // Keep the method declared in one of the parent classes in the
1046 // lineage.
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001047 if (!done)
Yegappan Lakshmananb1027282023-08-19 11:26:42 +02001048 {
1049 // If "ifcl" is not the immediate parent of "cl", then search in
1050 // the intermediate parent classes.
1051 if (cl->class_extends != ifcl)
1052 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001053 class_T *parent = cl->class_extends;
1054 int method_offset = objmethods->ga_len;
Yegappan Lakshmananb1027282023-08-19 11:26:42 +02001055
1056 while (!done && parent != NULL && parent != ifcl)
1057 {
1058
1059 for (int cl_i = 0;
1060 cl_i < parent->class_obj_method_count_child; ++cl_i)
1061 {
1062 if (STRCMP(ifcl->class_obj_methods[if_i]->uf_name,
1063 parent->class_obj_methods[cl_i]->uf_name)
1064 == 0)
1065 {
1066 int *table = (int *)(if2cl + 1);
1067 table[if_i] = method_offset + cl_i;
1068 done = TRUE;
1069 break;
1070 }
1071 }
1072 method_offset += parent->class_obj_method_count_child;
1073 parent = parent->class_extends;
1074 }
1075 }
1076
1077 if (!done)
1078 {
1079 int *table = (int *)(if2cl + 1);
1080 table[if_i] = pobj_method_offset + if_i;
1081 }
1082 }
1083 }
1084
1085 return OK;
1086}
1087
1088/*
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001089 * Update the member and object method lookup tables for a new class in the
1090 * interface class.
1091 * For each interface add a lookup table for the member index on the interface
1092 * to the member index in the new class. And a lookup table for the object
1093 * method index on the interface to the object method index in the new class.
1094 */
1095 static int
1096add_lookup_tables(class_T *cl, class_T *extends_cl, garray_T *objmethods_gap)
1097{
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001098 // update the lookup table for all the implemented interfaces
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001099 for (int i = 0; i < cl->class_interface_count; ++i)
1100 {
1101 class_T *ifcl = cl->class_interfaces_cl[i];
1102
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001103 // update the lookup table for this interface and all its super
1104 // interfaces.
1105 while (ifcl != NULL)
1106 {
1107 if (update_member_method_lookup_table(ifcl, cl, objmethods_gap,
1108 0) == FAIL)
1109 return FAIL;
1110 ifcl = ifcl->class_extends;
1111 }
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001112 }
1113
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001114 // Update the lookup table for the extended class, if any
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001115 if (extends_cl != NULL)
1116 {
1117 class_T *pclass = extends_cl;
1118 int pobj_method_offset = objmethods_gap->ga_len;
1119
1120 // Update the entire lineage of extended classes.
1121 while (pclass != NULL)
1122 {
1123 if (update_member_method_lookup_table(pclass, cl,
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001124 objmethods_gap, pobj_method_offset) == FAIL)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001125 return FAIL;
1126
1127 pobj_method_offset += pclass->class_obj_method_count_child;
1128 pclass = pclass->class_extends;
1129 }
1130 }
1131
1132 return OK;
1133}
1134
1135/*
1136 * Add class members to a new class. Allocate a typval for each class member
1137 * and initialize it.
1138 */
1139 static void
1140add_class_members(class_T *cl, exarg_T *eap)
1141{
1142 // Allocate a typval for each class member and initialize it.
1143 cl->class_members_tv = ALLOC_CLEAR_MULT(typval_T,
1144 cl->class_class_member_count);
1145 if (cl->class_members_tv == NULL)
1146 return;
1147
1148 for (int i = 0; i < cl->class_class_member_count; ++i)
1149 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001150 ocmember_T *m = &cl->class_class_members[i];
1151 typval_T *tv = &cl->class_members_tv[i];
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001152 if (m->ocm_init != NULL)
1153 {
1154 typval_T *etv = eval_expr(m->ocm_init, eap);
1155 if (etv != NULL)
1156 {
1157 *tv = *etv;
1158 vim_free(etv);
1159 }
1160 }
1161 else
1162 {
1163 // TODO: proper default value
1164 tv->v_type = m->ocm_type->tt_type;
1165 tv->vval.v_string = NULL;
1166 }
1167 }
1168}
1169
1170/*
Yegappan Lakshmanan6ac15442023-08-20 18:20:17 +02001171 * Add a default constructor method (new()) to the class "cl".
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001172 */
1173 static void
1174add_default_constructor(
1175 class_T *cl,
1176 garray_T *classfunctions_gap,
1177 garray_T *type_list_gap)
1178{
1179 garray_T fga;
1180
1181 ga_init2(&fga, 1, 1000);
1182 ga_concat(&fga, (char_u *)"new(");
1183 for (int i = 0; i < cl->class_obj_member_count; ++i)
1184 {
1185 if (i > 0)
1186 ga_concat(&fga, (char_u *)", ");
1187 ga_concat(&fga, (char_u *)"this.");
1188 ocmember_T *m = cl->class_obj_members + i;
1189 ga_concat(&fga, (char_u *)m->ocm_name);
1190 ga_concat(&fga, (char_u *)" = v:none");
1191 }
1192 ga_concat(&fga, (char_u *)")\nenddef\n");
1193 ga_append(&fga, NUL);
1194
1195 exarg_T fea;
1196 CLEAR_FIELD(fea);
1197 fea.cmdidx = CMD_def;
1198 fea.cmd = fea.arg = fga.ga_data;
1199
1200 garray_T lines_to_free;
1201 ga_init2(&lines_to_free, sizeof(char_u *), 50);
1202
1203 ufunc_T *nf = define_function(&fea, NULL, &lines_to_free, CF_CLASS);
1204
1205 ga_clear_strings(&lines_to_free);
1206 vim_free(fga.ga_data);
1207
1208 if (nf != NULL && ga_grow(classfunctions_gap, 1) == OK)
1209 {
1210 ((ufunc_T **)classfunctions_gap->ga_data)[classfunctions_gap->ga_len]
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001211 = nf;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001212 ++classfunctions_gap->ga_len;
1213
1214 nf->uf_flags |= FC_NEW;
1215 nf->uf_ret_type = get_type_ptr(type_list_gap);
1216 if (nf->uf_ret_type != NULL)
1217 {
1218 nf->uf_ret_type->tt_type = VAR_OBJECT;
1219 nf->uf_ret_type->tt_class = cl;
1220 nf->uf_ret_type->tt_argcount = 0;
1221 nf->uf_ret_type->tt_args = NULL;
1222 }
1223 }
1224}
1225
1226/*
Yegappan Lakshmanane2deb7e2023-09-16 18:05:07 +02001227 * Add the class methods and object methods to the new class "cl".
1228 * When extending a class "extends_cl", add the instance methods from the
1229 * parent class also.
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001230 */
1231 static int
1232add_classfuncs_objmethods(
1233 class_T *cl,
1234 class_T *extends_cl,
1235 garray_T *classfunctions_gap,
1236 garray_T *objmethods_gap)
1237{
1238 // loop 1: class functions, loop 2: object methods
1239 for (int loop = 1; loop <= 2; ++loop)
1240 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001241 garray_T *gap = loop == 1 ? classfunctions_gap : objmethods_gap;
1242 int *fcount = loop == 1 ? &cl->class_class_function_count
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001243 : &cl->class_obj_method_count;
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001244 ufunc_T ***fup = loop == 1 ? &cl->class_class_functions
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001245 : &cl->class_obj_methods;
1246
1247 int parent_count = 0;
1248 if (extends_cl != NULL)
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001249 // Include object methods from the parent.
1250 // Don't include the parent class methods.
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001251 parent_count = loop == 1
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001252 ? 0
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001253 : extends_cl->class_obj_method_count;
1254
1255 *fcount = parent_count + gap->ga_len;
1256 if (*fcount == 0)
1257 {
1258 *fup = NULL;
1259 continue;
1260 }
1261 *fup = ALLOC_MULT(ufunc_T *, *fcount);
1262 if (*fup == NULL)
1263 return FAIL;
1264
1265 if (gap->ga_len != 0)
1266 mch_memmove(*fup, gap->ga_data, sizeof(ufunc_T *) * gap->ga_len);
1267 vim_free(gap->ga_data);
1268 if (loop == 1)
1269 cl->class_class_function_count_child = gap->ga_len;
1270 else
1271 cl->class_obj_method_count_child = gap->ga_len;
1272
Yegappan Lakshmanane2deb7e2023-09-16 18:05:07 +02001273 if (loop == 2)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001274 {
Yegappan Lakshmanane2deb7e2023-09-16 18:05:07 +02001275 // Copy instance methods from the parent.
1276
1277 for (int i = 0; i < parent_count; ++i)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001278 {
Yegappan Lakshmanane2deb7e2023-09-16 18:05:07 +02001279 // Can't use the same parent function, because "uf_class" is
1280 // different and compilation will have a different result.
1281 // Put them after the functions in the current class, object
1282 // methods may be overruled, then "super.Method()" is used to
1283 // find a method from the parent.
1284 ufunc_T *pf = (extends_cl->class_obj_methods)[i];
1285 (*fup)[gap->ga_len + i] = copy_function(pf);
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001286
1287 // If the child class overrides a function from the parent
1288 // the signature must be equal.
1289 char_u *pname = pf->uf_name;
1290 for (int ci = 0; ci < gap->ga_len; ++ci)
1291 {
1292 ufunc_T *cf = (*fup)[ci];
1293 char_u *cname = cf->uf_name;
1294 if (STRCMP(pname, cname) == 0)
1295 {
1296 where_T where = WHERE_INIT;
1297 where.wt_func_name = (char *)pname;
1298 where.wt_kind = WT_METHOD;
1299 (void)check_type(pf->uf_func_type, cf->uf_func_type,
1300 TRUE, where);
1301 }
1302 }
1303 }
1304 }
1305
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001306 // Set the class pointer on all the functions and object methods.
1307 for (int i = 0; i < *fcount; ++i)
1308 {
1309 ufunc_T *fp = (*fup)[i];
1310 fp->uf_class = cl;
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001311 if (i < gap->ga_len)
1312 fp->uf_defclass = cl;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001313 if (loop == 2)
1314 fp->uf_flags |= FC_OBJECT;
1315 }
1316 }
1317
1318 return OK;
1319}
1320
1321/*
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001322 * Handle ":class" and ":abstract class" up to ":endclass".
Bram Moolenaar554d0312023-01-05 19:59:18 +00001323 * Handle ":interface" up to ":endinterface".
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001324 */
1325 void
1326ex_class(exarg_T *eap)
1327{
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001328 int is_class = eap->cmdidx == CMD_class; // FALSE for :interface
1329 long start_lnum = SOURCING_LNUM;
1330 char_u *arg = eap->arg;
1331 int is_abstract = eap->cmdidx == CMD_abstract;
Bram Moolenaar554d0312023-01-05 19:59:18 +00001332
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001333 if (is_abstract)
1334 {
1335 if (STRNCMP(arg, "class", 5) != 0 || !VIM_ISWHITE(arg[5]))
1336 {
1337 semsg(_(e_invalid_argument_str), arg);
1338 return;
1339 }
1340 arg = skipwhite(arg + 5);
Bram Moolenaar24a8d062023-01-14 13:12:06 +00001341 is_class = TRUE;
1342 }
1343
1344 if (!current_script_is_vim9()
1345 || (cmdmod.cmod_flags & CMOD_LEGACY)
1346 || !getline_equal(eap->getline, eap->cookie, getsourceline))
1347 {
1348 if (is_class)
1349 emsg(_(e_class_can_only_be_defined_in_vim9_script));
1350 else
1351 emsg(_(e_interface_can_only_be_defined_in_vim9_script));
1352 return;
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001353 }
1354
1355 if (!ASCII_ISUPPER(*arg))
1356 {
Bram Moolenaar554d0312023-01-05 19:59:18 +00001357 if (is_class)
1358 semsg(_(e_class_name_must_start_with_uppercase_letter_str), arg);
1359 else
1360 semsg(_(e_interface_name_must_start_with_uppercase_letter_str),
1361 arg);
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001362 return;
1363 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001364 char_u *name_end = find_name_end(arg, NULL, NULL, FNE_CHECK_START);
1365 if (!IS_WHITE_OR_NUL(*name_end))
1366 {
Bram Moolenaar554d0312023-01-05 19:59:18 +00001367 semsg(_(e_white_space_required_after_name_str), arg);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001368 return;
1369 }
Bram Moolenaar94674f22023-01-06 18:42:20 +00001370 char_u *name_start = arg;
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001371
Bram Moolenaara86655a2023-01-12 17:06:27 +00001372 // "export class" gets used when creating the class, don't use "is_export"
1373 // for the items inside the class.
1374 int class_export = is_export;
1375 is_export = FALSE;
1376
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001377 // TODO:
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001378 // generics: <Tkey, Tentry>
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001379
Bram Moolenaar83677162023-01-08 19:54:10 +00001380 // Name for "extends BaseClass"
1381 char_u *extends = NULL;
1382
Bram Moolenaar94674f22023-01-06 18:42:20 +00001383 // Names for "implements SomeInterface"
1384 garray_T ga_impl;
1385 ga_init2(&ga_impl, sizeof(char_u *), 5);
1386
1387 arg = skipwhite(name_end);
1388 while (*arg != NUL && *arg != '#' && *arg != '\n')
1389 {
1390 // TODO:
Bram Moolenaar94674f22023-01-06 18:42:20 +00001391 // specifies SomeInterface
Bram Moolenaar83677162023-01-08 19:54:10 +00001392 if (STRNCMP(arg, "extends", 7) == 0 && IS_WHITE_OR_NUL(arg[7]))
1393 {
1394 if (extends != NULL)
1395 {
1396 emsg(_(e_duplicate_extends));
1397 goto early_ret;
1398 }
1399 arg = skipwhite(arg + 7);
1400 char_u *end = find_name_end(arg, NULL, NULL, FNE_CHECK_START);
1401 if (!IS_WHITE_OR_NUL(*end))
1402 {
1403 semsg(_(e_white_space_required_after_name_str), arg);
1404 goto early_ret;
1405 }
1406 extends = vim_strnsave(arg, end - arg);
1407 if (extends == NULL)
1408 goto early_ret;
1409
1410 arg = skipwhite(end + 1);
1411 }
1412 else if (STRNCMP(arg, "implements", 10) == 0
1413 && IS_WHITE_OR_NUL(arg[10]))
Bram Moolenaar94674f22023-01-06 18:42:20 +00001414 {
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001415 if (!is_class)
1416 {
1417 emsg(_(e_interface_cannot_use_implements));
1418 goto early_ret;
1419 }
1420
Bram Moolenaardf8f9472023-01-07 14:51:03 +00001421 if (ga_impl.ga_len > 0)
1422 {
1423 emsg(_(e_duplicate_implements));
1424 goto early_ret;
1425 }
Bram Moolenaar94674f22023-01-06 18:42:20 +00001426 arg = skipwhite(arg + 10);
Bram Moolenaardf8f9472023-01-07 14:51:03 +00001427
1428 for (;;)
Bram Moolenaar94674f22023-01-06 18:42:20 +00001429 {
Bram Moolenaardf8f9472023-01-07 14:51:03 +00001430 char_u *impl_end = find_name_end(arg, NULL, NULL,
1431 FNE_CHECK_START);
1432 if (!IS_WHITE_OR_NUL(*impl_end) && *impl_end != ',')
1433 {
1434 semsg(_(e_white_space_required_after_name_str), arg);
1435 goto early_ret;
1436 }
1437 char_u *iname = vim_strnsave(arg, impl_end - arg);
1438 if (iname == NULL)
1439 goto early_ret;
1440 for (int i = 0; i < ga_impl.ga_len; ++i)
1441 if (STRCMP(((char_u **)ga_impl.ga_data)[i], iname) == 0)
1442 {
1443 semsg(_(e_duplicate_interface_after_implements_str),
1444 iname);
1445 vim_free(iname);
1446 goto early_ret;
1447 }
1448 if (ga_add_string(&ga_impl, iname) == FAIL)
1449 {
1450 vim_free(iname);
1451 goto early_ret;
1452 }
1453 if (*impl_end != ',')
1454 {
1455 arg = skipwhite(impl_end);
1456 break;
1457 }
1458 arg = skipwhite(impl_end + 1);
Bram Moolenaar94674f22023-01-06 18:42:20 +00001459 }
Bram Moolenaar94674f22023-01-06 18:42:20 +00001460 }
1461 else
1462 {
1463 semsg(_(e_trailing_characters_str), arg);
1464early_ret:
Bram Moolenaar83677162023-01-08 19:54:10 +00001465 vim_free(extends);
Bram Moolenaar94674f22023-01-06 18:42:20 +00001466 ga_clear_strings(&ga_impl);
1467 return;
1468 }
1469 }
1470
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001471 garray_T type_list; // list of pointers to allocated types
1472 ga_init2(&type_list, sizeof(type_T *), 10);
1473
Bram Moolenaard505d172022-12-18 21:42:55 +00001474 // Growarray with class members declared in the class.
1475 garray_T classmembers;
1476 ga_init2(&classmembers, sizeof(ocmember_T), 10);
1477
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001478 // Growarray with functions declared in the class.
1479 garray_T classfunctions;
1480 ga_init2(&classfunctions, sizeof(ufunc_T *), 10);
Bram Moolenaard505d172022-12-18 21:42:55 +00001481
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001482 // Growarray with object members declared in the class.
1483 garray_T objmembers;
Bram Moolenaard505d172022-12-18 21:42:55 +00001484 ga_init2(&objmembers, sizeof(ocmember_T), 10);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001485
1486 // Growarray with object methods declared in the class.
1487 garray_T objmethods;
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001488 ga_init2(&objmethods, sizeof(ufunc_T *), 10);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001489
1490 /*
Bram Moolenaar554d0312023-01-05 19:59:18 +00001491 * Go over the body of the class/interface until "endclass" or
1492 * "endinterface" is found.
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001493 */
1494 char_u *theline = NULL;
1495 int success = FALSE;
1496 for (;;)
1497 {
1498 vim_free(theline);
1499 theline = eap->getline(':', eap->cookie, 0, GETLINE_CONCAT_ALL);
1500 if (theline == NULL)
1501 break;
1502 char_u *line = skipwhite(theline);
1503
Bram Moolenaar418b5472022-12-20 13:38:22 +00001504 // Skip empty and comment lines.
1505 if (*line == NUL)
1506 continue;
1507 if (*line == '#')
1508 {
1509 if (vim9_bad_comment(line))
1510 break;
1511 continue;
1512 }
1513
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001514 char_u *p = line;
Bram Moolenaar554d0312023-01-05 19:59:18 +00001515 char *end_name = is_class ? "endclass" : "endinterface";
1516 if (checkforcmd(&p, end_name, is_class ? 4 : 5))
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001517 {
Bram Moolenaar554d0312023-01-05 19:59:18 +00001518 if (STRNCMP(line, end_name, is_class ? 8 : 12) != 0)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001519 semsg(_(e_command_cannot_be_shortened_str), line);
1520 else if (*p == '|' || !ends_excmd2(line, p))
1521 semsg(_(e_trailing_characters_str), p);
Bram Moolenaar98aeb212022-12-08 22:09:14 +00001522 else
1523 success = TRUE;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001524 break;
1525 }
Bram Moolenaar554d0312023-01-05 19:59:18 +00001526 char *wrong_name = is_class ? "endinterface" : "endclass";
1527 if (checkforcmd(&p, wrong_name, is_class ? 5 : 4))
1528 {
Bram Moolenaar657aea72023-01-27 13:16:19 +00001529 semsg(_(e_invalid_command_str_expected_str), line, end_name);
Bram Moolenaar554d0312023-01-05 19:59:18 +00001530 break;
1531 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001532
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001533 int has_public = FALSE;
1534 if (checkforcmd(&p, "public", 3))
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001535 {
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001536 if (STRNCMP(line, "public", 6) != 0)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001537 {
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001538 semsg(_(e_command_cannot_be_shortened_str), line);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001539 break;
1540 }
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001541 has_public = TRUE;
1542 p = skipwhite(line + 6);
1543
Bram Moolenaard505d172022-12-18 21:42:55 +00001544 if (STRNCMP(p, "this", 4) != 0 && STRNCMP(p, "static", 6) != 0)
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001545 {
Bram Moolenaard505d172022-12-18 21:42:55 +00001546 emsg(_(e_public_must_be_followed_by_this_or_static));
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001547 break;
1548 }
1549 }
Bram Moolenaard505d172022-12-18 21:42:55 +00001550
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001551 int abstract_method = FALSE;
1552 char_u *pa = p;
1553 if (checkforcmd(&p, "abstract", 3))
1554 {
1555 if (STRNCMP(pa, "abstract", 8) != 0)
1556 {
1557 semsg(_(e_command_cannot_be_shortened_str), pa);
1558 break;
1559 }
1560
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001561 if (!is_class)
1562 // ignore "abstract" in an interface (as all the methods in an
1563 // interface are abstract.
1564 p = skipwhite(pa + 8);
1565 else
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001566 {
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001567 if (!is_abstract)
1568 {
1569 semsg(_(e_abstract_method_in_concrete_class), pa);
1570 break;
1571 }
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001572
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001573 abstract_method = TRUE;
1574 p = skipwhite(pa + 8);
1575 if (STRNCMP(p, "def", 3) != 0 && STRNCMP(p, "static", 6) != 0)
1576 {
1577 emsg(_(e_abstract_must_be_followed_by_def_or_static));
1578 break;
1579 }
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001580 }
1581 }
1582
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001583 int has_static = FALSE;
1584 char_u *ps = p;
1585 if (checkforcmd(&p, "static", 4))
1586 {
1587 if (STRNCMP(ps, "static", 6) != 0)
1588 {
1589 semsg(_(e_command_cannot_be_shortened_str), ps);
1590 break;
1591 }
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001592
1593 if (!is_class)
1594 {
1595 emsg(_(e_static_cannot_be_used_in_interface));
1596 break;
1597 }
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001598 has_static = TRUE;
1599 p = skipwhite(ps + 6);
1600 }
1601
Bram Moolenaard505d172022-12-18 21:42:55 +00001602 // object members (public, read access, private):
1603 // "this._varname"
1604 // "this.varname"
1605 // "public this.varname"
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001606 if (STRNCMP(p, "this", 4) == 0)
1607 {
1608 if (p[4] != '.' || !eval_isnamec1(p[5]))
1609 {
1610 semsg(_(e_invalid_object_member_declaration_str), p);
1611 break;
1612 }
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +02001613 if (has_static)
1614 {
1615 emsg(_(e_static_cannot_be_followed_by_this));
1616 break;
1617 }
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001618 char_u *varname = p + 5;
Bram Moolenaard505d172022-12-18 21:42:55 +00001619 char_u *varname_end = NULL;
Bram Moolenaar74e12742022-12-13 21:14:28 +00001620 type_T *type = NULL;
Bram Moolenaard505d172022-12-18 21:42:55 +00001621 char_u *init_expr = NULL;
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001622
1623 if (!is_class && *varname == '_')
1624 {
1625 // private variables are not supported in an interface
1626 semsg(_(e_private_variable_str_in_interface), varname);
1627 break;
1628 }
1629
Bram Moolenaard505d172022-12-18 21:42:55 +00001630 if (parse_member(eap, line, varname, has_public,
Bram Moolenaar554d0312023-01-05 19:59:18 +00001631 &varname_end, &type_list, &type,
1632 is_class ? &init_expr: NULL) == FAIL)
Bram Moolenaard505d172022-12-18 21:42:55 +00001633 break;
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +02001634 if (is_duplicate_member(&objmembers, varname, varname_end))
1635 {
1636 vim_free(init_expr);
1637 break;
1638 }
Bram Moolenaard505d172022-12-18 21:42:55 +00001639 if (add_member(&objmembers, varname, varname_end,
1640 has_public, type, init_expr) == FAIL)
Bram Moolenaar74e12742022-12-13 21:14:28 +00001641 {
Bram Moolenaard505d172022-12-18 21:42:55 +00001642 vim_free(init_expr);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001643 break;
1644 }
Bram Moolenaard505d172022-12-18 21:42:55 +00001645 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001646
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001647 // constructors:
1648 // def new()
1649 // enddef
1650 // def newOther()
1651 // enddef
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001652 // object methods and class functions:
1653 // def SomeMethod()
1654 // enddef
1655 // static def ClassFunction()
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001656 // enddef
1657 // TODO:
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001658 // def <Tval> someMethod()
1659 // enddef
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001660 else if (checkforcmd(&p, "def", 3))
1661 {
1662 exarg_T ea;
1663 garray_T lines_to_free;
1664
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001665 // TODO: error for "public static def Func()"?
1666
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001667 CLEAR_FIELD(ea);
1668 ea.cmd = line;
1669 ea.arg = p;
1670 ea.cmdidx = CMD_def;
1671 ea.getline = eap->getline;
1672 ea.cookie = eap->cookie;
1673
1674 ga_init2(&lines_to_free, sizeof(char_u *), 50);
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001675 int class_flags;
1676 if (is_class)
1677 class_flags = abstract_method ? CF_ABSTRACT_METHOD : CF_CLASS;
1678 else
1679 class_flags = CF_INTERFACE;
Bram Moolenaar554d0312023-01-05 19:59:18 +00001680 ufunc_T *uf = define_function(&ea, NULL, &lines_to_free,
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001681 class_flags);
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001682 ga_clear_strings(&lines_to_free);
1683
Bram Moolenaar6acf7572023-01-01 19:53:30 +00001684 if (uf != NULL)
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001685 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001686 char_u *name = uf->uf_name;
1687 int is_new = STRNCMP(name, "new", 3) == 0;
Gianmaria Bajo4b9777a2023-08-29 22:26:30 +02001688
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001689 if (!is_class && *name == '_')
1690 {
1691 // private variables are not supported in an interface
1692 semsg(_(e_private_method_str_in_interface), name);
1693 func_clear_free(uf, FALSE);
1694 break;
1695 }
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001696 if (is_new && !is_valid_constructor(uf, is_abstract,
1697 has_static))
Bram Moolenaar24a8d062023-01-14 13:12:06 +00001698 {
Yegappan Lakshmananb1027282023-08-19 11:26:42 +02001699 func_clear_free(uf, FALSE);
Bram Moolenaar24a8d062023-01-14 13:12:06 +00001700 break;
1701 }
Gianmaria Bajo4b9777a2023-08-29 22:26:30 +02001702
Bram Moolenaar58b40092023-01-11 15:59:05 +00001703 // Check the name isn't used already.
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001704 if (is_duplicate_method(&classfunctions, &objmethods, name))
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +02001705 {
1706 success = FALSE;
1707 func_clear_free(uf, FALSE);
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +02001708 break;
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +02001709 }
Bram Moolenaar58b40092023-01-11 15:59:05 +00001710
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001711 garray_T *fgap = has_static || is_new
1712 ? &classfunctions : &objmethods;
Bram Moolenaar6acf7572023-01-01 19:53:30 +00001713 if (ga_grow(fgap, 1) == OK)
1714 {
1715 if (is_new)
1716 uf->uf_flags |= FC_NEW;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00001717
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001718 if (abstract_method)
1719 uf->uf_flags |= FC_ABSTRACT;
1720
Bram Moolenaar6acf7572023-01-01 19:53:30 +00001721 ((ufunc_T **)fgap->ga_data)[fgap->ga_len] = uf;
1722 ++fgap->ga_len;
1723 }
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001724 }
1725 }
1726
1727 // class members
1728 else if (has_static)
1729 {
1730 // class members (public, read access, private):
1731 // "static _varname"
1732 // "static varname"
1733 // "public static varname"
1734 char_u *varname = p;
1735 char_u *varname_end = NULL;
1736 type_T *type = NULL;
1737 char_u *init_expr = NULL;
1738 if (parse_member(eap, line, varname, has_public,
Bram Moolenaar554d0312023-01-05 19:59:18 +00001739 &varname_end, &type_list, &type,
1740 is_class ? &init_expr : NULL) == FAIL)
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001741 break;
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +02001742 if (is_duplicate_member(&classmembers, varname, varname_end))
1743 {
1744 vim_free(init_expr);
1745 break;
1746 }
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001747 if (add_member(&classmembers, varname, varname_end,
1748 has_public, type, init_expr) == FAIL)
1749 {
1750 vim_free(init_expr);
1751 break;
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001752 }
1753 }
1754
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001755 else
1756 {
Bram Moolenaar554d0312023-01-05 19:59:18 +00001757 if (is_class)
1758 semsg(_(e_not_valid_command_in_class_str), line);
1759 else
1760 semsg(_(e_not_valid_command_in_interface_str), line);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001761 break;
1762 }
1763 }
1764 vim_free(theline);
1765
Bram Moolenaar83677162023-01-08 19:54:10 +00001766 class_T *extends_cl = NULL; // class from "extends" argument
1767
1768 /*
1769 * Check a few things before defining the class.
1770 */
1771
1772 // Check the "extends" class is valid.
1773 if (success && extends != NULL)
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001774 success = validate_extends_class(extends, &extends_cl, is_class);
Bram Moolenaar83677162023-01-08 19:54:10 +00001775 VIM_CLEAR(extends);
1776
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001777 // Check the new object methods to make sure their access (public or
1778 // private) is the same as that in the extended class lineage.
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +02001779 if (success && extends_cl != NULL)
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001780 success = validate_extends_methods(&objmethods, extends_cl);
1781
1782 // Check the new class and object variables are not duplicates of the
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001783 // variables in the extended class lineage. If an interface is extending
1784 // another interface, then it can duplicate the member variables.
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001785 if (success && extends_cl != NULL)
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001786 {
1787 if (is_class)
1788 success = extends_check_dup_members(&objmembers, extends_cl);
1789 else
1790 success = extends_check_intf_var_type(&objmembers, extends_cl);
1791 }
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +02001792
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001793 // When extending an abstract class, make sure all the abstract methods in
1794 // the parent class are implemented. If the current class is an abstract
1795 // class, then there is no need for this check.
1796 if (success && !is_abstract && extends_cl != NULL
1797 && (extends_cl->class_flags & CLASS_ABSTRACT))
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001798 success = validate_abstract_class_methods(&classfunctions,
1799 &objmethods, extends_cl);
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001800
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00001801 class_T **intf_classes = NULL;
1802
Bram Moolenaar83677162023-01-08 19:54:10 +00001803 // Check all "implements" entries are valid.
Bram Moolenaar94674f22023-01-06 18:42:20 +00001804 if (success && ga_impl.ga_len > 0)
1805 {
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00001806 intf_classes = ALLOC_CLEAR_MULT(class_T *, ga_impl.ga_len);
1807
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001808 success = validate_implements_classes(&ga_impl, intf_classes,
1809 &classfunctions, &classmembers,
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001810 &objmethods, &objmembers,
1811 extends_cl);
Bram Moolenaar94674f22023-01-06 18:42:20 +00001812 }
1813
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001814 // Check no function argument name is used as a class member.
Bram Moolenaard40f00c2023-01-13 17:36:49 +00001815 if (success)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001816 success = check_func_arg_names(&classfunctions, &objmethods,
1817 &classmembers);
Bram Moolenaard40f00c2023-01-13 17:36:49 +00001818
Bram Moolenaareb533502022-12-14 15:06:11 +00001819 class_T *cl = NULL;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001820 if (success)
1821 {
Bram Moolenaard505d172022-12-18 21:42:55 +00001822 // "endclass" encountered without failures: Create the class.
1823
Bram Moolenaareb533502022-12-14 15:06:11 +00001824 cl = ALLOC_CLEAR_ONE(class_T);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001825 if (cl == NULL)
1826 goto cleanup;
Bram Moolenaar554d0312023-01-05 19:59:18 +00001827 if (!is_class)
1828 cl->class_flags = CLASS_INTERFACE;
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001829 else if (is_abstract)
1830 cl->class_flags = CLASS_ABSTRACT;
Bram Moolenaar554d0312023-01-05 19:59:18 +00001831
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001832 cl->class_refcount = 1;
Bram Moolenaar94674f22023-01-06 18:42:20 +00001833 cl->class_name = vim_strnsave(name_start, name_end - name_start);
Bram Moolenaard505d172022-12-18 21:42:55 +00001834 if (cl->class_name == NULL)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001835 goto cleanup;
Bram Moolenaard505d172022-12-18 21:42:55 +00001836
Bram Moolenaard0200c82023-01-28 15:19:40 +00001837 if (extends_cl != NULL)
1838 {
1839 cl->class_extends = extends_cl;
1840 extends_cl->class_flags |= CLASS_EXTENDED;
1841 }
Bram Moolenaar83677162023-01-08 19:54:10 +00001842
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001843 // Add class and object variables to "cl".
Bram Moolenaard505d172022-12-18 21:42:55 +00001844 if (add_members_to_class(&classmembers,
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001845 NULL,
1846 0,
Bram Moolenaar83677162023-01-08 19:54:10 +00001847 &cl->class_class_members,
1848 &cl->class_class_member_count) == FAIL
Bram Moolenaard505d172022-12-18 21:42:55 +00001849 || add_members_to_class(&objmembers,
Bram Moolenaar83677162023-01-08 19:54:10 +00001850 extends_cl == NULL ? NULL
1851 : extends_cl->class_obj_members,
1852 extends_cl == NULL ? 0
1853 : extends_cl->class_obj_member_count,
1854 &cl->class_obj_members,
1855 &cl->class_obj_member_count) == FAIL)
Bram Moolenaard505d172022-12-18 21:42:55 +00001856 goto cleanup;
1857
Bram Moolenaar29ac5df2023-01-16 19:43:47 +00001858 if (ga_impl.ga_len > 0)
1859 {
1860 // Move the "implements" names into the class.
1861 cl->class_interface_count = ga_impl.ga_len;
1862 cl->class_interfaces = ALLOC_MULT(char_u *, ga_impl.ga_len);
1863 if (cl->class_interfaces == NULL)
1864 goto cleanup;
1865 for (int i = 0; i < ga_impl.ga_len; ++i)
1866 cl->class_interfaces[i] = ((char_u **)ga_impl.ga_data)[i];
1867 VIM_CLEAR(ga_impl.ga_data);
1868 ga_impl.ga_len = 0;
1869
Bram Moolenaard0200c82023-01-28 15:19:40 +00001870 cl->class_interfaces_cl = intf_classes;
1871 intf_classes = NULL;
1872 }
1873
1874 if (cl->class_interface_count > 0 || extends_cl != NULL)
1875 {
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001876 // Add a method and member lookup table to each of the interface
1877 // classes.
1878 if (add_lookup_tables(cl, extends_cl, &objmethods) == FAIL)
1879 goto cleanup;
Bram Moolenaar29ac5df2023-01-16 19:43:47 +00001880 }
1881
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001882 // Allocate a typval for each class member and initialize it.
Bram Moolenaar554d0312023-01-05 19:59:18 +00001883 if (is_class && cl->class_class_member_count > 0)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001884 add_class_members(cl, eap);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001885
Yegappan Lakshmanan6ac15442023-08-20 18:20:17 +02001886 int have_new = FALSE;
1887 ufunc_T *class_func = NULL;
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001888 for (int i = 0; i < classfunctions.ga_len; ++i)
Yegappan Lakshmanan6ac15442023-08-20 18:20:17 +02001889 {
1890 class_func = ((ufunc_T **)classfunctions.ga_data)[i];
1891 if (STRCMP(class_func->uf_name, "new") == 0)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001892 {
1893 have_new = TRUE;
1894 break;
1895 }
Yegappan Lakshmanan6ac15442023-08-20 18:20:17 +02001896 }
1897
1898 if (have_new)
1899 // The return type of new() is an object of class "cl"
1900 class_func->uf_ret_type->tt_class = cl;
1901 else if (is_class && !is_abstract && !have_new)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001902 // No new() method was defined, add the default constructor.
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001903 add_default_constructor(cl, &classfunctions, &type_list);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001904
Bram Moolenaar58b40092023-01-11 15:59:05 +00001905 // Move all the functions into the created class.
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001906 if (add_classfuncs_objmethods(cl, extends_cl, &classfunctions,
1907 &objmethods) == FAIL)
1908 goto cleanup;
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001909
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001910 cl->class_type.tt_type = VAR_CLASS;
Bram Moolenaarb1e32ac2023-02-21 12:38:51 +00001911 cl->class_type.tt_class = cl;
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001912 cl->class_object_type.tt_type = VAR_OBJECT;
Bram Moolenaarb1e32ac2023-02-21 12:38:51 +00001913 cl->class_object_type.tt_class = cl;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001914 cl->class_type_list = type_list;
1915
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02001916 class_created(cl);
1917
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001918 // TODO:
Bram Moolenaard505d172022-12-18 21:42:55 +00001919 // - Fill hashtab with object members and methods ?
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001920
1921 // Add the class to the script-local variables.
Bram Moolenaar94674f22023-01-06 18:42:20 +00001922 // TODO: handle other context, e.g. in a function
Ernie Rael21d32122023-09-02 15:09:18 +02001923 // TODO: does uf_hash need to be cleared?
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001924 typval_T tv;
1925 tv.v_type = VAR_CLASS;
1926 tv.vval.v_class = cl;
Bram Moolenaara86655a2023-01-12 17:06:27 +00001927 is_export = class_export;
Bram Moolenaar83ae6152023-02-25 19:59:31 +00001928 SOURCING_LNUM = start_lnum;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001929 set_var_const(cl->class_name, current_sctx.sc_sid,
Bram Moolenaar83ae6152023-02-25 19:59:31 +00001930 NULL, &tv, FALSE, 0, 0);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001931 return;
1932 }
1933
1934cleanup:
Bram Moolenaareb533502022-12-14 15:06:11 +00001935 if (cl != NULL)
1936 {
1937 vim_free(cl->class_name);
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001938 vim_free(cl->class_class_functions);
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00001939 if (cl->class_interfaces != NULL)
1940 {
1941 for (int i = 0; i < cl->class_interface_count; ++i)
1942 vim_free(cl->class_interfaces[i]);
1943 vim_free(cl->class_interfaces);
1944 }
1945 if (cl->class_interfaces_cl != NULL)
1946 {
1947 for (int i = 0; i < cl->class_interface_count; ++i)
1948 class_unref(cl->class_interfaces_cl[i]);
1949 vim_free(cl->class_interfaces_cl);
1950 }
Bram Moolenaareb533502022-12-14 15:06:11 +00001951 vim_free(cl->class_obj_members);
1952 vim_free(cl->class_obj_methods);
1953 vim_free(cl);
1954 }
1955
Bram Moolenaar83677162023-01-08 19:54:10 +00001956 vim_free(extends);
1957 class_unref(extends_cl);
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00001958
1959 if (intf_classes != NULL)
1960 {
1961 for (int i = 0; i < ga_impl.ga_len; ++i)
1962 class_unref(intf_classes[i]);
1963 vim_free(intf_classes);
1964 }
Bram Moolenaar94674f22023-01-06 18:42:20 +00001965 ga_clear_strings(&ga_impl);
1966
Bram Moolenaard505d172022-12-18 21:42:55 +00001967 for (int round = 1; round <= 2; ++round)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001968 {
Bram Moolenaard505d172022-12-18 21:42:55 +00001969 garray_T *gap = round == 1 ? &classmembers : &objmembers;
1970 if (gap->ga_len == 0 || gap->ga_data == NULL)
1971 continue;
1972
1973 for (int i = 0; i < gap->ga_len; ++i)
1974 {
1975 ocmember_T *m = ((ocmember_T *)gap->ga_data) + i;
1976 vim_free(m->ocm_name);
1977 vim_free(m->ocm_init);
1978 }
1979 ga_clear(gap);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001980 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001981
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001982 for (int i = 0; i < objmethods.ga_len; ++i)
1983 {
1984 ufunc_T *uf = ((ufunc_T **)objmethods.ga_data)[i];
1985 func_clear_free(uf, FALSE);
1986 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001987 ga_clear(&objmethods);
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001988
1989 for (int i = 0; i < classfunctions.ga_len; ++i)
1990 {
1991 ufunc_T *uf = ((ufunc_T **)classfunctions.ga_data)[i];
1992 func_clear_free(uf, FALSE);
1993 }
1994 ga_clear(&classfunctions);
1995
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001996 clear_type_list(&type_list);
1997}
1998
1999/*
Bram Moolenaarf54cedd2022-12-23 17:56:27 +00002000 * Find member "name" in class "cl", set "member_idx" to the member index and
2001 * return its type.
Yegappan Lakshmanan3775f772023-09-01 22:05:45 +02002002 * When "is_object" is TRUE, then look for object members. Otherwise look for
2003 * class members.
Bram Moolenaarf54cedd2022-12-23 17:56:27 +00002004 * When not found "member_idx" is set to -1 and t_any is returned.
Ernie Rael456ae552023-09-01 18:54:54 +02002005 * Set *p_m ocmmember_T if not NULL
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002006 */
2007 type_T *
2008class_member_type(
Yegappan Lakshmananeb91e242023-08-31 18:10:46 +02002009 class_T *cl,
Yegappan Lakshmanan3775f772023-09-01 22:05:45 +02002010 int is_object,
Yegappan Lakshmananeb91e242023-08-31 18:10:46 +02002011 char_u *name,
2012 char_u *name_end,
2013 int *member_idx,
Ernie Rael456ae552023-09-01 18:54:54 +02002014 ocmember_T **p_m)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002015{
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002016 size_t len = name_end - name;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002017 ocmember_T *m;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002018
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002019 *member_idx = -1; // not found (yet)
2020
2021 m = member_lookup(cl, is_object ? VAR_OBJECT : VAR_CLASS, name, len,
2022 member_idx);
2023 if (m == NULL)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002024 {
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002025 char_u *varname = vim_strnsave(name, len);
2026 if (varname != NULL)
2027 {
2028 if (is_object && class_member_idx(cl, name, len) >= 0)
2029 // A class variable with this name is present
2030 semsg(_(e_class_member_str_accessible_only_inside_class_str),
2031 varname, cl->class_name);
2032 else if (!is_object && object_member_idx(cl, name, len) >= 0)
2033 // An instance variable with this name is present
2034 semsg(_(e_object_member_str_accessible_only_using_object_str),
2035 varname, cl->class_name);
2036 else
2037 semsg(_(e_unknown_variable_str), varname);
2038 }
2039 vim_free(varname);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002040 return &t_any;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002041 }
Bram Moolenaarf54cedd2022-12-23 17:56:27 +00002042
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002043 if (p_m != NULL)
2044 *p_m = m;
2045
2046 return m->ocm_type;
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00002047}
2048
2049/*
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00002050 * Handle ":enum" up to ":endenum".
2051 */
2052 void
2053ex_enum(exarg_T *eap UNUSED)
2054{
2055 // TODO
2056}
2057
2058/*
2059 * Handle ":type".
2060 */
2061 void
2062ex_type(exarg_T *eap UNUSED)
2063{
2064 // TODO
2065}
2066
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002067/*
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002068 * Returns OK if a member variable named "name" is present in the class "cl".
2069 * Otherwise returns FAIL. If found, the member variable typval is set in
2070 * "rettv". If "is_object" is TRUE, then the object member variable table is
2071 * searched. Otherwise the class member variable table is searched.
2072 */
2073 static int
2074get_member_tv(
2075 class_T *cl,
2076 int is_object,
2077 char_u *name,
2078 size_t namelen,
2079 typval_T *rettv)
2080{
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002081 ocmember_T *m;
2082 int m_idx;
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002083
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002084 m = member_lookup(cl, is_object ? VAR_OBJECT : VAR_CLASS, name, namelen,
2085 &m_idx);
2086 if (m == NULL)
2087 return FAIL;
2088
2089 if (*name == '_')
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002090 {
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002091 semsg(_(e_cannot_access_private_member_str), m->ocm_name);
2092 return FAIL;
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002093 }
2094
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002095 // The object only contains a pointer to the class, the member
2096 // values array follows right after that.
2097 object_T *obj = rettv->vval.v_object;
2098 if (is_object)
2099 {
2100 typval_T *tv = (typval_T *)(obj + 1) + m_idx;
2101 copy_tv(tv, rettv);
2102 }
2103 else
2104 copy_tv(&cl->class_members_tv[m_idx], rettv);
2105
2106 object_unref(obj);
2107
2108 return OK;
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002109}
2110
2111/*
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002112 * Evaluate what comes after a class:
2113 * - class member: SomeClass.varname
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00002114 * - class function: SomeClass.SomeMethod()
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002115 * - class constructor: SomeClass.new()
2116 * - object member: someObject.varname
2117 * - object method: someObject.SomeMethod()
2118 *
2119 * "*arg" points to the '.'.
2120 * "*arg" is advanced to after the member name or method call.
2121 *
2122 * Returns FAIL or OK.
2123 */
2124 int
2125class_object_index(
2126 char_u **arg,
2127 typval_T *rettv,
2128 evalarg_T *evalarg,
2129 int verbose UNUSED) // give error messages
2130{
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002131 if (VIM_ISWHITE((*arg)[1]))
2132 {
2133 semsg(_(e_no_white_space_allowed_after_str_str), ".", *arg);
2134 return FAIL;
2135 }
2136
2137 ++*arg;
2138 char_u *name = *arg;
2139 char_u *name_end = find_name_end(name, NULL, NULL, FNE_CHECK_START);
2140 if (name_end == name)
2141 return FAIL;
2142 size_t len = name_end - name;
2143
Bram Moolenaar552bdca2023-02-17 21:08:50 +00002144 class_T *cl;
2145 if (rettv->v_type == VAR_CLASS)
2146 cl = rettv->vval.v_class;
2147 else // VAR_OBJECT
2148 {
2149 if (rettv->vval.v_object == NULL)
2150 {
2151 emsg(_(e_using_null_object));
2152 return FAIL;
2153 }
2154 cl = rettv->vval.v_object->obj_class;
2155 }
2156
Bram Moolenaard13dd302023-03-11 20:56:35 +00002157 if (cl == NULL)
2158 {
2159 emsg(_(e_incomplete_type));
2160 return FAIL;
2161 }
2162
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002163 if (*name_end == '(')
2164 {
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002165 ufunc_T *fp;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002166
Ernie Rael4d00b832023-09-11 19:54:42 +02002167 fp = method_lookup(cl, rettv->v_type, name, len, NULL);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002168 if (fp == NULL)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002169 {
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002170 method_not_found_msg(cl, rettv->v_type, name, len);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002171 return FAIL;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002172 }
2173
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002174 typval_T argvars[MAX_FUNC_ARGS + 1];
2175 int argcount = 0;
2176
2177 if (*fp->uf_name == '_')
2178 {
2179 // Cannot access a private method outside of a class
2180 semsg(_(e_cannot_access_private_method_str), name);
2181 return FAIL;
2182 }
2183
2184 char_u *argp = name_end;
2185 int ret = get_func_arguments(&argp, evalarg, 0,
2186 argvars, &argcount);
2187 if (ret == FAIL)
2188 return FAIL;
2189
2190 funcexe_T funcexe;
2191 CLEAR_FIELD(funcexe);
2192 funcexe.fe_evaluate = TRUE;
2193 if (rettv->v_type == VAR_OBJECT)
2194 {
2195 funcexe.fe_object = rettv->vval.v_object;
2196 ++funcexe.fe_object->obj_refcount;
2197 }
2198
2199 // Clear the class or object after calling the function, in
2200 // case the refcount is one.
2201 typval_T tv_tofree = *rettv;
2202 rettv->v_type = VAR_UNKNOWN;
2203
2204 // Call the user function. Result goes into rettv;
2205 int error = call_user_func_check(fp, argcount, argvars,
2206 rettv, &funcexe, NULL);
2207
2208 // Clear the previous rettv and the arguments.
2209 clear_tv(&tv_tofree);
2210 for (int idx = 0; idx < argcount; ++idx)
2211 clear_tv(&argvars[idx]);
2212
2213 if (error != FCERR_NONE)
2214 {
2215 user_func_error(error, printable_func_name(fp),
2216 funcexe.fe_found_var);
2217 return FAIL;
2218 }
2219 *arg = argp;
2220 return OK;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002221 }
2222
2223 else if (rettv->v_type == VAR_OBJECT)
2224 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002225 // Search in the object member variable table and the class member
2226 // variable table.
Yegappan Lakshmanan23c92d92023-09-09 11:33:29 +02002227 if (get_member_tv(cl, TRUE, name, len, rettv) == OK)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002228 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002229 *arg = name_end;
2230 return OK;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002231 }
2232
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002233 member_not_found_msg(cl, VAR_OBJECT, name, len);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002234 }
2235
Bram Moolenaard505d172022-12-18 21:42:55 +00002236 else if (rettv->v_type == VAR_CLASS)
2237 {
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002238 int m_idx;
2239
Bram Moolenaard505d172022-12-18 21:42:55 +00002240 // class member
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002241 ocmember_T *m = class_member_lookup(cl, name, len, &m_idx);
2242 if (m == NULL)
Bram Moolenaard505d172022-12-18 21:42:55 +00002243 {
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002244 member_not_found_msg(cl, VAR_CLASS, name, len);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002245 return FAIL;
Bram Moolenaard505d172022-12-18 21:42:55 +00002246 }
2247
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002248 if (*name == '_')
2249 {
2250 semsg(_(e_cannot_access_private_member_str), m->ocm_name);
2251 return FAIL;
2252 }
2253 if ((cl->class_flags & CLASS_INTERFACE) != 0)
2254 {
2255 semsg(_(e_interface_static_direct_access_str),
2256 cl->class_name, m->ocm_name);
2257 return FAIL;
2258 }
2259
2260 typval_T *tv = &cl->class_members_tv[m_idx];
2261 copy_tv(tv, rettv);
2262 class_unref(cl);
2263
2264 *arg = name_end;
2265 return OK;
Bram Moolenaard505d172022-12-18 21:42:55 +00002266 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002267
2268 return FAIL;
2269}
2270
2271/*
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002272 * If "arg" points to a class or object method, return it.
2273 * Otherwise return NULL.
2274 */
2275 ufunc_T *
2276find_class_func(char_u **arg)
2277{
2278 char_u *name = *arg;
2279 char_u *name_end = find_name_end(name, NULL, NULL, FNE_CHECK_START);
2280 if (name_end == name || *name_end != '.')
2281 return NULL;
2282
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002283 ufunc_T *fp = NULL;
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002284 size_t len = name_end - name;
2285 typval_T tv;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002286 tv.v_type = VAR_UNKNOWN;
Bram Moolenaar993dbc32023-01-01 20:31:30 +00002287 if (eval_variable(name, (int)len,
2288 0, &tv, NULL, EVAL_VAR_NOAUTOLOAD) == FAIL)
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002289 return NULL;
2290 if (tv.v_type != VAR_CLASS && tv.v_type != VAR_OBJECT)
Bram Moolenaareb533502022-12-14 15:06:11 +00002291 goto fail_after_eval;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002292
2293 class_T *cl = tv.v_type == VAR_CLASS ? tv.vval.v_class
2294 : tv.vval.v_object->obj_class;
2295 if (cl == NULL)
Bram Moolenaareb533502022-12-14 15:06:11 +00002296 goto fail_after_eval;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002297 char_u *fname = name_end + 1;
2298 char_u *fname_end = find_name_end(fname, NULL, NULL, FNE_CHECK_START);
2299 if (fname_end == fname)
Bram Moolenaareb533502022-12-14 15:06:11 +00002300 goto fail_after_eval;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002301 len = fname_end - fname;
2302
Ernie Rael4d00b832023-09-11 19:54:42 +02002303 fp = method_lookup(cl, tv.v_type, fname, len, NULL);
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002304
Bram Moolenaareb533502022-12-14 15:06:11 +00002305fail_after_eval:
2306 clear_tv(&tv);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002307 return fp;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002308}
2309
2310/*
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002311 * Returns the index of class variable "name" in the class "cl".
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002312 * Returns -1, if the variable is not found.
2313 * If "namelen" is zero, then it is assumed that "name" is NUL terminated.
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00002314 */
2315 int
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002316class_member_idx(class_T *cl, char_u *name, size_t namelen)
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00002317{
Ernie Rael4d00b832023-09-11 19:54:42 +02002318 int idx;
2319 class_member_lookup(cl, name, namelen, &idx);
2320 return idx;
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00002321}
2322
2323/*
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002324 * Returns a pointer to the class member variable "name" in the class "cl".
2325 * Returns NULL if the variable is not found.
2326 * The member variable index is set in "idx".
2327 */
2328 ocmember_T *
2329class_member_lookup(class_T *cl, char_u *name, size_t namelen, int *idx)
2330{
Ernie Rael4d00b832023-09-11 19:54:42 +02002331 ocmember_T *ret_m = NULL;
2332 int ret_idx = -1;
2333 for (int i = 0; i < cl->class_class_member_count; ++i)
2334 {
2335 ocmember_T *m = &cl->class_class_members[i];
2336 if (namelen)
2337 {
2338 if (STRNCMP(name, m->ocm_name, namelen) == 0
2339 && m->ocm_name[namelen] == NUL)
2340 {
2341 ret_m = m;
2342 ret_idx = i;
2343 break;
2344 }
2345 }
2346 else if (STRCMP(name, m->ocm_name) == 0)
2347 {
2348 ret_m = m;
2349 ret_idx = i;
2350 break;
2351 }
2352 }
2353 if (idx != NULL)
2354 *idx = ret_idx;
2355 return ret_m;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002356}
2357
2358/*
2359 * Returns the index of class method "name" in the class "cl".
2360 * Returns -1, if the method is not found.
Yegappan Lakshmanan342f4f62023-09-09 11:37:23 +02002361 */
2362 int
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002363class_method_idx(class_T *cl, char_u *name, size_t namelen)
Yegappan Lakshmanan342f4f62023-09-09 11:37:23 +02002364{
Ernie Rael4d00b832023-09-11 19:54:42 +02002365 int idx;
2366 class_method_lookup(cl, name, namelen, &idx);
2367 return idx;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002368}
2369
2370/*
2371 * Returns a pointer to the class method "name" in class "cl".
2372 * Returns NULL if the method is not found.
2373 * The method index is set in "idx".
2374 */
2375 ufunc_T *
2376class_method_lookup(class_T *cl, char_u *name, size_t namelen, int *idx)
2377{
Ernie Rael4d00b832023-09-11 19:54:42 +02002378 ufunc_T *ret_fp = NULL;
2379 int ret_idx = -1;
2380 for (int i = 0; i < cl->class_class_function_count; ++i)
2381 {
2382 ufunc_T *fp = cl->class_class_functions[i];
2383 char_u *ufname = (char_u *)fp->uf_name;
2384 if (STRNCMP(name, ufname, namelen) == 0 && ufname[namelen] == NUL)
2385 {
2386 ret_fp = fp;
2387 ret_idx = i;
2388 break;
2389 }
2390 }
2391 if (idx != NULL)
2392 *idx = ret_idx;
2393 return ret_fp;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002394}
2395
2396/*
2397 * Returns the index of object member variable "name" in the class "cl".
2398 * Returns -1, if the variable is not found.
2399 * If "namelen" is zero, then it is assumed that "name" is NUL terminated.
2400 */
2401 int
2402object_member_idx(class_T *cl, char_u *name, size_t namelen)
2403{
Ernie Rael4d00b832023-09-11 19:54:42 +02002404 int idx;
2405 object_member_lookup(cl, name, namelen, &idx);
2406 return idx;
Yegappan Lakshmanan342f4f62023-09-09 11:37:23 +02002407}
2408
2409/*
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002410 * Returns a pointer to the object member variable "name" in the class "cl".
2411 * Returns NULL if the variable is not found.
2412 * The object member variable index is set in "idx".
2413 */
2414 ocmember_T *
2415object_member_lookup(class_T *cl, char_u *name, size_t namelen, int *idx)
2416{
Ernie Rael4d00b832023-09-11 19:54:42 +02002417 ocmember_T *ret_m = NULL;
2418 int ret_idx = -1;
2419 for (int i = 0; i < cl->class_obj_member_count; ++i)
2420 {
2421 ocmember_T *m = &cl->class_obj_members[i];
2422 if (namelen)
2423 {
2424 if (STRNCMP(name, m->ocm_name, namelen) == 0
2425 && m->ocm_name[namelen] == NUL)
2426 {
2427 ret_m = m;
2428 ret_idx = i;
2429 break;
2430 }
2431 }
2432 else if (STRCMP(name, m->ocm_name) == 0)
2433 {
2434 ret_m = m;
2435 ret_idx = i;
2436 break;
2437 }
2438 }
2439 if (idx != NULL)
2440 *idx = ret_idx;
2441 return ret_m;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002442}
2443
2444/*
2445 * Returns the index of object method "name" in the class "cl".
2446 * Returns -1, if the method is not found.
2447 */
2448 int
2449object_method_idx(class_T *cl, char_u *name, size_t namelen)
2450{
Ernie Rael4d00b832023-09-11 19:54:42 +02002451 int idx;
2452 object_method_lookup(cl, name, namelen, &idx);
2453 return idx;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002454}
2455
2456/*
2457 * Returns a pointer to the object method "name" in class "cl".
2458 * Returns NULL if the method is not found.
2459 * The object method index is set in "idx".
2460 */
2461 ufunc_T *
2462object_method_lookup(class_T *cl, char_u *name, size_t namelen, int *idx)
2463{
Ernie Rael4d00b832023-09-11 19:54:42 +02002464 ufunc_T *ret_fp = NULL;
2465 int ret_idx = -1;
2466 for (int i = 0; i < cl->class_obj_method_count; ++i)
2467 {
2468 ufunc_T *fp = cl->class_obj_methods[i];
2469 // Use a separate pointer to avoid that ASAN complains about
2470 // uf_name[] only being 4 characters.
2471 char_u *ufname = (char_u *)fp->uf_name;
2472 if (STRNCMP(name, ufname, namelen) == 0 && ufname[namelen] == NUL)
2473 {
2474 ret_fp = fp;
2475 ret_idx = i;
2476 break;
2477 }
2478 }
2479 if (idx != NULL)
2480 *idx = ret_idx;
2481 return ret_fp;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002482}
2483
2484/*
2485 * Lookup a class or object member variable by name. If v_type is VAR_CLASS,
2486 * then lookup a class member variable and if it is VAR_OBJECT, then lookup a
2487 * object member variable.
2488 *
2489 * Returns a pointer to the member variable structure if variable is found.
2490 * Otherwise returns NULL. The member variable index is set in "*idx".
2491 */
2492 ocmember_T *
2493member_lookup(
2494 class_T *cl,
2495 vartype_T v_type,
2496 char_u *name,
2497 size_t namelen,
2498 int *idx)
2499{
2500 if (v_type == VAR_CLASS)
2501 return class_member_lookup(cl, name, namelen, idx);
2502 else
2503 return object_member_lookup(cl, name, namelen, idx);
2504}
2505
2506/*
2507 * Lookup a class or object method by name. If v_type is VAR_CLASS, then
2508 * lookup a class method and if it is VAR_OBJECT, then lookup a object method.
2509 *
2510 * Returns a pointer to the method structure if variable is found.
2511 * Otherwise returns NULL. The method variable index is set in "*idx".
2512 */
2513 ufunc_T *
2514method_lookup(
2515 class_T *cl,
2516 vartype_T v_type,
2517 char_u *name,
2518 size_t namelen,
2519 int *idx)
2520{
2521 if (v_type == VAR_CLASS)
2522 return class_method_lookup(cl, name, namelen, idx);
2523 else
2524 return object_method_lookup(cl, name, namelen, idx);
2525}
2526
2527/*
Bram Moolenaar62a69232023-01-24 15:07:04 +00002528 * Return TRUE if current context "cctx_arg" is inside class "cl".
2529 * Return FALSE if not.
2530 */
2531 int
2532inside_class(cctx_T *cctx_arg, class_T *cl)
2533{
2534 for (cctx_T *cctx = cctx_arg; cctx != NULL; cctx = cctx->ctx_outer)
Ernie Raelcf138d42023-09-06 20:45:03 +02002535 if (cctx->ctx_ufunc != NULL
2536 && class_instance_of(cctx->ctx_ufunc->uf_class, cl))
Bram Moolenaar62a69232023-01-24 15:07:04 +00002537 return TRUE;
2538 return FALSE;
2539}
2540
2541/*
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002542 * Make a copy of an object.
2543 */
2544 void
2545copy_object(typval_T *from, typval_T *to)
2546{
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +02002547 if (from->vval.v_object == NULL)
2548 to->vval.v_object = NULL;
2549 else
2550 {
2551 to->vval.v_object = from->vval.v_object;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002552 ++to->vval.v_object->obj_refcount;
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +02002553 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002554}
2555
2556/*
2557 * Free an object.
2558 */
2559 static void
2560object_clear(object_T *obj)
2561{
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002562 // Avoid a recursive call, it can happen if "obj" has a circular reference.
2563 obj->obj_refcount = INT_MAX;
2564
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002565 class_T *cl = obj->obj_class;
2566
Jia-Ju Bai5b0889b2023-08-13 20:04:04 +02002567 if (!cl)
Yegappan Lakshmanand4e4ecb2023-08-27 18:35:45 +02002568 return;
Jia-Ju Bai5b0889b2023-08-13 20:04:04 +02002569
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002570 // the member values are just after the object structure
2571 typval_T *tv = (typval_T *)(obj + 1);
2572 for (int i = 0; i < cl->class_obj_member_count; ++i)
2573 clear_tv(tv + i);
2574
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002575 // Remove from the list headed by "first_object".
2576 object_cleared(obj);
2577
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002578 vim_free(obj);
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002579 class_unref(cl);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002580}
2581
2582/*
2583 * Unreference an object.
2584 */
2585 void
2586object_unref(object_T *obj)
2587{
2588 if (obj != NULL && --obj->obj_refcount <= 0)
2589 object_clear(obj);
2590}
2591
2592/*
2593 * Make a copy of a class.
2594 */
2595 void
2596copy_class(typval_T *from, typval_T *to)
2597{
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +02002598 if (from->vval.v_class == NULL)
2599 to->vval.v_class = NULL;
2600 else
2601 {
2602 to->vval.v_class = from->vval.v_class;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002603 ++to->vval.v_class->class_refcount;
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +02002604 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002605}
2606
2607/*
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02002608 * Free the class "cl" and its contents.
2609 */
2610 static void
2611class_free(class_T *cl)
2612{
2613 // Freeing what the class contains may recursively come back here.
2614 // Clear "class_name" first, if it is NULL the class does not need to
2615 // be freed.
2616 VIM_CLEAR(cl->class_name);
2617
2618 class_unref(cl->class_extends);
2619
2620 for (int i = 0; i < cl->class_interface_count; ++i)
2621 {
2622 vim_free(((char_u **)cl->class_interfaces)[i]);
2623 if (cl->class_interfaces_cl[i] != NULL)
2624 class_unref(cl->class_interfaces_cl[i]);
2625 }
2626 vim_free(cl->class_interfaces);
2627 vim_free(cl->class_interfaces_cl);
2628
2629 itf2class_T *next;
2630 for (itf2class_T *i2c = cl->class_itf2class; i2c != NULL; i2c = next)
2631 {
2632 next = i2c->i2c_next;
2633 vim_free(i2c);
2634 }
2635
2636 for (int i = 0; i < cl->class_class_member_count; ++i)
2637 {
2638 ocmember_T *m = &cl->class_class_members[i];
2639 vim_free(m->ocm_name);
2640 vim_free(m->ocm_init);
2641 if (cl->class_members_tv != NULL)
2642 clear_tv(&cl->class_members_tv[i]);
2643 }
2644 vim_free(cl->class_class_members);
2645 vim_free(cl->class_members_tv);
2646
2647 for (int i = 0; i < cl->class_obj_member_count; ++i)
2648 {
2649 ocmember_T *m = &cl->class_obj_members[i];
2650 vim_free(m->ocm_name);
2651 vim_free(m->ocm_init);
2652 }
2653 vim_free(cl->class_obj_members);
2654
2655 for (int i = 0; i < cl->class_class_function_count; ++i)
2656 {
2657 ufunc_T *uf = cl->class_class_functions[i];
2658 func_clear_free(uf, FALSE);
2659 }
2660 vim_free(cl->class_class_functions);
2661
2662 for (int i = 0; i < cl->class_obj_method_count; ++i)
2663 {
2664 ufunc_T *uf = cl->class_obj_methods[i];
2665 func_clear_free(uf, FALSE);
2666 }
2667 vim_free(cl->class_obj_methods);
2668
2669 clear_type_list(&cl->class_type_list);
2670
2671 class_cleared(cl);
2672
2673 vim_free(cl);
2674}
2675
2676/*
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002677 * Unreference a class. Free it when the reference count goes down to zero.
2678 */
2679 void
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002680class_unref(class_T *cl)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002681{
Bram Moolenaard505d172022-12-18 21:42:55 +00002682 if (cl != NULL && --cl->class_refcount <= 0 && cl->class_name != NULL)
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02002683 class_free(cl);
2684}
2685
2686/*
2687 * Go through the list of all classes and free items without "copyID".
2688 */
2689 int
2690class_free_nonref(int copyID)
2691{
2692 int did_free = FALSE;
2693
2694 for (class_T *cl = first_class; cl != NULL; cl = next_nonref_class)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002695 {
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02002696 next_nonref_class = cl->class_next_used;
2697 if ((cl->class_copyID & COPYID_MASK) != (copyID & COPYID_MASK))
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00002698 {
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02002699 // Free the class and items it contains.
2700 class_free(cl);
2701 did_free = TRUE;
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00002702 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002703 }
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02002704
2705 next_nonref_class = NULL;
2706 return did_free;
2707}
2708
2709 int
2710set_ref_in_classes(int copyID)
2711{
2712 for (class_T *cl = first_class; cl != NULL; cl = cl->class_next_used)
2713 set_ref_in_item_class(cl, copyID, NULL, NULL);
2714
2715 return FALSE;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002716}
2717
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002718static object_T *first_object = NULL;
2719
2720/*
2721 * Call this function when an object has been created. It will be added to the
2722 * list headed by "first_object".
2723 */
2724 void
2725object_created(object_T *obj)
2726{
2727 if (first_object != NULL)
2728 {
2729 obj->obj_next_used = first_object;
2730 first_object->obj_prev_used = obj;
2731 }
2732 first_object = obj;
2733}
2734
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002735static object_T *next_nonref_obj = NULL;
2736
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002737/*
2738 * Call this function when an object has been cleared and is about to be freed.
2739 * It is removed from the list headed by "first_object".
2740 */
2741 void
2742object_cleared(object_T *obj)
2743{
2744 if (obj->obj_next_used != NULL)
2745 obj->obj_next_used->obj_prev_used = obj->obj_prev_used;
2746 if (obj->obj_prev_used != NULL)
2747 obj->obj_prev_used->obj_next_used = obj->obj_next_used;
2748 else if (first_object == obj)
2749 first_object = obj->obj_next_used;
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002750
2751 // update the next object to check if needed
2752 if (obj == next_nonref_obj)
2753 next_nonref_obj = obj->obj_next_used;
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002754}
2755
2756/*
2757 * Go through the list of all objects and free items without "copyID".
2758 */
2759 int
2760object_free_nonref(int copyID)
2761{
2762 int did_free = FALSE;
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002763
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002764 for (object_T *obj = first_object; obj != NULL; obj = next_nonref_obj)
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002765 {
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002766 next_nonref_obj = obj->obj_next_used;
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002767 if ((obj->obj_copyID & COPYID_MASK) != (copyID & COPYID_MASK))
2768 {
2769 // Free the object and items it contains.
2770 object_clear(obj);
2771 did_free = TRUE;
2772 }
2773 }
2774
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002775 next_nonref_obj = NULL;
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002776 return did_free;
2777}
2778
LemonBoyafe04662023-08-23 21:08:11 +02002779/*
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002780 * Echo a class or object method not found message.
2781 */
2782 void
2783method_not_found_msg(class_T *cl, vartype_T v_type, char_u *name, size_t len)
2784{
2785 char_u *method_name = vim_strnsave(name, len);
2786 if ((v_type == VAR_OBJECT)
2787 && (class_method_idx(cl, name, len) >= 0))
2788 {
2789 // If this is a class method, then give a different error
2790 if (*name == '_')
2791 semsg(_(e_cannot_access_private_method_str), method_name);
2792 else
2793 semsg(_(e_class_member_str_accessible_only_using_class_str),
2794 method_name, cl->class_name);
2795 }
2796 else if ((v_type == VAR_CLASS)
2797 && (object_method_idx(cl, name, len) >= 0))
2798 {
2799 // If this is an object method, then give a different error
2800 if (*name == '_')
2801 semsg(_(e_cannot_access_private_method_str), method_name);
2802 else
2803 semsg(_(e_object_member_str_accessible_only_using_object_str),
2804 method_name, cl->class_name);
2805 }
2806 else
2807 semsg(_(e_method_not_found_on_class_str_str), cl->class_name,
2808 method_name);
2809 vim_free(method_name);
2810}
2811
2812/*
2813 * Echo a class or object member not found message.
2814 */
2815 void
2816member_not_found_msg(class_T *cl, vartype_T v_type, char_u *name, size_t len)
2817{
2818 char_u *varname = len ? vim_strnsave(name, len) : vim_strsave(name);
2819
2820 if (v_type == VAR_OBJECT)
2821 {
2822 if (class_member_idx(cl, name, len) >= 0)
2823 semsg(_(e_class_member_str_accessible_only_using_class_str),
2824 varname, cl->class_name);
2825 else
2826 semsg(_(e_member_not_found_on_object_str_str), cl->class_name,
2827 varname);
2828 }
2829 else
2830 {
2831 if (object_member_idx(cl, name, len) >= 0)
2832 semsg(_(e_object_member_str_accessible_only_using_object_str),
2833 varname, cl->class_name);
2834 else
2835 semsg(_(e_class_member_str_not_found_in_class_str),
2836 varname, cl->class_name);
2837 }
2838 vim_free(varname);
2839}
2840
2841/*
Yegappan Lakshmanand4e4ecb2023-08-27 18:35:45 +02002842 * Return TRUE when the class "cl", its base class or one of the implemented
2843 * interfaces matches the class "other_cl".
LemonBoyafe04662023-08-23 21:08:11 +02002844 */
2845 int
2846class_instance_of(class_T *cl, class_T *other_cl)
2847{
2848 if (cl == other_cl)
2849 return TRUE;
2850
2851 // Recursively check the base classes.
2852 for (; cl != NULL; cl = cl->class_extends)
2853 {
2854 if (cl == other_cl)
2855 return TRUE;
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02002856 // Check the implemented interfaces and the super interfaces
LemonBoyafe04662023-08-23 21:08:11 +02002857 for (int i = cl->class_interface_count - 1; i >= 0; --i)
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02002858 {
2859 class_T *intf = cl->class_interfaces_cl[i];
2860 while (intf != NULL)
2861 {
2862 if (intf == other_cl)
2863 return TRUE;
2864 // check the super interfaces
2865 intf = intf->class_extends;
2866 }
2867 }
LemonBoyafe04662023-08-23 21:08:11 +02002868 }
2869
2870 return FALSE;
2871}
2872
2873/*
2874 * "instanceof(object, classinfo)" function
2875 */
2876 void
2877f_instanceof(typval_T *argvars, typval_T *rettv)
2878{
2879 typval_T *object_tv = &argvars[0];
2880 typval_T *classinfo_tv = &argvars[1];
2881 listitem_T *li;
2882
2883 rettv->vval.v_number = VVAL_FALSE;
2884
2885 if (check_for_object_arg(argvars, 0) == FAIL
2886 || check_for_class_or_list_arg(argvars, 1) == FAIL)
2887 return;
2888
2889 if (classinfo_tv->v_type == VAR_LIST)
2890 {
2891 FOR_ALL_LIST_ITEMS(classinfo_tv->vval.v_list, li)
2892 {
2893 if (li->li_tv.v_type != VAR_CLASS)
2894 {
2895 emsg(_(e_class_required));
2896 return;
2897 }
2898
2899 if (class_instance_of(object_tv->vval.v_object->obj_class,
2900 li->li_tv.vval.v_class) == TRUE)
2901 {
2902 rettv->vval.v_number = VVAL_TRUE;
2903 return;
2904 }
2905 }
2906 }
2907 else if (classinfo_tv->v_type == VAR_CLASS)
2908 {
2909 rettv->vval.v_number = class_instance_of(object_tv->vval.v_object->obj_class,
2910 classinfo_tv->vval.v_class);
2911 }
2912}
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00002913
2914#endif // FEAT_EVAL