package guard.api.sensitive.ui.sapiguard.components;

import android.content.ContentProvider;
import android.content.ContentUris;
import android.content.ContentValues;
import android.content.Context;
import android.content.UriMatcher;
import android.database.Cursor;
import android.database.SQLException;
import android.database.sqlite.SQLiteDatabase;
import android.database.sqlite.SQLiteOpenHelper;
import android.database.sqlite.SQLiteQueryBuilder;
import android.net.Uri;
import android.text.TextUtils;

import java.util.HashMap;

public class PkgContentProvider extends ContentProvider {
    static final String PROVIDER_NAME = "guard.api.sensitive.ui.sapiguard.components.PkgContentProvider";
    static final String URL = "content://" + PROVIDER_NAME + "/pkginfo";
    static final Uri CONTENT_URI = Uri.parse(URL);
    private static HashMap<String, String> PKGINFO_PROJECTION_MAP;

    static final String _ID = "_id";

    static final int NOMATCH = -1;
    static final int PKG_INFO = 1;
    static final int PKG_ID = 2;


    private static UriMatcher uriMatcher = new UriMatcher(NOMATCH);
    static{
        uriMatcher.addURI(PROVIDER_NAME, "pkginfo", PKG_INFO);
        uriMatcher.addURI(PROVIDER_NAME, "pkginfo/#", PKG_ID);

    }


    private SQLiteDatabase db;
    static final int DATABASE_VERSION = 1;

    static final String DATABASE_NAME = "hookpkg";
    static final String PKGINFO_TABLE_NAME = "pkginfo";
    static final String CREATE_DB_TABLE =
            " CREATE TABLE " + PKGINFO_TABLE_NAME +
                    " (_id INTEGER PRIMARY KEY AUTOINCREMENT, " +
                    " name TEXT NOT NULL);" ;


    /**
     * Helper class that actually creates and manages
     * the provider's underlying data repository.
     */

    private static class DatabaseHelper extends SQLiteOpenHelper {
        DatabaseHelper(Context context){
            super(context, DATABASE_NAME, null, DATABASE_VERSION);
        }

        @Override
        public void onCreate(SQLiteDatabase db) {
            db.execSQL(CREATE_DB_TABLE);
        }

        @Override
        public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
            db.execSQL("DROP TABLE IF EXISTS " +  PKGINFO_TABLE_NAME);
            onCreate(db);
        }
    }

    @Override
    public boolean onCreate() {
        Context context = getContext();
        DatabaseHelper dbHelper = new DatabaseHelper(context);

        /**
         * Create a write able database which will trigger its
         * creation if it doesn't already exist.
         */

        db = dbHelper.getWritableDatabase();
        return (db == null)? false:true;
    }

    public PkgContentProvider() {
    }

    @Override
    public int delete(Uri uri, String selection, String[] selectionArgs) {
        int count = 0;
        switch (uriMatcher.match(uri)){
            case PKG_INFO:
                count = db.delete(PKGINFO_TABLE_NAME, selection, selectionArgs);
                break;

            case PKG_ID:
                String id = uri.getPathSegments().get(1);
                count = db.delete( PKGINFO_TABLE_NAME, _ID +  " = " + id +
                                (!TextUtils.isEmpty(selection) ? " AND (" + selection + ')' : ""), selectionArgs);
                break;
            default:
                throw new IllegalArgumentException("Unknown URI " + uri);
        }

        getContext().getContentResolver().notifyChange(uri, null);
        return count;
    }

    @Override
    public String getType(Uri uri) {
        switch (uriMatcher.match(uri)){
            /**
             * Get all package records
             */
            case PKG_INFO:
                return "vnd.android.cursor.dir/vnd.example.package";
            /**
             * Get a particular package
             */
            case PKG_ID:
                return "vnd.android.cursor.item/vnd.example.package";
            default:
                throw new IllegalArgumentException("Unsupported URI: " + uri);
        }
    }

    @Override
    public Uri insert(Uri uri, ContentValues values) {
        /**
         * Add a new pkginfo record
         */
        long rowID = db.insert(	PKGINFO_TABLE_NAME, "", values);

        /**
         * If record is added successfully
         */
        if (rowID > 0) {
            Uri _uri = ContentUris.withAppendedId(CONTENT_URI, rowID);
            getContext().getContentResolver().notifyChange(_uri, null);
            return _uri;
        }

        throw new SQLException("Failed to add a record into " + uri);
    }



    @Override
    public Cursor query(Uri uri, String[] projection, String selection,
                        String[] selectionArgs, String sortOrder) {
            SQLiteQueryBuilder qb = new SQLiteQueryBuilder();
            qb.setTables(PKGINFO_TABLE_NAME);

            switch (uriMatcher.match(uri)) {
                case PKG_INFO:
                    qb.setProjectionMap(PKGINFO_PROJECTION_MAP);
                    break;
//                case STUDENT_ID:
//                    qb.appendWhere( _ID + "=" + uri.getPathSegments().get(1));
//                    break;

                default:
            }

           Cursor c = qb.query(db,	projection,	selection,
                    selectionArgs,null, null, sortOrder);
            /**
             * register to watch a content URI for changes
             */
            c.setNotificationUri(getContext().getContentResolver(), uri);
            return c;
    }

    @Override
    public int update(Uri uri, ContentValues values, String selection,
                      String[] selectionArgs) {
            int count = 0;
            switch (uriMatcher.match(uri)) {
                case PKG_INFO:
                    count = db.update(PKGINFO_TABLE_NAME, values, selection, selectionArgs);
                    break;

                case PKG_ID:
                    count = db.update(PKGINFO_TABLE_NAME, values,
                            _ID + " = " + uri.getPathSegments().get(1) +
                                    (!TextUtils.isEmpty(selection) ? " AND (" +selection + ')' : ""), selectionArgs);
                    break;
                default:
                    throw new IllegalArgumentException("Unknown URI " + uri );
            }

            getContext().getContentResolver().notifyChange(uri, null);
            return count;
    }
}
