/*
 * Copyright (C) 2015-2016 The CyanogenMod 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 org.cyanogenmod.themes.provider.util;

import android.content.Context;
import android.content.Intent;
import android.database.Cursor;
import android.net.Uri;
import android.text.TextUtils;

import cyanogenmod.platform.Manifest;
import cyanogenmod.providers.ThemesContract;
import cyanogenmod.providers.ThemesContract.MixnMatchColumns;
import cyanogenmod.providers.ThemesContract.PreviewColumns;
import cyanogenmod.providers.ThemesContract.ThemesColumns;
import cyanogenmod.themes.ThemeManager;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class ProviderUtils {
    /**
     * Convenience method for determining if a theme exists in the provider
     * @param context
     * @param pkgName
     * @return True if the theme exists, false otherwise
     */
    public static boolean themeExistsInProvider(Context context, String pkgName) {
        boolean exists = false;
        String[] projection = new String[] { ThemesColumns.PKG_NAME };
        String selection = ThemesColumns.PKG_NAME + "=?";
        String[] selectionArgs = new String[] { pkgName };
        Cursor c = context.getContentResolver().query(ThemesColumns.CONTENT_URI,
                projection, selection, selectionArgs, null);

        if (c != null) {
            exists = c.getCount() >= 1;
            c.close();
        }
        return exists;
    }

    /**
     * Queries the {@link ThemeManager} to check if the theme is currently
     * being processed by {@link org.cyanogenmod.platform.internal.ThemeManagerService}
     * @param context
     * @param pkgName
     * @return True if the theme is being processed or queued up for processing
     */
    public static boolean isThemeBeingProcessed(Context context, String pkgName) {
        ThemeManager tm = ThemeManager.getInstance();
        return tm.isThemeBeingProcessed(pkgName);
    }

    /**
     * Convenience method for getting the install state of a theme in the provider
     * @param context
     * @param pkgName
     * @return
     */
    public static int getInstallStateForTheme(Context context, String pkgName) {
        if (context == null || pkgName == null) return ThemesColumns.InstallState.UNKNOWN;

        String[] projection = new String[] { ThemesColumns.INSTALL_STATE };
        String selection = ThemesColumns.PKG_NAME + "=?";
        String[] selectionArgs = new String[] { pkgName };
        Cursor c = context.getContentResolver().query(ThemesColumns.CONTENT_URI,
                projection, selection, selectionArgs, null);

        int state = ThemesColumns.InstallState.UNKNOWN;
        if (c != null) {
            if (c.moveToFirst()) {
                state = c.getInt(c.getColumnIndex(ThemesColumns.INSTALL_STATE));
            }
            c.close();
        }
        return state;
    }

    public static String getCurrentThemeForComponent(Context context, String selection,
            String[] selectionArgs) {
        if (context == null || selection == null || selectionArgs == null) {
            return null;
        }

        String[] projection = new String[] {MixnMatchColumns.COL_VALUE};
        Cursor c = context.getContentResolver().query(MixnMatchColumns.CONTENT_URI,
                projection, selection, selectionArgs, null);

        String themePkgName = null;
        if (c != null) {
            if (c.moveToFirst()) {
                themePkgName = c.getString(c.getColumnIndex(MixnMatchColumns.COL_VALUE));
            }
            c.close();
        }
        return themePkgName;
    }

    public static String[] modifyPreviewsProjection(String[] projection) {
        if (projection == null) return null;

        Set<String> validKeys =
                new HashSet<String>(Arrays.asList(ThemesContract.PreviewColumns.VALID_KEYS));
        ArrayList<String> newProjection = new ArrayList<String>();
        for (String item : projection) {
            if (validKeys.contains(item)) {
                newProjection.add(getProjectionFromKeyValue(item));
            } else {
                newProjection.add(item);
            }
        }
        return newProjection.toArray(new String[newProjection.size()]);
    }

    public static String modifyDefaultPreviewsSelection(String selection, String[] projection) {
        String newSelection = modifyPreviewsSelection(selection, projection);
        if (!TextUtils.isEmpty(newSelection)) {
            newSelection += " AND ";
        } else {
            newSelection = "";
        }
        newSelection += PreviewColumns.COMPONENT_ID + "=0";
        return newSelection;
    }

    public static String modifyPreviewsSelection(String selection, String[] projection) {
        if (selection == null && projection == null) return null;

        String newSelection = "";
        if (!TextUtils.isEmpty(selection)) {
            newSelection += "(" + selection;
        }
        List<String> projectionItems = getPreviewProjectionItems(projection);
        if (projectionItems != null && projectionItems.size() > 0) {
            if (!TextUtils.isEmpty(newSelection)) {
                newSelection += " AND ";
            }
            newSelection += "(";
            for (int i = 0; i < projectionItems.size(); i++) {
                newSelection += PreviewColumns.COL_KEY + "=?";
                if (i < projectionItems.size() - 1) {
                    newSelection += " OR ";
                }
            }
            newSelection += ")";
        }
        if (!TextUtils.isEmpty(selection)) {
            newSelection += ")";
        }
        return newSelection;
    }

    public static String[] modifyPreviewsSelectionArgs(String[] selectionArgs,
            String[] projection) {
        if (selectionArgs == null && projection == null) return null;

        ArrayList<String> newSelectionArgs = new ArrayList<String>();
        if (selectionArgs != null) {
            newSelectionArgs.addAll(Arrays.asList(selectionArgs));
        }
        List<String> projectionItems = getPreviewProjectionItems(projection);
        if (projectionItems != null) {
            newSelectionArgs.addAll(projectionItems);
        }
        return newSelectionArgs.toArray(new String[newSelectionArgs.size()]);
    }

    public static List<String> getPreviewProjectionItems(String[] projection) {
        if (projection == null) return null;
        Set<String> validKeys =
                new HashSet<String>(Arrays.asList(PreviewColumns.VALID_KEYS));
        ArrayList<String> newProjection = new ArrayList<String>();
        for (String item : projection) {
            if (validKeys.contains(item)) {
                newProjection.add(getProjectionFromKeyValue(item));
            }
        }
        return newProjection;
    }

    /**
     * This allows pivoting key/value pairs as column/entry pairs.
     * This is only needed when querying multiple keys at a time.
     * @param keyValue
     * @return
     */
    public static String getProjectionFromKeyValue(String keyValue) {
        return String.format("MAX( CASE %s WHEN '%s' THEN %s ELSE NULL END) AS %s",
                ThemesContract.PreviewColumns.COL_KEY, keyValue,
                ThemesContract.PreviewColumns.COL_VALUE, keyValue);
    }

    /**
     * Sends the {@link cyanogenmod.content.Intent#ACTION_THEME_INSTALLED} action
     * @param context
     * @param pkgName
     */
    public static void sendThemeInstalledBroadcast(Context context, String pkgName) {
        Intent intent = new Intent(cyanogenmod.content.Intent.ACTION_THEME_INSTALLED,
                Uri.fromParts(cyanogenmod.content.Intent.URI_SCHEME_PACKAGE, pkgName, null));
        context.sendBroadcast(intent, Manifest.permission.READ_THEMES);
    }

    /**
     * Sends the {@link cyanogenmod.content.Intent#ACTION_THEME_UPDATED} action
     * @param context
     * @param pkgName
     */
    public static void sendThemeUpdatedBroadcast(Context context, String pkgName) {
        Intent intent = new Intent(cyanogenmod.content.Intent.ACTION_THEME_UPDATED,
                Uri.fromParts(cyanogenmod.content.Intent.URI_SCHEME_PACKAGE, pkgName, null));
        context.sendBroadcast(intent, Manifest.permission.READ_THEMES);
    }

    /**
     * Sends the {@link cyanogenmod.content.Intent#ACTION_THEME_REMOVED} action
     * @param context
     * @param pkgName
     */
    public static void sendThemeRemovedBroadcast(Context context, String pkgName) {
        Intent intent = new Intent(cyanogenmod.content.Intent.ACTION_THEME_REMOVED,
                Uri.fromParts(cyanogenmod.content.Intent.URI_SCHEME_PACKAGE, pkgName, null));
        context.sendBroadcast(intent, Manifest.permission.READ_THEMES);
    }
}