1 files changed, 35 insertions, 16 deletions

diff --git a/src/nvim/eval.c b/src/nvim/eval.c
index 16a96ba29e..6e9f89bbb5 100644
--- a/src/nvim/eval.c
+++ b/src/nvim/eval.c
@@ -6950,6 +6950,7 @@ static struct fst {
   { "winrestview",       1, 1, f_winrestview },
   { "winsaveview",       0, 0, f_winsaveview },
   { "winwidth",          1, 1, f_winwidth },
+  { "wordcount",         0, 0, f_wordcount },
   { "writefile",         2, 3, f_writefile },
   { "xor",               2, 2, f_xor },
 };
@@ -8338,6 +8339,7 @@ static void f_cursor(typval_T *argvars, typval_T *rettv)
 {
   long line, col;
   long coladd = 0;
+  bool set_curswant = true;
 
   rettv->vval.v_number = -1;
   if (argvars[1].v_type == VAR_UNKNOWN) {
@@ -8354,30 +8356,35 @@ static void f_cursor(typval_T *argvars, typval_T *rettv)
     coladd = pos.coladd;
     if (curswant >= 0) {
       curwin->w_curswant = curswant - 1;
+      set_curswant = false;
     }
   } else {
     line = get_tv_lnum(argvars);
     col = get_tv_number_chk(&argvars[1], NULL);
-    if (argvars[2].v_type != VAR_UNKNOWN)
+    if (argvars[2].v_type != VAR_UNKNOWN) {
       coladd = get_tv_number_chk(&argvars[2], NULL);
+    }
   }
   if (line < 0 || col < 0
-      || coladd < 0
-      )
-    return;             /* type error; errmsg already given */
-  if (line > 0)
+      || coladd < 0) {
+    return;             // type error; errmsg already given
+  }
+  if (line > 0) {
     curwin->w_cursor.lnum = line;
-  if (col > 0)
+  }
+  if (col > 0) {
     curwin->w_cursor.col = col - 1;
+  }
   curwin->w_cursor.coladd = coladd;
 
-  /* Make sure the cursor is in a valid position. */
+  // Make sure the cursor is in a valid position.
   check_cursor();
-  /* Correct cursor for multi-byte character. */
-  if (has_mbyte)
+  // Correct cursor for multi-byte character.
+  if (has_mbyte) {
     mb_adjust_cursor();
+  }
 
-  curwin->w_set_curswant = TRUE;
+  curwin->w_set_curswant = set_curswant;
   rettv->vval.v_number = 0;
 }
 
@@ -14683,25 +14690,30 @@ static void f_setpos(typval_T *argvars, typval_T *rettv)
   name = get_tv_string_chk(argvars);
   if (name != NULL) {
     if (list2fpos(&argvars[1], &pos, &fnum, &curswant) == OK) {
-      if (--pos.col < 0)
+      if (--pos.col < 0) {
         pos.col = 0;
+      }
       if (name[0] == '.' && name[1] == NUL) {
-        /* set cursor */
+        // set cursor
         if (fnum == curbuf->b_fnum) {
           curwin->w_cursor = pos;
           if (curswant >= 0) {
             curwin->w_curswant = curswant - 1;
+            curwin->w_set_curswant = false;
           }
           check_cursor();
           rettv->vval.v_number = 0;
-        } else
+        } else {
           EMSG(_(e_invarg));
+        }
       } else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)   {
-        /* set mark */
-        if (setmark_pos(name[1], &pos, fnum) == OK)
+        // set mark
+        if (setmark_pos(name[1], &pos, fnum) == OK) {
           rettv->vval.v_number = 0;
-      } else
+        }
+      } else {
         EMSG(_(e_invarg));
+      }
     }
   }
 }
@@ -16926,6 +16938,13 @@ static void f_winwidth(typval_T *argvars, typval_T *rettv)
     rettv->vval.v_number = wp->w_width;
 }
 
+/// "wordcount()" function
+static void f_wordcount(typval_T *argvars, typval_T *rettv)
+{
+  rettv_dict_alloc(rettv);
+  cursor_pos_info(rettv->vval.v_dict);
+}
+
 /// "writefile()" function
 static void f_writefile(typval_T *argvars, typval_T *rettv)
 {