real-ladybug 0.0.1.dev1__cp312-cp312-win_amd64.whl

real_ladybug-source/scripts/get-clangd-diagnostics.py

@@ -0,0 +1,233 @@
+#!/usr/bin/env python3
+import argparse
+import io
+import json
+import multiprocessing
+import os
+import subprocess
+import sys
+
+
+class LSPClient:
+    def __init__(self, *, compile_commands_dir=None, jobs=None, verbose=False):
+        args = ["clangd", "-j", str(jobs)]
+        if jobs is None:
+            jobs = multiprocessing.cpu_count()
+
+        if compile_commands_dir is not None:
+            args += ["--compile-commands-dir", compile_commands_dir]
+
+        self.id = 0
+        self.child = subprocess.Popen(
+            args,
+            stdin=subprocess.PIPE,
+            stdout=subprocess.PIPE,
+            # Let clangd inherit our stderr, or suppress it entirely.
+            stderr=None if verbose else subprocess.DEVNULL,
+        )
+        self.stdin = io.TextIOWrapper(self.child.stdin, newline="\r\n")
+        self.stdout = io.TextIOWrapper(self.child.stdout, newline="\r\n")
+
+    def request(self, method, params):
+        self.send_request(method, params)
+        return self.recv_response()
+
+    def send_request(self, method, params):
+        self.id += 1
+        self.send_json(
+            dict(
+                id=self.id,
+                jsonrpc="2.0",
+                method=method,
+                params=params,
+            )
+        )
+
+    def send_json(self, json_data):
+        data = json.dumps(json_data)
+        bindata = data.encode("utf-8")
+        header = f"Content-Length: {len(bindata)}\r\n\r\n"
+
+        self.stdin.write(header + data)
+        self.stdin.flush()
+
+    def recv_response(self):
+        json_data = self.recv_json()
+        assert json_data["id"] == self.id
+        assert "error" not in json_data
+        return json_data["result"]
+
+    def send_notif(self, method, params):
+        self.send_json(
+            dict(
+                jsonrpc="2.0",
+                method=method,
+                params=params,
+            )
+        )
+
+    def expect_notif(self, method):
+        json_data = self.recv_json()
+        assert json_data["method"] == method
+        assert "error" not in json_data
+        return json_data["params"]
+
+    def recv_json(self):
+        header = self.stdout.readline()
+        content_len_header = "Content-Length: "
+        assert header.startswith(content_len_header)
+        assert header.endswith("\r\n")
+        data_len = int(header[len(content_len_header) : -2])
+
+        # Expect end of header
+        assert self.stdout.read(2) == "\r\n"
+
+        data = self.stdout.read(data_len)
+        return json.loads(data)
+
+    def initialize(self, project):
+        return self.request(
+            "initialize",
+            dict(
+                processId=os.getpid(),
+                rootUri="file://" + project,
+                capabilities={},
+            ),
+        )
+
+    def open_file(self, file):
+        with open(file) as f:
+            file_content = f.read()
+
+        self.send_notif(
+            "textDocument/didOpen",
+            dict(
+                textDocument=dict(
+                    uri="file://" + os.path.realpath(file),
+                    languageId="cpp",
+                    version=1,
+                    text=file_content,
+                )
+            ),
+        )
+
+    def show_diagnostics(self):
+        diagnostics_response = self.expect_notif("textDocument/publishDiagnostics")
+        uri = diagnostics_response["uri"]
+        file_prefix = "file://"
+        assert uri.startswith(file_prefix)
+        file = uri[len(file_prefix) :]
+        diagnostics = diagnostics_response["diagnostics"]
+        for diagnostic in diagnostics:
+            LSPClient.show_diagnostic(file, diagnostic)
+
+        return len(diagnostics) != 0
+
+    def show_diagnostic(file, diagnostic):
+        range = diagnostic["range"]
+        start = range["start"]
+        line = start["line"]
+        message = diagnostic["message"]
+        print(f"{file}:{line}:{message}")
+
+    def send_shutdown(self):
+        self.send_request("shutdown", None)
+
+    def recv_shutdown_send_exit(self):
+        self.send_notif("exit", None)
+
+    def wait(self):
+        assert self.child.wait() == 0
+
+
+def shutdown_all(clients):
+    # Shutdown the clients in parallel. This drastically speeds up cleanup time.
+    for client in clients:
+        client.send_shutdown()
+
+    for client in clients:
+        client.recv_shutdown_send_exit()
+
+    for client in clients:
+        client.wait()
+
+
+def get_clients(client_count, compile_commands_dir, total_jobs, verbose):
+    # Distribute jobs evenly.
+    # Clients near the front get more jobs, but they also may get more files.
+    job_count = [0] * client_count
+    for i in range(total_jobs):
+        job_count[i % client_count] += 1
+    return [LSPClient(compile_commands_dir=compile_commands_dir, jobs=jobs, verbose=verbose) for jobs in job_count]
+
+
+def get_diagnostics(files, *, client_count, compile_commands_dir, total_jobs, verbose):
+    if client_count > total_jobs:
+        print(f"Client count {client_count} is greater than total jobs {total_jobs}. Forcing the client count to {total_jobs}.", file=sys.stderr)
+        client_count = total_jobs
+
+    project = os.getcwd()
+
+    client_count = min(client_count, len(files))
+    clients = get_clients(client_count, compile_commands_dir, total_jobs, verbose)
+    for client in clients:
+        client.initialize(project)
+
+    # Similar to distributing jobs, we distribute files evenly.
+    file_counts = [0] * client_count
+    for i, file in enumerate(files):
+        client_idx = i % client_count
+        clients[client_idx].open_file(file)
+        file_counts[client_idx] += 1
+
+    any_diagnostic = 0
+    for file_count, client in zip(file_counts, clients):
+        for _ in range(file_count):
+            any_diagnostic |= client.show_diagnostics()
+
+    shutdown_all(clients)
+    return any_diagnostic
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        prog="get-clangd-diagnostics.py",
+        description="Scan project for any clangd diagnostics (including warnings) and outputs them.",
+    )
+    parser.add_argument("files", nargs="+", help="Files to scan")
+    parser.add_argument(
+        "--instances",
+        type=int,
+        default=4,
+        help="Number of clangd instances to spawn in parallel. Defaults to 4.",
+    )
+    parser.add_argument(
+        "-j",
+        "--jobs",
+        type=int,
+        help="Number of total jobs across all servers. Defaults to the CPU count.",
+    )
+    parser.add_argument(
+        "-p",
+        "--compile-commands-dir",
+        help="Directory containining compile_commands.json",
+    )
+    parser.add_argument(
+        "-v",
+        "--verbose",
+        action="store_true",
+        help="Show clangd debug output",
+    )
+
+    args = parser.parse_args(sys.argv[1:])
+    jobs = args.jobs if args.jobs is not None else multiprocessing.cpu_count()
+    return get_diagnostics(
+        args.files,
+        client_count=args.instances,
+        compile_commands_dir=args.compile_commands_dir,
+        total_jobs=jobs,
+        verbose=args.verbose,
+    )
+
+
+sys.exit(main())

real_ladybug-source/scripts/migrate-lbug-db.py

@@ -0,0 +1,308 @@
+#!/usr/bin/env python3
+"""
+Lbug Database Migration Script
+
+This script helps migrate Lbug databases between versions.
+- Sets up isolated Python environments for each Lbug version
+- Exports data from the source database using the old version
+- Imports data into the target database using the new version
+- If `overwrite` is enabled, the target database will replace the source database and the source database will be backed up with an `_old` suffix
+- If `delete-old` is enabled, the source database will be deleted
+
+Usage Examples:
+    # Basic migration from 0.9.0 to 0.11.0
+    python migrate-lbug-db.py --old-version 0.9.0 --new-version 0.11.0 --old-db /path/to/old/database --new-db /path/to/new/database
+
+Notes:
+- Can only be used to migrate to newer Lbug versions, from 0.11.0 onwards
+"""
+
+import tempfile
+import sys
+import struct
+import shutil
+import subprocess
+import argparse
+import os
+
+# Database file extensions
+LBUG_FILE_EXTENSIONS = ["", ".wal", ".shadow"]
+
+
+# FIXME: Replace this with a Lbug query to get the mapping when available.
+lbug_version_mapping = {
+    34: "0.7.0",
+    35: "0.7.1",
+    36: "0.8.2",
+    37: "0.9.0",
+    38: "0.10.1",
+    39: "0.11.0",
+}
+
+minimum_lbug_migration_version = "0.11.0"
+
+
+def lbug_version_comparison(version: str, target: str) -> bool:
+    """Return True if Lbug *v* is greater or equal to target version"""
+    # Transform version string to version tuple to use in version tuple comparison
+    # NOTE: If version info contains non digit info (like dev release info 0.11.0.dev1) set the value of the non digit
+    # tuple part to be 0 (transform it to 0.11.0.0)
+    target = tuple(int(part) if part.isdigit() else 0 for part in target.split("."))
+    current = tuple(int(part) if part.isdigit() else 0 for part in version.split("."))
+    return current >= target
+
+
+def read_lbug_storage_version(lbug_db_path: str) -> int:
+    """
+    Reads the Lbug storage version.
+
+    :param lbug_db_path: Path to the Lbug database file/directory.
+    :return: Storage version code as an integer.
+    """
+    if os.path.isdir(lbug_db_path):
+        lbug_version_file_path = os.path.join(lbug_db_path, "catalog.kz")
+        if not os.path.isfile(lbug_version_file_path):
+            raise FileNotFoundError("Lbug catalog.kz file does not exist")
+    else:
+        lbug_version_file_path = lbug_db_path
+
+    with open(lbug_version_file_path, "rb") as f:
+        f.seek(4)
+        # Read the next 8 bytes as a little-endian unsigned 64-bit integer
+        data = f.read(8)
+        if len(data) < 8:
+            raise ValueError(
+                f"File '{lbug_version_file_path}' does not contain a storage version code."
+            )
+        version_code = struct.unpack("<Q", data)[0]
+
+    if version_code in lbug_version_mapping:
+        return lbug_version_mapping[version_code]
+    else:
+        raise ValueError(f"Could not map version_code {version_code} to proper Lbug version.")
+
+
+def ensure_env(version: str, export_dir) -> str:
+    """
+    Creates a venv at `{export_dir}/.lbug_envs/{version}` and installs `lbug=={version}`
+    Returns the venv's python executable path.
+    """
+    # Use temp directory to create venv
+    lbug_envs_dir = os.path.join(export_dir, ".lbug_envs")
+
+    # venv base under the script directory
+    base = os.path.join(lbug_envs_dir, version)
+    py_bin = os.path.join(base, "bin", "python")
+    # If environment already exists clean it
+    if os.path.isdir(base):
+        shutil.rmtree(base)
+
+    print(f"→ Setting up venv for Lbug {version}...", file=sys.stderr)
+    # Create venv
+    # NOTE: Running python in debug mode can cause issues with creating a virtual environment from that python instance
+    subprocess.run([sys.executable, "-m", "venv", base], check=True)
+    # Install the specific Lbug version
+    subprocess.run([py_bin, "-m", "pip", "install", "--upgrade", "pip"], check=True)
+    subprocess.run([py_bin, "-m", "pip", "install", f"lbug=={version}"], check=True)
+    return py_bin
+
+
+def run_migration_step(python_exe: str, db_path: str, cypher: str):
+    """
+    Uses `python_exe` to connect to the Lbug database at `db_path` and run the `cypher` query.
+    """
+    snippet = f"""
+import lbug
+db = lbug.Database(r"{db_path}")
+conn = lbug.Connection(db)
+conn.execute(r\"\"\"{cypher}\"\"\")
+"""
+    proc = subprocess.run([python_exe, "-c", snippet], capture_output=True, text=True)
+    if proc.returncode != 0:
+        print(f"Error: query failed:\n{cypher}\n{proc.stderr}", file=sys.stderr)
+        sys.exit(proc.returncode)
+
+
+def lbug_migration(
+    new_db, old_db, new_version, old_version=None, overwrite=None, delete_old=None
+):
+    """
+    Main migration function that handles the complete migration process.
+    """
+    if new_db == old_db:
+        raise ValueError(
+            "The new database path cannot be the same as the old database path. Please provide a different path for the new database."
+        )
+
+    if not lbug_version_comparison(
+        version=new_version, target=minimum_lbug_migration_version
+    ):
+        raise ValueError(
+            f"New version for lbug is not supported, has to be equal or higher than version: {minimum_lbug_migration_version}"
+        )
+
+    print(
+        f"Migrating Lbug database from {old_version} to {new_version}", file=sys.stderr
+    )
+    print(f"Source: {old_db}", file=sys.stderr)
+    print("", file=sys.stderr)
+
+    # If version of old lbug db is not provided try to determine it based on file info
+    if not old_version:
+        old_version = read_lbug_storage_version(old_db)
+
+    # Check if old database exists
+    if not os.path.exists(old_db):
+        raise FileNotFoundError(f"Source database '{old_db}' does not exist.")
+
+    # Prepare target - ensure parent directory exists but remove target if it exists
+    parent_dir = os.path.dirname(new_db)
+    if parent_dir:
+        os.makedirs(parent_dir, exist_ok=True)
+
+    if os.path.exists(new_db):
+        raise FileExistsError(
+            f"File already exists at {new_db}, remove file or change new database file path to continue"
+        )
+
+    # Use temp directory for all processing, it will be cleaned up after with statement
+    with tempfile.TemporaryDirectory() as export_dir:
+        # Set up environments
+        print(f"Setting up Lbug {old_version} environment...", file=sys.stderr)
+        old_py = ensure_env(old_version, export_dir)
+        print(f"Setting up Lbug {new_version} environment...", file=sys.stderr)
+        new_py = ensure_env(new_version, export_dir)
+
+        export_file = os.path.join(export_dir, "lbug_export")
+        print(f"Exporting old DB → {export_dir}", file=sys.stderr)
+        run_migration_step(old_py, old_db, f"EXPORT DATABASE '{export_file}'")
+        print("Export complete.", file=sys.stderr)
+
+        # Check if export files were created and have content
+        schema_file = os.path.join(export_file, "schema.cypher")
+        if not os.path.exists(schema_file) or os.path.getsize(schema_file) == 0:
+            raise ValueError(f"Schema file not found: {schema_file}")
+
+        print(f"Importing into new DB at {new_db}", file=sys.stderr)
+        run_migration_step(new_py, new_db, f"IMPORT DATABASE '{export_file}'")
+        print("Import complete.", file=sys.stderr)
+
+    # Rename new lbug database to old lbug database name if enabled
+    if overwrite or delete_old:
+        # Remove lbug lock from migrated DB
+        lock_file = new_db + ".lock"
+        if os.path.exists(lock_file):
+            os.remove(lock_file)
+        rename_databases(old_db, old_version, new_db, delete_old)
+
+    print("Lbug graph database migration finished successfully!")
+
+
+def rename_databases(old_db: str, old_version: str, new_db: str, delete_old: bool):
+    """
+    When overwrite is enabled, back up the original old_db (file with .shadow and .wal or directory)
+    by renaming it to *_old, and replace it with the newly imported new_db files.
+
+    When delete_old is enabled, replace the old database with the new one and delete old database.
+
+    :raises FileNotFoundError: If the original database path is not found
+    :raises OSError: If file operations fail
+    """
+    base_dir = os.path.dirname(old_db)
+    name = os.path.basename(old_db.rstrip(os.sep))
+    # Add _old_ and version info to backup graph database
+    backup_database_name = f"{name}_old_" + old_version.replace(".", "_")
+    backup_base = os.path.join(base_dir, backup_database_name)
+
+    if os.path.isfile(old_db):
+        # File-based database: handle main file and accompanying lock/WAL
+        for ext in LBUG_FILE_EXTENSIONS:
+            src = old_db + ext
+            dst = backup_base + ext
+            if os.path.exists(src):
+                if delete_old:
+                    os.remove(src)
+                else:
+                    os.rename(src, dst)
+                    print(f"Renamed '{src}' to '{dst}'", file=sys.stderr)
+    elif os.path.isdir(old_db):
+        # Directory-based Lbug database
+        backup_dir = backup_base
+        if delete_old:
+            shutil.rmtree(old_db)
+        else:
+            os.rename(old_db, backup_dir)
+            print(f"Renamed directory '{old_db}' to '{backup_dir}'", file=sys.stderr)
+    else:
+        print(
+            f"Original database path '{old_db}' not found for renaming.",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+
+    # Now move new files into place
+    for ext in ["", ".wal", ".shadow"]:
+        src_new = new_db + ext
+        dst_new = os.path.join(base_dir, name + ext)
+        if os.path.exists(src_new):
+            os.rename(src_new, dst_new)
+            print(f"Renamed '{src_new}' to '{dst_new}'", file=sys.stderr)
+
+
+def main():
+    p = argparse.ArgumentParser(
+        description="Migrate Lbug DB via PyPI versions",
+        epilog="""
+Examples:
+  %(prog)s --old-version 0.9.0 --new-version 0.11.0 \\
+    --old-db /path/to/old/db --new-db /path/to/new/db --overwrite
+
+Note: This script will create temporary virtual environments in .lbug_envs/ directory
+to isolate different Lbug versions.
+        """,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    p.add_argument(
+        "--old-version",
+        required=False,
+        default=None,
+        help="Source Lbug version (e.g., 0.9.0). If not provided, automatic lbug version detection will be attempted.",
+    )
+    p.add_argument(
+        "--new-version", required=True, help="Target Lbug version (e.g., 0.11.0)"
+    )
+    p.add_argument("--old-db", required=True, help="Path to source database directory")
+    p.add_argument(
+        "--new-db",
+        required=True,
+        help="Path to target database directory, it can't be the same path as the old database. Use the overwrite flag if you want to replace the old database with the new one.",
+    )
+    p.add_argument(
+        "--overwrite",
+        required=False,
+        action="store_true",
+        default=False,
+        help="Rename new-db to the old-db name and location, and keeps old-db as backup if delete-old is not True",
+    )
+    p.add_argument(
+        "--delete-old",
+        required=False,
+        action="store_true",
+        default=False,
+        help="When overwrite and delete-old are True, old-db will not be stored as backup",
+    )
+
+    args = p.parse_args()
+
+    lbug_migration(
+        new_db=args.new_db,
+        old_db=args.old_db,
+        new_version=args.new_version,
+        old_version=args.old_version,
+        overwrite=args.overwrite,
+        delete_old=args.delete_old,
+    )
+
+
+if __name__ == "__main__":
+    main()

real_ladybug-source/scripts/multiplatform-test-helper/collect-results.py

@@ -0,0 +1,71 @@
+#!/usr/bin/env python3
+
+import os
+import sys
+import pandas as pd
+import json
+
+
+def main():
+    if len(sys.argv) != 2:
+        print("Usage: collect-results.py <results_dir>")
+        sys.exit(1)
+    if not os.path.isdir(sys.argv[1]):
+        print(f"Error: {sys.argv[1]} is not a directory")
+        sys.exit(1)
+    results_dir = sys.argv[1]
+    results_df_hash = {}
+    results_exit_codes_hash = {}
+    results_summary = {}
+    stages = []
+    for root, _, files in os.walk(results_dir):
+        for csv_file in files:
+            if not csv_file.endswith(".csv"):
+                continue
+            name_split = csv_file.split(".")
+            name_split.pop()
+            platform = ".".join(name_split)
+            df = pd.read_csv(os.path.join(root, csv_file), header=None)
+            df.columns = ["stage", "exit_code"]
+            results_df_hash[platform] = df
+
+    for platform, df in results_df_hash.items():
+        for stage, exit_code in df.values:
+            if stage not in stages:
+                stages.append(stage)
+            if platform not in results_exit_codes_hash:
+                results_exit_codes_hash[platform] = {}
+            results_exit_codes_hash[platform][stage] = int(exit_code)
+
+    for platform in results_df_hash.keys():
+        results_summary[platform] = []
+        for stage in stages:
+            if stage in results_exit_codes_hash[platform]:
+                if results_exit_codes_hash[platform][stage] == 0:
+                    status = "✅"
+                else:
+                    status = "❌"
+            else:
+                status = "❓"
+            results_summary[platform].append({"stage": stage, "status": status})
+
+    summary_df = {"stage": stages}
+    for platform, summary in results_summary.items():
+        df = pd.DataFrame(summary)
+        status = df["status"]
+        summary_df[platform] = status
+    summary_df = pd.DataFrame(summary_df)
+    summary_df.index = summary_df["stage"]
+    del summary_df["stage"]
+    summary_df.index.name = None
+
+    markdown = summary_df.to_markdown()
+    with open("results.md", "w") as f:
+        f.write(markdown)
+
+    with open("results.json", "w") as f:
+        json.dump(results_summary, f, indent=4)
+
+
+if __name__ == "__main__":
+    main()

real_ladybug-source/scripts/multiplatform-test-helper/notify-discord.py

@@ -0,0 +1,68 @@
+import discord
+import os
+import json
+import sys
+
+TOKEN = os.getenv("DISCORD_BOT_TOKEN")
+CHANNEL_ID = os.getenv("DISCORD_CHANNEL_ID")
+GITHUB_URL = os.getenv("GITHUB_URL")
+
+messages = []
+
+
+if __name__ == "__main__":
+    if not len(sys.argv) == 2:
+        print("Usage: python send-discord-notification.py <result.json>")
+        sys.exit(1)
+    if not os.path.isfile(sys.argv[1]):
+        print(f"Error: {sys.argv[1]} is not a file")
+        sys.exit(1)
+    if not TOKEN:
+        print("Error: DISCORD_BOT_TOKEN is not set")
+        sys.exit(1)
+
+    if not CHANNEL_ID:
+        print("Error: DISCORD_CHANNEL_ID is not set")
+        sys.exit(1)
+    client = discord.Client(intents=discord.Intents.default())
+
+    @client.event
+    async def on_ready():
+        channel = client.get_channel(int(CHANNEL_ID))
+        for message in messages:
+            try:
+                await channel.send(message)
+            except Exception as e:
+                print(f"Error: {e}")
+                sys.exit(1)
+        await client.close()
+
+    message = "## Multiplatform test result:\n"
+    with open(sys.argv[1], "r") as f:
+        result = json.load(f)
+        failures = {}
+        stages = []
+        for platform in sorted(result.keys()):
+            for r in result[platform]:
+                if r['stage'] not in stages:
+                    stages.append(r['stage'])
+                if r['status'] != "✅":
+                    failures.setdefault(platform, list()).append(r)
+        message += f"- Platforms:\n  - {', '.join(sorted(result.keys()))}\n"
+        message += f"- Stages:\n  - {', '.join(stages)}\n"
+        # Add only the failures.
+        if len(failures) > 0:
+            message += "### Failures:\n"
+            for platform in sorted(failures.keys()):
+                if len(message) >= 1500:
+                    messages.append(message)
+                    message = ""
+                message += f"- **{platform}**:\n"
+                for r in failures[platform]:
+                    message += f"  - {r['stage']}: {r['status']}\n"
+    if GITHUB_URL:
+        message += "\n"
+        message += f"  [Github]({GITHUB_URL})"
+    if message:
+        messages.append(message)
+    client.run(TOKEN)