ReVanced-Mirror/VancedMicroG
1
0
Fork 0
mirror of https://github.com/TeamVanced/VancedMicroG synced 2024-12-29 06:05:47 +01:00
Code Issues Projects Releases Wiki Activity

Add exposure summary to exposure intent

Browse Source
This commit is contained in:
Christian Grigis 2020-10-09 13:54:32 +02:00 committed by Marvin W
parent 4e3c03cfaa
commit da7a4e0fdb
1 changed files with 28 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
play-services-nearby-core/src/main/kotlin/org/microg/gms/nearby/exposurenotification/ExposureNotificationServiceImpl.kt
View File

@ -157,6 +157,27 @@ class ExposureNotificationServiceImpl(private val context: Context, private val
digest() digest()
} }
private fun buildExposureSummary(token: String): ExposureSummary = ExposureDatabase.with(context) { database ->
val pair = database.loadConfiguration(packageName, token)
val (configuration, exposures) = if (pair != null) {
pair.second to database.findAllMeasuredExposures(pair.first).merge()
} else {
ExposureConfiguration.ExposureConfigurationBuilder().build() to emptyList()
}
ExposureSummary.ExposureSummaryBuilder()
.setDaysSinceLastExposure(exposures.map { it.daysSinceExposure }.min()?.toInt() ?: 0)
.setMatchedKeyCount(exposures.map { it.key }.distinct().size)
.setMaximumRiskScore(exposures.map { it.getRiskScore(configuration) }.max()?.toInt() ?: 0)
.setAttenuationDurations(intArrayOf(
exposures.map { it.getAttenuationDurations(configuration)[0] }.sum(),
exposures.map { it.getAttenuationDurations(configuration)[1] }.sum(),
exposures.map { it.getAttenuationDurations(configuration)[2] }.sum()
))
.setSummationRiskScore(exposures.map { it.getRiskScore(configuration) }.sum())
.build()
}
override fun provideDiagnosisKeys(params: ProvideDiagnosisKeysParams) { override fun provideDiagnosisKeys(params: ProvideDiagnosisKeysParams) {
Log.w(TAG, "provideDiagnosisKeys() with $packageName/${params.token}") Log.w(TAG, "provideDiagnosisKeys() with $packageName/${params.token}")
val tid = ExposureDatabase.with(context) { database -> val tid = ExposureDatabase.with(context) { database ->
@ -256,10 +277,14 @@ class ExposureNotificationServiceImpl(private val context: Context, private val
put("request_keys_count", keys) put("request_keys_count", keys)
}.toString()) }.toString())
val match = database.findAllMeasuredExposures(tid).isNotEmpty() val exposureSummary = buildExposureSummary(params.token)
try { try {
val intent = Intent(if (match) ACTION_EXPOSURE_STATE_UPDATED else ACTION_EXPOSURE_NOT_FOUND) val intent = if (exposureSummary.matchedKeyCount > 0) {
Intent(ACTION_EXPOSURE_STATE_UPDATED).putExtra(EXTRA_EXPOSURE_SUMMARY, exposureSummary)
} else {
Intent(ACTION_EXPOSURE_NOT_FOUND)
}
intent.putExtra(EXTRA_TOKEN, params.token) intent.putExtra(EXTRA_TOKEN, params.token)
intent.`package` = packageName intent.`package` = packageName
Log.d(TAG, "Sending $intent") Log.d(TAG, "Sending $intent")
@ -272,23 +297,7 @@ class ExposureNotificationServiceImpl(private val context: Context, private val
} }
override fun getExposureSummary(params: GetExposureSummaryParams): Unit = ExposureDatabase.with(context) { database -> override fun getExposureSummary(params: GetExposureSummaryParams): Unit = ExposureDatabase.with(context) { database ->
val pair = database.loadConfiguration(packageName, params.token) val response = buildExposureSummary(params.token)
val (configuration, exposures) = if (pair != null) {
pair.second to database.findAllMeasuredExposures(pair.first).merge()
} else {
ExposureConfiguration.ExposureConfigurationBuilder().build() to emptyList()
}
val response = ExposureSummary.ExposureSummaryBuilder()
.setDaysSinceLastExposure(exposures.map { it.daysSinceExposure }.min()?.toInt() ?: 0)
.setMatchedKeyCount(exposures.map { it.key }.distinct().size)
.setMaximumRiskScore(exposures.map { it.getRiskScore(configuration) }.max()?.toInt() ?: 0)
.setAttenuationDurations(intArrayOf(
exposures.map { it.getAttenuationDurations(configuration)[0] }.sum(),
exposures.map { it.getAttenuationDurations(configuration)[1] }.sum(),
exposures.map { it.getAttenuationDurations(configuration)[2] }.sum()
))
.setSummationRiskScore(exposures.map { it.getRiskScore(configuration) }.sum())
.build()
database.noteAppAction(packageName, "getExposureSummary", JSONObject().apply { database.noteAppAction(packageName, "getExposureSummary", JSONObject().apply {
put("request_token", params.token) put("request_token", params.token)