minor reorganization

bin/autojump

@@ -23,6 +23,7 @@ from __future__ import print_function
 
 from collections import namedtuple
 from functools import partial
+from itertools import chain
 # FIXME(ting|2013-12-17): fix imports for Python 3 compatability
 from itertools import ifilter
 from itertools import imap
@@ -84,7 +85,7 @@ def parse_environment(config):
     return config
 
 
-def eval_arguments(config):
+def parse_arguments():
     """Evaluate arguments and run appropriate logic, returning an error code."""
     parser = ArgumentParser(
             description='Automatically jump to directory passed as an argument.',
@@ -119,49 +120,7 @@ def eval_arguments(config):
             '-v', '--version', action="version", version="%(prog)s " +
             VERSION, help='show version information')
 
-    args = parser.parse_args()
+    return parser.parse_args()
-
-    if args.add:
-        add_path(config, args.add)
-        return 0
-
-    # if args.complete:
-        # config['match_cnt'] = 9
-        # config['ignore_case'] = True
-
-    if args.decrease:
-        try:
-            print_entry(decrease_path(config, os.getcwdu(), args.decrease))
-            return 0
-        except OSError:
-            print("Current directory no longer exists.", file=sys.stderr)
-            return 1
-
-    if args.increase:
-        try:
-            print_entry(add_path(config, os.getcwdu(), args.increase))
-            return 0
-        except OSError:
-            print("Current directory no longer exists.", file=sys.stderr)
-            return 1
-
-    if args.purge:
-        print("Purged %d entries." % purge_missing_paths(config))
-        return 0
-
-    if args.stat:
-        print_stats(config)
-        return 0
-
-    # default behavior, no optional arguments
-    result = first(find_matches(config, args.directory))
-    if result:
-        print(encode_local(result.path))
-    else:
-        # always return something so the calling shell function has something
-        # to `cd` to
-        print(encode_local('.'))
-    return 0
 
 
 def add_path(config, path, increment=10):
@@ -206,7 +165,7 @@ def detect_smartcase(needles):
 
 
 def find_matches(config, needles):
-    """Return an iterator to a matching entries."""
+    """Return an iterator to matching entries."""
     entriefy = lambda tup: Entry(*tup)
     not_cwd = lambda entry: entry.path != os.getcwdu()
     data = sorted(
@@ -221,10 +180,11 @@ def find_matches(config, needles):
     needles = map(sanitize, needles)
     ignore_case = detect_smartcase(needles)
 
-    exact_matches = match_consecutive(needles, data, ignore_case)
+    consecutive_matches = match_consecutive(needles, data, ignore_case)
+    fuzzy_matches = match_fuzzy(needles, data, ignore_case)
 
     exists = lambda entry: os.path.exists(entry.path)
-    return ifilter(exists, exact_matches)
+    return ifilter(exists, chain(consecutive_matches, fuzzy_matches))
 
 
 def match_consecutive(needles, haystack, ignore_case=False):
@@ -267,24 +227,10 @@ def match_consecutive(needles, haystack, ignore_case=False):
     return ifilter(found, haystack)
 
 
-def match_regex(needles, haystack, ignore_case=False):
+def match_fuzzy(needles, haystack, ignore_case=False):
     """
-    Performs an exact match by combining all arguments into a single regex
-    expression and finding matches.
-
-    For example:
-        needles = ['qui', 'fox']
-        regex needle = r'.*qui.*fox.*'
-        haystack = [
-            (path="foobar", 10.0),
-            (path="The quick brown fox jumped over the lazy dog", 12.3)]
-
-        result = [(path="The quick brown fox jumped over the lazy dog", 12.3)]
     """
-    regex_needle = '.*' + '.*'.join(needles) + '.*'
+    return []
-    regex_flags = re.IGNORECASE | re.UNICODE if ignore_case else re.UNICODE
-    has_needle = lambda haystack: re.search(regex_needle, haystack.path, flags=regex_flags)
-    return ifilter(has_needle, haystack)
 
 
 def purge_missing_paths(config):
@@ -315,7 +261,42 @@ def print_stats(config):
 
 
 def main():
-    return eval_arguments(parse_environment(set_defaults()))
+    config = parse_environment(set_defaults())
+    args = parse_arguments()
+
+    if args.add:
+        add_path(config, args.add)
+    # elif args.complete:
+        # config['match_cnt'] = 9
+        # config['ignore_case'] = True
+    elif args.decrease:
+        try:
+            print_entry(decrease_path(config, os.getcwdu(), args.decrease))
+        except OSError:
+            print("Current directory no longer exists.", file=sys.stderr)
+            return 1
+    elif args.increase:
+        try:
+            print_entry(add_path(config, os.getcwdu(), args.increase))
+        except OSError:
+            print("Current directory no longer exists.", file=sys.stderr)
+            return 1
+    elif args.purge:
+        print("Purged %d entries." % purge_missing_paths(config))
+    elif args.stat:
+        print_stats(config)
+    else:
+        # default behavior, no optional arguments
+        result = first(find_matches(load(config).iteritems(), args.directory))
+
+        if result:
+            print(encode_local(result.path))
+        else:
+            # always return something so the calling shell function has an
+            # argument to `cd` to
+            print(encode_local('.'))
+
+    return 0
 
 
 if __name__ == "__main__":