From 0f7c49a5f66fc5a44a5600367f24d3b5bf290193 Mon Sep 17 00:00:00 2001
From: e <68205167+BubbaJuice@users.noreply.github.com>
Date: Thu, 1 Aug 2024 12:33:53 -0700
Subject: [PATCH] Adding rudimentary logging feature

---
 .idea/deploymentTargetSelector.xml            |   2 +-
 app/build.gradle                              |  20 ++
 app/src/main/AndroidManifest.xml              |   7 +
 .../bubbajuice/cellinfo/MainActivity.kt       | 334 +++++++++++++++++-
 gradle/libs.versions.toml                     |   6 +
 5 files changed, 366 insertions(+), 3 deletions(-)

diff --git a/.idea/deploymentTargetSelector.xml b/.idea/deploymentTargetSelector.xml
index bd7dd59..570f9bd 100644
--- a/.idea/deploymentTargetSelector.xml
+++ b/.idea/deploymentTargetSelector.xml
@@ -4,7 +4,7 @@
     <selectionStates>
       <SelectionState runConfigName="app">
         <option name="selectionMode" value="DROPDOWN" />
-        <DropdownSelection timestamp="2024-07-31T16:31:41.911572400Z">
+        <DropdownSelection timestamp="2024-08-01T17:31:26.990896800Z">
           <Target type="DEFAULT_BOOT">
             <handle>
               <DeviceId pluginId="PhysicalDevice" identifier="serial=ZY22GBG5DD" />
diff --git a/app/build.gradle b/app/build.gradle
index 93b52d6..a010109 100644
--- a/app/build.gradle
+++ b/app/build.gradle
@@ -2,6 +2,8 @@ plugins {
     alias(libs.plugins.android.application)
     alias(libs.plugins.jetbrains.kotlin.android)
     id 'org.jetbrains.kotlin.plugin.serialization' version '2.0.20-RC'
+    id 'kotlin-android'
+    id 'kotlin-kapt'
 }
 
 android {
@@ -19,6 +21,16 @@ android {
         vectorDrawables {
             useSupportLibrary true
         }
+
+        javaCompileOptions {
+            annotationProcessorOptions {
+                arguments += [
+                        "room.schemaLocation": "$projectDir/schemas".toString(),
+                        "room.incremental": "true",
+                        "room.expandProjection": "true"
+                ]
+            }
+        }
     }
 
     signingConfigs {
@@ -58,6 +70,8 @@ android {
 }
 
 dependencies {
+    implementation libs.androidx.databinding.adapters
+    def room_version = "2.5.1"
     implementation libs.androidx.core.ktx
     implementation libs.androidx.lifecycle.runtime.ktx
     implementation libs.androidx.activity.compose
@@ -69,6 +83,12 @@ dependencies {
     implementation "org.burnoutcrew.composereorderable:reorderable:0.9.6"
     implementation "org.jetbrains.kotlinx:kotlinx-serialization-json:1.3.2"
     implementation libs.androidx.runtime.livedata
+    implementation libs.androidx.room.common
+    implementation libs.androidx.room.ktx
+    implementation "androidx.room:room-runtime:$room_version"
+    annotationProcessor "androidx.room:room-compiler:$room_version"
+    kapt "androidx.room:room-compiler:$room_version"
+    implementation "androidx.room:room-ktx:$room_version"
     testImplementation libs.junit
     androidTestImplementation libs.androidx.junit
     androidTestImplementation libs.androidx.espresso.core
diff --git a/app/src/main/AndroidManifest.xml b/app/src/main/AndroidManifest.xml
index 03c8555..35bbba0 100644
--- a/app/src/main/AndroidManifest.xml
+++ b/app/src/main/AndroidManifest.xml
@@ -4,6 +4,9 @@
     <uses-permission android:name="android.permission.READ_PHONE_STATE"/>
     <uses-permission android:name="android.permission.ACCESS_FINE_LOCATION"/>
     <uses-permission android:name="android.permission.ACCESS_COARSE_LOCATION"/>
+    <uses-permission android:name="android.permission.FOREGROUND_SERVICE_LOCATION"/>
+    <uses-permission android:name="android.permission.FOREGROUND_SERVICE" />
+    <uses-permission android:name="android.permission.POST_NOTIFICATIONS" />
 
     <application
         android:allowBackup="true"
@@ -25,6 +28,10 @@
                 <category android:name="android.intent.category.LAUNCHER" />
             </intent-filter>
         </activity>
+        <service
+            android:name=".CellLoggingService"
+            android:exported="false"
+            android:foregroundServiceType="location" />
     </application>
 
 </manifest>
diff --git a/app/src/main/java/io/github/bubbajuice/cellinfo/MainActivity.kt b/app/src/main/java/io/github/bubbajuice/cellinfo/MainActivity.kt
index ea22b5c..4a92412 100644
--- a/app/src/main/java/io/github/bubbajuice/cellinfo/MainActivity.kt
+++ b/app/src/main/java/io/github/bubbajuice/cellinfo/MainActivity.kt
@@ -2,14 +2,23 @@ package io.github.bubbajuice.cellinfo
 
 import android.Manifest
 import android.app.Application
+import android.app.Notification
+import android.app.NotificationChannel
+import android.app.NotificationManager
+import android.app.PendingIntent
+import android.app.Service
 import android.content.Context
+import android.content.Intent
 import android.content.pm.PackageManager
+import android.icu.text.SimpleDateFormat
 import android.os.Bundle
+import android.os.IBinder
 import android.telephony.CellIdentityNr
 import android.telephony.CellInfo
 import android.telephony.CellInfoLte
 import android.telephony.CellInfoNr
 import android.telephony.TelephonyManager
+import android.util.Log
 import androidx.activity.ComponentActivity
 import androidx.activity.compose.setContent
 import androidx.activity.result.contract.ActivityResultContracts
@@ -20,6 +29,7 @@ import androidx.compose.foundation.combinedClickable
 import androidx.compose.foundation.layout.*
 import androidx.compose.foundation.lazy.itemsIndexed
 import androidx.compose.foundation.lazy.LazyColumn
+import androidx.compose.foundation.lazy.items
 import androidx.compose.material.icons.Icons
 import androidx.compose.material.icons.filled.Call
 import androidx.compose.material.icons.filled.KeyboardArrowDown
@@ -33,23 +43,324 @@ import androidx.compose.runtime.*
 import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
 import androidx.compose.ui.platform.LocalConfiguration
+import androidx.compose.ui.text.font.FontWeight
 import androidx.compose.ui.text.style.TextAlign
 import androidx.compose.ui.unit.dp
 import androidx.compose.ui.window.Dialog
 import androidx.lifecycle.AndroidViewModel
 import androidx.lifecycle.viewModelScope
 import androidx.core.app.ActivityCompat
+import androidx.core.app.NotificationCompat
+import androidx.room.*
 import io.github.bubbajuice.cellinfo.ui.theme.MyApplicationTheme
 import kotlin.math.round
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.Job
+import kotlinx.coroutines.cancel
 import kotlinx.coroutines.delay
+import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.asStateFlow
+import kotlinx.coroutines.isActive
 import kotlinx.coroutines.launch
+import kotlinx.coroutines.withContext
 import kotlinx.serialization.decodeFromString
 import kotlinx.serialization.encodeToString
 import kotlinx.serialization.json.Json
 import kotlinx.serialization.Serializable
 import org.burnoutcrew.reorderable.*
+import java.util.Date
+import java.util.Locale
+
+@Entity(tableName = "logged_cells")
+data class LoggedCell(
+    @PrimaryKey val cellId: String,
+    val type: String,
+    val timestamp: Long,
+    val enbId: String?,
+    val earfcn: String?,
+    val pci: String?,
+    val cellSector: String?,
+    val bandNumber: String?,
+    val tac: String?,
+    val mcc: String?,
+    val mnc: String?,
+    val operator: String?
+)
+
+class CellLoggingService : Service() {
+    private lateinit var cellDatabase: CellDatabase
+    private lateinit var telephonyManager: TelephonyManager
+    private lateinit var settingsViewModel: SettingsViewModel
+    private val serviceScope = CoroutineScope(Dispatchers.Default)
+    private var loggingJob: Job? = null
+
+    private fun clearDatabase() {
+        CoroutineScope(Dispatchers.IO).launch {
+            cellDatabase.clearAllTables()
+        }
+    }
+
+    override fun onCreate() {
+        super.onCreate()
+        cellDatabase = CellDatabase.getInstance(applicationContext)
+        clearDatabase()
+        telephonyManager = getSystemService(Context.TELEPHONY_SERVICE) as TelephonyManager
+        settingsViewModel = SettingsViewModel(application)
+    }
+
+    override fun onStartCommand(intent: Intent?, flags: Int, startId: Int): Int {
+        startForeground(NOTIFICATION_ID, createNotification())
+        startLogging()
+        return START_STICKY
+    }
+
+    override fun onBind(intent: Intent?): IBinder? = null
+
+    override fun onDestroy() {
+        super.onDestroy()
+        stopLogging()
+    }
+
+    private fun startLogging() {
+        loggingJob = serviceScope.launch {
+            while (isActive) {
+                logCells()
+                delay(1000) // Log every second
+            }
+        }
+    }
+
+    private fun stopLogging() {
+        loggingJob?.cancel()
+        serviceScope.cancel()
+    }
+
+    private suspend fun logCells() {
+        if (ActivityCompat.checkSelfPermission(
+                this,
+                Manifest.permission.ACCESS_FINE_LOCATION
+            ) == PackageManager.PERMISSION_GRANTED
+        ) {
+            val cellInfoList = withContext(Dispatchers.Default) {
+                telephonyManager.allCellInfo
+            }
+            cellInfoList.forEach { cellInfo ->
+                when (cellInfo) {
+                    is CellInfoNr -> logNrCell(cellInfo)
+                    is CellInfoLte -> logLteCell(cellInfo)
+                }
+            }
+        }
+    }
+
+    private suspend fun logNrCell(cellInfo: CellInfoNr) {
+        val cellIdentity = cellInfo.cellIdentity as? CellIdentityNr ?: return
+        val cellId = cellIdentity.nci.toString()
+        logCell(
+            "NR",
+            cellId,
+            null, // eNB ID not applicable for NR
+            cellIdentity.nrarfcn.toString(),
+            cellIdentity.pci.toString(),
+            null, // Cell sector not applicable for NR
+            getNrBandFromArfcn(cellIdentity.nrarfcn).toString(),
+            cellIdentity.tac.toString(),
+            cellIdentity.mccString,
+            cellIdentity.mncString,
+            cellIdentity.operatorAlphaLong?.toString()
+        )
+    }
+
+    private suspend fun logLteCell(cellInfo: CellInfoLte) {
+        val cellIdentity = cellInfo.cellIdentity
+        val cellId = cellIdentity.ci.toString()
+        logCell(
+            "LTE",
+            cellId,
+            calculateENBId(cellId),
+            cellIdentity.earfcn.toString(),
+            cellIdentity.pci.toString(),
+            calculateCellSectorId(cellId),
+            getLTEBandFromEArfcn(cellIdentity.earfcn).toString(),
+            cellIdentity.tac.toString(),
+            cellIdentity.mccString,
+            cellIdentity.mncString,
+            cellIdentity.operatorAlphaLong?.toString()
+        )
+    }
+
+    private fun logCell(
+        type: String,
+        cellId: String,
+        enbId: String?,
+        earfcn: String?,
+        pci: String?,
+        cellSector: String?,
+        bandNumber: String?,
+        tac: String?,
+        mcc: String?,
+        mnc: String?,
+        operator: String?
+    ) {
+        val loggedCell = LoggedCell(
+            cellId,
+            type,
+            System.currentTimeMillis(),
+            enbId,
+            earfcn,
+            pci,
+            cellSector,
+            bandNumber,
+            tac,
+            mcc,
+            mnc,
+            operator
+        )
+        serviceScope.launch(Dispatchers.IO) {
+            cellDatabase.cellDao().insert(loggedCell)
+            checkForNewCell(loggedCell)
+        }
+    }
+
+    private fun checkForNewCell(loggedCell: LoggedCell) {
+        serviceScope.launch(Dispatchers.IO) {
+            val isNewCell = cellDatabase.cellDao().getCellCount(loggedCell.cellId) == 1
+            if (isNewCell) {
+                withContext(Dispatchers.Main) {
+                    showNewCellNotification(loggedCell)
+                }
+            }
+        }
+    }
+
+    private fun showNewCellNotification(loggedCell: LoggedCell) {
+        val notificationManager = getSystemService(Context.NOTIFICATION_SERVICE) as NotificationManager
+        val notification = NotificationCompat.Builder(this, CHANNEL_ID)
+            .setContentTitle("New Cell Detected")
+            .setContentText("${loggedCell.type} cell with ID ${loggedCell.cellId}")
+            .setSmallIcon(R.drawable.ic_launcher_foreground)
+            .setPriority(NotificationCompat.PRIORITY_DEFAULT)
+            .setAutoCancel(true)
+            .build()
+
+        notificationManager.notify(NEW_CELL_NOTIFICATION_ID, notification)
+    }
+
+    private fun createNotification(): Notification {
+        val notificationIntent = Intent(this, MainActivity::class.java)
+        val pendingIntent = PendingIntent.getActivity(
+            this, 0, notificationIntent, PendingIntent.FLAG_IMMUTABLE
+        )
+
+        return NotificationCompat.Builder(this, CHANNEL_ID)
+            .setContentTitle("Cell Logging Service")
+            .setContentText("Logging cells in the background")
+            .setSmallIcon(R.drawable.ic_launcher_foreground)
+            .setContentIntent(pendingIntent)
+            .build()
+    }
+
+    companion object {
+        const val CHANNEL_ID = "CellLoggingChannel"
+        private const val NOTIFICATION_ID = 1
+        private const val NEW_CELL_NOTIFICATION_ID = 2
+    }
+}
+
+@Dao
+interface CellDao {
+    @Insert(onConflict = OnConflictStrategy.REPLACE)
+    fun insert(cell: LoggedCell)
+
+    @Query("SELECT COUNT(*) FROM logged_cells WHERE cellId = :cellId")
+    fun getCellCount(cellId: String): Int
+
+    @Query("SELECT * FROM logged_cells ORDER BY timestamp DESC")
+    fun getAllCells(): Flow<List<LoggedCell>>
+}
+
+@Database(entities = [LoggedCell::class], version = 2, exportSchema = false)
+abstract class CellDatabase : RoomDatabase() {
+    abstract fun cellDao(): CellDao
+
+    companion object {
+        private var instance: CellDatabase? = null
+
+        fun getInstance(context: Context): CellDatabase {
+            return instance ?: synchronized(this) {
+                instance ?: buildDatabase(context).also { instance = it }
+            }
+        }
+
+        private fun buildDatabase(context: Context): CellDatabase {
+            return Room.databaseBuilder(
+                context.applicationContext,
+                CellDatabase::class.java,
+                "cell_database"
+            )
+                .fallbackToDestructiveMigration() // This will recreate the database if the version changes
+                .build()
+        }
+    }
+}
+
+class Converters {
+    @TypeConverter
+    fun fromTimestamp(value: Long?): Date? {
+        return value?.let { Date(it) }
+    }
+
+    @TypeConverter
+    fun dateToTimestamp(date: Date?): Long? {
+        return date?.time
+    }
+}
+
+@Composable
+fun LogPage(cellDatabase: CellDatabase) {
+    val cells by cellDatabase.cellDao().getAllCells().collectAsState(initial = emptyList())
+
+    LazyColumn {
+        items(cells) { cell ->
+            LoggedCellItem(cell)
+        }
+    }
+}
+
+@Composable
+fun LoggedCellItem(cell: LoggedCell) {
+    // Skip cells with cellId 268435455
+    if (cell.cellId == "268435455") {
+        return
+    }
+
+    Card(
+        modifier = Modifier
+            .fillMaxWidth()
+            .padding(8.dp),
+        elevation = CardDefaults.cardElevation(4.dp)
+    ) {
+        Column(modifier = Modifier.padding(16.dp)) {
+            Text("${cell.type} Cell", style = MaterialTheme.typography.titleMedium)
+            Text("Cell ID: ${cell.cellId}", style = MaterialTheme.typography.bodyMedium)
+            Text("eNB ID: ${cell.enbId ?: "N/A"}", style = MaterialTheme.typography.bodyMedium)
+            Text("EARFCN: ${cell.earfcn ?: "N/A"}", style = MaterialTheme.typography.bodyMedium)
+            Text("PCI: ${cell.pci ?: "N/A"}", style = MaterialTheme.typography.bodyMedium)
+            Text("Cell Sector: ${cell.cellSector ?: "N/A"}", style = MaterialTheme.typography.bodyMedium)
+            Text("Band Number: ${cell.bandNumber ?: "N/A"}", style = MaterialTheme.typography.bodyMedium)
+            Text("TAC: ${cell.tac ?: "N/A"}", style = MaterialTheme.typography.bodyMedium)
+            Text("MCC: ${cell.mcc ?: "N/A"}", style = MaterialTheme.typography.bodyMedium)
+            Text("MNC: ${cell.mnc ?: "N/A"}", style = MaterialTheme.typography.bodyMedium)
+            Text("Operator: ${cell.operator ?: "N/A"}", style = MaterialTheme.typography.bodyMedium)
+            Text(
+                "Logged: ${SimpleDateFormat("yyyy-MM-dd HH:mm:ss", Locale.getDefault()).format(Date(cell.timestamp))}",
+                style = MaterialTheme.typography.bodySmall
+            )
+        }
+    }
+}
 
 @Serializable
 data class CellComponent(
@@ -359,6 +670,7 @@ fun ComponentList(
 
 class MainActivity : ComponentActivity() {
     private val settingsViewModel: SettingsViewModel by viewModels()
+    private lateinit var cellDatabase: CellDatabase
     private val permissions = arrayOf(
         Manifest.permission.READ_PHONE_STATE,
         Manifest.permission.ACCESS_FINE_LOCATION,
@@ -367,6 +679,10 @@ class MainActivity : ComponentActivity() {
 
     override fun onCreate(savedInstanceState: Bundle?) {
         super.onCreate(savedInstanceState)
+        cellDatabase = CellDatabase.getInstance(applicationContext)
+
+        createNotificationChannel()
+        startCellLoggingService()
 
         setContent {
             MyApplicationTheme {
@@ -393,8 +709,7 @@ class MainActivity : ComponentActivity() {
                             Text("Map page (not yet implemented)", modifier = Modifier.padding(innerPadding))
                         }
                         "Log" -> {
-                            // Placeholder for Log page
-                            Text("Log page (not yet implemented)", modifier = Modifier.padding(innerPadding))
+                            LogPage(cellDatabase)
                         }
                         "Live" -> {
                             CellInfoScreen(
@@ -423,6 +738,21 @@ class MainActivity : ComponentActivity() {
         }
     }
 
+    private fun createNotificationChannel() {
+        val channel = NotificationChannel(
+            CellLoggingService.CHANNEL_ID,
+            "Cell Logging Channel",
+            NotificationManager.IMPORTANCE_DEFAULT
+        )
+        val notificationManager = getSystemService(Context.NOTIFICATION_SERVICE) as NotificationManager
+        notificationManager.createNotificationChannel(channel)
+    }
+
+    private fun startCellLoggingService() {
+        val serviceIntent = Intent(this, CellLoggingService::class.java)
+        startForegroundService(serviceIntent)
+    }
+
     private val requestPermissions =
         registerForActivityResult(ActivityResultContracts.RequestMultiplePermissions()) { permissions ->
             val allPermissionsGranted = permissions.entries.all { it.value }
diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml
index dd543db..8e7f63c 100644
--- a/gradle/libs.versions.toml
+++ b/gradle/libs.versions.toml
@@ -9,6 +9,9 @@ lifecycleRuntimeKtx = "2.8.4"
 activityCompose = "1.9.1"
 composeBom = "2024.04.01"
 runtimeLivedata = "1.6.8"
+roomCommon = "2.6.1"
+roomKtx = "2.6.1"
+databindingAdapters = "8.5.1"
 
 [libraries]
 androidx-core-ktx = { group = "androidx.core", name = "core-ktx", version.ref = "coreKtx" }
@@ -26,6 +29,9 @@ androidx-ui-test-manifest = { group = "androidx.compose.ui", name = "ui-test-man
 androidx-ui-test-junit4 = { group = "androidx.compose.ui", name = "ui-test-junit4" }
 androidx-material3 = { group = "androidx.compose.material3", name = "material3" }
 androidx-runtime-livedata = { group = "androidx.compose.runtime", name = "runtime-livedata", version.ref = "runtimeLivedata" }
+androidx-room-common = { group = "androidx.room", name = "room-common", version.ref = "roomCommon" }
+androidx-room-ktx = { group = "androidx.room", name = "room-ktx", version.ref = "roomKtx" }
+androidx-databinding-adapters = { group = "androidx.databinding", name = "databinding-adapters", version.ref = "databindingAdapters" }
 
 [plugins]
 android-application = { id = "com.android.application", version.ref = "agp" }