dev.forcetower.breaker.Orchestra.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of snowpiercer Show documentation
Breaks through the snow!
There is a newer version: 2.1.1
package dev.forcetower.breaker

import com.google.gson.GsonBuilder
import dev.forcetower.breaker.json.StringJsonPattern
import dev.forcetower.breaker.model.*
import dev.forcetower.breaker.operation.*
import dev.forcetower.breaker.result.Outcome
import dev.forcetower.breaker.service.TechNoAPI
import okhttp3.Credentials
import okhttp3.OkHttpClient
import retrofit2.Retrofit
import retrofit2.converter.gson.GsonConverterFactory
import java.util.concurrent.TimeUnit

class Orchestra(
    baseClient: OkHttpClient,
    agent: String
) {
    private var credentials: String? = null

    private val gson = GsonBuilder()
        .registerTypeAdapter(String::class.java, StringJsonPattern.STRING_TECH_DESERIALIZER)
        .create()

    private val client = baseClient
        .newBuilder()
        .addInterceptor { chain ->
            val url = chain.request().url
            val request = chain.request()
                .newBuilder()
                .apply {
                    val cred = credentials
                    if (cred != null) header("Authorization", cred)
                    val timestampedUrl = url.newBuilder()
                        .addQueryParameter("_", "${System.currentTimeMillis()}")
                        .build()
                    url(timestampedUrl)
                    addHeader("User-Agent", agent)
                    addHeader("X-Requested-With", "com.tecnotrends.sagres.mobile")
                    addHeader("Accept", "application/json, text/javascript, */*; q=0.01")
                    addHeader("Accept-Encoding", "gzip, deflate")
                    addHeader("Accept-Language", "en-US,en;q=0.9")
                }
                .build()
            return@addInterceptor chain.proceed(request)
        }.build()

    private val service = Retrofit.Builder()
        .baseUrl("http://academico2.uefs.br/Api/SagresApi/")
        .client(client)
        .addConverterFactory(GsonConverterFactory.create(gson))
        .build()
        .create(TechNoAPI::class.java)

    fun setAuthorization(authorization: Authorization) {
        credentials = Credentials.basic(authorization.username, authorization.password)
    }

    suspend fun login(): Outcome {
        return LoginOperation().execute(service)
    }

    suspend fun semesters(profileId: Long): Outcome> {
        return SemestersOperation(profileId).execute(service)
    }

    suspend fun grades(profileId: Long, semesterId: Long): Outcome> {
        return GradesOperation(profileId, semesterId).execute(service)
    }

    suspend fun messages(profileId: Long, until: String = ""): Outcome {
        return MessagesOperation(profileId, until).execute(service)
    }

    suspend fun lectures(classId: Long, limit: Int, offset: Int): Outcome> {
        return LectureOperation(classId, limit, offset).execute(service)
    }

    class Builder {
        private var client: OkHttpClient? = null
        private var agent: String? = null

        fun client(client: OkHttpClient?): Builder {
            this.client = client
            return this
        }

        fun userAgent(agent: String): Builder {
            this.agent = agent
            return this
        }

        private fun createClient(): OkHttpClient {
            return OkHttpClient.Builder()
                .callTimeout(2, TimeUnit.MINUTES)
                .readTimeout(2, TimeUnit.MINUTES)
                .writeTimeout(2, TimeUnit.MINUTES)
                .callTimeout(2, TimeUnit.MINUTES)
                .build()
        }

        fun build(): Orchestra {
            val client = this.client ?: createClient()
            val agent = this.agent ?: "Mozilla/5.0 (Linux; Android 10; MI 9 Build/QKQ1.190825.002; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/84.0.4147.125 Mobile Safari/537.36"
            return Orchestra(client, agent)
        }
    }
}