From 06a3dc62562a5ccdca836492191a6e03048b3d92 Mon Sep 17 00:00:00 2001
From: Rumperuu <me@bengoldsworthy.uk>
Date: Wed, 19 Sep 2018 08:39:16 +0100
Subject: [PATCH] giag

---
 .../uk/ac/lancaster/auditor/AuditActivity.kt  | 31 +++++++
 .../lancaster/auditor/ChooseRecordActivity.kt | 20 +++++
 .../uk/ac/lancaster/auditor/MainActivity.kt   |  6 +-
 .../auditor/barcode/BarcodeScanActivity.kt    | 89 +++++++++++--------
 app/src/main/res/layout/activity_audit.xml    | 44 +++++++++
 .../res/layout/activity_choose_record.xml     | 21 +++++
 app/src/main/res/layout/activity_main.xml     | 15 ++--
 7 files changed, 180 insertions(+), 46 deletions(-)
 create mode 100644 app/src/main/java/uk/ac/lancaster/auditor/AuditActivity.kt
 create mode 100644 app/src/main/java/uk/ac/lancaster/auditor/ChooseRecordActivity.kt
 create mode 100644 app/src/main/res/layout/activity_audit.xml
 create mode 100644 app/src/main/res/layout/activity_choose_record.xml

diff --git a/app/src/main/java/uk/ac/lancaster/auditor/AuditActivity.kt b/app/src/main/java/uk/ac/lancaster/auditor/AuditActivity.kt
new file mode 100644
index 0000000..724adaf
--- /dev/null
+++ b/app/src/main/java/uk/ac/lancaster/auditor/AuditActivity.kt
@@ -0,0 +1,31 @@
+package uk.ac.lancaster.auditor
+
+import android.content.Intent
+import android.os.Bundle
+import android.support.v7.app.AppCompatActivity
+import android.widget.Button
+import uk.ac.lancaster.auditor.barcode.BarcodeScanActivity
+
+class AuditActivity : AppCompatActivity() {
+    override fun onCreate(savedInstanceState: Bundle?) {
+        super.onCreate(savedInstanceState)
+        setContentView(R.layout.activity_audit)
+
+        findViewById<Button>(R.id.button).setOnClickListener {
+            val intent = Intent(applicationContext, ChooseRecordActivity::class.java)
+            startActivityForResult(intent, 1)
+        }
+        findViewById<Button>(R.id.button2).setOnClickListener {
+            val intent = Intent(applicationContext, ChooseRecordActivity::class.java)
+            startActivityForResult(intent, 1)
+        }
+        findViewById<Button>(R.id.button3).setOnClickListener {
+            val intent = Intent(applicationContext, ChooseRecordActivity::class.java)
+            startActivityForResult(intent, 1)
+        }
+    }
+
+    override fun onActivityResult(data: Intent?) {
+
+    }
+}
\ No newline at end of file
diff --git a/app/src/main/java/uk/ac/lancaster/auditor/ChooseRecordActivity.kt b/app/src/main/java/uk/ac/lancaster/auditor/ChooseRecordActivity.kt
new file mode 100644
index 0000000..94bf81b
--- /dev/null
+++ b/app/src/main/java/uk/ac/lancaster/auditor/ChooseRecordActivity.kt
@@ -0,0 +1,20 @@
+package uk.ac.lancaster.auditor
+
+import android.os.Bundle
+import android.support.v7.app.AppCompatActivity
+import android.R.array
+import android.widget.ArrayAdapter
+
+
+
+class ChooseRecordActivity : AppCompatActivity() {
+    override fun onCreate(savedInstanceState: Bundle?) {
+        super.onCreate(savedInstanceState)
+        setContentView(R.layout.activity_choose_record)
+
+        var fileList = getFilesDir().listFiles()
+
+        val adapter = ArrayAdapter<String>(this, android.R.layout.simple_list_item_1, fileList)
+        (R.id.list).setAdapter(aa)
+    }
+}
\ No newline at end of file
diff --git a/app/src/main/java/uk/ac/lancaster/auditor/MainActivity.kt b/app/src/main/java/uk/ac/lancaster/auditor/MainActivity.kt
index 74c4a2e..595534f 100644
--- a/app/src/main/java/uk/ac/lancaster/auditor/MainActivity.kt
+++ b/app/src/main/java/uk/ac/lancaster/auditor/MainActivity.kt
@@ -11,9 +11,13 @@ class MainActivity : AppCompatActivity() {
         super.onCreate(savedInstanceState)
         setContentView(R.layout.activity_main)
 
-        findViewById<Button>(R.id.scan_ballot_qr).setOnClickListener {
+        findViewById<Button>(R.id.cast_vote).setOnClickListener {
             val intent = Intent(applicationContext, BarcodeScanActivity::class.java)
             startActivity(intent)
         }
+        findViewById<Button>(R.id.audit_election).setOnClickListener {
+            val intent = Intent(applicationContext, AuditActivity::class.java)
+            startActivity(intent)
+        }
     }
 }
diff --git a/app/src/main/java/uk/ac/lancaster/auditor/barcode/BarcodeScanActivity.kt b/app/src/main/java/uk/ac/lancaster/auditor/barcode/BarcodeScanActivity.kt
index 4e7037a..c5cc6c8 100644
--- a/app/src/main/java/uk/ac/lancaster/auditor/barcode/BarcodeScanActivity.kt
+++ b/app/src/main/java/uk/ac/lancaster/auditor/barcode/BarcodeScanActivity.kt
@@ -1,6 +1,7 @@
 package uk.ac.lancaster.auditor.barcode
 
 import android.content.Intent
+import android.content.Context
 import android.os.Bundle
 import android.os.Environment
 import android.support.v7.app.AppCompatActivity
@@ -9,7 +10,7 @@ import android.util.Log
 import android.widget.Button
 import android.widget.TextView
 import java.io.File
-import java.io.FileWriter
+import java.io.FileOutputStream
 import com.google.android.gms.common.api.CommonStatusCodes
 import com.google.android.gms.vision.barcode.Barcode
 import uk.ac.lancaster.auditor.BallotVerifyActivity
@@ -19,6 +20,18 @@ import uk.ac.lancaster.auditor.R
 class BarcodeScanActivity : AppCompatActivity() {
 
     private lateinit var mResultTextView: TextView
+    private var stage = 0
+
+    data class Record(val recordID: Int) {
+        var voterID: String = ""
+        var eventID: String = ""
+        var pollID: String = ""
+        var hashes: String = ""
+        var handle: String = ""
+        var hmac: String = ""
+    }
+
+    private val record = Record(0)
 
     override fun onCreate(savedInstanceState: Bundle?) {
         super.onCreate(savedInstanceState)
@@ -38,50 +51,32 @@ class BarcodeScanActivity : AppCompatActivity() {
                 if (data != null) {
                     val barcode = data.getParcelableExtra<Barcode>(BarcodeCaptureActivity.BarcodeObject)
                     mResultTextView.text = barcode.displayValue
-                    val handle = barcode.displayValue.split(";")[0]
-                    val hmac = String(Base64.decode(barcode.displayValue.split(";")[1], Base64.DEFAULT))
 
-                    // save the ballot
-                    if (Environment.getExternalStorageState() == Environment.MEDIA_MOUNTED) {
-                        val dir = File(Environment.getExternalStoragePublicDirectory(Environment.DIRECTORY_DOCUMENTS), "ballots")
-                        if (!dir?.mkdirs()) {
-                            Log.e(LOG_TAG, "Directory not created")
+                    // Store the relevant info from whatever this QR code is for
+                    when (stage) {
+                        // Stage 0 is getting the voter ID, poll ID and event ID
+                        0 -> {
+                            val barcodeVal = barcode.displayValue.split(";")
+                            record.voterID = barcodeVal[0]
+                            record.eventID = barcodeVal[1]
+                            record.pollID = barcodeVal[2]
                         }
-
-                        val ballotFile = File(dir, "$handle.ballot")
-                        if (ballotFile.exists()) {
-                            ballotFile.delete()
+                        // Stage 1 is getting the hashes of both ballots
+                        1 -> {
+                            record.hashes = barcode.displayValue
                         }
-
-                        try {
-                            val out = FileWriter(ballotFile)
-                            out.write(handle)
-                            out.flush()
-                            out.close()
-                        } catch (e: Exception) {
-                            e.printStackTrace()
+                        // Stage 2 is getting the handle of the unselected ballot on the server
+                        2 -> {
+                            record.handle = barcode.displayValue.split(";")[0]
+                            record.hmac = String(Base64.decode(barcode.displayValue.split(";")[1], Base64.DEFAULT))
+                            saveRecord()
+                            finish()
                         }
-
-                        val hmacFile = File(dir, "$handle.hmac")
-                        if (hmacFile.exists()) {
-                            hmacFile.delete()
+                        else -> {
+                            //something's gone wrong if this gets called
                         }
-
-                        try {
-                            val out = FileWriter(hmacFile)
-                            out.write(hmac)
-                            out.flush()
-                            out.close()
-                        } catch (e: Exception) {
-                            e.printStackTrace()
-                        }
-
                     }
-
-                    //if some test of the QR = ballot handle
-                    val intent = Intent(baseContext, BallotVerifyActivity::class.java)
-                    intent.putExtra("ballotHandle", handle)
-                    startActivity(intent)
+                    stage++
                 } else
                     mResultTextView.setText(R.string.no_barcode_captured)
             } else
@@ -91,6 +86,22 @@ class BarcodeScanActivity : AppCompatActivity() {
             super.onActivityResult(requestCode, resultCode, data)
     }
 
+    private fun saveRecord() {
+        // save the ballot
+        val filename = record.handle
+        val fileContents = record.toString()
+        val outputStream: FileOutputStream
+
+        try {
+            outputStream = openFileOutput(filename, Context.MODE_PRIVATE)
+            outputStream.write(fileContents.toByteArray())
+            outputStream.close()
+        } catch (e: Exception) {
+            e.printStackTrace()
+        }
+
+    }
+
     companion object {
         private val LOG_TAG = MainActivity::class.java.simpleName
         private val BARCODE_READER_REQUEST_CODE = 1
diff --git a/app/src/main/res/layout/activity_audit.xml b/app/src/main/res/layout/activity_audit.xml
new file mode 100644
index 0000000..24e56ce
--- /dev/null
+++ b/app/src/main/res/layout/activity_audit.xml
@@ -0,0 +1,44 @@
+<?xml version="1.0" encoding="utf-8"?>
+<android.support.constraint.ConstraintLayout
+    xmlns:android="http://schemas.android.com/apk/res/android"
+    xmlns:app="http://schemas.android.com/apk/res-auto"
+    xmlns:tools="http://schemas.android.com/tools"
+    android:layout_width="match_parent"
+    android:layout_height="match_parent">
+
+    <Button
+        android:id="@+id/button"
+        android:layout_width="wrap_content"
+        android:layout_height="wrap_content"
+        android:layout_marginEnd="8dp"
+        android:layout_marginStart="8dp"
+        android:layout_marginTop="8dp"
+        android:text="Check ballot exist"
+        app:layout_constraintEnd_toEndOf="parent"
+        app:layout_constraintStart_toStartOf="parent"
+        app:layout_constraintTop_toTopOf="parent" />
+
+    <Button
+        android:id="@+id/button2"
+        android:layout_width="wrap_content"
+        android:layout_height="wrap_content"
+        android:layout_marginEnd="8dp"
+        android:layout_marginStart="8dp"
+        android:layout_marginTop="8dp"
+        android:text="Verify cast-as-recorded"
+        app:layout_constraintEnd_toEndOf="parent"
+        app:layout_constraintStart_toStartOf="parent"
+        app:layout_constraintTop_toBottomOf="@+id/button" />
+
+    <Button
+        android:id="@+id/button3"
+        android:layout_width="wrap_content"
+        android:layout_height="wrap_content"
+        android:layout_marginEnd="8dp"
+        android:layout_marginStart="8dp"
+        android:layout_marginTop="8dp"
+        android:text="Verify tallied-as-cast"
+        app:layout_constraintEnd_toEndOf="parent"
+        app:layout_constraintStart_toStartOf="parent"
+        app:layout_constraintTop_toBottomOf="@+id/button2" />
+</android.support.constraint.ConstraintLayout>
\ No newline at end of file
diff --git a/app/src/main/res/layout/activity_choose_record.xml b/app/src/main/res/layout/activity_choose_record.xml
new file mode 100644
index 0000000..5541952
--- /dev/null
+++ b/app/src/main/res/layout/activity_choose_record.xml
@@ -0,0 +1,21 @@
+<?xml version="1.0" encoding="utf-8"?>
+<android.support.constraint.ConstraintLayout
+    xmlns:android="http://schemas.android.com/apk/res/android"
+    xmlns:app="http://schemas.android.com/apk/res-auto"
+    xmlns:tools="http://schemas.android.com/tools"
+    android:layout_width="match_parent"
+    android:layout_height="match_parent">
+
+    <ListView
+        android:id="@+id/list"
+        android:layout_width="368dp"
+        android:layout_height="495dp"
+        android:layout_marginBottom="8dp"
+        android:layout_marginEnd="8dp"
+        android:layout_marginStart="8dp"
+        app:layout_constraintBottom_toBottomOf="parent"
+        app:layout_constraintEnd_toEndOf="parent"
+        app:layout_constraintHorizontal_bias="1.0"
+        app:layout_constraintStart_toStartOf="parent"
+        app:layout_constraintTop_toTopOf="parent" />
+</android.support.constraint.ConstraintLayout>
\ No newline at end of file
diff --git a/app/src/main/res/layout/activity_main.xml b/app/src/main/res/layout/activity_main.xml
index 5af7069..83f52f2 100644
--- a/app/src/main/res/layout/activity_main.xml
+++ b/app/src/main/res/layout/activity_main.xml
@@ -2,30 +2,33 @@
 <android.support.constraint.ConstraintLayout
     xmlns:android="http://schemas.android.com/apk/res/android"
     xmlns:app="http://schemas.android.com/apk/res-auto"
+    xmlns:tools="http://schemas.android.com/tools"
     android:layout_width="match_parent"
     android:layout_height="match_parent">
 
     <Button
-        android:id="@+id/scan_hash_qr"
+        android:id="@+id/cast_vote"
         android:layout_width="218dp"
         android:layout_height="64dp"
         android:layout_marginEnd="8dp"
         android:layout_marginStart="8dp"
         android:layout_marginTop="8dp"
-        android:text="Scan Hash QR"
+        android:text="I'm casting a vote"
         app:layout_constraintEnd_toEndOf="parent"
         app:layout_constraintStart_toStartOf="parent"
         app:layout_constraintTop_toTopOf="parent" />
 
     <Button
-        android:id="@+id/scan_ballot_qr"
+        android:id="@+id/audit_election"
         android:layout_width="218dp"
         android:layout_height="64dp"
+        android:layout_marginBottom="8dp"
         android:layout_marginEnd="8dp"
         android:layout_marginStart="8dp"
-        android:layout_marginTop="8dp"
-        android:text="Scan Ballot QR"
+        android:layout_marginTop="336dp"
+        android:text="I'm auditing an election"
+        app:layout_constraintBottom_toBottomOf="parent"
         app:layout_constraintEnd_toEndOf="parent"
         app:layout_constraintStart_toStartOf="parent"
-        app:layout_constraintTop_toBottomOf="@+id/scan_hash_qr" />
+        app:layout_constraintTop_toBottomOf="@+id/cast_vote" />
 </android.support.constraint.ConstraintLayout>
\ No newline at end of file