sequenceDiagram
    actor UserScript as "User/Script"
    participant CurrentCH as "Current CH Node"
    participant ZooKeeper

    Note over UserScript, CurrentCH: In 'finally' block of clean() function
    opt Not keep_paths (Cleanup requested)
        CurrentCH->>CurrentCH: effective_listing_table // This is the table name string (local or remote) defined earlier in clean()
        CurrentCH->>CurrentCH: Call _drop_table(effective_listing_table, user_password) // May target a remote owner

        alt has_zk() is true AND check_zk_node(ZK_PATH_LISTING_TABLE_INFO) is true
            CurrentCH->>ZooKeeper: delete_zk_node(ZK_PATH_LISTING_TABLE_INFO)
        end
    end

erDiagram
    listing_table {
        String obj_path PK "S3 Object Path"
        UInt64 obj_size "Object Size (bytes)"
        String storage_policy "ClickHouse storage policy for this table's data"
    }
    note "ClickHouse MergeTree table storing S3 object paths and sizes. Ordered by obj_path." for listing_table

    zk_listing_table_info {
        String replica_owner PK "Hostname of CH replica owning the listing_table"
    }
    note "Data in ZooKeeper node at ZK_PATH_LISTING_TABLE_INFO. Coordinates ownership of the 'listing_table' in a multi-replica setup." for zk_listing_table_info

classDiagram
    class clean_object_storage_module {
      <<Python Module>>
      +ZK_PATH_LISTING_TABLE_INFO : string
      +clean(ctx, use_saved_list, keep_paths, ...)
      -_clean_object_storage(ctx, ch_client, user_name, user_password, ...)
      -_traverse_object_storage(ctx, listing_table, user_password, ...)
      -_insert_listing_batch(ctx, obj_paths_batch, listing_table, user_password)
      -_drop_table(ctx, table_name, user_password)
    }
    note for clean_object_storage_module "Functions updated to use Query objects for secure credential handling and ZooKeeper for multi-replica coordination."

    class Query {
      <<Value Object>>
      +statement : string
      +sensitive_args : map
      +Query(statement: string, sensitive_args: map)
    }
    note for Query "New abstraction for SQL queries, separating sensitive arguments like passwords."

    class ch_tools.common.clickhouse.client {
      <<Utility Module>>
      +execute_query(ctx, query_or_string, sensitive_args, format_): any
    }
    note for ch_tools.common.clickhouse.client "execute_query is used to run SQL, now handles Query objects."

    class ch_tools.chadmin.internal.zookeeper {
      <<Utility Module>>
      +has_zk(ctx) : bool
      +check_zk_node(ctx, path) : bool
      +create_zk_nodes(ctx, paths, make_parents)
      +update_zk_nodes(ctx, path, data_bytes)
      +get_zk_node(ctx, path) : bytes
      +delete_zk_node(ctx, path)
    }
    note for ch_tools.chadmin.internal.zookeeper "Provides ZooKeeper interaction helpers, including new has_zk function."

    class ch_tools.common.clickhouse.client.ClickhouseClient {
        +user: string
        +password: string
        +host: string
        +check_port(ClickhousePort): bool
    }
    note for ch_tools.common.clickhouse.client.ClickhouseClient "Used to fetch credentials, host, and check port for remoteSecure selection."

    clean_object_storage_module ..> Query : creates >
    clean_object_storage_module ..> ch_tools.common.clickhouse.client : uses execute_query >
    clean_object_storage_module ..> ch_tools.chadmin.internal.zookeeper : uses ZK functions >
    clean_object_storage_module ..> ch_tools.common.clickhouse.client.ClickhouseClient : uses >