Parse options of documentclass

with updated __pprint and test file

Author: kiryph

autoload/vimtex/state/class.vim

@@ -34,7 +34,8 @@ function! vimtex#state#class#new(opts) abort " {{{1
         \ ? vimtex#parser#preamble(l:new.tex, {'root' : l:new.root})
         \ : []
 
-  let l:new.documentclass = s:parse_documentclass(l:preamble)
+  let [l:new.documentclass, l:new.documentclass_options] =
+        \ s:parse_documentclass(l:preamble)
   let l:new.packages = s:parse_packages(l:preamble)
   let l:new.graphicspath = s:parse_graphicspath(l:preamble, l:new.root)
   let l:new.glossaries = s:parse_glossaries(
@@ -74,6 +75,16 @@ function! s:vimtex.__pprint() abort dict " {{{1
     call add(l:items, ['document class', self.documentclass])
   endif
 
+  if exists('self.documentclass_options')
+    let l:string = join(map(sort(keys(self.documentclass_options)),
+          \ {_, key -> key.."="..
+          \ (self.documentclass_options[key]== v:true ? 'true' :
+          \  self.documentclass_options[key] == v:false ? 'false' :
+          \  self.documentclass_options[key])
+          \ }))
+    call add(l:items, ['document class options', l:string])
+  endif
+
   if !empty(self.packages)
     call add(l:items, ['packages', join(sort(keys(self.packages)))])
   endif
@@ -194,9 +205,37 @@ endfunction
 
 
 function! s:parse_documentclass(preamble) abort " {{{1
-  let l:preamble_lines = filter(copy(a:preamble), {_, x -> x !~# '^\s*%'})
-  return matchstr(join(l:preamble_lines, ''),
-        \ '\\documentclass[^{]*{\zs[^}]\+\ze}')
+  " Remove EOL comments and then join
+  let l:preamble_joined = join(map(copy(a:preamble),
+        \ {_, x -> split(x..' ', '%')[0]}), '')
+
+  let l:docclass = matchstr(l:preamble_joined, '\\documentclass[^{]*{\zs[^}]\+\ze}')
+
+  let l:docclass_options = {}
+  let l:unparsed = matchstr(l:preamble_joined, '\\documentclass[^\[]*\[\zs[^\]]\+\ze\]')
+  for l:el in map(split(l:unparsed, ','), {_, x -> trim(x)})
+    if l:el == ''
+      " Empty option
+      continue
+    elseif l:el =~ '='
+      " Key-value option
+      let [l:key, l:value] = map(split(l:el, '='), {_, x -> trim(x)} )
+
+      if l:value ==? 'true'
+        let l:docclass_options[l:key] = v:true
+      elseif l:value ==? 'false'
+        let l:docclass_options[l:key] = v:false
+      else
+        let l:docclass_options[l:key] = l:value
+      endif
+
+    else
+      " Key-only option
+      let l:docclass_options[l:el] = v:true
+    endif
+  endfor
+
+  return [l:docclass, l:docclass_options]
 endfunction
 
 " }}}1

test/test-state/test_parse_documentclass_options.tex

@@ -0,0 +1,34 @@
+% Leading comments here
+
+\documentclass[% Options of scrbook
+  %
+  % draft,
+  fontsize=12pt,
+  % smallheadings,
+  headings=big,
+  english,
+  paper=a4,
+  twoside,
+  open=right,
+  DIV=14,
+  BCOR=20mm,
+  headinclude=false,
+  footinclude=false,
+  mpinclude=false,
+  % pagesize,
+  titlepage,
+  parskip=half,
+  headsepline,
+  chapterprefix=false,
+  appendixprefix=Appendix,
+  appendixwithprefixline=true,
+  bibliography=totoc,
+  toc=graduated,
+  numbers=noenddot,
+]{scrbook}
+
+\begin{document}
+
+Hello, world!
+
+\end{document}

test/test-state/test_parse_documentclass_options.vim

@@ -0,0 +1,39 @@
+set nocompatible
+let &rtp = '../..,' . &rtp
+filetype plugin on
+
+nnoremap q :qall!<cr>
+
+call vimtex#log#set_silent()
+
+silent edit test_parse_documentclass_options.tex
+
+if empty($INMAKE) | finish | endif
+
+call assert_equal('scrbook', b:vimtex.documentclass)
+
+let s:options = {
+      \  'fontsize': '12pt',
+      \  'headings': 'big',
+      \  'english': v:true,
+      \  'paper': 'a4',
+      \  'twoside': v:true,
+      \  'open': 'right',
+      \  'DIV': '14',
+      \  'BCOR': '20mm',
+      \  'headinclude': v:false,
+      \  'footinclude': v:false,
+      \  'mpinclude': v:false,
+      \  'titlepage': v:true,
+      \  'parskip': 'half',
+      \  'headsepline': v:true,
+      \  'chapterprefix': v:false,
+      \  'appendixprefix': 'Appendix',
+      \  'appendixwithprefixline': v:true,
+      \  'bibliography': 'totoc',
+      \  'toc': 'graduated',
+      \  'numbers': 'noenddot',
+      \ }
+call assert_equal(s:options, b:vimtex.documentclass_options)
+
+call vimtex#test#finished()