searx/engines/google_images.py

@@ -1,20 +1,28 @@
 # SPDX-License-Identifier: AGPL-3.0-or-later
 # lint: pylint
-"""This is the implementation of the google images engine using the google
-internal API used the Google Go Android app.
+"""This is the implementation of the google images engine.
 
-This internal API offer results in
+.. admonition:: Content-Security-Policy (CSP)
 
-- JSON (_fmt:json)
-- Protobuf (_fmt:pb)
-- Protobuf compressed? (_fmt:pc)
-- HTML (_fmt:html)
-- Protobuf encoded in JSON (_fmt:jspb).
+   This engine needs to allow images from the `data URLs`_ (prefixed with the
+   ``data:`` scheme)::
 
+       Header set Content-Security-Policy "img-src 'self' data: ;"
+
+.. _data URLs:
+   https://developer.mozilla.org/en-US/docs/Web/HTTP/Basics_of_HTTP/Data_URIs
 """
 
-from urllib.parse import urlencode
-from json import loads
+import re
+from urllib.parse import urlencode, unquote
+from lxml import html
+
+from searx.utils import (
+    eval_xpath,
+    eval_xpath_list,
+    eval_xpath_getindex,
+    extract_text,
+)
 
 from searx.engines.google import (
     get_lang_info,
@@ -34,12 +42,12 @@ about = {
     "official_api_documentation": 'https://developers.google.com/custom-search',
     "use_official_api": False,
     "require_api_key": False,
-    "results": 'JSON',
+    "results": 'HTML',
 }
 
 # engine dependent config
 categories = ['images', 'web']
-paging = True
+paging = False
 use_locale_domain = True
 time_range_support = True
 safesearch = True
@@ -48,8 +56,74 @@ send_accept_language_header = True
 filter_mapping = {0: 'images', 1: 'active', 2: 'active'}
 
 
+def scrap_out_thumbs(dom):
+    """Scrap out thumbnail data from <script> tags."""
+    ret_val = {}
+    for script in eval_xpath(dom, '//script[contains(., "_setImgSrc(")]'):
+        _script = script.text
+        # _setImgSrc('0','data:image\/jpeg;base64,\/9j\/4AAQSkZJR ....');
+        _thumb_no, _img_data = _script[len("_setImgSrc(") : -2].split(",", 1)
+        _thumb_no = _thumb_no.replace("'", "")
+        _img_data = _img_data.replace("'", "")
+        _img_data = _img_data.replace(r"\/", r"/")
+        ret_val[_thumb_no] = _img_data.replace(r"\x3d", "=")
+    return ret_val
+
+
+# [0, "-H96xjSoW5DsgM", ["https://encrypted-tbn0.gstatic.com/images?q...", 155, 324]
+# , ["https://assets.cdn.moviepilot.de/files/d3bf..", 576, 1200],
+_RE_JS_IMAGE_URL = re.compile(
+    r'"'
+    r'([^"]*)'  # -H96xjSoW5DsgM
+    r'",\s*\["'
+    r'https://[^\.]*\.gstatic.com/images[^"]*'  # https://encrypted-tbn0.gstatic.com/images?q...
+    r'[^\[]*\["'
+    r'(https?://[^"]*)'  # https://assets.cdn.moviepilot.de/files/d3bf...
+)
+
+
+def parse_urls_img_from_js(dom):
+
+    # There are two HTML script tags starting with a JS function
+    # 'AF_initDataCallback(...)'
+    #
+    # <script nonce="zscm+Ab/JzBk1Qd4GY6wGQ">
+    #   AF_initDataCallback({key: 'ds:0', hash: '1', data:[], sideChannel: {}});
+    # </script>
+    # <script nonce="zscm+Ab/JzBk1Qd4GY6wGQ">
+    #   AF_initDataCallback({key: 'ds:1', hash: '2', data:[null,[[["online_chips",[["the big",
+    #     ["https://encrypted-tbn0.gstatic.com/images?q...",null,null,true,[null,0],f
+    #   ...
+    # </script>
+    #
+    # The second script contains the URLs of the images.
+
+    # The AF_initDataCallback(..) is called with very large dictionary, that
+    # looks like JSON but it is not JSON since it contains JS variables and
+    # constants like 'null' (we can't use a JSON parser for).
+    #
+    # The alternative is to parse the entire <script> and find all image URLs by
+    # a regular expression.
+
+    img_src_script = eval_xpath_getindex(dom, '//script[contains(., "AF_initDataCallback({key: ")]', 1).text
+    data_id_to_img_url = {}
+    for data_id, url in _RE_JS_IMAGE_URL.findall(img_src_script):
+        data_id_to_img_url[data_id] = url
+    return data_id_to_img_url
+
+
+def get_img_url_by_data_id(data_id_to_img_url, img_node):
+    """Get full image URL by @data-id from parent element."""
+
+    data_id = eval_xpath_getindex(img_node, '../../../@data-id', 0)
+    img_url = data_id_to_img_url.get(data_id, '')
+    img_url = unquote(img_url.replace(r'\u00', r'%'))
+
+    return img_url
+
+
 def request(query, params):
-    """Google-Image search request"""
+    """Google-Video search request"""
 
     lang_info = get_lang_info(params, supported_languages, language_aliases, False)
 
@@ -58,17 +132,7 @@ def request(query, params):
         + lang_info['subdomain']
         + '/search'
         + "?"
-        + urlencode(
-            {
-                'q': query,
-                'tbm': "isch",
-                **lang_info['params'],
-                'ie': "utf8",
-                'oe': "utf8",
-                'asearch': 'isch',
-                'async': '_fmt:json,p:1,ijn:' + str(params['pageno']),
-            }
-        )
+        + urlencode({'q': query, 'tbm': "isch", **lang_info['params'], 'ie': "utf8", 'oe': "utf8", 'num': 30})
     )
 
     if params['time_range'] in time_range_dict:
@@ -77,9 +141,9 @@ def request(query, params):
         query_url += '&' + urlencode({'safe': filter_mapping[params['safesearch']]})
     params['url'] = query_url
 
+    params['cookies']['CONSENT'] = "YES+"
     params['headers'].update(lang_info['headers'])
-    params['headers']['User-Agent'] = 'NSTN/3.60.474802233.release Dalvik/2.1.0 (Linux; U; Android 12; US) gzip'
-    params['headers']['Accept'] = '*/*'
+    params['headers']['Accept'] = 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8'
     return params
 
 
@@ -89,34 +153,78 @@ def response(resp):
 
     detect_google_sorry(resp)
 
-    json_start = resp.text.find('{"ischj":')
-    json_data = loads(resp.text[json_start:])
+    # convert the text to dom
+    dom = html.fromstring(resp.text)
+    img_bas64_map = scrap_out_thumbs(dom)
+    data_id_to_img_url = parse_urls_img_from_js(dom)
 
-    for item in json_data["ischj"]["metadata"]:
+    # parse results
+    #
+    # root element::
+    #     <div id="islmp" ..>
+    # result div per image::
+    #     <div jsmodel="tTXmib"> / <div jsaction="..." data-id="..."
+    #     The data-id matches to a item in a json-data structure in::
+    #         <script nonce="I+vqelcy/01CKiBJi5Z1Ow">AF_initDataCallback({key: 'ds:1', ... data:function(){return [ ...
+    #     In this structure the link to the origin PNG, JPG or whatever is given
+    # first link per image-div contains a <img> with the data-iid for bas64 encoded image data::
+    #      <img class="rg_i Q4LuWd" data-iid="0"
+    # second link per image-div is the target link::
+    #      <a class="VFACy kGQAp" href="https://en.wikipedia.org/wiki/The_Sacrament_of_the_Last_Supper">
+    # the second link also contains two div tags with the *description* and *publisher*::
+    #      <div class="WGvvNb">The Sacrament of the Last Supper ...</div>
+    #      <div class="fxgdke">en.wikipedia.org</div>
 
-        result_item = {
-            'url': item["result"]["referrer_url"],
-            'title': item["result"]["page_title"],
-            'content': item["text_in_grid"]["snippet"],
-            'source': item["result"]["site_title"],
-            'img_format': f'{item["original_image"]["width"]} x {item["original_image"]["height"]}',
-            'img_src': item["original_image"]["url"],
-            'thumbnail_src': item["thumbnail"]["url"],
-            'template': 'images.html',
-        }
+    root = eval_xpath(dom, '//div[@id="islmp"]')
+    if not root:
+        logger.error("did not find root element id='islmp'")
+        return results
 
-        author = item["result"].get('iptc', {}).get('creator')
-        if author:
-            result_item['author'] = ', '.join(author)
+    root = root[0]
+    for img_node in eval_xpath_list(root, './/img[contains(@class, "rg_i")]'):
 
-        copyright_notice = item["result"].get('iptc', {}).get('copyright_notice')
-        if copyright_notice:
-            result_item['source'] += ' / ' + copyright_notice
+        img_alt = eval_xpath_getindex(img_node, '@alt', 0)
 
-        file_size = item.get('gsa', {}).get('file_size')
-        if file_size:
-            result_item['source'] += ' (%s)' % file_size
+        img_base64_id = eval_xpath(img_node, '@data-iid')
+        if img_base64_id:
+            img_base64_id = img_base64_id[0]
+            thumbnail_src = img_bas64_map[img_base64_id]
+        else:
+            thumbnail_src = eval_xpath(img_node, '@src')
+            if not thumbnail_src:
+                thumbnail_src = eval_xpath(img_node, '@data-src')
+            if thumbnail_src:
+                thumbnail_src = thumbnail_src[0]
+            else:
+                thumbnail_src = ''
 
-        results.append(result_item)
+        link_node = eval_xpath_getindex(img_node, '../../../a[2]', 0)
+        url = eval_xpath_getindex(link_node, '@href', 0, None)
+        if url is None:
+            logger.error("missing @href in node: %s", html.tostring(link_node))
+            continue
+
+        pub_nodes = eval_xpath(link_node, './div/div')
+        pub_descr = img_alt
+        pub_source = ''
+        if pub_nodes:
+            pub_descr = extract_text(pub_nodes[0])
+            pub_source = extract_text(pub_nodes[1])
+
+        src_url = get_img_url_by_data_id(data_id_to_img_url, img_node)
+        if not src_url:
+            src_url = thumbnail_src
+
+        results.append(
+            {
+                'url': url,
+                'title': img_alt,
+                'content': pub_descr,
+                'source': pub_source,
+                'img_src': src_url,
+                'thumbnail_src': thumbnail_src,
+                'template': 'images.html',
+            }
+        )
 
     return results