/*
 * Copyright (C) 2022 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package android.adservices.common;

import android.annotation.NonNull;
import android.annotation.Nullable;
import android.net.Uri;
import android.os.Parcel;
import android.os.Parcelable;

import com.android.adservices.AdServicesParcelableUtil;

import java.util.HashSet;
import java.util.Objects;
import java.util.Set;

/** Represents data specific to an ad that is necessary for ad selection and rendering. */
public final class AdData implements Parcelable {
    @NonNull private final Uri mRenderUri;
    @NonNull private final String mMetadata;
    @NonNull private final Set<String> mAdCounterKeys;
    @Nullable private final AdFilters mAdFilters;

    @NonNull
    public static final Creator<AdData> CREATOR =
            new Creator<AdData>() {
                @Override
                public AdData createFromParcel(@NonNull Parcel in) {
                    Objects.requireNonNull(in);

                    return new AdData(in);
                }

                @Override
                public AdData[] newArray(int size) {
                    return new AdData[size];
                }
            };

    private AdData(@NonNull AdData.Builder builder) {
        Objects.requireNonNull(builder);

        mRenderUri = builder.mRenderUri;
        mMetadata = builder.mMetadata;
        mAdCounterKeys = builder.mAdCounterKeys;
        mAdFilters = builder.mAdFilters;
    }

    private AdData(@NonNull Parcel in) {
        Objects.requireNonNull(in);

        mRenderUri = Uri.CREATOR.createFromParcel(in);
        mMetadata = in.readString();
        mAdCounterKeys =
                AdServicesParcelableUtil.readNullableFromParcel(
                        in, AdServicesParcelableUtil::readStringSetFromParcel);
        mAdFilters =
                AdServicesParcelableUtil.readNullableFromParcel(
                        in, AdFilters.CREATOR::createFromParcel);
    }

    @Override
    public void writeToParcel(@NonNull Parcel dest, int flags) {
        Objects.requireNonNull(dest);

        mRenderUri.writeToParcel(dest, flags);
        dest.writeString(mMetadata);
        AdServicesParcelableUtil.writeNullableToParcel(
                dest, mAdCounterKeys, AdServicesParcelableUtil::writeStringSetToParcel);
        AdServicesParcelableUtil.writeNullableToParcel(
                dest,
                mAdFilters,
                (targetParcel, sourceFilters) -> sourceFilters.writeToParcel(targetParcel, flags));
    }

    /** @hide */
    @Override
    public int describeContents() {
        return 0;
    }

    /** Gets the URI that points to the ad's rendering assets. The URI must use HTTPS. */
    @NonNull
    public Uri getRenderUri() {
        return mRenderUri;
    }

    /**
     * Gets the buyer ad metadata used during the ad selection process.
     *
     * <p>The metadata should be a valid JSON object serialized as a string. Metadata represents
     * ad-specific bidding information that will be used during ad selection as part of bid
     * generation and used in buyer JavaScript logic, which is executed in an isolated execution
     * environment.
     *
     * <p>If the metadata is not a valid JSON object that can be consumed by the buyer's JS, the ad
     * will not be eligible for ad selection.
     */
    @NonNull
    public String getMetadata() {
        return mMetadata;
    }

    /**
     * Gets the set of keys used in counting events.
     *
     * <p>The keys and counts per key are used in frequency cap filtering during ad selection to
     * disqualify associated ads from being submitted to bidding.
     *
     * <p>Note that these keys can be overwritten along with the ads and other bidding data for a
     * custom audience during the custom audience's daily update.
     *
     * @hide
     */
    // TODO(b/221876775): Unhide for frequency cap API review
    @NonNull
    public Set<String> getAdCounterKeys() {
        return mAdCounterKeys;
    }

    /**
     * Gets all {@link AdFilters} associated with the ad.
     *
     * <p>The filters, if met or exceeded, exclude the associated ad from participating in ad
     * selection. They are optional and if {@code null} specify that no filters apply to this ad.
     *
     * @hide
     */
    // TODO(b/221876775): Unhide for app install/frequency cap API review
    @Nullable
    public AdFilters getAdFilters() {
        return mAdFilters;
    }

    /** Checks whether two {@link AdData} objects contain the same information. */
    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof AdData)) return false;
        AdData adData = (AdData) o;
        return mRenderUri.equals(adData.mRenderUri)
                && mMetadata.equals(adData.mMetadata)
                && mAdCounterKeys.equals(adData.mAdCounterKeys)
                && Objects.equals(mAdFilters, adData.mAdFilters);
    }

    /** Returns the hash of the {@link AdData} object's data. */
    @Override
    public int hashCode() {
        return Objects.hash(mRenderUri, mMetadata, mAdCounterKeys, mAdFilters);
    }

    @Override
    public String toString() {
        return "AdData{"
                + "mRenderUri="
                + mRenderUri
                + ", mMetadata='"
                + mMetadata
                + '\''
                + generateAdCounterKeyString()
                + generateAdFilterString()
                + '}';
    }

    private String generateAdCounterKeyString() {
        // TODO(b/221876775) Add ad counter keys String when unhidden
        return "";
    }

    private String generateAdFilterString() {
        // TODO(b/266837113) Add ad filters String when unhidden
        return "";
    }

    /** Builder for {@link AdData} objects. */
    public static final class Builder {
        @Nullable private Uri mRenderUri;
        @Nullable private String mMetadata;
        @NonNull private Set<String> mAdCounterKeys = new HashSet<>();
        @Nullable private AdFilters mAdFilters;

        // TODO(b/232883403): We may need to add @NonNUll members as args.
        public Builder() {}

        /**
         * Sets the URI that points to the ad's rendering assets. The URI must use HTTPS.
         *
         * <p>See {@link #getRenderUri()} for detail.
         */
        @NonNull
        public AdData.Builder setRenderUri(@NonNull Uri renderUri) {
            Objects.requireNonNull(renderUri);
            mRenderUri = renderUri;
            return this;
        }

        /**
         * Sets the buyer ad metadata used during the ad selection process.
         *
         * <p>The metadata should be a valid JSON object serialized as a string. Metadata represents
         * ad-specific bidding information that will be used during ad selection as part of bid
         * generation and used in buyer JavaScript logic, which is executed in an isolated execution
         * environment.
         *
         * <p>If the metadata is not a valid JSON object that can be consumed by the buyer's JS, the
         * ad will not be eligible for ad selection.
         *
         * <p>See {@link #getMetadata()} for detail.
         */
        @NonNull
        public AdData.Builder setMetadata(@NonNull String metadata) {
            Objects.requireNonNull(metadata);
            mMetadata = metadata;
            return this;
        }

        /**
         * Sets the set of keys used in counting events.
         *
         * <p>See {@link #getAdCounterKeys()} for more information.
         *
         * @hide
         */
        // TODO(b/221876775): Unhide for frequency cap API review
        @NonNull
        public AdData.Builder setAdCounterKeys(@NonNull Set<String> adCounterKeys) {
            Objects.requireNonNull(adCounterKeys);
            mAdCounterKeys = adCounterKeys;
            return this;
        }

        /**
         * Sets all {@link AdFilters} associated with the ad.
         *
         * <p>See {@link #getAdFilters()} for more information.
         *
         * @hide
         */
        // TODO(b/221876775): Unhide for app install/frequency cap API review
        @NonNull
        public AdData.Builder setAdFilters(@Nullable AdFilters adFilters) {
            mAdFilters = adFilters;
            return this;
        }

        /**
         * Builds the {@link AdData} object.
         *
         * @throws NullPointerException if any required parameters are {@code null} when built
         */
        @NonNull
        public AdData build() {
            Objects.requireNonNull(mRenderUri);
            // TODO(b/231997523): Add JSON field validation.
            Objects.requireNonNull(mMetadata);

            return new AdData(this);
        }
    }
}