build / 3pp_common / scripthash.py [blame]

# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import hashlib
import logging
import os
import pathlib
import sys

_SRC_ROOT = pathlib.Path(__file__).resolve().parents[2]


def _find_deps():
    module_paths = (os.path.abspath(m.__file__) for m in sys.modules.values()
                    if m and getattr(m, '__file__', None))
    ret = set()
    for path in module_paths:
        if path.startswith(str(_SRC_ROOT)):
            if (path.endswith('.pyc')
                    or (path.endswith('c') and not os.path.splitext(path)[1])):
                path = path[:-1]
            ret.add(path)
    return list(ret)


def compute(extra_paths=None):
    """Compute a hash of loaded Python modules and given |extra_paths|."""
    all_paths = _find_deps() + (extra_paths or [])
    all_paths = [os.path.relpath(p, _SRC_ROOT) for p in all_paths]
    all_paths.sort()
    md5 = hashlib.md5()
    for path in all_paths:
        md5.update((_SRC_ROOT / path).read_bytes())
        md5.update(path.encode('utf-8'))
    logging.info('Script hash from: \n%s\n', '\n'.join(all_paths))
    return md5.hexdigest()[:10]