vim9script
# Maintainer: Maxim Kim <habamax@gmail.com>
-# Last Update: 2024 Oct 05
+# Last Update: 2025 Mar 21
#
# Toggle comments
# Usage:
noautocmd keepjumps setline(lnum1, lines)
return ''
enddef
+
+
+# Comment text object
+# Usage:
+# import autoload 'dist/comment.vim'
+# onoremap <silent>ic <scriptcmd>comment.ObjComment(v:true)<CR>
+# onoremap <silent>ac <scriptcmd>comment.ObjComment(v:false)<CR>
+# xnoremap <silent>ic <esc><scriptcmd>comment.ObjComment(v:true)<CR>
+# xnoremap <silent>ac <esc><scriptcmd>comment.ObjComment(v:false)<CR>
+export def ObjComment(inner: bool)
+ def IsComment(): bool
+ var stx = map(synstack(line('.'), col('.')), 'synIDattr(v:val, "name")')->join()
+ return stx =~? 'Comment'
+ enddef
+
+ # requires syntax support
+ if !exists("g:syntax_on")
+ return
+ endif
+
+ var pos_init = getcurpos()
+
+ # If not in comment, search next one,
+ if !IsComment()
+ if search('\v\k+', 'W', line(".") + 100, 100, () => !IsComment()) <= 0
+ return
+ endif
+ endif
+
+ # Search for the beginning of the comment block
+ if IsComment()
+ if search('\v%(\S+)|$', 'bW', 0, 200, IsComment) > 0
+ search('\v%(\S)|$', 'W', 0, 200, () => !IsComment())
+ else
+ cursor(1, 1)
+ search('\v\S+', 'cW', 0, 200)
+ endif
+ endif
+
+ var pos_start = getcurpos()
+
+ if !inner
+ var col = pos_start[2]
+ var prefix = getline(pos_start[1])[ : col - 2]
+ while col > 0 && prefix[col - 2] =~ '\s'
+ col -= 1
+ endwhile
+ pos_start[2] = col
+ endif
+
+ # Search for the comment end.
+ if pos_init[1] > pos_start[1]
+ cursor(pos_init[1], pos_init[2])
+ endif
+ if search('\v%(\S+)|$', 'W', 0, 200, IsComment) > 0
+ search('\S', 'beW', 0, 200, () => !IsComment())
+ else
+ if search('\%$', 'W', 0, 200) > 0
+ search('\ze\S', 'beW', 0, 200, () => !IsComment())
+ endif
+ endif
+
+ var pos_end = getcurpos()
+
+ if !inner
+ var spaces = matchstr(getline(pos_end[1]), '\%>.c\s*')
+ pos_end[2] += spaces->len()
+ if getline(pos_end[1])[pos_end[2] : ] =~ '^\s*$'
+ && (pos_start[2] == 1 || getline(pos_start[1])[ : pos_start[2]] =~ '^\s*$')
+ if search('\v\s*\_$(\s*\n)+', 'eW', 0, 200) > 0
+ pos_end = getcurpos()
+ endif
+ endif
+ endif
+
+ if (pos_end[2] == (getline(pos_end[1])->len() ?? 1)) && pos_start[2] == 1
+ cursor(pos_end[1], 1)
+ normal! V
+ cursor(pos_start[1], 1)
+ else
+ cursor(pos_end[1], pos_end[2])
+ normal! v
+ cursor(pos_start[1], pos_start[2])
+ endif
+enddef
call assert_equal(["# vim9script", "", "# def Hello()", '# echo "Hello"', "# enddef"], result)
endfunc
-
func Test_basic_uncomment()
CheckScreendump
let lines =<< trim END
let result = readfile(output_file)
- call assert_equal(["# vim9script", "", "def Hello()", ' echo "Hello"', "enddef"], result)
+ call assert_equal(["# vim9script", "", "def Hello()", ' echo "Hello"', "enddef"], result)
endfunc
func Test_bothends_comment()
call assert_equal(["/* int main() { */", "\t/* if 1 { */", "\t /* return 0; */", "\t/* } */", " /* return 1; */", "/* } */"], result)
endfunc
+
+func Test_textobj_icomment()
+ CheckScreendump
+ let lines =<< trim END
+ for x in range(10):
+ print(x) # printing stuff
+ # print(x*x)
+ #print(x*x*x)
+ print(x*x*x*x) # printing stuff
+ print(x*x*x*x*x) # printing stuff
+ # print(x*x)
+ #print(x*x*x)
+
+ print(x*x*x*x*x)
+ END
+
+ let input_file = "test_textobj_icomment_input.py"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "dic..")
+ let output_file = "comment_textobj_icomment.py"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["for x in range(10):", " print(x) ", " print(x*x*x*x) ", " print(x*x*x*x*x) ", "", " print(x*x*x*x*x)"], result)
+endfunc
+
+func Test_textobj_icomment2()
+ CheckScreendump
+ let lines =<< trim END
+ #include <stdio.h>
+
+ int main() {
+ printf("hello"); /* hello world */ printf(" world\n");
+ /* if 1 {
+ return 1;
+ }*/
+
+ return 0;
+ }
+ END
+
+ let input_file = "test_textobj_icomment2_input.c"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "dic..")
+ let output_file = "comment_textobj_icomment2.c"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["#include <stdio.h>", "", "int main() {", " printf(\"hello\"); printf(\" world\\n\");", " ", "", " return 0;", "}"], result)
+endfunc
+
+func Test_textobj_icomment3()
+ CheckScreendump
+ let lines =<< trim END
+ #include <stdio.h>
+
+ int main() {
+ printf("hello");/*hello world*/printf(" world\n");
+ return 0;
+ }
+ END
+
+ let input_file = "test_textobj_icomment3_input.c"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "jjjdic")
+ let output_file = "comment_textobj_icomment3.c"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["#include <stdio.h>", "", "int main() {", " printf(\"hello\");printf(\" world\\n\");", " return 0;", "}"], result)
+endfunc
+
+func Test_textobj_acomment()
+ CheckScreendump
+ let lines =<< trim END
+ for x in range(10):
+ print(x) # printing stuff
+ # print(x*x)
+ #print(x*x*x)
+ print(x*x*x*x) # printing stuff
+ print(x*x*x*x*x) # printing stuff
+ # print(x*x)
+ #print(x*x*x)
+
+ print(x*x*x*x*x)
+ END
+
+ let input_file = "test_textobj_acomment_input.py"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "dac..")
+ let output_file = "comment_textobj_acomment.py"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["for x in range(10):", " print(x)", " print(x*x*x*x)", " print(x*x*x*x*x)", "", " print(x*x*x*x*x)"], result)
+endfunc
+
+func Test_textobj_acomment2()
+ CheckScreendump
+ let lines =<< trim END
+ #include <stdio.h>
+
+ int main() {
+ printf("hello"); /* hello world */ printf(" world\n");
+ /* if 1 {
+ return 1;
+ }*/
+
+ return 0;
+ }
+ END
+
+ let input_file = "test_textobj_acomment2_input.c"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "dac.")
+ let output_file = "comment_textobj_acomment2.c"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["#include <stdio.h>", "", "int main() {", " printf(\"hello\");printf(\" world\\n\");", " return 0;", "}"], result)
+endfunc
+
+func Test_textobj_acomment3()
+ CheckScreendump
+ let lines =<< trim END
+ #include <stdio.h>
+
+ int main() {
+ printf("hello");/*hello world*/printf(" world\n");
+ return 0;
+ }
+ END
+
+ let input_file = "test_textobj_acomment3_input.c"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "jjjdac")
+ let output_file = "comment_textobj_acomment3.c"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["#include <stdio.h>", "", "int main() {", " printf(\"hello\");printf(\" world\\n\");", " return 0;", "}"], result)
+endfunc
+
+func Test_textobj_firstline_comment()
+ CheckScreendump
+ let lines =<< trim END
+ /*#include <stdio.h>*/
+
+ int main() {}
+ END
+
+ let input_file = "test_textobj_firstlinecomment_input.c"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "dac")
+ let output_file = "comment_textobj_firstline_comment.c"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["int main() {}"], result)
+endfunc
+
+func Test_textobj_noleading_space_comment()
+ CheckScreendump
+ let lines =<< trim END
+ int main() {// main start
+ }/* main end */
+ END
+
+ let input_file = "test_textobj_noleading_space_input.c"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "dacdic")
+ let output_file = "comment_textobj_noleading_space_comment.c"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["int main() {", "}"], result)
+endfunc
+
+func Test_textobj_noleading_space_comment2()
+ CheckScreendump
+ let lines =<< trim END
+ int main() {// main start
+ } /* main end */
+ END
+
+ let input_file = "test_textobj_noleading_space_input2.c"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "dac.")
+ let output_file = "comment_textobj_noleading_space_comment2.c"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["int main() {", "}"], result)
+endfunc
+
+func Test_textobj_cursor_on_leading_space_comment()
+ CheckScreendump
+ let lines =<< trim END
+ int main() {
+ // multilple comments
+ // cursor is between them
+ }
+ END
+
+ let input_file = "test_textobj_cursor_on_leading_space_comment_input.c"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "jjdac")
+ let output_file = "comment_textobj_cursor_on_leading_space_comment.c"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["int main() {", "", "}"], result)
+endfunc
+
+func Test_textobj_conseq_comment()
+ CheckScreendump
+ let lines =<< trim END
+ int main() {
+ printf("hello"); // hello
+ // world
+ printf("world");
+ }
+ END
+
+ let input_file = "test_textobj_conseq_comment_input.c"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "dac")
+ let output_file = "comment_textobj_conseq_comment.c"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["int main() {", " printf(\"hello\");", " printf(\"world\");", "}"], result)
+endfunc
+
+func Test_textobj_conseq_comment2()
+ CheckScreendump
+ let lines =<< trim END
+ int main() {
+ printf("hello"); // hello
+
+ // world
+ printf("world");
+ }
+ END
+
+ let input_file = "test_textobj_conseq_comment_input2.c"
+ call writefile(lines, input_file, "D")
+
+ let buf = RunVimInTerminal('-c "packadd comment" ' .. input_file, {})
+
+ call term_sendkeys(buf, "dac")
+ let output_file = "comment_textobj_conseq_comment2.c"
+ call term_sendkeys(buf, $":w {output_file}\<CR>")
+ defer delete(output_file)
+
+ call StopVimInTerminal(buf)
+
+ let result = readfile(output_file)
+
+ call assert_equal(["int main() {", " printf(\"hello\");", "", " // world", " printf(\"world\");", "}"], result)
+endfunc
projects / thirdparty / vim.git / commitdiff
