blob: 8b07d76498db7ef0dd6e67717ee5bde448f11625 [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 {
RestorerZ7fe8f432023-09-24 23:21:24 +020082 semsg(_(e_public_variable_name_cannot_start_with_underscore_str), line);
Bram Moolenaard505d172022-12-18 21:42:55 +000083 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 {
RestorerZ7fe8f432023-09-24 23:21:24 +0200116 emsg(_(e_cannot_initialize_variable_in_interface));
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +0200117 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 {
RestorerZ7fe8f432023-09-24 23:21:24 +0200429 semsg(_(e_duplicate_variable_str), c_m->ocm_name);
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +0200430 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];
Yegappan Lakshmanan1db15142023-09-19 20:34:05 +0200537 if (!IS_ABSTRACT_METHOD(uf))
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +0200538 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 {
RestorerZ7fe8f432023-09-24 23:21:24 +0200589 semsg(_(e_variable_str_of_interface_str_has_different_access),
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +0200590 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 {
RestorerZ7fe8f432023-09-24 23:21:24 +0200665 semsg(_(e_variable_str_of_interface_str_not_implemented),
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +0200666 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 {
RestorerZ7fe8f432023-09-24 23:21:24 +0200911 semsg(_(e_duplicate_variable_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 {
RestorerZ7fe8f432023-09-24 23:21:24 +0200960 emsg(_(e_cannot_define_new_method_in_abstract_class));
Gianmaria Bajo4b9777a2023-08-29 22:26:30 +0200961 return FALSE;
962 }
963 // A constructor is always static, no need to define it so.
964 if (has_static)
965 {
RestorerZ7fe8f432023-09-24 23:21:24 +0200966 emsg(_(e_cannot_define_new_method_as_static));
Gianmaria Bajo4b9777a2023-08-29 22:26:30 +0200967 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 {
RestorerZ7fe8f432023-09-24 23:21:24 +0200973 emsg(_(e_cannot_use_a_return_type_with_new_method));
Gianmaria Bajo4b9777a2023-08-29 22:26:30 +0200974 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
h-eastb895b0f2023-09-24 15:46:31 +02001203 ufunc_T *nf = define_function(&fea, NULL, &lines_to_free, CF_CLASS,
1204 cl->class_obj_members, cl->class_obj_member_count);
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001205
1206 ga_clear_strings(&lines_to_free);
1207 vim_free(fga.ga_data);
1208
1209 if (nf != NULL && ga_grow(classfunctions_gap, 1) == OK)
1210 {
1211 ((ufunc_T **)classfunctions_gap->ga_data)[classfunctions_gap->ga_len]
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001212 = nf;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001213 ++classfunctions_gap->ga_len;
1214
1215 nf->uf_flags |= FC_NEW;
1216 nf->uf_ret_type = get_type_ptr(type_list_gap);
1217 if (nf->uf_ret_type != NULL)
1218 {
1219 nf->uf_ret_type->tt_type = VAR_OBJECT;
1220 nf->uf_ret_type->tt_class = cl;
1221 nf->uf_ret_type->tt_argcount = 0;
1222 nf->uf_ret_type->tt_args = NULL;
1223 }
1224 }
1225}
1226
1227/*
Yegappan Lakshmanane2deb7e2023-09-16 18:05:07 +02001228 * Add the class methods and object methods to the new class "cl".
1229 * When extending a class "extends_cl", add the instance methods from the
1230 * parent class also.
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001231 */
1232 static int
1233add_classfuncs_objmethods(
1234 class_T *cl,
1235 class_T *extends_cl,
1236 garray_T *classfunctions_gap,
1237 garray_T *objmethods_gap)
1238{
1239 // loop 1: class functions, loop 2: object methods
1240 for (int loop = 1; loop <= 2; ++loop)
1241 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001242 garray_T *gap = loop == 1 ? classfunctions_gap : objmethods_gap;
1243 int *fcount = loop == 1 ? &cl->class_class_function_count
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001244 : &cl->class_obj_method_count;
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001245 ufunc_T ***fup = loop == 1 ? &cl->class_class_functions
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001246 : &cl->class_obj_methods;
1247
1248 int parent_count = 0;
1249 if (extends_cl != NULL)
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001250 // Include object methods from the parent.
1251 // Don't include the parent class methods.
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001252 parent_count = loop == 1
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001253 ? 0
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001254 : extends_cl->class_obj_method_count;
1255
1256 *fcount = parent_count + gap->ga_len;
1257 if (*fcount == 0)
1258 {
1259 *fup = NULL;
1260 continue;
1261 }
1262 *fup = ALLOC_MULT(ufunc_T *, *fcount);
1263 if (*fup == NULL)
1264 return FAIL;
1265
1266 if (gap->ga_len != 0)
1267 mch_memmove(*fup, gap->ga_data, sizeof(ufunc_T *) * gap->ga_len);
1268 vim_free(gap->ga_data);
1269 if (loop == 1)
1270 cl->class_class_function_count_child = gap->ga_len;
1271 else
1272 cl->class_obj_method_count_child = gap->ga_len;
1273
Yegappan Lakshmanane2deb7e2023-09-16 18:05:07 +02001274 if (loop == 2)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001275 {
Yegappan Lakshmanane2deb7e2023-09-16 18:05:07 +02001276 // Copy instance methods from the parent.
1277
1278 for (int i = 0; i < parent_count; ++i)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001279 {
Yegappan Lakshmanane2deb7e2023-09-16 18:05:07 +02001280 // Can't use the same parent function, because "uf_class" is
1281 // different and compilation will have a different result.
1282 // Put them after the functions in the current class, object
1283 // methods may be overruled, then "super.Method()" is used to
1284 // find a method from the parent.
1285 ufunc_T *pf = (extends_cl->class_obj_methods)[i];
1286 (*fup)[gap->ga_len + i] = copy_function(pf);
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001287
1288 // If the child class overrides a function from the parent
1289 // the signature must be equal.
1290 char_u *pname = pf->uf_name;
1291 for (int ci = 0; ci < gap->ga_len; ++ci)
1292 {
1293 ufunc_T *cf = (*fup)[ci];
1294 char_u *cname = cf->uf_name;
1295 if (STRCMP(pname, cname) == 0)
1296 {
1297 where_T where = WHERE_INIT;
1298 where.wt_func_name = (char *)pname;
1299 where.wt_kind = WT_METHOD;
1300 (void)check_type(pf->uf_func_type, cf->uf_func_type,
1301 TRUE, where);
1302 }
1303 }
1304 }
1305 }
1306
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001307 // Set the class pointer on all the functions and object methods.
1308 for (int i = 0; i < *fcount; ++i)
1309 {
1310 ufunc_T *fp = (*fup)[i];
1311 fp->uf_class = cl;
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001312 if (i < gap->ga_len)
1313 fp->uf_defclass = cl;
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001314 if (loop == 2)
1315 fp->uf_flags |= FC_OBJECT;
1316 }
1317 }
1318
1319 return OK;
1320}
1321
1322/*
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001323 * Handle ":class" and ":abstract class" up to ":endclass".
Bram Moolenaar554d0312023-01-05 19:59:18 +00001324 * Handle ":interface" up to ":endinterface".
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001325 */
1326 void
1327ex_class(exarg_T *eap)
1328{
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001329 int is_class = eap->cmdidx == CMD_class; // FALSE for :interface
1330 long start_lnum = SOURCING_LNUM;
1331 char_u *arg = eap->arg;
1332 int is_abstract = eap->cmdidx == CMD_abstract;
Bram Moolenaar554d0312023-01-05 19:59:18 +00001333
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001334 if (is_abstract)
1335 {
1336 if (STRNCMP(arg, "class", 5) != 0 || !VIM_ISWHITE(arg[5]))
1337 {
1338 semsg(_(e_invalid_argument_str), arg);
1339 return;
1340 }
1341 arg = skipwhite(arg + 5);
Bram Moolenaar24a8d062023-01-14 13:12:06 +00001342 is_class = TRUE;
1343 }
1344
1345 if (!current_script_is_vim9()
1346 || (cmdmod.cmod_flags & CMOD_LEGACY)
1347 || !getline_equal(eap->getline, eap->cookie, getsourceline))
1348 {
1349 if (is_class)
1350 emsg(_(e_class_can_only_be_defined_in_vim9_script));
1351 else
1352 emsg(_(e_interface_can_only_be_defined_in_vim9_script));
1353 return;
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001354 }
1355
1356 if (!ASCII_ISUPPER(*arg))
1357 {
Bram Moolenaar554d0312023-01-05 19:59:18 +00001358 if (is_class)
1359 semsg(_(e_class_name_must_start_with_uppercase_letter_str), arg);
1360 else
1361 semsg(_(e_interface_name_must_start_with_uppercase_letter_str),
1362 arg);
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001363 return;
1364 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001365 char_u *name_end = find_name_end(arg, NULL, NULL, FNE_CHECK_START);
1366 if (!IS_WHITE_OR_NUL(*name_end))
1367 {
Bram Moolenaar554d0312023-01-05 19:59:18 +00001368 semsg(_(e_white_space_required_after_name_str), arg);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001369 return;
1370 }
Bram Moolenaar94674f22023-01-06 18:42:20 +00001371 char_u *name_start = arg;
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001372
Bram Moolenaara86655a2023-01-12 17:06:27 +00001373 // "export class" gets used when creating the class, don't use "is_export"
1374 // for the items inside the class.
1375 int class_export = is_export;
1376 is_export = FALSE;
1377
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001378 // TODO:
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001379 // generics: <Tkey, Tentry>
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00001380
Bram Moolenaar83677162023-01-08 19:54:10 +00001381 // Name for "extends BaseClass"
1382 char_u *extends = NULL;
1383
Bram Moolenaar94674f22023-01-06 18:42:20 +00001384 // Names for "implements SomeInterface"
1385 garray_T ga_impl;
1386 ga_init2(&ga_impl, sizeof(char_u *), 5);
1387
1388 arg = skipwhite(name_end);
1389 while (*arg != NUL && *arg != '#' && *arg != '\n')
1390 {
1391 // TODO:
Bram Moolenaar94674f22023-01-06 18:42:20 +00001392 // specifies SomeInterface
Bram Moolenaar83677162023-01-08 19:54:10 +00001393 if (STRNCMP(arg, "extends", 7) == 0 && IS_WHITE_OR_NUL(arg[7]))
1394 {
1395 if (extends != NULL)
1396 {
1397 emsg(_(e_duplicate_extends));
1398 goto early_ret;
1399 }
1400 arg = skipwhite(arg + 7);
1401 char_u *end = find_name_end(arg, NULL, NULL, FNE_CHECK_START);
1402 if (!IS_WHITE_OR_NUL(*end))
1403 {
1404 semsg(_(e_white_space_required_after_name_str), arg);
1405 goto early_ret;
1406 }
1407 extends = vim_strnsave(arg, end - arg);
1408 if (extends == NULL)
1409 goto early_ret;
1410
1411 arg = skipwhite(end + 1);
1412 }
1413 else if (STRNCMP(arg, "implements", 10) == 0
1414 && IS_WHITE_OR_NUL(arg[10]))
Bram Moolenaar94674f22023-01-06 18:42:20 +00001415 {
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001416 if (!is_class)
1417 {
1418 emsg(_(e_interface_cannot_use_implements));
1419 goto early_ret;
1420 }
1421
Bram Moolenaardf8f9472023-01-07 14:51:03 +00001422 if (ga_impl.ga_len > 0)
1423 {
1424 emsg(_(e_duplicate_implements));
1425 goto early_ret;
1426 }
Bram Moolenaar94674f22023-01-06 18:42:20 +00001427 arg = skipwhite(arg + 10);
Bram Moolenaardf8f9472023-01-07 14:51:03 +00001428
1429 for (;;)
Bram Moolenaar94674f22023-01-06 18:42:20 +00001430 {
Bram Moolenaardf8f9472023-01-07 14:51:03 +00001431 char_u *impl_end = find_name_end(arg, NULL, NULL,
1432 FNE_CHECK_START);
Yegappan Lakshmanan2dede3d2023-09-27 19:02:01 +02001433 if ((!IS_WHITE_OR_NUL(*impl_end) && *impl_end != ',')
1434 || (*impl_end == ','
1435 && !IS_WHITE_OR_NUL(*(impl_end + 1))))
Bram Moolenaardf8f9472023-01-07 14:51:03 +00001436 {
1437 semsg(_(e_white_space_required_after_name_str), arg);
1438 goto early_ret;
1439 }
Yegappan Lakshmanan2dede3d2023-09-27 19:02:01 +02001440 if (impl_end - arg == 0)
1441 {
1442 emsg(_(e_missing_name_after_implements));
1443 goto early_ret;
1444 }
1445
Bram Moolenaardf8f9472023-01-07 14:51:03 +00001446 char_u *iname = vim_strnsave(arg, impl_end - arg);
1447 if (iname == NULL)
1448 goto early_ret;
1449 for (int i = 0; i < ga_impl.ga_len; ++i)
1450 if (STRCMP(((char_u **)ga_impl.ga_data)[i], iname) == 0)
1451 {
1452 semsg(_(e_duplicate_interface_after_implements_str),
1453 iname);
1454 vim_free(iname);
1455 goto early_ret;
1456 }
1457 if (ga_add_string(&ga_impl, iname) == FAIL)
1458 {
1459 vim_free(iname);
1460 goto early_ret;
1461 }
1462 if (*impl_end != ',')
1463 {
1464 arg = skipwhite(impl_end);
1465 break;
1466 }
1467 arg = skipwhite(impl_end + 1);
Bram Moolenaar94674f22023-01-06 18:42:20 +00001468 }
Bram Moolenaar94674f22023-01-06 18:42:20 +00001469 }
1470 else
1471 {
1472 semsg(_(e_trailing_characters_str), arg);
1473early_ret:
Bram Moolenaar83677162023-01-08 19:54:10 +00001474 vim_free(extends);
Bram Moolenaar94674f22023-01-06 18:42:20 +00001475 ga_clear_strings(&ga_impl);
1476 return;
1477 }
1478 }
1479
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001480 garray_T type_list; // list of pointers to allocated types
1481 ga_init2(&type_list, sizeof(type_T *), 10);
1482
Bram Moolenaard505d172022-12-18 21:42:55 +00001483 // Growarray with class members declared in the class.
1484 garray_T classmembers;
1485 ga_init2(&classmembers, sizeof(ocmember_T), 10);
1486
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001487 // Growarray with functions declared in the class.
1488 garray_T classfunctions;
1489 ga_init2(&classfunctions, sizeof(ufunc_T *), 10);
Bram Moolenaard505d172022-12-18 21:42:55 +00001490
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001491 // Growarray with object members declared in the class.
1492 garray_T objmembers;
Bram Moolenaard505d172022-12-18 21:42:55 +00001493 ga_init2(&objmembers, sizeof(ocmember_T), 10);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001494
1495 // Growarray with object methods declared in the class.
1496 garray_T objmethods;
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001497 ga_init2(&objmethods, sizeof(ufunc_T *), 10);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001498
1499 /*
Bram Moolenaar554d0312023-01-05 19:59:18 +00001500 * Go over the body of the class/interface until "endclass" or
1501 * "endinterface" is found.
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001502 */
1503 char_u *theline = NULL;
1504 int success = FALSE;
1505 for (;;)
1506 {
1507 vim_free(theline);
1508 theline = eap->getline(':', eap->cookie, 0, GETLINE_CONCAT_ALL);
1509 if (theline == NULL)
1510 break;
1511 char_u *line = skipwhite(theline);
1512
Bram Moolenaar418b5472022-12-20 13:38:22 +00001513 // Skip empty and comment lines.
1514 if (*line == NUL)
1515 continue;
1516 if (*line == '#')
1517 {
1518 if (vim9_bad_comment(line))
1519 break;
1520 continue;
1521 }
1522
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001523 char_u *p = line;
Bram Moolenaar554d0312023-01-05 19:59:18 +00001524 char *end_name = is_class ? "endclass" : "endinterface";
1525 if (checkforcmd(&p, end_name, is_class ? 4 : 5))
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001526 {
Bram Moolenaar554d0312023-01-05 19:59:18 +00001527 if (STRNCMP(line, end_name, is_class ? 8 : 12) != 0)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001528 semsg(_(e_command_cannot_be_shortened_str), line);
1529 else if (*p == '|' || !ends_excmd2(line, p))
1530 semsg(_(e_trailing_characters_str), p);
Bram Moolenaar98aeb212022-12-08 22:09:14 +00001531 else
1532 success = TRUE;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001533 break;
1534 }
Bram Moolenaar554d0312023-01-05 19:59:18 +00001535 char *wrong_name = is_class ? "endinterface" : "endclass";
1536 if (checkforcmd(&p, wrong_name, is_class ? 5 : 4))
1537 {
Bram Moolenaar657aea72023-01-27 13:16:19 +00001538 semsg(_(e_invalid_command_str_expected_str), line, end_name);
Bram Moolenaar554d0312023-01-05 19:59:18 +00001539 break;
1540 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001541
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001542 int has_public = FALSE;
1543 if (checkforcmd(&p, "public", 3))
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001544 {
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001545 if (STRNCMP(line, "public", 6) != 0)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001546 {
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001547 semsg(_(e_command_cannot_be_shortened_str), line);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001548 break;
1549 }
Yegappan Lakshmanan2dede3d2023-09-27 19:02:01 +02001550 if (!is_class)
1551 {
1552 emsg(_(e_public_member_not_supported_in_interface));
1553 break;
1554 }
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001555 has_public = TRUE;
1556 p = skipwhite(line + 6);
1557
Bram Moolenaard505d172022-12-18 21:42:55 +00001558 if (STRNCMP(p, "this", 4) != 0 && STRNCMP(p, "static", 6) != 0)
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001559 {
Bram Moolenaard505d172022-12-18 21:42:55 +00001560 emsg(_(e_public_must_be_followed_by_this_or_static));
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001561 break;
1562 }
1563 }
Bram Moolenaard505d172022-12-18 21:42:55 +00001564
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001565 int abstract_method = FALSE;
1566 char_u *pa = p;
1567 if (checkforcmd(&p, "abstract", 3))
1568 {
1569 if (STRNCMP(pa, "abstract", 8) != 0)
1570 {
1571 semsg(_(e_command_cannot_be_shortened_str), pa);
1572 break;
1573 }
1574
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001575 if (!is_class)
1576 // ignore "abstract" in an interface (as all the methods in an
1577 // interface are abstract.
1578 p = skipwhite(pa + 8);
1579 else
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001580 {
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001581 if (!is_abstract)
1582 {
1583 semsg(_(e_abstract_method_in_concrete_class), pa);
1584 break;
1585 }
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001586
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001587 abstract_method = TRUE;
1588 p = skipwhite(pa + 8);
1589 if (STRNCMP(p, "def", 3) != 0 && STRNCMP(p, "static", 6) != 0)
1590 {
1591 emsg(_(e_abstract_must_be_followed_by_def_or_static));
1592 break;
1593 }
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001594 }
1595 }
1596
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001597 int has_static = FALSE;
1598 char_u *ps = p;
1599 if (checkforcmd(&p, "static", 4))
1600 {
1601 if (STRNCMP(ps, "static", 6) != 0)
1602 {
1603 semsg(_(e_command_cannot_be_shortened_str), ps);
1604 break;
1605 }
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001606
1607 if (!is_class)
1608 {
Yegappan Lakshmanan00cd1822023-09-18 19:56:49 +02001609 emsg(_(e_static_member_not_supported_in_interface));
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001610 break;
1611 }
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001612 has_static = TRUE;
1613 p = skipwhite(ps + 6);
1614 }
1615
Bram Moolenaard505d172022-12-18 21:42:55 +00001616 // object members (public, read access, private):
1617 // "this._varname"
1618 // "this.varname"
1619 // "public this.varname"
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001620 if (STRNCMP(p, "this", 4) == 0)
1621 {
1622 if (p[4] != '.' || !eval_isnamec1(p[5]))
1623 {
RestorerZ7fe8f432023-09-24 23:21:24 +02001624 semsg(_(e_invalid_object_variable_declaration_str), p);
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001625 break;
1626 }
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +02001627 if (has_static)
1628 {
1629 emsg(_(e_static_cannot_be_followed_by_this));
1630 break;
1631 }
Bram Moolenaar3d473ee2022-12-14 20:59:32 +00001632 char_u *varname = p + 5;
Bram Moolenaard505d172022-12-18 21:42:55 +00001633 char_u *varname_end = NULL;
Bram Moolenaar74e12742022-12-13 21:14:28 +00001634 type_T *type = NULL;
Bram Moolenaard505d172022-12-18 21:42:55 +00001635 char_u *init_expr = NULL;
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001636
1637 if (!is_class && *varname == '_')
1638 {
1639 // private variables are not supported in an interface
Yegappan Lakshmanan00cd1822023-09-18 19:56:49 +02001640 semsg(_(e_private_variable_not_supported_in_interface),
1641 varname);
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001642 break;
1643 }
1644
Bram Moolenaard505d172022-12-18 21:42:55 +00001645 if (parse_member(eap, line, varname, has_public,
Bram Moolenaar554d0312023-01-05 19:59:18 +00001646 &varname_end, &type_list, &type,
1647 is_class ? &init_expr: NULL) == FAIL)
Bram Moolenaard505d172022-12-18 21:42:55 +00001648 break;
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +02001649 if (is_duplicate_member(&objmembers, varname, varname_end))
1650 {
1651 vim_free(init_expr);
1652 break;
1653 }
Bram Moolenaard505d172022-12-18 21:42:55 +00001654 if (add_member(&objmembers, varname, varname_end,
1655 has_public, type, init_expr) == FAIL)
Bram Moolenaar74e12742022-12-13 21:14:28 +00001656 {
Bram Moolenaard505d172022-12-18 21:42:55 +00001657 vim_free(init_expr);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001658 break;
1659 }
Bram Moolenaard505d172022-12-18 21:42:55 +00001660 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001661
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001662 // constructors:
1663 // def new()
1664 // enddef
1665 // def newOther()
1666 // enddef
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001667 // object methods and class functions:
1668 // def SomeMethod()
1669 // enddef
1670 // static def ClassFunction()
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001671 // enddef
1672 // TODO:
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001673 // def <Tval> someMethod()
1674 // enddef
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001675 else if (checkforcmd(&p, "def", 3))
1676 {
1677 exarg_T ea;
1678 garray_T lines_to_free;
1679
Yegappan Lakshmanan2dede3d2023-09-27 19:02:01 +02001680 if (has_public)
1681 {
1682 // "public" keyword is not supported when defining an object or
1683 // class method
1684 emsg(_(e_public_keyword_not_supported_for_method));
1685 break;
1686 }
1687
1688 if (*p == NUL)
1689 {
1690 // No method name following def
1691 semsg(_(e_not_valid_command_in_class_str), line);
1692 break;
1693 }
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001694
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001695 CLEAR_FIELD(ea);
1696 ea.cmd = line;
1697 ea.arg = p;
1698 ea.cmdidx = CMD_def;
1699 ea.getline = eap->getline;
1700 ea.cookie = eap->cookie;
1701
1702 ga_init2(&lines_to_free, sizeof(char_u *), 50);
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001703 int class_flags;
1704 if (is_class)
1705 class_flags = abstract_method ? CF_ABSTRACT_METHOD : CF_CLASS;
1706 else
1707 class_flags = CF_INTERFACE;
Bram Moolenaar554d0312023-01-05 19:59:18 +00001708 ufunc_T *uf = define_function(&ea, NULL, &lines_to_free,
h-eastb895b0f2023-09-24 15:46:31 +02001709 class_flags, objmembers.ga_data, objmembers.ga_len);
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001710 ga_clear_strings(&lines_to_free);
1711
Bram Moolenaar6acf7572023-01-01 19:53:30 +00001712 if (uf != NULL)
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001713 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02001714 char_u *name = uf->uf_name;
1715 int is_new = STRNCMP(name, "new", 3) == 0;
Gianmaria Bajo4b9777a2023-08-29 22:26:30 +02001716
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001717 if (!is_class && *name == '_')
1718 {
1719 // private variables are not supported in an interface
Yegappan Lakshmanan00cd1822023-09-18 19:56:49 +02001720 semsg(_(e_private_method_not_supported_in_interface),
1721 name);
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001722 func_clear_free(uf, FALSE);
1723 break;
1724 }
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001725 if (is_new && !is_valid_constructor(uf, is_abstract,
1726 has_static))
Bram Moolenaar24a8d062023-01-14 13:12:06 +00001727 {
Yegappan Lakshmananb1027282023-08-19 11:26:42 +02001728 func_clear_free(uf, FALSE);
Bram Moolenaar24a8d062023-01-14 13:12:06 +00001729 break;
1730 }
Gianmaria Bajo4b9777a2023-08-29 22:26:30 +02001731
Bram Moolenaar58b40092023-01-11 15:59:05 +00001732 // Check the name isn't used already.
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001733 if (is_duplicate_method(&classfunctions, &objmethods, name))
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +02001734 {
1735 success = FALSE;
1736 func_clear_free(uf, FALSE);
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +02001737 break;
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +02001738 }
Bram Moolenaar58b40092023-01-11 15:59:05 +00001739
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001740 garray_T *fgap = has_static || is_new
1741 ? &classfunctions : &objmethods;
Bram Moolenaar6acf7572023-01-01 19:53:30 +00001742 if (ga_grow(fgap, 1) == OK)
1743 {
1744 if (is_new)
1745 uf->uf_flags |= FC_NEW;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00001746
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001747 if (abstract_method)
1748 uf->uf_flags |= FC_ABSTRACT;
1749
Bram Moolenaar6acf7572023-01-01 19:53:30 +00001750 ((ufunc_T **)fgap->ga_data)[fgap->ga_len] = uf;
1751 ++fgap->ga_len;
1752 }
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001753 }
1754 }
1755
1756 // class members
1757 else if (has_static)
1758 {
1759 // class members (public, read access, private):
1760 // "static _varname"
1761 // "static varname"
1762 // "public static varname"
1763 char_u *varname = p;
1764 char_u *varname_end = NULL;
1765 type_T *type = NULL;
1766 char_u *init_expr = NULL;
1767 if (parse_member(eap, line, varname, has_public,
Bram Moolenaar554d0312023-01-05 19:59:18 +00001768 &varname_end, &type_list, &type,
1769 is_class ? &init_expr : NULL) == FAIL)
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001770 break;
Yegappan Lakshmanan2ba9d2e2023-08-28 21:26:23 +02001771 if (is_duplicate_member(&classmembers, varname, varname_end))
1772 {
1773 vim_free(init_expr);
1774 break;
1775 }
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001776 if (add_member(&classmembers, varname, varname_end,
1777 has_public, type, init_expr) == FAIL)
1778 {
1779 vim_free(init_expr);
1780 break;
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001781 }
1782 }
1783
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001784 else
1785 {
Bram Moolenaar554d0312023-01-05 19:59:18 +00001786 if (is_class)
1787 semsg(_(e_not_valid_command_in_class_str), line);
1788 else
1789 semsg(_(e_not_valid_command_in_interface_str), line);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001790 break;
1791 }
1792 }
1793 vim_free(theline);
1794
Bram Moolenaar83677162023-01-08 19:54:10 +00001795 class_T *extends_cl = NULL; // class from "extends" argument
1796
1797 /*
1798 * Check a few things before defining the class.
1799 */
1800
1801 // Check the "extends" class is valid.
1802 if (success && extends != NULL)
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001803 success = validate_extends_class(extends, &extends_cl, is_class);
Bram Moolenaar83677162023-01-08 19:54:10 +00001804 VIM_CLEAR(extends);
1805
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001806 // Check the new object methods to make sure their access (public or
1807 // private) is the same as that in the extended class lineage.
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +02001808 if (success && extends_cl != NULL)
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001809 success = validate_extends_methods(&objmethods, extends_cl);
1810
1811 // Check the new class and object variables are not duplicates of the
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001812 // variables in the extended class lineage. If an interface is extending
1813 // another interface, then it can duplicate the member variables.
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001814 if (success && extends_cl != NULL)
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001815 {
1816 if (is_class)
1817 success = extends_check_dup_members(&objmembers, extends_cl);
1818 else
1819 success = extends_check_intf_var_type(&objmembers, extends_cl);
1820 }
Yegappan Lakshmanane3b6c782023-08-29 22:32:02 +02001821
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001822 // When extending an abstract class, make sure all the abstract methods in
1823 // the parent class are implemented. If the current class is an abstract
1824 // class, then there is no need for this check.
1825 if (success && !is_abstract && extends_cl != NULL
1826 && (extends_cl->class_flags & CLASS_ABSTRACT))
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001827 success = validate_abstract_class_methods(&classfunctions,
1828 &objmethods, extends_cl);
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001829
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00001830 class_T **intf_classes = NULL;
1831
Bram Moolenaar83677162023-01-08 19:54:10 +00001832 // Check all "implements" entries are valid.
Bram Moolenaar94674f22023-01-06 18:42:20 +00001833 if (success && ga_impl.ga_len > 0)
1834 {
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00001835 intf_classes = ALLOC_CLEAR_MULT(class_T *, ga_impl.ga_len);
1836
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001837 success = validate_implements_classes(&ga_impl, intf_classes,
1838 &classfunctions, &classmembers,
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02001839 &objmethods, &objmembers,
1840 extends_cl);
Bram Moolenaar94674f22023-01-06 18:42:20 +00001841 }
1842
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001843 // Check no function argument name is used as a class member.
Bram Moolenaard40f00c2023-01-13 17:36:49 +00001844 if (success)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001845 success = check_func_arg_names(&classfunctions, &objmethods,
1846 &classmembers);
Bram Moolenaard40f00c2023-01-13 17:36:49 +00001847
Bram Moolenaareb533502022-12-14 15:06:11 +00001848 class_T *cl = NULL;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001849 if (success)
1850 {
Bram Moolenaard505d172022-12-18 21:42:55 +00001851 // "endclass" encountered without failures: Create the class.
1852
Bram Moolenaareb533502022-12-14 15:06:11 +00001853 cl = ALLOC_CLEAR_ONE(class_T);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001854 if (cl == NULL)
1855 goto cleanup;
Bram Moolenaar554d0312023-01-05 19:59:18 +00001856 if (!is_class)
1857 cl->class_flags = CLASS_INTERFACE;
Yegappan Lakshmanan7bcd25c2023-09-08 19:27:51 +02001858 else if (is_abstract)
1859 cl->class_flags = CLASS_ABSTRACT;
Bram Moolenaar554d0312023-01-05 19:59:18 +00001860
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001861 cl->class_refcount = 1;
Bram Moolenaar94674f22023-01-06 18:42:20 +00001862 cl->class_name = vim_strnsave(name_start, name_end - name_start);
Bram Moolenaard505d172022-12-18 21:42:55 +00001863 if (cl->class_name == NULL)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001864 goto cleanup;
Bram Moolenaard505d172022-12-18 21:42:55 +00001865
Bram Moolenaard0200c82023-01-28 15:19:40 +00001866 if (extends_cl != NULL)
1867 {
1868 cl->class_extends = extends_cl;
1869 extends_cl->class_flags |= CLASS_EXTENDED;
1870 }
Bram Moolenaar83677162023-01-08 19:54:10 +00001871
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001872 // Add class and object variables to "cl".
Bram Moolenaard505d172022-12-18 21:42:55 +00001873 if (add_members_to_class(&classmembers,
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02001874 NULL,
1875 0,
Bram Moolenaar83677162023-01-08 19:54:10 +00001876 &cl->class_class_members,
1877 &cl->class_class_member_count) == FAIL
Bram Moolenaard505d172022-12-18 21:42:55 +00001878 || add_members_to_class(&objmembers,
Bram Moolenaar83677162023-01-08 19:54:10 +00001879 extends_cl == NULL ? NULL
1880 : extends_cl->class_obj_members,
1881 extends_cl == NULL ? 0
1882 : extends_cl->class_obj_member_count,
1883 &cl->class_obj_members,
1884 &cl->class_obj_member_count) == FAIL)
Bram Moolenaard505d172022-12-18 21:42:55 +00001885 goto cleanup;
1886
Bram Moolenaar29ac5df2023-01-16 19:43:47 +00001887 if (ga_impl.ga_len > 0)
1888 {
1889 // Move the "implements" names into the class.
1890 cl->class_interface_count = ga_impl.ga_len;
1891 cl->class_interfaces = ALLOC_MULT(char_u *, ga_impl.ga_len);
1892 if (cl->class_interfaces == NULL)
1893 goto cleanup;
1894 for (int i = 0; i < ga_impl.ga_len; ++i)
1895 cl->class_interfaces[i] = ((char_u **)ga_impl.ga_data)[i];
1896 VIM_CLEAR(ga_impl.ga_data);
1897 ga_impl.ga_len = 0;
1898
Bram Moolenaard0200c82023-01-28 15:19:40 +00001899 cl->class_interfaces_cl = intf_classes;
1900 intf_classes = NULL;
1901 }
1902
1903 if (cl->class_interface_count > 0 || extends_cl != NULL)
1904 {
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001905 // Add a method and member lookup table to each of the interface
1906 // classes.
1907 if (add_lookup_tables(cl, extends_cl, &objmethods) == FAIL)
1908 goto cleanup;
Bram Moolenaar29ac5df2023-01-16 19:43:47 +00001909 }
1910
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001911 // Allocate a typval for each class member and initialize it.
Bram Moolenaar554d0312023-01-05 19:59:18 +00001912 if (is_class && cl->class_class_member_count > 0)
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001913 add_class_members(cl, eap);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001914
Yegappan Lakshmanan6ac15442023-08-20 18:20:17 +02001915 int have_new = FALSE;
1916 ufunc_T *class_func = NULL;
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001917 for (int i = 0; i < classfunctions.ga_len; ++i)
Yegappan Lakshmanan6ac15442023-08-20 18:20:17 +02001918 {
1919 class_func = ((ufunc_T **)classfunctions.ga_data)[i];
1920 if (STRCMP(class_func->uf_name, "new") == 0)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001921 {
1922 have_new = TRUE;
1923 break;
1924 }
Yegappan Lakshmanan6ac15442023-08-20 18:20:17 +02001925 }
1926
1927 if (have_new)
1928 // The return type of new() is an object of class "cl"
1929 class_func->uf_ret_type->tt_class = cl;
1930 else if (is_class && !is_abstract && !have_new)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001931 // No new() method was defined, add the default constructor.
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001932 add_default_constructor(cl, &classfunctions, &type_list);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001933
Bram Moolenaar58b40092023-01-11 15:59:05 +00001934 // Move all the functions into the created class.
Yegappan Lakshmanan4b1cc792023-08-19 22:39:33 +02001935 if (add_classfuncs_objmethods(cl, extends_cl, &classfunctions,
1936 &objmethods) == FAIL)
1937 goto cleanup;
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001938
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001939 cl->class_type.tt_type = VAR_CLASS;
Bram Moolenaarb1e32ac2023-02-21 12:38:51 +00001940 cl->class_type.tt_class = cl;
Bram Moolenaarffdaca92022-12-09 21:41:48 +00001941 cl->class_object_type.tt_type = VAR_OBJECT;
Bram Moolenaarb1e32ac2023-02-21 12:38:51 +00001942 cl->class_object_type.tt_class = cl;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001943 cl->class_type_list = type_list;
1944
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02001945 class_created(cl);
1946
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001947 // TODO:
Bram Moolenaard505d172022-12-18 21:42:55 +00001948 // - Fill hashtab with object members and methods ?
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001949
1950 // Add the class to the script-local variables.
Bram Moolenaar94674f22023-01-06 18:42:20 +00001951 // TODO: handle other context, e.g. in a function
Ernie Rael21d32122023-09-02 15:09:18 +02001952 // TODO: does uf_hash need to be cleared?
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001953 typval_T tv;
1954 tv.v_type = VAR_CLASS;
1955 tv.vval.v_class = cl;
Bram Moolenaara86655a2023-01-12 17:06:27 +00001956 is_export = class_export;
Bram Moolenaar83ae6152023-02-25 19:59:31 +00001957 SOURCING_LNUM = start_lnum;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001958 set_var_const(cl->class_name, current_sctx.sc_sid,
Bram Moolenaar83ae6152023-02-25 19:59:31 +00001959 NULL, &tv, FALSE, 0, 0);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001960 return;
1961 }
1962
1963cleanup:
Bram Moolenaareb533502022-12-14 15:06:11 +00001964 if (cl != NULL)
1965 {
1966 vim_free(cl->class_name);
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00001967 vim_free(cl->class_class_functions);
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00001968 if (cl->class_interfaces != NULL)
1969 {
1970 for (int i = 0; i < cl->class_interface_count; ++i)
1971 vim_free(cl->class_interfaces[i]);
1972 vim_free(cl->class_interfaces);
1973 }
1974 if (cl->class_interfaces_cl != NULL)
1975 {
1976 for (int i = 0; i < cl->class_interface_count; ++i)
1977 class_unref(cl->class_interfaces_cl[i]);
1978 vim_free(cl->class_interfaces_cl);
1979 }
Bram Moolenaareb533502022-12-14 15:06:11 +00001980 vim_free(cl->class_obj_members);
1981 vim_free(cl->class_obj_methods);
1982 vim_free(cl);
1983 }
1984
Bram Moolenaar83677162023-01-08 19:54:10 +00001985 vim_free(extends);
1986 class_unref(extends_cl);
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00001987
1988 if (intf_classes != NULL)
1989 {
1990 for (int i = 0; i < ga_impl.ga_len; ++i)
1991 class_unref(intf_classes[i]);
1992 vim_free(intf_classes);
1993 }
Bram Moolenaar94674f22023-01-06 18:42:20 +00001994 ga_clear_strings(&ga_impl);
1995
Bram Moolenaard505d172022-12-18 21:42:55 +00001996 for (int round = 1; round <= 2; ++round)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00001997 {
Bram Moolenaard505d172022-12-18 21:42:55 +00001998 garray_T *gap = round == 1 ? &classmembers : &objmembers;
1999 if (gap->ga_len == 0 || gap->ga_data == NULL)
2000 continue;
2001
2002 for (int i = 0; i < gap->ga_len; ++i)
2003 {
2004 ocmember_T *m = ((ocmember_T *)gap->ga_data) + i;
2005 vim_free(m->ocm_name);
2006 vim_free(m->ocm_init);
2007 }
2008 ga_clear(gap);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002009 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002010
Bram Moolenaarffdaca92022-12-09 21:41:48 +00002011 for (int i = 0; i < objmethods.ga_len; ++i)
2012 {
2013 ufunc_T *uf = ((ufunc_T **)objmethods.ga_data)[i];
2014 func_clear_free(uf, FALSE);
2015 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002016 ga_clear(&objmethods);
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00002017
2018 for (int i = 0; i < classfunctions.ga_len; ++i)
2019 {
2020 ufunc_T *uf = ((ufunc_T **)classfunctions.ga_data)[i];
2021 func_clear_free(uf, FALSE);
2022 }
2023 ga_clear(&classfunctions);
2024
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002025 clear_type_list(&type_list);
2026}
2027
2028/*
Bram Moolenaarf54cedd2022-12-23 17:56:27 +00002029 * Find member "name" in class "cl", set "member_idx" to the member index and
2030 * return its type.
Yegappan Lakshmanan3775f772023-09-01 22:05:45 +02002031 * When "is_object" is TRUE, then look for object members. Otherwise look for
2032 * class members.
Bram Moolenaarf54cedd2022-12-23 17:56:27 +00002033 * When not found "member_idx" is set to -1 and t_any is returned.
Ernie Rael456ae552023-09-01 18:54:54 +02002034 * Set *p_m ocmmember_T if not NULL
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002035 */
2036 type_T *
2037class_member_type(
Yegappan Lakshmananeb91e242023-08-31 18:10:46 +02002038 class_T *cl,
Yegappan Lakshmanan3775f772023-09-01 22:05:45 +02002039 int is_object,
Yegappan Lakshmananeb91e242023-08-31 18:10:46 +02002040 char_u *name,
2041 char_u *name_end,
Yegappan Lakshmanan00cd1822023-09-18 19:56:49 +02002042 int *member_idx)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002043{
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002044 size_t len = name_end - name;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002045 ocmember_T *m;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002046
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002047 *member_idx = -1; // not found (yet)
2048
2049 m = member_lookup(cl, is_object ? VAR_OBJECT : VAR_CLASS, name, len,
2050 member_idx);
2051 if (m == NULL)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002052 {
Yegappan Lakshmanan00cd1822023-09-18 19:56:49 +02002053 member_not_found_msg(cl, is_object ? VAR_OBJECT : VAR_CLASS, name,
2054 len);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002055 return &t_any;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002056 }
Bram Moolenaarf54cedd2022-12-23 17:56:27 +00002057
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002058 return m->ocm_type;
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00002059}
2060
2061/*
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00002062 * Handle ":enum" up to ":endenum".
2063 */
2064 void
2065ex_enum(exarg_T *eap UNUSED)
2066{
2067 // TODO
2068}
2069
2070/*
2071 * Handle ":type".
2072 */
2073 void
2074ex_type(exarg_T *eap UNUSED)
2075{
2076 // TODO
2077}
2078
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002079/*
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002080 * Returns OK if a member variable named "name" is present in the class "cl".
2081 * Otherwise returns FAIL. If found, the member variable typval is set in
2082 * "rettv". If "is_object" is TRUE, then the object member variable table is
2083 * searched. Otherwise the class member variable table is searched.
2084 */
2085 static int
2086get_member_tv(
2087 class_T *cl,
2088 int is_object,
2089 char_u *name,
2090 size_t namelen,
2091 typval_T *rettv)
2092{
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002093 ocmember_T *m;
2094 int m_idx;
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002095
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002096 m = member_lookup(cl, is_object ? VAR_OBJECT : VAR_CLASS, name, namelen,
2097 &m_idx);
2098 if (m == NULL)
2099 return FAIL;
2100
2101 if (*name == '_')
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002102 {
RestorerZ7fe8f432023-09-24 23:21:24 +02002103 semsg(_(e_cannot_access_private_variable_str), m->ocm_name);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002104 return FAIL;
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002105 }
2106
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002107 // The object only contains a pointer to the class, the member
2108 // values array follows right after that.
2109 object_T *obj = rettv->vval.v_object;
2110 if (is_object)
2111 {
2112 typval_T *tv = (typval_T *)(obj + 1) + m_idx;
2113 copy_tv(tv, rettv);
2114 }
2115 else
2116 copy_tv(&cl->class_members_tv[m_idx], rettv);
2117
2118 object_unref(obj);
2119
2120 return OK;
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002121}
2122
2123/*
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002124 * Evaluate what comes after a class:
2125 * - class member: SomeClass.varname
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00002126 * - class function: SomeClass.SomeMethod()
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002127 * - class constructor: SomeClass.new()
2128 * - object member: someObject.varname
2129 * - object method: someObject.SomeMethod()
2130 *
2131 * "*arg" points to the '.'.
2132 * "*arg" is advanced to after the member name or method call.
2133 *
2134 * Returns FAIL or OK.
2135 */
2136 int
2137class_object_index(
2138 char_u **arg,
2139 typval_T *rettv,
2140 evalarg_T *evalarg,
2141 int verbose UNUSED) // give error messages
2142{
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002143 if (VIM_ISWHITE((*arg)[1]))
2144 {
2145 semsg(_(e_no_white_space_allowed_after_str_str), ".", *arg);
2146 return FAIL;
2147 }
2148
2149 ++*arg;
2150 char_u *name = *arg;
2151 char_u *name_end = find_name_end(name, NULL, NULL, FNE_CHECK_START);
2152 if (name_end == name)
2153 return FAIL;
2154 size_t len = name_end - name;
2155
Bram Moolenaar552bdca2023-02-17 21:08:50 +00002156 class_T *cl;
2157 if (rettv->v_type == VAR_CLASS)
2158 cl = rettv->vval.v_class;
2159 else // VAR_OBJECT
2160 {
2161 if (rettv->vval.v_object == NULL)
2162 {
2163 emsg(_(e_using_null_object));
2164 return FAIL;
2165 }
2166 cl = rettv->vval.v_object->obj_class;
2167 }
2168
Bram Moolenaard13dd302023-03-11 20:56:35 +00002169 if (cl == NULL)
2170 {
2171 emsg(_(e_incomplete_type));
2172 return FAIL;
2173 }
2174
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002175 if (*name_end == '(')
2176 {
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002177 ufunc_T *fp;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002178
Ernie Rael4d00b832023-09-11 19:54:42 +02002179 fp = method_lookup(cl, rettv->v_type, name, len, NULL);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002180 if (fp == NULL)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002181 {
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002182 method_not_found_msg(cl, rettv->v_type, name, len);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002183 return FAIL;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002184 }
2185
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002186 typval_T argvars[MAX_FUNC_ARGS + 1];
2187 int argcount = 0;
2188
2189 if (*fp->uf_name == '_')
2190 {
2191 // Cannot access a private method outside of a class
2192 semsg(_(e_cannot_access_private_method_str), name);
2193 return FAIL;
2194 }
2195
2196 char_u *argp = name_end;
2197 int ret = get_func_arguments(&argp, evalarg, 0,
2198 argvars, &argcount);
2199 if (ret == FAIL)
2200 return FAIL;
2201
2202 funcexe_T funcexe;
2203 CLEAR_FIELD(funcexe);
2204 funcexe.fe_evaluate = TRUE;
2205 if (rettv->v_type == VAR_OBJECT)
2206 {
2207 funcexe.fe_object = rettv->vval.v_object;
2208 ++funcexe.fe_object->obj_refcount;
2209 }
2210
2211 // Clear the class or object after calling the function, in
2212 // case the refcount is one.
2213 typval_T tv_tofree = *rettv;
2214 rettv->v_type = VAR_UNKNOWN;
2215
2216 // Call the user function. Result goes into rettv;
2217 int error = call_user_func_check(fp, argcount, argvars,
2218 rettv, &funcexe, NULL);
2219
2220 // Clear the previous rettv and the arguments.
2221 clear_tv(&tv_tofree);
2222 for (int idx = 0; idx < argcount; ++idx)
2223 clear_tv(&argvars[idx]);
2224
2225 if (error != FCERR_NONE)
2226 {
2227 user_func_error(error, printable_func_name(fp),
2228 funcexe.fe_found_var);
2229 return FAIL;
2230 }
2231 *arg = argp;
2232 return OK;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002233 }
2234
2235 else if (rettv->v_type == VAR_OBJECT)
2236 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002237 // Search in the object member variable table and the class member
2238 // variable table.
Yegappan Lakshmanan23c92d92023-09-09 11:33:29 +02002239 if (get_member_tv(cl, TRUE, name, len, rettv) == OK)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002240 {
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002241 *arg = name_end;
2242 return OK;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002243 }
2244
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002245 member_not_found_msg(cl, VAR_OBJECT, name, len);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002246 }
2247
Bram Moolenaard505d172022-12-18 21:42:55 +00002248 else if (rettv->v_type == VAR_CLASS)
2249 {
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002250 int m_idx;
2251
Bram Moolenaard505d172022-12-18 21:42:55 +00002252 // class member
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002253 ocmember_T *m = class_member_lookup(cl, name, len, &m_idx);
2254 if (m == NULL)
Bram Moolenaard505d172022-12-18 21:42:55 +00002255 {
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002256 member_not_found_msg(cl, VAR_CLASS, name, len);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002257 return FAIL;
Bram Moolenaard505d172022-12-18 21:42:55 +00002258 }
2259
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002260 if (*name == '_')
2261 {
RestorerZ7fe8f432023-09-24 23:21:24 +02002262 semsg(_(e_cannot_access_private_variable_str), m->ocm_name);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002263 return FAIL;
2264 }
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002265
2266 typval_T *tv = &cl->class_members_tv[m_idx];
2267 copy_tv(tv, rettv);
2268 class_unref(cl);
2269
2270 *arg = name_end;
2271 return OK;
Bram Moolenaard505d172022-12-18 21:42:55 +00002272 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002273
2274 return FAIL;
2275}
2276
2277/*
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002278 * If "arg" points to a class or object method, return it.
2279 * Otherwise return NULL.
2280 */
2281 ufunc_T *
2282find_class_func(char_u **arg)
2283{
2284 char_u *name = *arg;
2285 char_u *name_end = find_name_end(name, NULL, NULL, FNE_CHECK_START);
2286 if (name_end == name || *name_end != '.')
2287 return NULL;
2288
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002289 ufunc_T *fp = NULL;
Yegappan Lakshmanan1689e842023-09-06 20:23:23 +02002290 size_t len = name_end - name;
2291 typval_T tv;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002292 tv.v_type = VAR_UNKNOWN;
Bram Moolenaar993dbc32023-01-01 20:31:30 +00002293 if (eval_variable(name, (int)len,
2294 0, &tv, NULL, EVAL_VAR_NOAUTOLOAD) == FAIL)
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002295 return NULL;
2296 if (tv.v_type != VAR_CLASS && tv.v_type != VAR_OBJECT)
Bram Moolenaareb533502022-12-14 15:06:11 +00002297 goto fail_after_eval;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002298
2299 class_T *cl = tv.v_type == VAR_CLASS ? tv.vval.v_class
2300 : tv.vval.v_object->obj_class;
2301 if (cl == NULL)
Bram Moolenaareb533502022-12-14 15:06:11 +00002302 goto fail_after_eval;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002303 char_u *fname = name_end + 1;
2304 char_u *fname_end = find_name_end(fname, NULL, NULL, FNE_CHECK_START);
2305 if (fname_end == fname)
Bram Moolenaareb533502022-12-14 15:06:11 +00002306 goto fail_after_eval;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002307 len = fname_end - fname;
2308
Ernie Rael4d00b832023-09-11 19:54:42 +02002309 fp = method_lookup(cl, tv.v_type, fname, len, NULL);
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002310
Bram Moolenaareb533502022-12-14 15:06:11 +00002311fail_after_eval:
2312 clear_tv(&tv);
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002313 return fp;
Bram Moolenaar7ce7daf2022-12-10 18:42:12 +00002314}
2315
2316/*
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002317 * Returns the index of class variable "name" in the class "cl".
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002318 * Returns -1, if the variable is not found.
2319 * If "namelen" is zero, then it is assumed that "name" is NUL terminated.
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00002320 */
2321 int
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002322class_member_idx(class_T *cl, char_u *name, size_t namelen)
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00002323{
Ernie Rael4d00b832023-09-11 19:54:42 +02002324 int idx;
2325 class_member_lookup(cl, name, namelen, &idx);
2326 return idx;
Bram Moolenaar6bafdd42023-01-01 12:58:33 +00002327}
2328
2329/*
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002330 * Returns a pointer to the class member variable "name" in the class "cl".
2331 * Returns NULL if the variable is not found.
2332 * The member variable index is set in "idx".
2333 */
2334 ocmember_T *
2335class_member_lookup(class_T *cl, char_u *name, size_t namelen, int *idx)
2336{
Ernie Rael4d00b832023-09-11 19:54:42 +02002337 ocmember_T *ret_m = NULL;
2338 int ret_idx = -1;
2339 for (int i = 0; i < cl->class_class_member_count; ++i)
2340 {
2341 ocmember_T *m = &cl->class_class_members[i];
2342 if (namelen)
2343 {
2344 if (STRNCMP(name, m->ocm_name, namelen) == 0
2345 && m->ocm_name[namelen] == NUL)
2346 {
2347 ret_m = m;
2348 ret_idx = i;
2349 break;
2350 }
2351 }
2352 else if (STRCMP(name, m->ocm_name) == 0)
2353 {
2354 ret_m = m;
2355 ret_idx = i;
2356 break;
2357 }
2358 }
2359 if (idx != NULL)
2360 *idx = ret_idx;
2361 return ret_m;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002362}
2363
2364/*
2365 * Returns the index of class method "name" in the class "cl".
2366 * Returns -1, if the method is not found.
Yegappan Lakshmanan342f4f62023-09-09 11:37:23 +02002367 */
2368 int
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002369class_method_idx(class_T *cl, char_u *name, size_t namelen)
Yegappan Lakshmanan342f4f62023-09-09 11:37:23 +02002370{
Ernie Rael4d00b832023-09-11 19:54:42 +02002371 int idx;
2372 class_method_lookup(cl, name, namelen, &idx);
2373 return idx;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002374}
2375
2376/*
2377 * Returns a pointer to the class method "name" in class "cl".
2378 * Returns NULL if the method is not found.
2379 * The method index is set in "idx".
2380 */
2381 ufunc_T *
2382class_method_lookup(class_T *cl, char_u *name, size_t namelen, int *idx)
2383{
Ernie Rael4d00b832023-09-11 19:54:42 +02002384 ufunc_T *ret_fp = NULL;
2385 int ret_idx = -1;
2386 for (int i = 0; i < cl->class_class_function_count; ++i)
2387 {
2388 ufunc_T *fp = cl->class_class_functions[i];
2389 char_u *ufname = (char_u *)fp->uf_name;
2390 if (STRNCMP(name, ufname, namelen) == 0 && ufname[namelen] == NUL)
2391 {
2392 ret_fp = fp;
2393 ret_idx = i;
2394 break;
2395 }
2396 }
2397 if (idx != NULL)
2398 *idx = ret_idx;
2399 return ret_fp;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002400}
2401
2402/*
2403 * Returns the index of object member variable "name" in the class "cl".
2404 * Returns -1, if the variable is not found.
2405 * If "namelen" is zero, then it is assumed that "name" is NUL terminated.
2406 */
2407 int
2408object_member_idx(class_T *cl, char_u *name, size_t namelen)
2409{
Ernie Rael4d00b832023-09-11 19:54:42 +02002410 int idx;
2411 object_member_lookup(cl, name, namelen, &idx);
2412 return idx;
Yegappan Lakshmanan342f4f62023-09-09 11:37:23 +02002413}
2414
2415/*
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002416 * Returns a pointer to the object member variable "name" in the class "cl".
2417 * Returns NULL if the variable is not found.
2418 * The object member variable index is set in "idx".
2419 */
2420 ocmember_T *
2421object_member_lookup(class_T *cl, char_u *name, size_t namelen, int *idx)
2422{
Ernie Rael4d00b832023-09-11 19:54:42 +02002423 ocmember_T *ret_m = NULL;
2424 int ret_idx = -1;
2425 for (int i = 0; i < cl->class_obj_member_count; ++i)
2426 {
2427 ocmember_T *m = &cl->class_obj_members[i];
2428 if (namelen)
2429 {
2430 if (STRNCMP(name, m->ocm_name, namelen) == 0
2431 && m->ocm_name[namelen] == NUL)
2432 {
2433 ret_m = m;
2434 ret_idx = i;
2435 break;
2436 }
2437 }
2438 else if (STRCMP(name, m->ocm_name) == 0)
2439 {
2440 ret_m = m;
2441 ret_idx = i;
2442 break;
2443 }
2444 }
2445 if (idx != NULL)
2446 *idx = ret_idx;
2447 return ret_m;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002448}
2449
2450/*
2451 * Returns the index of object method "name" in the class "cl".
2452 * Returns -1, if the method is not found.
2453 */
2454 int
2455object_method_idx(class_T *cl, char_u *name, size_t namelen)
2456{
Ernie Rael4d00b832023-09-11 19:54:42 +02002457 int idx;
2458 object_method_lookup(cl, name, namelen, &idx);
2459 return idx;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002460}
2461
2462/*
2463 * Returns a pointer to the object method "name" in class "cl".
2464 * Returns NULL if the method is not found.
2465 * The object method index is set in "idx".
2466 */
2467 ufunc_T *
2468object_method_lookup(class_T *cl, char_u *name, size_t namelen, int *idx)
2469{
Ernie Rael4d00b832023-09-11 19:54:42 +02002470 ufunc_T *ret_fp = NULL;
2471 int ret_idx = -1;
2472 for (int i = 0; i < cl->class_obj_method_count; ++i)
2473 {
2474 ufunc_T *fp = cl->class_obj_methods[i];
2475 // Use a separate pointer to avoid that ASAN complains about
2476 // uf_name[] only being 4 characters.
2477 char_u *ufname = (char_u *)fp->uf_name;
2478 if (STRNCMP(name, ufname, namelen) == 0 && ufname[namelen] == NUL)
2479 {
2480 ret_fp = fp;
2481 ret_idx = i;
2482 break;
2483 }
2484 }
2485 if (idx != NULL)
2486 *idx = ret_idx;
2487 return ret_fp;
Yegappan Lakshmananf36bbcd2023-09-10 18:19:06 +02002488}
2489
2490/*
2491 * Lookup a class or object member variable by name. If v_type is VAR_CLASS,
2492 * then lookup a class member variable and if it is VAR_OBJECT, then lookup a
2493 * object member variable.
2494 *
2495 * Returns a pointer to the member variable structure if variable is found.
2496 * Otherwise returns NULL. The member variable index is set in "*idx".
2497 */
2498 ocmember_T *
2499member_lookup(
2500 class_T *cl,
2501 vartype_T v_type,
2502 char_u *name,
2503 size_t namelen,
2504 int *idx)
2505{
2506 if (v_type == VAR_CLASS)
2507 return class_member_lookup(cl, name, namelen, idx);
2508 else
2509 return object_member_lookup(cl, name, namelen, idx);
2510}
2511
2512/*
2513 * Lookup a class or object method by name. If v_type is VAR_CLASS, then
2514 * lookup a class method and if it is VAR_OBJECT, then lookup a object method.
2515 *
2516 * Returns a pointer to the method structure if variable is found.
2517 * Otherwise returns NULL. The method variable index is set in "*idx".
2518 */
2519 ufunc_T *
2520method_lookup(
2521 class_T *cl,
2522 vartype_T v_type,
2523 char_u *name,
2524 size_t namelen,
2525 int *idx)
2526{
2527 if (v_type == VAR_CLASS)
2528 return class_method_lookup(cl, name, namelen, idx);
2529 else
2530 return object_method_lookup(cl, name, namelen, idx);
2531}
2532
2533/*
Bram Moolenaar62a69232023-01-24 15:07:04 +00002534 * Return TRUE if current context "cctx_arg" is inside class "cl".
2535 * Return FALSE if not.
2536 */
2537 int
2538inside_class(cctx_T *cctx_arg, class_T *cl)
2539{
2540 for (cctx_T *cctx = cctx_arg; cctx != NULL; cctx = cctx->ctx_outer)
Ernie Raelcf138d42023-09-06 20:45:03 +02002541 if (cctx->ctx_ufunc != NULL
2542 && class_instance_of(cctx->ctx_ufunc->uf_class, cl))
Bram Moolenaar62a69232023-01-24 15:07:04 +00002543 return TRUE;
2544 return FALSE;
2545}
2546
2547/*
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002548 * Make a copy of an object.
2549 */
2550 void
2551copy_object(typval_T *from, typval_T *to)
2552{
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +02002553 if (from->vval.v_object == NULL)
2554 to->vval.v_object = NULL;
2555 else
2556 {
2557 to->vval.v_object = from->vval.v_object;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002558 ++to->vval.v_object->obj_refcount;
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +02002559 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002560}
2561
2562/*
2563 * Free an object.
2564 */
2565 static void
2566object_clear(object_T *obj)
2567{
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002568 // Avoid a recursive call, it can happen if "obj" has a circular reference.
2569 obj->obj_refcount = INT_MAX;
2570
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002571 class_T *cl = obj->obj_class;
2572
Jia-Ju Bai5b0889b2023-08-13 20:04:04 +02002573 if (!cl)
Yegappan Lakshmanand4e4ecb2023-08-27 18:35:45 +02002574 return;
Jia-Ju Bai5b0889b2023-08-13 20:04:04 +02002575
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002576 // the member values are just after the object structure
2577 typval_T *tv = (typval_T *)(obj + 1);
2578 for (int i = 0; i < cl->class_obj_member_count; ++i)
2579 clear_tv(tv + i);
2580
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002581 // Remove from the list headed by "first_object".
2582 object_cleared(obj);
2583
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002584 vim_free(obj);
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002585 class_unref(cl);
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002586}
2587
2588/*
2589 * Unreference an object.
2590 */
2591 void
2592object_unref(object_T *obj)
2593{
2594 if (obj != NULL && --obj->obj_refcount <= 0)
2595 object_clear(obj);
2596}
2597
2598/*
2599 * Make a copy of a class.
2600 */
2601 void
2602copy_class(typval_T *from, typval_T *to)
2603{
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +02002604 if (from->vval.v_class == NULL)
2605 to->vval.v_class = NULL;
2606 else
2607 {
2608 to->vval.v_class = from->vval.v_class;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002609 ++to->vval.v_class->class_refcount;
Yegappan Lakshmanan618e47d2023-08-22 21:29:28 +02002610 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002611}
2612
2613/*
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02002614 * Free the class "cl" and its contents.
2615 */
2616 static void
2617class_free(class_T *cl)
2618{
2619 // Freeing what the class contains may recursively come back here.
2620 // Clear "class_name" first, if it is NULL the class does not need to
2621 // be freed.
2622 VIM_CLEAR(cl->class_name);
2623
2624 class_unref(cl->class_extends);
2625
2626 for (int i = 0; i < cl->class_interface_count; ++i)
2627 {
2628 vim_free(((char_u **)cl->class_interfaces)[i]);
2629 if (cl->class_interfaces_cl[i] != NULL)
2630 class_unref(cl->class_interfaces_cl[i]);
2631 }
2632 vim_free(cl->class_interfaces);
2633 vim_free(cl->class_interfaces_cl);
2634
2635 itf2class_T *next;
2636 for (itf2class_T *i2c = cl->class_itf2class; i2c != NULL; i2c = next)
2637 {
2638 next = i2c->i2c_next;
2639 vim_free(i2c);
2640 }
2641
2642 for (int i = 0; i < cl->class_class_member_count; ++i)
2643 {
2644 ocmember_T *m = &cl->class_class_members[i];
2645 vim_free(m->ocm_name);
2646 vim_free(m->ocm_init);
2647 if (cl->class_members_tv != NULL)
2648 clear_tv(&cl->class_members_tv[i]);
2649 }
2650 vim_free(cl->class_class_members);
2651 vim_free(cl->class_members_tv);
2652
2653 for (int i = 0; i < cl->class_obj_member_count; ++i)
2654 {
2655 ocmember_T *m = &cl->class_obj_members[i];
2656 vim_free(m->ocm_name);
2657 vim_free(m->ocm_init);
2658 }
2659 vim_free(cl->class_obj_members);
2660
2661 for (int i = 0; i < cl->class_class_function_count; ++i)
2662 {
2663 ufunc_T *uf = cl->class_class_functions[i];
2664 func_clear_free(uf, FALSE);
2665 }
2666 vim_free(cl->class_class_functions);
2667
2668 for (int i = 0; i < cl->class_obj_method_count; ++i)
2669 {
2670 ufunc_T *uf = cl->class_obj_methods[i];
2671 func_clear_free(uf, FALSE);
2672 }
2673 vim_free(cl->class_obj_methods);
2674
2675 clear_type_list(&cl->class_type_list);
2676
2677 class_cleared(cl);
2678
2679 vim_free(cl);
2680}
2681
2682/*
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002683 * Unreference a class. Free it when the reference count goes down to zero.
2684 */
2685 void
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002686class_unref(class_T *cl)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002687{
Bram Moolenaard505d172022-12-18 21:42:55 +00002688 if (cl != NULL && --cl->class_refcount <= 0 && cl->class_name != NULL)
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02002689 class_free(cl);
2690}
2691
2692/*
2693 * Go through the list of all classes and free items without "copyID".
2694 */
2695 int
2696class_free_nonref(int copyID)
2697{
2698 int did_free = FALSE;
2699
2700 for (class_T *cl = first_class; cl != NULL; cl = next_nonref_class)
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002701 {
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02002702 next_nonref_class = cl->class_next_used;
2703 if ((cl->class_copyID & COPYID_MASK) != (copyID & COPYID_MASK))
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00002704 {
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02002705 // Free the class and items it contains.
2706 class_free(cl);
2707 did_free = TRUE;
Bram Moolenaara94bd9d2023-01-12 15:01:32 +00002708 }
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002709 }
Yegappan Lakshmanane651e112023-09-04 07:51:01 +02002710
2711 next_nonref_class = NULL;
2712 return did_free;
2713}
2714
2715 int
2716set_ref_in_classes(int copyID)
2717{
2718 for (class_T *cl = first_class; cl != NULL; cl = cl->class_next_used)
2719 set_ref_in_item_class(cl, copyID, NULL, NULL);
2720
2721 return FALSE;
Bram Moolenaar00b28d62022-12-08 15:32:33 +00002722}
2723
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002724static object_T *first_object = NULL;
2725
2726/*
2727 * Call this function when an object has been created. It will be added to the
2728 * list headed by "first_object".
2729 */
2730 void
2731object_created(object_T *obj)
2732{
2733 if (first_object != NULL)
2734 {
2735 obj->obj_next_used = first_object;
2736 first_object->obj_prev_used = obj;
2737 }
2738 first_object = obj;
2739}
2740
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002741static object_T *next_nonref_obj = NULL;
2742
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002743/*
2744 * Call this function when an object has been cleared and is about to be freed.
2745 * It is removed from the list headed by "first_object".
2746 */
2747 void
2748object_cleared(object_T *obj)
2749{
2750 if (obj->obj_next_used != NULL)
2751 obj->obj_next_used->obj_prev_used = obj->obj_prev_used;
2752 if (obj->obj_prev_used != NULL)
2753 obj->obj_prev_used->obj_next_used = obj->obj_next_used;
2754 else if (first_object == obj)
2755 first_object = obj->obj_next_used;
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002756
2757 // update the next object to check if needed
2758 if (obj == next_nonref_obj)
2759 next_nonref_obj = obj->obj_next_used;
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002760}
2761
2762/*
2763 * Go through the list of all objects and free items without "copyID".
2764 */
2765 int
2766object_free_nonref(int copyID)
2767{
2768 int did_free = FALSE;
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002769
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002770 for (object_T *obj = first_object; obj != NULL; obj = next_nonref_obj)
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002771 {
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002772 next_nonref_obj = obj->obj_next_used;
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002773 if ((obj->obj_copyID & COPYID_MASK) != (copyID & COPYID_MASK))
2774 {
2775 // Free the object and items it contains.
2776 object_clear(obj);
2777 did_free = TRUE;
2778 }
2779 }
2780
Bram Moolenaarf7ca56f2023-06-05 16:53:25 +01002781 next_nonref_obj = NULL;
Bram Moolenaard28d7b92022-12-08 20:42:00 +00002782 return did_free;
2783}
2784
LemonBoyafe04662023-08-23 21:08:11 +02002785/*
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002786 * Echo a class or object method not found message.
2787 */
2788 void
2789method_not_found_msg(class_T *cl, vartype_T v_type, char_u *name, size_t len)
2790{
2791 char_u *method_name = vim_strnsave(name, len);
2792 if ((v_type == VAR_OBJECT)
2793 && (class_method_idx(cl, name, len) >= 0))
2794 {
2795 // If this is a class method, then give a different error
2796 if (*name == '_')
2797 semsg(_(e_cannot_access_private_method_str), method_name);
2798 else
RestorerZ7fe8f432023-09-24 23:21:24 +02002799 semsg(_(e_class_method_str_accessible_only_using_class_str),
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002800 method_name, cl->class_name);
2801 }
2802 else if ((v_type == VAR_CLASS)
2803 && (object_method_idx(cl, name, len) >= 0))
2804 {
2805 // If this is an object method, then give a different error
2806 if (*name == '_')
2807 semsg(_(e_cannot_access_private_method_str), method_name);
2808 else
RestorerZ7fe8f432023-09-24 23:21:24 +02002809 semsg(_(e_object_method_str_accessible_only_using_object_str),
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002810 method_name, cl->class_name);
2811 }
2812 else
2813 semsg(_(e_method_not_found_on_class_str_str), cl->class_name,
2814 method_name);
2815 vim_free(method_name);
2816}
2817
2818/*
2819 * Echo a class or object member not found message.
2820 */
2821 void
2822member_not_found_msg(class_T *cl, vartype_T v_type, char_u *name, size_t len)
2823{
2824 char_u *varname = len ? vim_strnsave(name, len) : vim_strsave(name);
2825
2826 if (v_type == VAR_OBJECT)
2827 {
2828 if (class_member_idx(cl, name, len) >= 0)
RestorerZ7fe8f432023-09-24 23:21:24 +02002829 semsg(_(e_class_variable_str_accessible_only_using_class_str),
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002830 varname, cl->class_name);
2831 else
RestorerZ7fe8f432023-09-24 23:21:24 +02002832 semsg(_(e_variable_not_found_on_object_str_str), cl->class_name,
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002833 varname);
2834 }
2835 else
2836 {
2837 if (object_member_idx(cl, name, len) >= 0)
RestorerZ7fe8f432023-09-24 23:21:24 +02002838 semsg(_(e_object_variable_str_accessible_only_using_object_str),
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002839 varname, cl->class_name);
2840 else
RestorerZ7fe8f432023-09-24 23:21:24 +02002841 semsg(_(e_class_variable_str_not_found_in_class_str),
Yegappan Lakshmananc30a90d2023-09-15 20:14:55 +02002842 varname, cl->class_name);
2843 }
2844 vim_free(varname);
2845}
2846
2847/*
Yegappan Lakshmanand4e4ecb2023-08-27 18:35:45 +02002848 * Return TRUE when the class "cl", its base class or one of the implemented
2849 * interfaces matches the class "other_cl".
LemonBoyafe04662023-08-23 21:08:11 +02002850 */
2851 int
2852class_instance_of(class_T *cl, class_T *other_cl)
2853{
2854 if (cl == other_cl)
2855 return TRUE;
2856
2857 // Recursively check the base classes.
2858 for (; cl != NULL; cl = cl->class_extends)
2859 {
2860 if (cl == other_cl)
2861 return TRUE;
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02002862 // Check the implemented interfaces and the super interfaces
LemonBoyafe04662023-08-23 21:08:11 +02002863 for (int i = cl->class_interface_count - 1; i >= 0; --i)
Yegappan Lakshmanan92d9ee52023-09-17 17:03:19 +02002864 {
2865 class_T *intf = cl->class_interfaces_cl[i];
2866 while (intf != NULL)
2867 {
2868 if (intf == other_cl)
2869 return TRUE;
2870 // check the super interfaces
2871 intf = intf->class_extends;
2872 }
2873 }
LemonBoyafe04662023-08-23 21:08:11 +02002874 }
2875
2876 return FALSE;
2877}
2878
2879/*
2880 * "instanceof(object, classinfo)" function
2881 */
2882 void
2883f_instanceof(typval_T *argvars, typval_T *rettv)
2884{
2885 typval_T *object_tv = &argvars[0];
2886 typval_T *classinfo_tv = &argvars[1];
2887 listitem_T *li;
2888
2889 rettv->vval.v_number = VVAL_FALSE;
2890
2891 if (check_for_object_arg(argvars, 0) == FAIL
2892 || check_for_class_or_list_arg(argvars, 1) == FAIL)
2893 return;
2894
Ernie Rael3da696d2023-09-19 20:14:18 +02002895 if (object_tv->vval.v_object == NULL)
2896 return;
2897
LemonBoyafe04662023-08-23 21:08:11 +02002898 if (classinfo_tv->v_type == VAR_LIST)
2899 {
2900 FOR_ALL_LIST_ITEMS(classinfo_tv->vval.v_list, li)
2901 {
2902 if (li->li_tv.v_type != VAR_CLASS)
2903 {
2904 emsg(_(e_class_required));
2905 return;
2906 }
2907
2908 if (class_instance_of(object_tv->vval.v_object->obj_class,
2909 li->li_tv.vval.v_class) == TRUE)
2910 {
2911 rettv->vval.v_number = VVAL_TRUE;
2912 return;
2913 }
2914 }
2915 }
2916 else if (classinfo_tv->v_type == VAR_CLASS)
2917 {
2918 rettv->vval.v_number = class_instance_of(object_tv->vval.v_object->obj_class,
2919 classinfo_tv->vval.v_class);
2920 }
2921}
Bram Moolenaarc1c365c2022-12-04 20:13:24 +00002922
2923#endif // FEAT_EVAL