LeOS-Genesis/app/src/main/java/com/darkweb/genesissearchengine/databaseManager/databaseController.java

package com.darkweb.genesissearchengine.databaseManager;

import android.content.ContentValues;
import android.database.Cursor;
import androidx.appcompat.app.AppCompatActivity;
import com.darkweb.genesissearchengine.appManager.activityContextManager;
import com.darkweb.genesissearchengine.appManager.bookmarkManager.bookmarkRowModel;
import com.darkweb.genesissearchengine.appManager.historyManager.historyRowModel;
import com.darkweb.genesissearchengine.appManager.homeManager.geckoManager.geckoSession;
import com.darkweb.genesissearchengine.appManager.tabManager.tabRowModel;
import com.darkweb.genesissearchengine.constants.constants;
import com.darkweb.genesissearchengine.constants.status;
import com.darkweb.genesissearchengine.dataManager.dataController;
import com.darkweb.genesissearchengine.dataManager.dataEnums;
import com.darkweb.genesissearchengine.helperManager.helperMethod;

import net.sqlcipher.database.SQLiteDatabaseHook;
import net.sqlcipher.database.SQLiteDatabase;

import org.mozilla.geckoview.GeckoSession;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Date;
import java.util.Locale;

import static com.darkweb.genesissearchengine.constants.constants.CONST_DATABASE_NAME;


public class databaseController
{

    /*Private Variables*/

    private static final databaseController sOurInstance = new databaseController();
    private static SQLiteDatabase mDatabaseInstance;

    public static databaseController getInstance()
    {
        return sOurInstance;
    }

    private databaseController()
    {
    }

    /*Initializations*/

    public void prepareDatabaseEnvironment(AppCompatActivity app_context) {
        File databaseFile = app_context.getDatabasePath(CONST_DATABASE_NAME);

        if (!databaseFile.exists()) {
            databaseFile.getParentFile().mkdirs();
        }
    }

    public void initialize(AppCompatActivity app_context)
    {
        try
        {
            SQLiteDatabase.loadLibs(app_context);
            prepareDatabaseEnvironment(app_context);
            mDatabaseInstance = mDatabaseInstance.openOrCreateDatabase(app_context.getDatabasePath(CONST_DATABASE_NAME), constants.CONST_ENCRYPTION_KEY_DATABASE,null, wrapHook(null));

            mDatabaseInstance.execSQL("CREATE TABLE IF NOT EXISTS " + "history" + " (id  INT(4) PRIMARY KEY,date DATETIME,url VARCHAR,title VARCHAR);");
            mDatabaseInstance.execSQL("CREATE TABLE IF NOT EXISTS " + "bookmark" + " (id INT(4) PRIMARY KEY,title VARCHAR,url VARCHAR);");
            mDatabaseInstance.execSQL("CREATE TABLE IF NOT EXISTS " + "tab" + " (mid INT(4) PRIMARY KEY,date,title VARCHAR,url VARCHAR,mThumbnail BLOB, theme VARCHAR, session VARCHAR);");

        }
        catch (Exception ex)
        {
            ex.printStackTrace();
        }

    }

    /*Helper Methods*/

    public SQLiteDatabaseHook wrapHook(final SQLiteDatabaseHook hook) {
        if (hook == null)
        {
            return keyHook;
        }
        return new SQLiteDatabaseHook() {
            @Override
            public void preKey(net.sqlcipher.database.SQLiteDatabase sqLiteDatabase) {
                keyHook.preKey(sqLiteDatabase);
                hook.preKey(sqLiteDatabase);
            }

            @Override
            public void postKey(net.sqlcipher.database.SQLiteDatabase sqLiteDatabase) {
                keyHook.postKey(sqLiteDatabase);
                hook.preKey(sqLiteDatabase);
            }
        };
    }

    SQLiteDatabaseHook keyHook = new SQLiteDatabaseHook() {
        @Override
        public void preKey(net.sqlcipher.database.SQLiteDatabase sqLiteDatabase) {

        }

        @Override
        public void postKey(net.sqlcipher.database.SQLiteDatabase sqLiteDatabase) {

        }
    };

    public void execSQL(String query,String[] params)
    {
        if(params==null)
        {
            mDatabaseInstance.execSQL(query);
        }
        else
        {
            mDatabaseInstance.execSQL(query,params);
        }
    }

    public void execTab(String query, ContentValues params,String pID)
    {
        if(params!=null)
        {
            mDatabaseInstance.update(query, params, "mid = ?", new String[]{pID});
        }
    }

    public ArrayList<historyRowModel> selectHistory(int startIndex,int endIndex){
        ArrayList<historyRowModel> tempmodel = new ArrayList<>();

        Cursor c = mDatabaseInstance.rawQuery("SELECT * FROM history ORDER BY date DESC LIMIT " + endIndex + " OFFSET "+startIndex, null);
        if (c.moveToFirst()){
            do {
                historyRowModel model = new historyRowModel(c.getString(3), c.getString(2),Integer.parseInt(c.getString(0)));
                try {
                    Date m_date = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS", Locale.US).parse(c.getString(1));
                    model.setDate(m_date);
                } catch (Exception e) {
                    e.printStackTrace();
                }
                if(Calendar.getInstance().getTime().getTime() < model.getDate().getTime()){
                    tempmodel.add(model);
                }else {
                    tempmodel.add(model);
                }
            } while(c.moveToNext());
        }
        c.close();

        return  tempmodel;
    }

    public ArrayList<tabRowModel> selectTabs(){
        ArrayList<tabRowModel> mTempListModel = new ArrayList<>();

        Cursor c = mDatabaseInstance.rawQuery("SELECT * FROM tab ORDER BY date DESC", null);
        if (c.moveToFirst()){
            do {
                geckoSession mSession =  activityContextManager.getInstance().getHomeController().onNewTabInit();
                tabRowModel model = new tabRowModel(c.getString(0), c.getString(1),c.getBlob(4));
                GeckoSession.SessionState session = null;
                try {
                    if(status.sRestoreTabs){
                        session = GeckoSession.SessionState.fromString(c.getString(6));
                    }
                } catch (Exception ex) {
                    ex.printStackTrace();
                }
                model.setSession(mSession, c.getString(3),c.getString(2), c.getString(5), session);
                model.getSession().setSessionID(model.getmId());
                mTempListModel.add(model);
            } while(c.moveToNext());
        }
        c.close();

        return  mTempListModel;
    }

    public int getLargestHistoryID(){
        int id = 0;
        Cursor c = mDatabaseInstance.rawQuery("SELECT max(id) FROM history", null);

        if (c.moveToFirst()){
            do {
                if(c.getString(0)==null){
                    break;
                }else {
                    id = Integer.parseInt(c.getString(0));
                }
            } while(c.moveToNext());
        }
        c.close();

        return  id;
    }

    public ArrayList<bookmarkRowModel> selectBookmark(){
        ArrayList<bookmarkRowModel> tempmodel = new ArrayList<>();
        Cursor c = mDatabaseInstance.rawQuery("SELECT * FROM bookmark ORDER BY id DESC ", null);

        if (c.moveToFirst()){
            do {
                tempmodel.add(new bookmarkRowModel(c.getString(1), c.getString(2),Integer.parseInt(c.getString(0))));
            } while(c.moveToNext());
        }
        c.close();

        return  tempmodel;
    }

    public void deleteFromList(int index,String table) {
        databaseController.getInstance().execSQL("delete from "+table+" where id="+index,null);
    }

}