191 lines
7.7 KiB
Kotlin
191 lines
7.7 KiB
Kotlin
package io.heckel.ntfy.msg
|
|
|
|
import android.os.Build
|
|
import io.heckel.ntfy.BuildConfig
|
|
import io.heckel.ntfy.db.Notification
|
|
import io.heckel.ntfy.db.User
|
|
import io.heckel.ntfy.util.*
|
|
import okhttp3.*
|
|
import okhttp3.RequestBody.Companion.toRequestBody
|
|
import java.io.IOException
|
|
import java.net.URLEncoder
|
|
import java.nio.charset.StandardCharsets.UTF_8
|
|
import java.util.concurrent.TimeUnit
|
|
import kotlin.random.Random
|
|
|
|
class ApiService {
|
|
private val client = OkHttpClient.Builder()
|
|
.callTimeout(15, TimeUnit.SECONDS) // Total timeout for entire request
|
|
.connectTimeout(15, TimeUnit.SECONDS)
|
|
.readTimeout(15, TimeUnit.SECONDS)
|
|
.writeTimeout(15, TimeUnit.SECONDS)
|
|
.build()
|
|
private val publishClient = OkHttpClient.Builder()
|
|
.callTimeout(5, TimeUnit.MINUTES) // Total timeout for entire request
|
|
.connectTimeout(15, TimeUnit.SECONDS)
|
|
.readTimeout(15, TimeUnit.SECONDS)
|
|
.writeTimeout(15, TimeUnit.SECONDS)
|
|
.build()
|
|
private val subscriberClient = OkHttpClient.Builder()
|
|
.readTimeout(77, TimeUnit.SECONDS) // Assuming that keepalive messages are more frequent than this
|
|
.build()
|
|
private val parser = NotificationParser()
|
|
|
|
fun publish(
|
|
baseUrl: String,
|
|
topic: String,
|
|
user: User? = null,
|
|
message: String,
|
|
title: String = "",
|
|
priority: Int = 3,
|
|
tags: List<String> = emptyList(),
|
|
delay: String = "",
|
|
body: RequestBody? = null,
|
|
filename: String = ""
|
|
) {
|
|
val url = topicUrl(baseUrl, topic)
|
|
val query = mutableListOf<String>()
|
|
if (priority in 1..5) {
|
|
query.add("priority=$priority")
|
|
}
|
|
if (tags.isNotEmpty()) {
|
|
query.add("tags=${URLEncoder.encode(tags.joinToString(","), "UTF-8")}")
|
|
}
|
|
if (title.isNotEmpty()) {
|
|
query.add("title=${URLEncoder.encode(title, "UTF-8")}")
|
|
}
|
|
if (delay.isNotEmpty()) {
|
|
query.add("delay=${URLEncoder.encode(delay, "UTF-8")}")
|
|
}
|
|
if (filename.isNotEmpty()) {
|
|
query.add("filename=${URLEncoder.encode(filename, "UTF-8")}")
|
|
}
|
|
if (body != null) {
|
|
query.add("message=${URLEncoder.encode(message.replace("\n", "\\n"), "UTF-8")}")
|
|
}
|
|
val urlWithQuery = if (query.isNotEmpty()) {
|
|
url + "?" + query.joinToString("&")
|
|
} else {
|
|
url
|
|
}
|
|
val request = requestBuilder(urlWithQuery, user)
|
|
.put(body ?: message.toRequestBody())
|
|
.build()
|
|
Log.d(TAG, "Publishing to $request")
|
|
publishClient.newCall(request).execute().use { response ->
|
|
if (response.code == 401 || response.code == 403) {
|
|
throw UnauthorizedException(user)
|
|
} else if (response.code == 413) {
|
|
throw EntityTooLargeException()
|
|
} else if (!response.isSuccessful) {
|
|
throw Exception("Unexpected response ${response.code} when publishing to $url")
|
|
}
|
|
Log.d(TAG, "Successfully published to $url")
|
|
}
|
|
}
|
|
|
|
fun poll(subscriptionId: Long, baseUrl: String, topic: String, user: User?, since: String? = null): List<Notification> {
|
|
val sinceVal = since ?: "all"
|
|
val url = topicUrlJsonPoll(baseUrl, topic, sinceVal)
|
|
Log.d(TAG, "Polling topic $url")
|
|
|
|
val request = requestBuilder(url, user).build()
|
|
client.newCall(request).execute().use { response ->
|
|
if (!response.isSuccessful) {
|
|
throw Exception("Unexpected response ${response.code} when polling topic $url")
|
|
}
|
|
val body = response.body?.string()?.trim()
|
|
if (body == null || body.isEmpty()) return emptyList()
|
|
val notifications = body.lines().mapNotNull { line ->
|
|
parser.parse(line, subscriptionId = subscriptionId, notificationId = 0) // No notification when we poll
|
|
}
|
|
|
|
Log.d(TAG, "Notifications: $notifications")
|
|
return notifications
|
|
}
|
|
}
|
|
|
|
fun subscribe(
|
|
baseUrl: String,
|
|
topics: String,
|
|
since: String?,
|
|
user: User?,
|
|
notify: (topic: String, Notification) -> Unit,
|
|
fail: (Exception) -> Unit
|
|
): Call {
|
|
val sinceVal = since ?: "all"
|
|
val url = topicUrlJson(baseUrl, topics, sinceVal)
|
|
Log.d(TAG, "Opening subscription connection to $url")
|
|
val request = requestBuilder(url, user).build()
|
|
val call = subscriberClient.newCall(request)
|
|
call.enqueue(object : Callback {
|
|
override fun onResponse(call: Call, response: Response) {
|
|
try {
|
|
if (!response.isSuccessful) {
|
|
throw Exception("Unexpected response ${response.code} when subscribing to topic $url")
|
|
}
|
|
val source = response.body?.source() ?: throw Exception("Unexpected response for $url: body is empty")
|
|
while (!source.exhausted()) {
|
|
val line = source.readUtf8Line() ?: throw Exception("Unexpected response for $url: line is null")
|
|
val notification = parser.parseWithTopic(line, notificationId = Random.nextInt(), subscriptionId = 0) // subscriptionId to be set downstream
|
|
if (notification != null) {
|
|
notify(notification.topic, notification.notification)
|
|
}
|
|
}
|
|
} catch (e: Exception) {
|
|
Log.e(TAG, "Connection to $url failed (1): ${e.message}", e)
|
|
fail(e)
|
|
}
|
|
}
|
|
override fun onFailure(call: Call, e: IOException) {
|
|
Log.e(TAG, "Connection to $url failed (2): ${e.message}", e)
|
|
fail(e)
|
|
}
|
|
})
|
|
return call
|
|
}
|
|
|
|
fun checkAuth(baseUrl: String, topic: String, user: User?): Boolean {
|
|
if (user == null) {
|
|
Log.d(TAG, "Checking anonymous read against ${topicUrl(baseUrl, topic)}")
|
|
} else {
|
|
Log.d(TAG, "Checking read access for user ${user.username} against ${topicUrl(baseUrl, topic)}")
|
|
}
|
|
val url = topicUrlAuth(baseUrl, topic)
|
|
val request = requestBuilder(url, user).build()
|
|
client.newCall(request).execute().use { response ->
|
|
if (response.isSuccessful) {
|
|
return true
|
|
} else if (user == null && response.code == 404) {
|
|
return true // Special case: Anonymous login to old servers return 404 since /<topic>/auth doesn't exist
|
|
} else if (response.code == 401 || response.code == 403) { // See server/server.go
|
|
return false
|
|
}
|
|
throw Exception("Unexpected server response ${response.code}")
|
|
}
|
|
}
|
|
|
|
class UnauthorizedException(val user: User?) : Exception()
|
|
class EntityTooLargeException() : Exception()
|
|
|
|
companion object {
|
|
val USER_AGENT = "ntfy/${BuildConfig.VERSION_NAME} (${BuildConfig.FLAVOR}; Android ${Build.VERSION.RELEASE}; SDK ${Build.VERSION.SDK_INT})"
|
|
private const val TAG = "NtfyApiService"
|
|
|
|
// These constants have corresponding values in the server codebase!
|
|
const val CONTROL_TOPIC = "~control"
|
|
const val EVENT_MESSAGE = "message"
|
|
const val EVENT_KEEPALIVE = "keepalive"
|
|
const val EVENT_POLL_REQUEST = "poll_request"
|
|
|
|
fun requestBuilder(url: String, user: User?): Request.Builder {
|
|
val builder = Request.Builder()
|
|
.url(url)
|
|
.addHeader("User-Agent", USER_AGENT)
|
|
if (user != null) {
|
|
builder.addHeader("Authorization", Credentials.basic(user.username, user.password, UTF_8))
|
|
}
|
|
return builder
|
|
}
|
|
}
|
|
}
|