summaryrefslogtreecommitdiff
path: root/searx/engines/devicons.py
blob: 16ae4cc38756564937e7010cf4756d52fd06027e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
# SPDX-License-Identifier: AGPL-3.0-or-later
"""Devicons (icons)"""

import typing as t

from searx.result_types import EngineResults

if t.TYPE_CHECKING:
    from extended_types import SXNG_Response
    from search.processors.online import OnlineParams


about = {
    "website": "https://devicon.dev/",
    "wikidata_id": None,
    "official_api_documentation": None,
    "use_official_api": True,
    "results": "JSON",
}

cdn_base_url = "https://cdn.jsdelivr.net/gh/devicons/devicon@latest"
categories = ["images", "icons"]


def request(query: str, params: "OnlineParams"):
    params["url"] = f"{cdn_base_url}/devicon.json"
    params['query'] = query
    return params


def response(resp: "SXNG_Response") -> EngineResults:
    res = EngineResults()
    query_parts = resp.search_params["query"].lower().split(" ")

    def is_result_match(result: dict[str, t.Any]) -> bool:
        for part in query_parts:
            if part in result["name"]:
                return True

            for tag in result["altnames"] + result["tags"]:
                if part in tag:
                    return True

        return False

    filtered_results = filter(is_result_match, resp.json())
    for result in filtered_results:
        for image_type in result["versions"]["svg"]:
            img_src = f"{cdn_base_url}/icons/{result['name']}/{result['name']}-{image_type}.svg"
            res.add(
                res.types.LegacyResult(
                    {
                        "template": "images.html",
                        "url": img_src,
                        "title": result["name"],
                        "content": f"Base color: {result['color']}",
                        "img_src": img_src,
                        "img_format": "SVG",
                    }
                )
            )

    return res