Add refresh problem list

Author: ianding1

README.md

@@ -70,7 +70,9 @@ Default value is `''`.
 
 ## Updates
 
-- 2019/07/27: Support LeetCode China
+- 2019/07/27:
+  + Support LeetCode China accounts
+  + Refresh the problem list with `r`
 - 2019/07/23: Support topics and companies
 
 ## FAQ

autoload/leetcode.vim

@@ -72,6 +72,7 @@ function! s:SetupProblemListBuffer() abort
     setlocal bufhidden=hide
     nnoremap <silent> <buffer> <return> :call <SID>HandleProblemListCR()<cr>
     nnoremap <silent> <buffer> s :call <SID>HandleProblemListS()<cr>
+    nnoremap <silent> <buffer> r :call <SID>HandleProblemListR()<cr>
 
     syn match lcEasy /| Easy /hs=s+2
     syn match lcMedium /| Medium /hs=s+2
@@ -107,6 +108,8 @@ function! s:PrintProblemList(problems) abort
                 \ '### Keys',
                 \ '  - <cr>  open the problem/go to the topic or company',
                 \ '  - s     view the submissions',
+                \ '  - r     refresh',
+                \ '',
                 \ '### Indicators',
                 \ '  - [P] = paid-only problems',
                 \ ''])
@@ -135,16 +138,23 @@ function! s:PrintProblemList(problems) abort
     let b:leetcode_problem_end_line = line('$')
 endfunction
 
-function! s:ListProblemsOfTopic(topic_slug) abort
+function! s:ListProblemsOfTopic(topic_slug, refresh) abort
     let buf_name = 'leetcode:///problems/topic/' . a:topic_slug
     if buflisted(buf_name)
         execute bufnr(buf_name) . 'buffer'
-        return
+        if a:refresh
+            setlocal modifiable
+            silent! normal! ggdG
+        else
+            return
+        endif
+    else
+        execute 'rightbelow new ' . buf_name
+        call s:SetupProblemListBuffer()
+        let b:leetcode_buffer_type = 'topic'
+        let b:leetcode_buffer_topic = a:topic_slug
     endif
 
-    execute 'rightbelow new ' . buf_name
-    call s:SetupProblemListBuffer()
-
     let expr = printf('leetcode.get_problems_of_topic("%s")', a:topic_slug)
     let problems = py3eval(expr)['problems']
 
@@ -164,16 +174,23 @@ function! s:ListProblemsOfTopic(topic_slug) abort
     silent! only
 endfunction
 
-function! s:ListProblemsOfCompany(company_slug) abort
+function! s:ListProblemsOfCompany(company_slug, refresh) abort
     let bufname = 'leetcode:///problems/company/' . a:company_slug
     if buflisted(bufname)
         execute bufnr(bufname) . 'buffer'
-        return
+        if a:refresh
+            setlocal modifiable
+            silent! normal! ggdG
+        else
+            return
+        endif
+    else
+        execute 'rightbelow new ' . bufname
+        call s:SetupProblemListBuffer()
+        let b:leetcode_buffer_type = 'company'
+        let b:leetcode_buffer_company = a:company_slug
     endif
 
-    execute 'rightbelow new ' . bufname
-    call s:SetupProblemListBuffer()
-
     let expr = printf('leetcode.get_problems_of_company("%s")', a:company_slug)
     let problems = py3eval(expr)['problems']
 
@@ -182,8 +199,6 @@ function! s:ListProblemsOfCompany(company_slug) abort
     let b:leetcode_company_start_line = 0
     let b:leetcode_company_end_line = 0
 
-    setlocal modifiable
-
     call append('$', ['# LeetCode [company:' . a:company_slug . ']', ''])
 
     call s:PrintProblemList(problems)
@@ -193,15 +208,24 @@ function! s:ListProblemsOfCompany(company_slug) abort
     silent! only
 endfunction
 
-function! leetcode#ListProblems() abort
+function! leetcode#ListProblems(refresh) abort
     if s:CheckSignIn() == v:false
         return
     endif
 
     let buf_name = 'leetcode:///problems/all'
     if buflisted(buf_name)
         execute bufnr(buf_name) . 'buffer'
-        return
+        if a:refresh
+            setlocal modifiable
+            silent! normal! ggdG
+        else
+            return
+        endif
+    else
+        execute 'rightbelow new ' . buf_name
+        call s:SetupProblemListBuffer()
+        let b:leetcode_buffer_type = 'all'
     endif
 
     let expr = printf('leetcode.get_problems(["all"])')
@@ -211,11 +235,6 @@ function! leetcode#ListProblems() abort
     let topics = topics_and_companies['topics']
     let companies = topics_and_companies['companies']
 
-    execute 'rightbelow new ' . buf_name
-    call s:SetupProblemListBuffer()
-
-    set modifiable
-
     " concatenate the topics into a string
     let topic_slugs = map(topics, 'v:val["topic_slug"]')
     let topic_lines = s:FormatIntoColumns(topic_slugs)
@@ -251,7 +270,7 @@ function! s:HandleProblemListCR() abort
                 \ line_nr < b:leetcode_topic_end_line
         let topic_slug = expand('<cWORD>')
         if topic_slug != ''
-            call s:ListProblemsOfTopic(topic_slug)
+            call s:ListProblemsOfTopic(topic_slug, 0)
         endif
         return
     endif
@@ -260,7 +279,7 @@ function! s:HandleProblemListCR() abort
                 \ line_nr < b:leetcode_company_end_line
         let company_slug = expand('<cWORD>')
         if company_slug != ''
-            call s:ListProblemsOfCompany(company_slug)
+            call s:ListProblemsOfCompany(company_slug, 0)
         endif
         return
     endif
@@ -282,6 +301,16 @@ function! s:HandleProblemListCR() abort
     endif
 endfunction
 
+function! s:HandleProblemListR() abort
+    if b:leetcode_buffer_type ==# 'all'
+        call leetcode#ListProblems(1)
+    elseif b:leetcode_buffer_type ==# 'topic'
+        call s:ListProblemsOfTopic(b:leetcode_buffer_topic, 1)
+    elseif b:leetcode_buffer_type ==# 'company'
+        call s:ListProblemsOfCompany(b:leetcode_buffer_company, 1)
+    endif
+endfunction
+
 let s:file_type_to_ext = {
             \ 'cpp': 'cpp',
             \ 'java': 'java',

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()
+command! -nargs=0 LeetCodeList call leetcode#ListProblems(0)
 command! -nargs=0 LeetCodeReset call leetcode#ResetSolution(0)
 command! -nargs=0 LeetCodeTest call leetcode#TestSolution()
 command! -nargs=0 LeetCodeSubmit call leetcode#SubmitSolution()