PatientDAO.java

/*
 * The contents of this file are subject to the OpenMRS Public License
 * Version 1.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://license.openmrs.org
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 *
 * Copyright (C) OpenMRS, LLC.  All Rights Reserved.
 */

package org.openmrs.mobile.dao;


import android.graphics.Bitmap;
import android.graphics.BitmapFactory;

import net.sqlcipher.Cursor;

import org.openmrs.mobile.application.OpenMRS;
import org.openmrs.mobile.databases.DBOpenHelper;
import org.openmrs.mobile.databases.OpenMRSDBOpenHelper;
import org.openmrs.mobile.databases.tables.PatientTable;
import org.openmrs.mobile.models.Patient;
import org.openmrs.mobile.models.PatientIdentifier;
import org.openmrs.mobile.models.Person;
import org.openmrs.mobile.models.PersonAddress;
import org.openmrs.mobile.models.PersonName;

import java.io.ByteArrayInputStream;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

import rx.Observable;

import static org.openmrs.mobile.databases.DBOpenHelper.createObservableIO;

public class PatientDAO {

    public Observable<Long> savePatient(Patient patient) {
        return createObservableIO(() -> new PatientTable().insert(patient));
    }

    public boolean updatePatient(long patientID, Patient patient) {
        return new PatientTable().update(patientID, patient) > 0;
    }

    public void deletePatient(long id) {
        OpenMRS.getInstance().getOpenMRSLogger().w("Patient deleted with id: " + id);
        DBOpenHelper openHelper = OpenMRSDBOpenHelper.getInstance().getDBOpenHelper();
        openHelper.getReadableDatabase().delete(PatientTable.TABLE_NAME, PatientTable.Column.ID
                + " = " + id, null);
    }

    public Observable<List<Patient>> getAllPatients() {
        return createObservableIO(() -> {
            List<Patient> patients = new ArrayList<Patient>();
            DBOpenHelper openHelper = OpenMRSDBOpenHelper.getInstance().getDBOpenHelper();
            Cursor cursor = openHelper.getReadableDatabase().query(PatientTable.TABLE_NAME,
                    null, null, null, null, null, null);

            if (null != cursor) {
                try {
                    while (cursor.moveToNext()) {
                        Patient patient = cursorToPatient(cursor);
                        patients.add(patient);
                    }
                } finally {
                    cursor.close();
                }
            }
            return patients;
        });
    }

    private Patient cursorToPatient(Cursor cursor) {
        Patient patient = new Patient();
        Person person = new Person();

        patient.setId(cursor.getLong(cursor.getColumnIndex(PatientTable.Column.ID)));
        patient.setDisplay(cursor.getString(cursor.getColumnIndex(PatientTable.Column.DISPLAY)));
        patient.setUuid(cursor.getString(cursor.getColumnIndex(PatientTable.Column.UUID)));
        patient.setEncounters(cursor.getString(cursor.getColumnIndex(PatientTable.Column.ENCOUNTERS)));


        PatientIdentifier patientIdentifier = new PatientIdentifier();
        patientIdentifier.setIdentifier(cursor.getString(cursor.getColumnIndex(PatientTable.Column.IDENTIFIER)));
        patient.getIdentifiers().add(patientIdentifier);

        PersonName personName = new PersonName();
        personName.setGivenName(cursor.getString(cursor.getColumnIndex(PatientTable.Column.GIVEN_NAME)));
        personName.setMiddleName(cursor.getString(cursor.getColumnIndex(PatientTable.Column.MIDDLE_NAME)));
        personName.setFamilyName(cursor.getString(cursor.getColumnIndex(PatientTable.Column.FAMILY_NAME)));
        person.getNames().add(personName);

        person.setGender(cursor.getString(cursor.getColumnIndex(PatientTable.Column.GENDER)));
        person.setBirthdate(cursor.getString(cursor.getColumnIndex(PatientTable.Column.BIRTH_DATE)));
        byte[] photoByteArray = cursor.getBlob(cursor.getColumnIndex(PatientTable.Column.PHOTO));
        if (photoByteArray != null)
            person.setPhoto(byteArrayToBitmap(photoByteArray));
        person.getAddresses().add(cursorToAddress(cursor));

        patient.setPerson(person);
        return patient;
    }

    public boolean isUserAlreadySaved(String uuid) {
        String where = String.format("%s = ?", PatientTable.Column.UUID);
        String[] whereArgs = new String[]{uuid};

        DBOpenHelper helper = OpenMRSDBOpenHelper.getInstance().getDBOpenHelper();
        final Cursor cursor = helper.getReadableDatabase().query(PatientTable.TABLE_NAME, null, where, whereArgs, null, null, null);
        String patientUUID = "";
        if (null != cursor) {
            try {
                if (cursor.moveToFirst()) {
                    int uuidColumnIndex = cursor.getColumnIndex(PatientTable.Column.UUID);
                    patientUUID = cursor.getString(uuidColumnIndex);
                }
            } finally {
                cursor.close();
            }
        }
        return uuid.equalsIgnoreCase(patientUUID);
    }

    public boolean userDoesNotExist(String uuid) {
        return !isUserAlreadySaved(uuid);
    }

    public Patient findPatientByUUID(String uuid) {
        Patient patient = new Patient();
        String where = String.format("%s = ?", PatientTable.Column.UUID);
        String[] whereArgs = new String[]{uuid};

        DBOpenHelper helper = OpenMRSDBOpenHelper.getInstance().getDBOpenHelper();
        final Cursor cursor = helper.getReadableDatabase().query(PatientTable.TABLE_NAME, null, where, whereArgs, null, null, null);
        if (null != cursor) {
            try {
                if (cursor.moveToFirst()) {
                    patient = cursorToPatient(cursor);
                }
            } finally {
                cursor.close();
            }
        }
        return patient;
    }

    public List<Patient> getUnsyncedPatients(){
        List<Patient> patientList = new LinkedList<>();
        String where = String.format("%s = ?", PatientTable.Column.SYNCED);
        String[] whereArgs = new String[]{"false"};

        DBOpenHelper helper = OpenMRSDBOpenHelper.getInstance().getDBOpenHelper();
        final Cursor cursor = helper.getReadableDatabase().query(PatientTable.TABLE_NAME, null , where, whereArgs, null, null, null);
        if (null != cursor) {
            try {
                while (cursor.moveToNext()) {
                    Patient patient = cursorToPatient(cursor);
                    if(!patient.isSynced()){
                        patientList.add(patient);
                    }
                }
            } finally {
                cursor.close();
            }
        }
        return patientList;
    }

    public Patient findPatientByID(String id) {
        Patient patient = new Patient();
        String where = String.format("%s = ?", PatientTable.Column.ID);
        String[] whereArgs = new String[]{id};

        DBOpenHelper helper = OpenMRSDBOpenHelper.getInstance().getDBOpenHelper();
        final Cursor cursor = helper.getReadableDatabase().query(PatientTable.TABLE_NAME, null, where, whereArgs, null, null, null);
        if (null != cursor) {
            try {
                if (cursor.moveToFirst()) {
                    patient = cursorToPatient(cursor);
                }
            } finally {
                cursor.close();
            }
        }
        return patient;
    }

    private PersonAddress cursorToAddress(Cursor cursor) {
        int address1ColumnIndex = cursor.getColumnIndex(PatientTable.Column.ADDRESS_1);
        int address2ColumnIndex = cursor.getColumnIndex(PatientTable.Column.ADDRESS_2);
        int postalColumnIndex = cursor.getColumnIndex(PatientTable.Column.POSTAL_CODE);
        int countryColumnIndex = cursor.getColumnIndex(PatientTable.Column.COUNTRY);
        int stateColumnIndex = cursor.getColumnIndex(PatientTable.Column.STATE);
        int cityColumnIndex = cursor.getColumnIndex(PatientTable.Column.CITY);

        PersonAddress personAddress = new PersonAddress();
        personAddress.setAddress1(cursor.getString(address1ColumnIndex));
        personAddress.setAddress2(cursor.getString(address2ColumnIndex));
        personAddress.setPostalCode(cursor.getString(postalColumnIndex));
        personAddress.setCountry( cursor.getString(countryColumnIndex));
        personAddress.setStateProvince(cursor.getString(stateColumnIndex));
        personAddress.setCityVillage(cursor.getString(cityColumnIndex));

        return personAddress;
    }

    private Bitmap byteArrayToBitmap(byte[] imageByteArray) {
        ByteArrayInputStream inputStream = new ByteArrayInputStream(imageByteArray);
        return BitmapFactory.decodeStream(inputStream);
    }

}