Homec4science
Diffusion Genome Privacy - SHCS App (mobile_app) Browse

genopri-shcs/project/PPPClientMobile/src/ch/epfl/pharma/database/DatabaseHelper.java
9913a583c025mobile_app

Last Change
Raw File

DatabaseHelper.java
View Options

package ch.epfl.pharma.database;
import java.io.IOException;
import java.math.BigInteger;
import java.util.ArrayList;
import java.util.LinkedList;
import android.content.ContentValues;
import android.content.Context;
import android.database.Cursor;
import android.database.sqlite.SQLiteDatabase;
import android.util.Log;
import ch.epfl.pharma.database.contract.ClinicalTestContract;
import ch.epfl.pharma.database.contract.CryptokeysContract;
import ch.epfl.pharma.database.contract.MarkerTestContract;
import ch.epfl.pharma.database.contract.PatientsContract;
import ch.epfl.pharma.database.contract.ReportsContract;
import ch.epfl.pharma.database.contract.ResultsContract;
import ch.epfl.pharma.database.contract.TestsContract;
import ch.epfl.pharma.database.contract.entry.DatabaseEntry.CryptokeyEntry;
import ch.epfl.pharma.database.contract.entry.DatabaseEntry.PatientEntry;
import ch.epfl.pharma.framework.Patient;
import ch.epfl.pharma.framework.Test;
import crypto.elgamal.PartialSecret;
import framework.ClinicalFactor;
import framework.WeightedMarker;
public class DatabaseHelper {
private static final String TAG = DatabaseHelper.class.getName();
private AssetDatabaseOpenHelper assetHelper;
private SQLiteDatabase mDatabase;
private Context mContext;
public DatabaseHelper(Context context, String assetDbName) {
this.mContext = context;
this.assetHelper = new AssetDatabaseOpenHelper(mContext, assetDbName);
}
public void createDatabase() throws IOException {
// import database
if (assetHelper.databaseExists()) {
openDatabase();
} else {
assetHelper.createDatabase();
openDatabase();
// create missing tables
mDatabase.execSQL(CryptokeysContract.SQL_TABLE_CREATE);
mDatabase.execSQL(PatientsContract.SQL_TABLE_CREATE);
mDatabase.execSQL(ReportsContract.SQL_TABLE_CREATE);
mDatabase.execSQL(ResultsContract.SQL_TABLE_CREATE);
}
close();
}
public void openDatabase() {
assetHelper.openDatabase();
mDatabase = assetHelper.getDatabase();
}
public void close() {
assetHelper.close();
mDatabase = null;
}
public LinkedList<Test> getTests() {
Cursor c = TestsContract.getTests(mDatabase);
LinkedList<Test> tests = new LinkedList<>();
if (c.moveToFirst()) {
do {
Test test = TestsContract.getTest(c);
tests.add(test);
} while (c.moveToNext());
}
c.close();
return tests;
}
public long insertPatient(Patient patient) {
ContentValues values = new ContentValues();
values.put(PatientEntry.ID_PATIENT, patient.getPatientId());
values.put(PatientEntry.LASTNAME, patient.getLastname());
values.put(PatientEntry.FIRSTNAME, patient.getFirstname());
values.put(PatientEntry.SEX, patient.getSexStr());
values.put(PatientEntry.BIRTHDAY, patient.getBirthday());
values.put(PatientEntry.ADDRESS, patient.getAddress());
values.put(PatientEntry.PICTURE, patient.getPictureUri());
long newRowId = mDatabase.insert(
PatientEntry.TABLE,
null,
values);
return newRowId;
}
public LinkedList<Patient> getPatients() {
Cursor c = PatientsContract.getPatients(mDatabase);
LinkedList<Patient> patients = new LinkedList<>();
if (c.moveToFirst()) {
do {
Patient patient = PatientsContract.getPatient(c);
patients.add(patient);
} while (c.moveToNext());
}
c.close();
return patients;
}
public Cursor getAllPatients() {
Cursor c = PatientsContract.getPatients(mDatabase);
return c;
}
public Cursor filterAllPatients(String query) {
Cursor c = PatientsContract.getPatients(mDatabase, query);
return c;
}
public ArrayList<WeightedMarker> getMarkers(int testId) {
Cursor c = MarkerTestContract.getMarkers(mDatabase, testId);
ArrayList<WeightedMarker> wMarkers = new ArrayList<>();
if (c.moveToFirst()) {
do {
WeightedMarker wMarker =
MarkerTestContract.getWeightedMarker(c);
wMarkers.add(wMarker);
} while (c.moveToNext());
}
c.close();
return wMarkers;
}
public ArrayList<ClinicalFactor> getCFs(int testId) {
Cursor c = ClinicalTestContract.getClinicals(mDatabase, testId);
ArrayList<ClinicalFactor> cfs = new ArrayList<>();
if (c.moveToFirst()) {
do {
ClinicalFactor cf =
ClinicalTestContract.getClinicalFactor(c);
cfs.add(cf);
} while (c.moveToNext());
}
c.close();
return cfs;
}
public PartialSecret getPrivateKey(int patientId) {
Cursor c = CryptokeysContract.getPrivateKey(mDatabase, patientId);
if (c.moveToFirst()) {
BigInteger secret = CryptokeysContract.getSecret(c);
PartialSecret partialSecret = new PartialSecret(PartialSecret.PART_MU, secret);
c.close();
return partialSecret;
} else {
c.close();
return null;
}
}
public long insertPrivateKey(int patientId, PartialSecret secret) {
ContentValues values = new ContentValues();
values.put(CryptokeyEntry.ID_PATIENT, patientId);
values.put(CryptokeyEntry.VALUE, secret.getPrivKey().toString());
values.put(CryptokeyEntry.ISPUBLIC, false);
long newRowId = mDatabase.insert(
CryptokeyEntry.TABLE,
null,
values);
return newRowId;
}
// public long insertResult(int patientId, int testId, double result) {
// ContentValues values = new ContentValues();
// values.put(ResultEntry.COL_ID_PATIENT, patientId);
// values.put(ResultEntry.COL_ID_TEST, testId);
// values.put(ResultEntry.COL_ID_RESULT, result);
// values.put(ResultEntry.COL_DATE, System.currentTimeMillis());
//
// long newRowId = mDatabase.insert(
// ResultEntry.TABLE_NAME,
// null,
// values);
//
// return newRowId;
// }
public boolean patientExists(Patient patient) {
Cursor c = PatientsContract.getPatient(mDatabase, patient.getPatientId());
int count = c.getCount();
Log.i(TAG, "Patient Exists? Count=" + count + " for patient=" + patient.getPatientId());
if (count == 1) {
// same patient id
return true;
} else if(count == 0) {
// no patient exists
return false;
} else {
// should not happen
throw new IllegalStateException("Found multiple patients for same patient identifier");
}
}
public Patient getPatient(long patientId) {
Cursor c = PatientsContract.getPatient(mDatabase, patientId);
c.moveToFirst();
Patient patient = PatientsContract.getPatient(c);
return patient;
}
public Cursor getAllReports() {
Cursor c = ReportsContract.getReports(mDatabase);
return c;
}
}
c4science · Help