Merge pull request #28 from Garulf/fix-auth-user-repo-search

Fix auth user repo search

Author: Garulf

src/query.py

@@ -2,33 +2,34 @@
 from pyflowlauncher.result import ResultResponse, send_results
 from pyflowlauncher.jsonrpc import JsonRPCClient
 
-import results
+from results import repo_results, scored_repo_results
 
 STARS_PREFIX = "*"
 SEPERATOR = "/"
 
+PER_PAGE = 100
+SEARCH_LIMIT = 15
 
-def query(query: str) -> ResultResponse:
-    if not query:
-        return send_results([])
 
+def query(query: str) -> ResultResponse:
     settings = JsonRPCClient().recieve().get("settings", {})
     token = settings.get("token", None) or None
-    gh = Github(login_or_token=token, per_page=15)
+    gh = Github(login_or_token=token, per_page=PER_PAGE)
 
     repo_query = query
     parsed_query = query.split(SEPERATOR)
 
-    if query.startswith(STARS_PREFIX) and token:
-        stars = query[1:]
-        repos = gh.get_user().get_starred()
-        return send_results(results.starred_repo_results(stars, repos))
+    if token:
+        if query.startswith("/") or query == "":
+            repos = gh.get_user().get_repos()
+            return send_results(scored_repo_results(query, repos))
+        elif query.startswith(STARS_PREFIX):
+            query = query[1:]
+            repos = gh.get_user().get_starred()
+            return send_results(scored_repo_results(query, repos))
+    if len(parsed_query) == 2:
+        user, repo_query = parsed_query
+        results = repo_results(gh.search_repositories(f"user:{user} {repo_query}")[:SEARCH_LIMIT])
     else:
-        if len(parsed_query) > 1:
-            user, search = parsed_query
-            if not user and token is not None:
-                user = gh.get_user().login
-            if user:
-                repo_query = f"user:{user} {search}"
-        repos = gh.search_repositories(repo_query)
-    return send_results(results.repo_results(repos))
+        results = repo_results(gh.search_repositories(repo_query)[:SEARCH_LIMIT])
+    return send_results(results)

src/results.py

@@ -21,11 +21,11 @@ def repo_result(repo: Repository) -> Result:
 
 
 def repo_results(repos: PaginatedList) -> Generator[Result, None, None]:
-    for repo in repos.get_page(0):
+    for repo in repos:
         yield repo_result(repo)
 
 
-def starred_repo_results(query: str, repos: PaginatedList) -> Generator[Result, None, None]:
+def scored_repo_results(query: str, repos: PaginatedList) -> Generator[Result, None, None]:
     yield from score_results(query, repo_results(repos), match_on_empty_query=True)