Setup for Vim Ollama

[dotfiles.git] / invoke / setup-linux / tasks.py

diff --git a/invoke/setup-linux/tasks.py b/invoke/setup-linux/tasks.py
index 3587e7e34df047e93770471df773ccfdec36b1ab..067c8115b2c680554b5e59484fc7c51dca02afb6 100644 (file)
--- a/invoke/setup-linux/tasks.py
+++ b/invoke/setup-linux/tasks.py
@@ -1,8 +1,11 @@
 from invoke import task
 
+from datetime import datetime as dt
 import os
 
 HOME = os.path.expanduser('~')
+TODAY = dt.now().strftime('%Y.%m.%d')
+
 
 @task()
 def defaults(c):
@@ -61,8 +64,39 @@ def link_to_home(c, path, add_dot=False):
     c.run(f'ls -la {HOME} | grep {filename}')
 
 
+@task
+def link_bashrc(c):
+    c.run(f'mv {HOME}/.bashrc {HOME}/bashrc_bak_{TODAY}')
+    link_to_home(c, 'files/pc/bashrc', add_dot=True)
+
 
 @task
 def link_all_for_invoke(c):
     link_to_home(c, 'files/pc/tasks.py')
     link_to_home(c, 'files/pc/bashrc', add_dot=True)
+
+@task
+def vim_nerdtree(c):
+    add_vim_plugin(c, 'https://github.com/preservim/nerdtree.git', 'nerdtree')
+
+@task
+def vim_markdown(c):
+    add_vim_plugin(c, 'https://github.com/preservim/vim-markdown.git', 'markdown')
+ 
+@task
+def vim_airline(c):
+    add_vim_plugin(c, 'https://github.com/vim-airline/vim-airline.git', 'airline')
+
+@task
+def vim_ollama(c):
+    add_vim_plugin(c, 'https://github.com/gergap/vim-ollama.git', 'vim-ollama')
+
+@task
+def add_vim_plugin(c, git_path, folder):
+    c.run(f'mkdir -p {HOME}/.vim/pack/')
+    c.run(f'mkdir -p {HOME}/.vim/pack/vendor')
+    c.run(f'mkdir -p {HOME}/.vim/pack/vendor/opt')
+    c.run(f'git clone --depth 1 {git_path} {HOME}/.vim/pack/vendor/opt/{folder}')
+    c.run(f'ls -l {HOME}/.vim/pack/vendor/opt')
+    print(f"Use `packadd {folder}` in .vimrc to enable")
+