git-branch-info.vim

   1 "
   2 " Git branch info
   3 " Last change: June 19 2008
   4 " Version> 0.0.7
   5 " Maintainer: Eustáquio 'TaQ' Rangel
   6 " License: GPL
   7 " URL: git://github.com/taq/vim-git-branch-info.git
   8 "
   9 " This plugin show branches information on the status line.
  10 " To install, just put this file on ~/.vim/plugins and set your status line:
  11 "
  12 " :set statusline=%{GitBranchInfoString}
  13 "
  14 " Of course you can append this configuration to an existing one and make all
  15 " the customization you want on the status line, like:
  16 "
  17 " :set statusline=%#ErrorMsg#%{GitBranchInfoString}%#StatusLine#
  18 "
  19 " The command above will show the Git branches info on the same color as the
  20 " error messages. You can choose any color scheme you want to. Use
  21 "
  22 " :help highlight-groups
  23 "
  24 " to check for some other options.
  25 "
  26 " There are some customization on the result string based on existing variables
  27 " like:
  28 "
  29 " let g:git_branch_status_head_current=1
  30 " This will show just the current head branch name
  31 "
  32 " let g:git_branch_status_text="text"
  33 " This will show 'text' before the branches. If not set ' Git ' (with a trailing
  34 " left space) will be displayed.
  35 "
  36 " let g:git_branch_status_nogit=""
  37 " The message when there is no Git repository on the current dir
  38 "
  39 " let g:git_branch_status_around=""
  40 " Characters to put around the branch strings. Need to be a pair or characters,
  41 " the first will be on the beginning of the branch string and the last on the
  42 " end.
  43 "
  44 " let g:git_branch_status_ignore_remotes=1
  45 " Ignore the remote branches. If you don't want information about them, this can
  46 " make things works faster.
  47 "
  48 " If you want to make your own customizations, you can use the GitBranchInfoTokens()
  49 " function. It returns an array with the current branch as the first element and
  50 " another array with the other branches as the second element, like:
  51 "
  52 " :set statusline=%#ErrorMsg#%{GitBranchInfoTokens()[0]}%#StatusLine#
  53 "
  54 " or
  55 "
  56 " :set statusline=%#StatusLineNC#\ Git\ %#ErrorMsg#\ %{GitBranchInfoTokens()[0]}\ %#StatusLine#
  57 "
  58 " will give you a nice custom formatted string.
  59 "
  60 " This will show you the current branch only. No prefix text, no characters
  61 " around it. You can also make another functions to use the returned array.
  62 "
  63 let s:menu_on   = 0
  64 let s:checking = ""
  65 let b:git_dir   = ""
  66 let b:git_load_branch = ""
  67
  68 autocmd BufEnter * call GitBranchInfoInit()
  69 autocmd BufWriteCmd * call GitBranchInfoWriteCheck()
  70
  71 function GitBranchInfoCheckGitDir()
  72         return exists("b:git_dir") && !empty(b:git_dir)
  73 endfunction
  74
  75 function GitBranchInfoCheckReadable()
  76         return filereadable(b:git_dir."/HEAD")
  77 endfunction
  78
  79 function GitBranchInfoWriteCheck()
  80         if !GitBranchInfoCheckGitDir()
  81                 exec "write"
  82                 return 1
  83         endif
  84         " if the branches are the same, no problem
  85         let l:current = GitBranchInfoTokens()[0]
  86         if l:current==b:git_load_branch
  87                 exec "write"
  88                 return 1
  89         endif
  90         " ask what we will do
  91         echohl ErrorMsg
  92         let l:answer = tolower(input("Loaded from \'".b:git_load_branch."\' branch but saving on \'".l:current."\' branch, confirm [y/n]? ","n"))
  93         echohl None
  94         let l:msg = "File ".(l:answer=="y" ? "" : "NOT ")."saved on branch \'".l:current."\'."
  95         " ok, save even with different branches
  96         if l:answer=="y"
  97                 exec "write"
  98         endif
  99         " show message
 100         echohl WarningMsg
 101         echo l:msg
 102         echohl None
 103         return l:answer=="y"
 104 endfunction
 105
 106 function GitBranchInfoInit()
 107         call GitBranchInfoFindDir()
 108         let l:current = GitBranchInfoTokens()
 109         let b:git_load_branch = l:current[0]
 110 endfunction
 111
 112 function GitBranchInfoFindDir()
 113         let l:bufname   = bufname("%")
 114         let l:buflist   = strlen(l:bufname)>0 ? split(l:bufname,"/") : [""]
 115         let l:prefix    = l:bufname =~ "^/" ? "/" : ""
 116         let b:git_dir   = ""
 117         for l:item in l:buflist
 118                 call remove(l:buflist,-1)
 119                 let l:path = l:prefix.join(l:buflist,"/").l:prefix.".git"
 120                 if !empty(finddir(l:path))
 121                         let b:git_dir = l:path
 122                         break
 123                 endif
 124         endfor
 125         return b:git_dir
 126 endfunction
 127
 128 function GitBranchInfoGitDir()
 129         return b:git_dir
 130 endfunction
 131
 132 function GitBranchInfoLoadBranch()
 133         return b:git_load_branch
 134 endfunction
 135
 136 function GitBranchInfoRenewMenu(current,heads,remotes)
 137         call GitBranchInfoRemoveMenu()
 138         call GitBranchInfoShowMenu(a:current,a:heads,a:remotes)
 139 endfunction
 140
 141 function GitBranchInfoCheckout(branch)
 142         let l:tokens    = GitBranchInfoTokens()
 143         let l:checkout  = "git\ checkout\ ".a:branch
 144         let l:where             = substitute(b:git_dir,".git$","","")
 145         let l:cmd               = strlen(l:where)>0 ? "!cd\ ".l:where.";\ ".l:checkout : "!".l:checkout
 146         exe l:cmd
 147         call GitBranchInfoRenewMenu(l:tokens[0],l:tokens[1],l:tokens[2])
 148 endfunction
 149
 150 function GitBranchInfoFetch(remote)
 151         let l:tokens    = GitBranchInfoTokens()
 152         let l:fetch             =  "git\ fetch\ ".a:remote
 153         let l:where             = substitute(b:git_dir,".git$","","")
 154         let l:cmd               = strlen(l:where)>0 ? "!cd\ ".l:where.";\ ".l:fetch : "!".l:fetch
 155         exe l:cmd
 156 endfunction
 157
 158 function GitBranchInfoShowMenu(current,heads,remotes)
 159         if !has("gui")
 160                 return
 161         endif
 162         let s:menu_on   = 1
 163         let l:compare   = a:current
 164         let l:current   = [a:current]
 165         let l:heads             = len(a:heads)>0         ? a:heads       : []
 166         let l:remotes   = len(a:remotes)>0 ? a:remotes : []
 167         let l:locals    = sort(extend(l:current,l:heads))
 168         for l:branch in l:locals
 169                 let l:moption   = (l:branch==l:compare ? "Working\\ \\on\\ " : "Checkout\\ ").l:branch
 170                 let l:mcom              = (l:branch==l:compare ? ":echo 'Already\ on\ branch\ \''".l:branch."\''.'<CR>" : "call GitBranchInfoCheckout('".l:branch."')<CR><CR>")
 171                 exe ":menu <silent> Plugin.Git\\ Info.".l:moption." :".l:mcom
 172         endfor
 173         exe ":menu <silent> Plugin.Git\\ Info.-Local- :"
 174         let l:lastone = ""
 175         for l:branch in l:remotes
 176                 let l:tokens    = split(l:branch,"/")
 177                 if l:tokens[0]==l:lastone
 178                         continue
 179                 endif
 180                 let l:lastone = l:tokens[0]
 181                 exe "menu <silent> Plugin.Git\\ Info.Fetch\\ ".l:tokens[0]." :call GitBranchInfoFetch('".l:tokens[0]."')<CR><CR>"
 182         endfor
 183 endfunction
 184
 185 function GitBranchInfoRemoveMenu()
 186         if !has("gui") || s:menu_on==0
 187                 return
 188         endif
 189         exe ":unmenu Plugin.Git\\ Info"
 190         let s:menu_on = 0
 191 endfunction
 192
 193 function GitBranchInfoString()
 194         let l:tokens    = GitBranchInfoTokens() " get the tokens
 195         if len(l:tokens)==1                                                     " no git here
 196                 call GitBranchInfoRemoveMenu()
 197                 return l:tokens[0]
 198         end
 199         let s:current   = l:tokens[0]                           " the current branch is the first one
 200         let l:branches  = l:tokens[1]                           " the other branches are the last one
 201         let l:remotes   = l:tokens[2]                           " remote branches
 202         " check for around characters
 203         let l:around    = exists("g:git_branch_status_around") ? (strlen(g:git_branch_status_around)==2 ? split(g:git_branch_status_around,'\zs') : ["",""]) : ["[","]"]
 204         " find the prefix text
 205         let l:text              = exists("g:git_branch_status_text")   ? g:git_branch_status_text : " Git "
 206         if s:menu_on == 0
 207                 call GitBranchInfoShowMenu(l:tokens[0],l:tokens[1],l:tokens[2])
 208         endif
 209         return l:text.l:around[0].s:current.l:around[1].(exists("g:git_branch_status_head_current")?"":l:around[0].join(l:branches,",").l:around[1])
 210 endfunction
 211
 212 function GitBranchInfoTokens()
 213         if !GitBranchInfoCheckGitDir()
 214                 let s:current = ''
 215                 return [exists("g:git_branch_status_nogit") ? g:git_branch_status_nogit : "No git."]
 216         endif
 217         if !GitBranchInfoCheckReadable()
 218                 let s:current = ''
 219                 return [s:current,[],[]]
 220         endif
 221         let s:current   = split(split(readfile(b:git_dir."/HEAD",'',1)[0])[1],"/")[2]
 222         if exists("g:git_branch_status_head_current")
 223                 let l:heads     = []
 224         else
 225                 let l:heads     = split(glob(b:git_dir."/refs/heads/*"),"\n")
 226                 call map(l:heads,'substitute(v:val,b:git_dir."/refs/heads/","","")')
 227                 call sort(filter(l:heads,'v:val !~ s:current'))
 228         endif
 229         if exists("g:git_branch_status_ignore_remotes")
 230                 let l:remotes = []
 231         else
 232                 let l:remotes   = split(glob(b:git_dir."/refs/remotes/*/**"),"\n")
 233                 call sort(map(l:remotes,'substitute(v:val,b:git_dir."/refs/remotes/","","")'))
 234         endif
 235         let l:checking = s:current.join(l:heads).join(l:remotes)
 236         if l:checking != s:checking && has("gui")
 237                 call GitBranchInfoRenewMenu(s:current,l:heads,l:remotes)
 238         endif
 239         let s:checking = l:checking
 240         return [s:current,l:heads,l:remotes]
 241 endfunction