AI that makes
enterprise sales
teams faster.

{{ config(
    materialized     = 'incremental',
    schema           = 'QUARANTINE',
    on_schema_change = 'fail'
) }}

WITH source AS (
    SELECT * FROM {{ ref('stg_yellow_taxi') }}
    {% if is_incremental() %}
    WHERE _loaded_at > (SELECT MAX(_loaded_at) FROM {{ this }})
    {% endif %}
),

validation AS (
    SELECT
        *,
        CASE
            WHEN dropoff_datetime <= pickup_datetime
                THEN 'dropoff_datetime <= pickup_datetime (trip ends before it starts)'
            WHEN trip_distance > 500
                THEN 'Unrealistic trip_distance > 500 miles: ' || CAST(trip_distance AS VARCHAR)
            WHEN passenger_count > 9
                THEN 'passenger_count exceeds max capacity: ' || CAST(passenger_count AS VARCHAR)
            -- … 8 more rules
            ELSE NULL
        END AS _rejection_reason
    FROM source
),

-- Only the INVALID rows (opposite of silver_trips.sql)
invalid_only AS (
    SELECT * FROM validation WHERE _rejection_reason IS NOT NULL
)

WITH bronze AS (
    SELECT COUNT(*) AS cnt FROM {{ source('taxi_raw', 'RAW_YELLOW_TAXI') }}
),
silver AS (
    SELECT COUNT(*) AS cnt FROM {{ ref('silver_trips') }}
),
quarantine AS (
    SELECT COUNT(*) AS cnt FROM {{ ref('quarantine_trips') }}
),
comparison AS (
    SELECT
        b.cnt                     AS bronze_count,
        s.cnt + q.cnt             AS silver_plus_quarantine_count,
        s.cnt                     AS silver_count,
        q.cnt                     AS quarantine_count,
        b.cnt - (s.cnt + q.cnt)   AS difference
    FROM bronze b, silver s, quarantine q
)

-- Return a row only if there's a discrepancy.
SELECT
    bronze_count,
    silver_plus_quarantine_count,
    -- …
    difference,
    -- …
FROM comparison
WHERE ABS(difference) > 200000

function createNFT(
    string memory name,
    string memory symbol,
    string memory baseURI
) external returns (address) {
    // Create new NFT contract with factory as temporary owner
    CustomNFT nft = new CustomNFT(name, symbol, address(this));

    address nftAddress = address(nft);

    // Mint first NFT (token ID 0) to the creator
    nft.safeMint(msg.sender, baseURI);

    // Transfer ownership to the actual user
    nft.transferOwnership(msg.sender);

    // Track the NFT
    userNFTs[msg.sender].push(nftAddress);
    allNFTs.push(nftAddress);

    emit NFTCreated(nftAddress, msg.sender, name, symbol);

    return nftAddress;
}