1 files changed, 42 insertions, 104 deletions
diff --git a/bin/vim-pack b/bin/vim-pack
index 25bc165..65c7fb4 100755
--- a/bin/vim-pack
+++ b/bin/vim-pack
@@ -1,104 +1,42 @@
-#!/usr/bin/env python
-
-import os
-import glob
-import asyncio
-import subprocess
-
-INTSTALL_DIR=os.path.expandvars("$HOME/.vim/pack/vendor")
-
-class Package(object):
- def __init__(self, name, location, type='opt', posthook=None):
- self.name = name
- self.location = location
- self.type = type
- self.posthook = posthook
- self.path = self.get_path()
-
- def fail(self):
- print("%s : FAIL" % self.name)
-
- def get_path(self):
- return os.path.expandvars("$HOME/.vim/pack/vendor/%s/%s" % (self.type, self.name))
-
- async def update(self):
- cmd = "cd %s && git pull" % self.path
- print("%s : Updating ..." % self.name)
-
- if not os.path.exists(self.path):
- os.mkdir(self.path)
- cmd = "git clone --depth 1 %s %s" % (self.location, self.path)
- if not os.path.isdir(self.path):
- print("ERROR: package directory error: %s" % self.name)
- self.fail()
- return
-
- process = await asyncio.create_subprocess_shell(
- cmd, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE,
- )
-
- stdout, stderr = await process.communicate()
-
- if process.returncode != 0:
- print("ERROR: git command error")
- print(stdout)
- print(stderr)
- self.fail()
- return
-
- if self.posthook:
- self.posthook()
-
- print("%s : Done" % (self.name))
-
- return ""
-
-packages = [
- Package(
- "nnn.vim", 
- "[email protected]:mcchrish/nnn.vim.git",
- "opt",
- ),
-]
-
-def update_packages(packages): 
- tasks = [ package.update() for package in packages ]
- loop = asyncio.get_event_loop()
- commands = asyncio.gather(*tasks)
- result = loop.run_until_complete(commands)
- loop.close()
- # print(result)
-
-def clean(packages):
- paths = [ package.path for package in packages ]
- to_clean = [ 
- path 
- for path in glob.glob(os.path.expandvars("$HOME/.vim/pack/vendor/*/*"))
- if path not in paths
- ]
-
- if not to_clean:
- return
-
- print("The following directory will be cleaned: [y/N]")
- print(to_clean)
- if input() not in ["y", "Yes", "yes"]:
- return
-
- cmd = ["rm", "-rf"] + to_clean
- result = subprocess.run(cmd, capture_output=True, shell=False)
- print(' '.join(cmd))
- if result.returncode != 0:
- print("FAIL on cleaning the following directory:")
- print(result.stdout)
- print(result.stderr)
- else:
- print("DONE")
-
-
-def main():
- update_packages(packages)
- clean(packages)
-
-if __name__ == "__main__":
- main()
+#!/usr/bin/env bash
+
+# package_dirs records directories of all installed packages. It is used
+# to help clean unused packages.
+package_dirs=()
+
+update_package () {
+ name=$1
+ remote_url=$2
+ load_type=$3
+
+ package_dir="$HOME/.vim/pack/vendor/$load_type/$name"
+ package_dirs+=($package_dir)
+ if [ ! -d $package_dir ]; then
+ echo "$name : Installing ... "
+ mkdir -p $package_dir
+ git clone --depth 1 $remote_url $package_dir
+ else
+ cd $package_dir
+ echo "$name : Updating ... "
+ git pull && echo "$name : Done" || echo "$name : FAIL"
+ fi
+}
+
+clean () {
+ for load_type in 'opt' 'start'; do
+ for package_name in $(ls $HOME/.vim/pack/vendor/$load_type); do
+ package_dir=$HOME/.vim/pack/vendor/$load_type/$package_name
+ if [[ ! " ${package_dirs[*]} " =~ " ${package_dir} " ]]; then
+ read -r -p "rm -rf $package_dir? [y/N]: "
+ response=${response,,}
+ if [[ "$response" =~ ^(yes|y)$ ]]; then
+ rm -rf $package_dir
+ fi
+ fi
+ done
+ done
+}
+
+update_package "nnn.vim" "[email protected]:mcchrish/nnn.vim.git" "opt"
+
+clean