Fixes minor same directory issue

This commit is contained in:
Athitya 2017-06-29 22:57:31 +05:30
parent db3c579b9c
commit e51e1f855c

View file

@ -39,7 +39,8 @@ class ColorLS
end
def load_from_yaml(filename)
path = $PROGRAM_NAME.gsub('/colorls.rb', '')
prog = $PROGRAM_NAME
path = prog.include?('/colorls.rb') ? prog.gsub('/colorls.rb', '') : '.'
YAML.safe_load(File.read("#{path}/#{filename}"))
end