update the problem state after submitted

Author: tenfyzhong

autoload/leetcode.vim

@@ -1069,6 +1069,7 @@ function! s:ShowRunResultInPreview(result) abort
     hi! link lcWarning lcFailure
 
     execute saved_winnr . 'wincmd w'
+    call <SID>UpdateSubmitState(a:result)
 endfunction
 
 function! s:CheckRunCodeTask(timer) abort
@@ -1295,3 +1296,29 @@ function! s:FormatIntoColumns(words) abort
 
     return lines
 endfunction
+
+function! s:UpdateSubmitState(result)
+    let state = '?'
+    if a:result['state'] == 'Accepted'
+        let state = 'X'
+    endif
+    let buffers = filter(range(1, bufnr('$')), 'buflisted(v:val)')
+    for b in buffers
+        let name = bufname(b)
+        if name !~ 'leetcode:\/\/\/problems'
+            continue
+        endif
+
+        let problems = getbufvar(b, 'leetcode_downloaded_problems')
+        if type(problems) != 3
+            continue
+        endif
+
+        for problem in problems
+            if problem['title'] == a:result['title'] 
+                let problem['state'] = state
+                break
+            endif
+        endfor
+    endfor
+endfunction

plugin/leetcode.vim

@@ -24,7 +24,7 @@ if !exists('g:leetcode_debug')
     let g:leetcode_debug = 0
 endif
 
-command! -nargs=0 LeetCodeList call leetcode#ListProblems('norefresh')
+command! -nargs=0 LeetCodeList call leetcode#ListProblems('redraw')
 command! -nargs=0 LeetCodeReset call leetcode#ResetSolution(0)
 command! -nargs=0 LeetCodeTest call leetcode#TestSolution()
 command! -nargs=0 LeetCodeSubmit call leetcode#SubmitSolution()