diff src/vim9compile.c @ 19423:f3e8e74cb747 v8.2.0269

patch 8.2.0269: Vim9: operator after list index does not work Commit: https://github.com/vim/vim/commit/b13af50f73dd89503c915f76fcf92be58789521a Author: Bram Moolenaar <Bram@vim.org> Date: Mon Feb 17 21:12:08 2020 +0100 patch 8.2.0269: Vim9: operator after list index does not work Problem: Vim9: operator after list index does not work. (Yasuhiro Matsumoto) Solution: After indexing a list change the type to the list member type. (closes #5651)
author Bram Moolenaar <Bram@vim.org>
date Mon, 17 Feb 2020 21:15:05 +0100
parents bc880a130120
children 5d34ae66118e
line wrap: on
line diff
--- a/src/vim9compile.c
+++ b/src/vim9compile.c
@@ -2330,6 +2330,9 @@ compile_subscript(
 	}
 	else if (**arg == '[')
 	{
+	    garray_T	*stack;
+	    type_T	**typep;
+
 	    // list index: list[123]
 	    // TODO: more arguments
 	    // TODO: dict member  dict['name']
@@ -2346,6 +2349,14 @@ compile_subscript(
 
 	    if (generate_instr_drop(cctx, ISN_INDEX, 1) == FAIL)
 		return FAIL;
+	    stack = &cctx->ctx_type_stack;
+	    typep = ((type_T **)stack->ga_data) + stack->ga_len - 1;
+	    if ((*typep)->tt_type != VAR_LIST && *typep != &t_any)
+	    {
+		emsg(_(e_listreq));
+		return FAIL;
+	    }
+	    *typep = (*typep)->tt_member;
 	}
 	else if (**arg == '.' && (*arg)[1] != '.')
 	{