584c98686d
Refactor api and autocomplete to use it and properly handle errors related to invalid project names.
83 lines
2.9 KiB
Python
Executable file
83 lines
2.9 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# This file is part of Elixir, a source code cross-referencer.
|
|
#
|
|
# Copyright (C) 2017--2020 Maxime Chretien <maxime.chretien@bootlin.com>
|
|
# and contributors.
|
|
#
|
|
# Elixir is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Elixir is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with Elixir. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import sys
|
|
import os
|
|
import json
|
|
from urllib import parse
|
|
from bsddb3.db import DB_SET_RANGE
|
|
import falcon
|
|
|
|
ELIXIR_DIR = os.path.dirname(os.path.realpath(__file__)) + '/..'
|
|
|
|
if ELIXIR_DIR not in sys.path:
|
|
sys.path = [ ELIXIR_DIR ] + sys.path
|
|
|
|
from lib import autoBytes
|
|
from query import get_query
|
|
|
|
class AutocompleteResource:
|
|
def on_get(self, req, resp):
|
|
# Get values from http GET
|
|
query_string = req.get_param('q')
|
|
query_family = req.get_param('f')
|
|
query_project = req.get_param('p')
|
|
|
|
query = get_query(req.context.config.project_dir, query_project)
|
|
if not query:
|
|
resp.status = falcon.HTTP_NOT_FOUND
|
|
return
|
|
|
|
latest = query.query('latest')
|
|
|
|
if query_family == 'B':
|
|
# DTS identifiers are stored quoted
|
|
process = lambda x: parse.unquote(x)
|
|
db = query.db.comps
|
|
else:
|
|
process = lambda x: x
|
|
db = query.db.defs
|
|
|
|
response = []
|
|
|
|
i = 0
|
|
cur = db.db.cursor()
|
|
query_bytes = autoBytes(parse.quote(query_string))
|
|
# Find "the smallest key greater than or equal to the specified key"
|
|
# https://docs.oracle.com/cd/E17276_01/html/api_reference/C/dbcget.html
|
|
# In practice this should mean "the key that starts with provided prefix"
|
|
# See docs about the default comparison function for B-Tree databases:
|
|
# https://docs.oracle.com/cd/E17276_01/html/api_reference/C/dbset_bt_compare.html
|
|
key, _ = cur.get(query_bytes, DB_SET_RANGE)
|
|
while i <= 10:
|
|
if key.startswith(query_bytes):
|
|
# If found key starts with the prefix, add to response
|
|
# and move to the next key
|
|
i += 1
|
|
response.append(process(key.decode("utf-8")))
|
|
key, _ = cur.next()
|
|
else:
|
|
# If found key does not start with the prefix, stop
|
|
break
|
|
|
|
resp.status = falcon.HTTP_200
|
|
resp.content_type = falcon.MEDIA_JSON
|
|
resp.media = response
|
|
|