12 Commits

Author SHA1 Message Date
xds
12afd1f90e recurrents 2025-11-17 15:02:47 +03:00
xds
d0cae182b7 build 2025-10-31 19:28:59 +03:00
xds
0f02b53bc0 build 2025-10-31 18:46:16 +03:00
xds
b08ab909c8 build 2025-10-31 18:45:28 +03:00
xds
a65f46aff3 build 2025-10-31 18:33:11 +03:00
xds
036ad00795 build 2025-10-31 18:29:31 +03:00
xds
1c3605623e build 2025-10-31 18:24:12 +03:00
xds
aaa12fcb86 build 2025-10-31 17:53:39 +03:00
xds
cef82c483f build 2025-10-31 17:46:35 +03:00
xds
e83e3a2b65 build 2025-10-31 17:40:57 +03:00
xds
c68e6afb8a build 2025-10-31 17:38:48 +03:00
xds
10b7c730ad build 2025-10-31 17:33:49 +03:00
53 changed files with 1503 additions and 149 deletions

View File

@@ -1,36 +1,15 @@
# ---------- build stage ----------
FROM gradle:8.9.0-jdk17-alpine AS build
WORKDIR /app
# Копируем wrapper + его папку (важно, что это две разные сущности)
COPY --chown=gradle:gradle gradlew ./gradlew
COPY --chown=gradle:gradle gradle/ ./gradle/
# Копируем скрипты сборки и исходники
COPY --chown=gradle:gradle build.gradle.kts settings.gradle.kts ./
COPY --chown=gradle:gradle src ./src
# Делаем gradlew исполняемым
RUN chmod +x gradlew
# Подкачаем зависимости (кэшируется) и соберём jar
RUN ./gradlew --no-daemon dependencies
RUN ./gradlew --no-daemon clean bootJar
# ---------- run stage ----------
FROM eclipse-temurin:17-jre AS runtime
WORKDIR /app
# (Опционально) установим curl для HEALTHCHECK
USER root
RUN apt-get update && apt-get install -y --no-install-recommends curl && rm -rf /var/lib/apt/lists/*
RUN groupadd --system --gid 1001 app && useradd --system --gid app --uid 1001 --shell /bin/bash --create-home app
RUN mkdir -p /app/static && chown -R app:app /app
COPY build/libs/luminic-space-v2.jar /app/luminic-space-v2.jar
USER app
COPY --from=build /app/build/libs/*.jar /app/app.jar
ENV JAVA_TOOL_OPTIONS="-XX:+UseContainerSupport -XX:MaxRAMPercentage=75.0"
EXPOSE 8080
HEALTHCHECK --interval=20s --timeout=3s --retries=3 CMD curl -fsS http://localhost:8080/actuator/health || exit 1
ENTRYPOINT ["java","-jar","/app/app.jar"]
ENTRYPOINT ["java","-jar","/app/luminic-space-v2.jar"]

View File

@@ -29,13 +29,16 @@ configurations {
repositories {
mavenCentral()
maven { url = uri("https://jitpack.io") }
}
dependencies {
// Spring
implementation("org.springframework.boot:spring-boot-starter-cache")
implementation("org.springframework.boot:spring-boot-starter-security")
implementation ("org.springframework.boot:spring-boot-starter-actuator")
implementation("org.springframework.boot:spring-boot-starter-actuator")
implementation("org.springframework.boot:spring-boot-starter-validation")
implementation("org.springframework.boot:spring-boot-starter-actuator")
implementation("org.springdoc:springdoc-openapi-starter-webmvc-ui:2.8.13")
@@ -53,6 +56,12 @@ dependencies {
implementation("commons-logging:commons-logging:1.3.4")
implementation("com.squareup.okhttp3:okhttp:4.12.0")
implementation("com.fasterxml.jackson.module:jackson-module-kotlin:2.17.0")
implementation("org.springframework.boot:spring-boot-starter-actuator")
implementation("io.micrometer:micrometer-registry-prometheus")
implementation("com.fasterxml.jackson.module:jackson-module-kotlin")
implementation("org.jetbrains.kotlin:kotlin-reflect")
implementation("org.jetbrains.kotlinx:kotlinx-serialization-json:1.7.3")
@@ -67,10 +76,11 @@ dependencies {
implementation("io.micrometer:micrometer-registry-prometheus")
implementation("org.telegram:telegrambots:6.9.7.1")
implementation("org.telegram:telegrambots-spring-boot-starter:6.9.7.1")
// implementation("org.telegram:telegrambots:6.9.7.1")
// implementation("org.telegram:telegrambots-spring-boot-starter:6.9.7.1")
implementation("com.opencsv:opencsv:5.10")
implementation("io.github.kotlin-telegram-bot.kotlin-telegram-bot:telegram:6.3.0")
compileOnly("org.projectlombok:lombok")
annotationProcessor("org.projectlombok:lombok")

10
deploy.sh Executable file
View File

@@ -0,0 +1,10 @@
#!/bin/bash
./gradlew bootJar || exit 1
scp build/libs/luminic-space-v2.jar root@213.226.71.138:/root/luminic/space/back
ssh root@213.226.71.138 "
cd /root/luminic/space/back &&
docker compose up -d --build &&
docker restart back-app-1
"

View File

@@ -4,6 +4,9 @@ networks:
services:
app:
image: back-app
volumes:
- ./luminic-space-v2.jar:/app/luminic-space-v2.jar
build:
context: .
dockerfile: Dockerfile
@@ -11,7 +14,6 @@ services:
SPRING_DATASOURCE_URL: jdbc:postgresql://postgres:5432/luminic-space-db
SPRING_DATASOURCE_USERNAME: luminicspace
SPRING_DATASOURCE_PASSWORD: LS1q2w3e4r!
MANAGEMENT_ENDPOINTS_WEB_EXPOSURE_INCLUDE: health,info
ports:
- "8089:8089"
restart: unless-stopped

View File

@@ -1,8 +1,8 @@
package space.luminic.finance.api
import org.apache.commons.codec.digest.DigestUtils.sha256
import org.apache.commons.codec.digest.HmacUtils.hmacSha256
import kotlinx.serialization.json.Json
import kotlinx.serialization.json.JsonBuilder
import org.slf4j.LoggerFactory
import org.springframework.beans.factory.annotation.Value
import org.springframework.security.core.context.SecurityContextHolder
@@ -13,6 +13,7 @@ import space.luminic.finance.dtos.UserDTO.RegisterUserDTO
import space.luminic.finance.mappers.UserMapper.toDto
import space.luminic.finance.mappers.UserMapper.toTelegramMap
import space.luminic.finance.services.AuthService
import java.net.URLDecoder
import java.security.MessageDigest
import java.time.Instant
import javax.crypto.Mac
@@ -27,28 +28,83 @@ class AuthController(
private val logger = LoggerFactory.getLogger(javaClass)
fun verifyTelegramAuth(data: Map<String, String>, botToken: String): Boolean {
val hash = data["hash"] ?: return false
fun verifyTelegramAuth(
loginData: Map<String, String>? = null, // from login widget
webAppInitData: String? = null
): Boolean {
val dataCheckString = data
.filterKeys { it != "hash" }
.toSortedMap()
.map { "${it.key}=${it.value}" }
.joinToString("\n")
// --- LOGIN WIDGET CHECK ---
if (loginData != null) {
val hash = loginData["hash"]
if (hash != null) {
val dataCheckString = loginData
.filterKeys { it != "hash" }
.toSortedMap()
.map { "${it.key}=${it.value}" }
.joinToString("\n")
val secretKey = sha256(botToken)
val hmacHex = hmacSha256(secretKey, dataCheckString)
val secretKey = MessageDigest.getInstance("SHA-256")
.digest(botToken.toByteArray())
if (hmacHex != hash) return false
val hmac = Mac.getInstance("HmacSHA256").apply {
init(SecretKeySpec(secretKey, "HmacSHA256"))
}.doFinal(dataCheckString.toByteArray())
.joinToString("") { "%02x".format(it) }
val authDate = data["auth_date"]?.toLongOrNull() ?: return false
val now = Instant.now().epochSecond
val authDate = loginData["auth_date"]?.toLongOrNull() ?: return false
if (Instant.now().epochSecond - authDate > 3600) return false
// Опционально — запрет старых ответов (например, старше 1 часа)
val maxAgeSeconds = 3600
if (now - authDate > maxAgeSeconds) return false
if (hmac == hash) return true
}
}
return true
// --- WEBAPP CHECK ---
// --- WEBAPP CHECK ---
if (webAppInitData != null) {
// Разбираем query string корректно (учитывая '=' внутри значения)
val pairs: Map<String, String> = webAppInitData.split("&")
.mapNotNull { part ->
val idx = part.indexOf('=')
if (idx <= 0) return@mapNotNull null
val k = part.substring(0, idx)
val v = part.substring(idx + 1)
k to URLDecoder.decode(v, Charsets.UTF_8.name())
}.toMap()
val receivedHash = pairs["hash"] ?: return false
// Строка для подписи: все поля КРОМЕ hash, отсортированные по ключу, в формате key=value с \n
val dataCheckString = pairs
.filterKeys { it != "hash" }
.toSortedMap()
.entries
.joinToString("\n") { (k, v) -> "$k=$v" }
// ВАЖНО: secret_key = HMAC_SHA256(message=botToken, key="WebAppData")
val secretKeyBytes = Mac.getInstance("HmacSHA256").apply {
init(SecretKeySpec("WebAppData".toByteArray(Charsets.UTF_8), "HmacSHA256"))
}.doFinal(botToken.toByteArray(Charsets.UTF_8))
// hash = HMAC_SHA256(message=data_check_string, key=secret_key)
val calcHashHex = Mac.getInstance("HmacSHA256").apply {
init(SecretKeySpec(secretKeyBytes, "HmacSHA256"))
}.doFinal(dataCheckString.toByteArray(Charsets.UTF_8))
.joinToString("") { "%02x".format(it) }
// опциональная проверка свежести
val authDate = pairs["auth_date"]?.toLongOrNull() ?: return false
val now = Instant.now().epochSecond
val ttl = 6 * 3600L // например, 6 часов для WebApp
val skew = 300L // допускаем до 5 минут будущего/прошлого
val diff = now - authDate
if (diff > ttl || diff < -skew) return false
if (calcHashHex == receivedHash) return true
}
return false
return false
}
private fun sha256(input: String): ByteArray =
@@ -62,6 +118,7 @@ class AuthController(
return hashBytes.joinToString("") { "%02x".format(it) }
}
@GetMapping("/test")
fun test(): String {
val authentication = SecurityContextHolder.getContext().authentication
@@ -80,16 +137,45 @@ class AuthController(
return authService.register(request.username, request.password, request.firstName).toDto()
}
private val json = Json { ignoreUnknownKeys = true }
@PostMapping("/tg-login")
fun tgLogin(@RequestBody tgUser: UserDTO.TelegramAuthDTO): String {
val ok = verifyTelegramAuth(tgUser.toTelegramMap(), botToken)
if (!ok) throw IllegalArgumentException("Invalid Telegram login")
return authService.tgAuth(tgUser)
fun tgLogin(@RequestBody tgUser: UserDTO.TelegramAuthDTO): Map<String, String> {
// println(tgUser.hash)
// println(botToken)
if (tgUser.initData == null) {
if (verifyTelegramAuth(
loginData = tgUser.toTelegramMap(),
)
) {
return mapOf("token" to authService.tgAuth(tgUser))
} else throw IllegalArgumentException("Invalid Telegram login")
} else {
if (verifyTelegramAuth(webAppInitData = tgUser.initData)) {
val params = tgUser.initData.split("&").associate {
val (k, v) = it.split("=", limit = 2)
k to URLDecoder.decode(v, "UTF-8")
}
val userJson = params["user"] ?: error("No user data")
val jsonUser = json.decodeFromString<UserDTO.TelegramUserData>(userJson)
val newUser = UserDTO.TelegramAuthDTO(
jsonUser.id,
jsonUser.first_name,
jsonUser.last_name,
jsonUser.username,
jsonUser.photo_url,
null,
hash = tgUser.hash,
initData = null,
)
return mapOf("token" to authService.tgAuth(newUser))
} else throw IllegalArgumentException("Invalid Telegram login")
}
}
@GetMapping("/me")
fun getMe(): UserDTO {
logger.info("Get Me")

View File

@@ -27,7 +27,7 @@ class SpaceController(
@GetMapping("/{spaceId}")
fun getSpace(@PathVariable spaceId: Int): SpaceDTO {
return spaceService.getSpace(spaceId).toDto()
return spaceService.getSpace(spaceId, null).toDto()
}
@PostMapping

View File

@@ -21,7 +21,7 @@ class BearerTokenFilter(
private val publicMatchers = listOf(
AntPathRequestMatcher("/auth/login", "POST"),
AntPathRequestMatcher("/auth/register", "POST"),
AntPathRequestMatcher("/auth/tgLogin", "POST"),
AntPathRequestMatcher("/auth/tg-login", "POST"),
AntPathRequestMatcher("/actuator/**"),
AntPathRequestMatcher("/static/**"),
AntPathRequestMatcher("/wishlistexternal/**"),

View File

@@ -31,7 +31,7 @@ class SecurityConfig(
.logout { it.disable() }
.authorizeHttpRequests {
it.requestMatchers(HttpMethod.POST, "/auth/login", "/auth/register", "/auth/tgLogin").permitAll()
it.requestMatchers(HttpMethod.POST, "/auth/login", "/auth/register", "/auth/tg-login").permitAll()
it.requestMatchers("/actuator/**", "/static/**").permitAll()
it.requestMatchers("/wishlistexternal/**").permitAll()
it.requestMatchers("/swagger-ui/**", "/swagger-ui.html", "/v3/api-docs/**").permitAll()
@@ -50,7 +50,7 @@ class SecurityConfig(
@Bean
fun corsConfigurationSource(): CorsConfigurationSource {
val cors = CorsConfiguration().apply {
allowedOrigins = listOf("https://luminic.space", "http://localhost:5173")
allowedOrigins = listOf("https://app.luminic.space", "http://localhost:5173")
allowedMethods = listOf("GET", "POST", "PUT", "DELETE", "OPTIONS", "PATCH")
allowedHeaders = listOf("*")
allowCredentials = true

View File

@@ -11,7 +11,7 @@ data class TransactionDTO(
var parentId: Int? = null,
val type: TransactionType = TransactionType.EXPENSE,
val kind: TransactionKind = TransactionKind.INSTANT,
val category: CategoryDTO,
val category: CategoryDTO? = null,
val comment: String,
val amount: BigDecimal,
val fees: BigDecimal = BigDecimal.ZERO,
@@ -23,17 +23,18 @@ data class TransactionDTO(
data class CreateTransactionDTO(
val type: TransactionType = TransactionType.EXPENSE,
val kind: TransactionKind = TransactionKind.INSTANT,
val categoryId: Int,
val categoryId: Int? = null,
val comment: String,
val amount: BigDecimal,
val fees: BigDecimal = BigDecimal.ZERO,
val date: LocalDate,
val recurrentId: Int? = null
)
data class UpdateTransactionDTO(
val type: TransactionType = TransactionType.EXPENSE,
val kind: TransactionKind = TransactionKind.INSTANT,
val categoryId: Int,
val categoryId: Int? = null,
val comment: String,
val amount: BigDecimal,
val fees: BigDecimal = BigDecimal.ZERO,

View File

@@ -1,8 +1,8 @@
package space.luminic.finance.dtos
import java.util.Date
import kotlinx.serialization.Serializable
data class UserDTO (
data class UserDTO(
var id: Int,
val username: String,
var firstName: String,
@@ -13,12 +13,12 @@ data class UserDTO (
) {
data class AuthUserDTO (
data class AuthUserDTO(
var username: String,
var password: String,
)
data class RegisterUserDTO (
data class RegisterUserDTO(
var username: String,
var firstName: String,
var password: String,
@@ -26,15 +26,27 @@ data class UserDTO (
)
data class TelegramAuthDTO(
val id: Long,
val id: Long?,
val first_name: String?,
val last_name: String?,
val username: String?,
val photo_url: String?,
val auth_date: Long,
val hash: String
val auth_date: Long?,
val hash: String,
val initData: String?,
)
@Serializable
class TelegramUserData(
val id: Long,
val first_name: String,
val last_name: String? = null,
val username: String? = null,
val photo_url: String? = null,
)
}

View File

@@ -11,7 +11,7 @@ object TransactionMapper {
parentId = this.parent?.id,
type = this.type,
kind = this.kind,
category = this.category.toDto(),
category = this.category?.toDto(),
comment = this.comment,
amount = this.amount,
fees = this.fees,

View File

@@ -0,0 +1,16 @@
package space.luminic.finance.models
data class State(
val user: User,
val state: StateCode = StateCode.AWAIT_SPACE_SELECT,
val data: Map<String, String> = mapOf()
) {
enum class StateCode {
AWAIT_SPACE_SELECT,
SPACE_SELECTED,
AWAIT_TRANSACTION,
}
}

View File

@@ -14,7 +14,7 @@ data class Transaction(
var parent: Transaction? = null,
val type: TransactionType = TransactionType.EXPENSE,
val kind: TransactionKind = TransactionKind.INSTANT,
val category: Category,
val category: Category? = null,
val comment: String,
val amount: BigDecimal,
val fees: BigDecimal = BigDecimal.ZERO,
@@ -25,6 +25,9 @@ data class Transaction(
@CreatedDate var createdAt: Instant? = null,
@LastModifiedBy var updatedBy: User? = null,
@LastModifiedDate var updatedAt: Instant? = null,
val tgChatId: Long? = null,
val tgMessageId: Long? = null,
val recurrentId: Int? = null,
) {

View File

@@ -0,0 +1,9 @@
package space.luminic.finance.repos
import space.luminic.finance.models.State
interface BotRepo {
fun getState(tgUserId: Long): State?
fun setState(userId: Int, stateCode: State.StateCode, stateData: Map<String, String>)
fun clearState(userId: Int)
}

View File

@@ -0,0 +1,114 @@
package space.luminic.finance.repos
import org.springframework.jdbc.core.ResultSetExtractor
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate
import org.springframework.stereotype.Repository
import space.luminic.finance.models.State
import space.luminic.finance.models.User
@Repository
class BotRepoImpl(
private val jdbcTemplate: NamedParameterJdbcTemplate,
) : BotRepo {
override fun getState(tgUserId: Long): State? {
val sql = """
select
bs.user_id as bs_user_id,
u.username as u_username,
u.first_name as u_first_name,
bs.state_code as bs_state_code,
bsd.data_code as bs_data_code,
bsd.data_value as bs_data_value
from finance.bot_states bs
join finance.users u on u.id = bs.user_id
left join finance.bot_states_data bsd on bsd.user_id = bs.user_id
where u.tg_id = :user_id
""".trimIndent()
val params = mapOf("user_id" to tgUserId)
return jdbcTemplate.query(sql, params, ResultSetExtractor { rs ->
var user: User? = null
var stateCode: State.StateCode? = null
val data = mutableMapOf<String, String>()
while (rs.next()) {
if (user == null) {
user = User(
id = rs.getInt("bs_user_id"),
username = rs.getString("u_username"),
firstName = rs.getString("u_first_name")
)
stateCode = rs.getString("bs_state_code")?.let { raw ->
runCatching { State.StateCode.valueOf(raw) }
.getOrElse { State.StateCode.AWAIT_SPACE_SELECT }
}
}
val code = rs.getString("bs_data_code")
val value = rs.getString("bs_data_value")
if (code != null && value != null) {
data[code] = value
}
}
user?.let {
State(
user = it,
state = stateCode ?: State.StateCode.AWAIT_SPACE_SELECT,
data = data.toMap()
)
}
})
}
override fun setState(
userId: Int,
stateCode: State.StateCode,
stateData: Map<String, String>
) {
// 1) UPSERT state (по user_id)
val upsertStateSql = """
INSERT INTO finance.bot_states (user_id, state_code)
VALUES (:user_id, :state_code)
ON CONFLICT (user_id) DO UPDATE
SET state_code = EXCLUDED.state_code
""".trimIndent()
jdbcTemplate.update(
upsertStateSql,
mapOf(
"user_id" to userId,
// если в БД enum — чаще всего ок передать name(); если колонка TEXT/VARCHAR — тоже ок
"state_code" to stateCode.name
)
)
// 2) Обновление data: вариант A — апсерты (рекомендуется)
if (stateData.isNotEmpty()) {
val upsertDataSql = """
INSERT INTO finance.bot_states_data (user_id, data_code, data_value)
VALUES (:user_id, :data_code, :data_value)
ON CONFLICT (user_id, data_code) DO UPDATE
SET data_value = EXCLUDED.data_value
""".trimIndent()
val batch = stateData.map { (code, value) ->
mapOf(
"user_id" to userId,
"data_code" to code,
"data_value" to value
)
}.toTypedArray()
jdbcTemplate.batchUpdate(upsertDataSql, batch)
}
// Если тебе принципиально "перезаписывать" состояние данных (вариант B):
// np.update("DELETE FROM finance.bot_states_data WHERE user_id = :user_id", mapOf("user_id" to userId))
// затем обычный INSERT batch без ON CONFLICT.
}
override fun clearState(userId: Int) {
TODO("Not yet implemented")
}
}

View File

@@ -24,7 +24,7 @@ class CategoryRepoImpl(
}
override fun findBySpaceId(spaceId: Int): List<Category> {
val query = "select * from finance.categories where space_id = :space_id order by id"
val query = "select * from finance.categories where space_id = :space_id order by name"
val params = mapOf("space_id" to spaceId)
return jdbcTemplate.query(query, params, categoryRowMapper())
}

View File

@@ -5,7 +5,9 @@ import space.luminic.finance.models.RecurrentOperation
interface RecurrentOperationRepo {
fun findAllBySpaceId(spaceId: Int): List<RecurrentOperation>
fun findBySpaceIdAndId(spaceId: Int, id: Int): RecurrentOperation?
fun findByDate( date: Int): List<RecurrentOperation>
fun create(operation: RecurrentOperation, createdById: Int): Int
fun update(operation: RecurrentOperation, updatedById: Int)
fun delete(id: Int)
fun findRecurrentsToCreate(spaceId: Int): List<RecurrentOperation>
}

View File

@@ -115,6 +115,40 @@ class RecurrentOperationRepoImpl(
return jdbcTemplate.query(sql, params, operationRowMapper()).firstOrNull()
}
override fun findByDate(
date: Int
): List<RecurrentOperation> {
val sql = """
select
ro.id as r_id,
ro.space_id AS r_space_id,
s.name AS s_name,
s.owner_id as s_owner_id,
su.username as su_username,
su.first_name AS su_first_name,
ro.category_id as r_category_id,
c.type AS c_type,
c.name AS c_name,
c.description AS c_description,
c.icon AS c_icon,
ro.name AS r_name,
ro.amount AS r_amount,
ro.date AS r_date,
ro.created_by_id as r_created_by_id,
r_created_by.username as r_created_by_username,
r_created_by.first_name as r_created_by_first_name,
ro.created_at as r_created_at
from finance.recurrent_operations ro
join finance.spaces s on ro.space_id = s.id
join finance.users su on s.owner_id = su.id
join finance.categories c on ro.category_id = c.id
join finance.users r_created_by on ro.created_by_id = r_created_by.id
where ro.date = :date
""".trimIndent()
val params = mapOf( "date" to date)
return jdbcTemplate.query(sql, params, operationRowMapper())
}
override fun create(operation: RecurrentOperation, createdById: Int): Int {
val sql = """
insert into finance.recurrent_operations (
@@ -175,4 +209,13 @@ class RecurrentOperationRepoImpl(
val params = mapOf("id" to id)
jdbcTemplate.update(sql, params)
}
override fun findRecurrentsToCreate(spaceId: Int): List<RecurrentOperation> {
val sql = """
select * from finance.transactions where space_id = :spaceId and t.date >
""".trimIndent()
TODO("Not ready")
}
}

View File

@@ -64,8 +64,8 @@ class TokenRepoImpl(
update finance.tokens set status = :status where token = :token
""".trimIndent()
val params = mapOf(
"token" to token,
"status" to token.status
"token" to token.token,
"status" to token.status.name
)
jdbcTemplate.update(sql, params)
return token

View File

@@ -6,7 +6,11 @@ interface TransactionRepo {
fun findAllBySpaceId(spaceId: Int): List<Transaction>
fun findBySpaceIdAndId(spaceId: Int, id: Int): Transaction?
fun create(transaction: Transaction, userId: Int): Int
fun createBatch(transactions: List<Transaction>, userId: Int)
fun update(transaction: Transaction): Int
fun delete(transactionId: Int)
fun deleteByRecurrentId(spaceId: Int, recurrentId: Int)
fun setCategory(txId:Int, categoryId: Int)
}

View File

@@ -13,21 +13,26 @@ class TransactionRepoImpl(
) : TransactionRepo {
private fun transactionRowMapper() = RowMapper { rs, _ ->
val category = if (rs.getString("c_id") == null) null else Category(
id = rs.getInt("c_id"),
type = Category.CategoryType.valueOf(rs.getString("c_type")),
name = rs.getString(("c_name")),
description = rs.getString(("c_description")),
icon = rs.getString("c_icon"),
isDeleted = rs.getBoolean(("c_is_deleted")),
createdAt = rs.getTimestamp("c_created_at").toInstant(),
updatedAt = rs.getTimestamp("c_updated_at")?.toInstant(),
)
val parent = if (rs.getInt("t_parent_id") != 0) findBySpaceIdAndId(
spaceId = rs.getInt("t_space_id"),
rs.getInt("t_parent_id")
) else null
Transaction(
id = rs.getInt("t_id"),
parent = findBySpaceIdAndId(spaceId = rs.getInt("t_space_id"), rs.getInt("t_parent_id")),
parent = parent,
type = Transaction.TransactionType.valueOf(rs.getString("t_type")),
kind = Transaction.TransactionKind.valueOf(rs.getString("t_kind")),
category = Category(
id = rs.getInt("c_id"),
type = Category.CategoryType.valueOf(rs.getString("c_type")),
name = rs.getString(("c_name")),
description = rs.getString(("c_description")),
icon = rs.getString("c_icon"),
isDeleted = rs.getBoolean(("c_is_deleted")),
createdAt = rs.getTimestamp("c_created_at").toInstant(),
updatedAt = rs.getTimestamp("c_updated_at").toInstant(),
),
category = category,
comment = rs.getString("t_comment"),
amount = rs.getBigDecimal("t_amount"),
fees = rs.getBigDecimal("t_fees"),
@@ -41,6 +46,9 @@ class TransactionRepoImpl(
),
createdAt = rs.getTimestamp("t_created_at").toInstant(),
updatedAt = rs.getTimestamp("t_updated_at").toInstant(),
tgChatId = rs.getLong("tg_chat_id"),
tgMessageId = rs.getLong("tg_message_id"),
recurrentId = rs.getInt("t_recurrent_id"),
)
}
@@ -70,13 +78,15 @@ class TransactionRepoImpl(
c.updated_at AS c_updated_at,
u.id AS u_id,
u.username AS u_username,
u.first_name AS u_first_name
u.first_name AS u_first_name,
t.tg_chat_id AS tg_chat_id,
t.tg_message_id AS tg_message_id,
t.recurrent_id AS t_recurrent_id
FROM finance.transactions t
JOIN finance.categories c ON t.category_id = c.id
LEFT JOIN finance.categories c ON t.category_id = c.id
JOIN finance.users u ON u.id = t.created_by_id
WHERE t.space_id = :spaceId and t.is_deleted = false
ORDER BY t.date
ORDER BY t.date, t.id
""".trimIndent()
val params = mapOf(
"spaceId" to spaceId,
@@ -99,6 +109,8 @@ class TransactionRepoImpl(
t.is_done AS t_is_done,
t.created_at AS t_created_at,
t.updated_at AS t_updated_at,
t.tg_chat_id AS tg_chat_id,
t.tg_message_id AS tg_message_id,
c.id AS c_id,
c.type AS c_type,
c.name AS c_name,
@@ -109,9 +121,10 @@ class TransactionRepoImpl(
c.updated_at AS c_updated_at,
u.id AS u_id,
u.username AS u_username,
u.first_name AS u_first_name
u.first_name AS u_first_name,
t.recurrent_id AS t_recurrent_id
FROM finance.transactions t
JOIN finance.categories c ON t.category_id = c.id
LEFT JOIN finance.categories c ON t.category_id = c.id
JOIN finance.users u ON u.id = t.created_by_id
WHERE t.space_id = :spaceId and t.id = :id and t.is_deleted = false""".trimMargin()
val params = mapOf(
@@ -123,7 +136,7 @@ class TransactionRepoImpl(
override fun create(transaction: Transaction, userId: Int): Int {
val sql = """
INSERT INTO finance.transactions (space_id, parent_id, type, kind, category_id, comment, amount, fees, date, is_deleted, is_done, created_by_id) VALUES (
INSERT INTO finance.transactions (space_id, parent_id, type, kind, category_id, comment, amount, fees, date, is_deleted, is_done, created_by_id, tg_chat_id, tg_message_id, recurrent_id) VALUES (
:spaceId,
:parentId,
:type,
@@ -135,7 +148,10 @@ class TransactionRepoImpl(
:date,
:is_deleted,
:is_done,
:createdById)
:createdById,
:tgChatId,
:tgMessageId,
:recurrentId)
returning id
""".trimIndent()
val params = mapOf(
@@ -143,20 +159,57 @@ class TransactionRepoImpl(
"parentId" to transaction.parent?.id,
"type" to transaction.type.name,
"kind" to transaction.kind.name,
"categoryId" to transaction.category.id,
"categoryId" to transaction.category?.id,
"comment" to transaction.comment,
"amount" to transaction.amount,
"fees" to transaction.fees,
"date" to transaction.date,
"is_deleted" to transaction.isDeleted,
"is_done" to transaction.isDone,
"createdById" to userId
"createdById" to userId,
"tgChatId" to transaction.tgChatId,
"tgMessageId" to transaction.tgMessageId,
"recurrentId" to transaction.recurrentId,
)
val createdTxId = jdbcTemplate.queryForObject(sql, params, Int::class.java)
transaction.id = createdTxId
return createdTxId!!
}
override fun createBatch(transactions: List<Transaction>, userId: Int) {
val sql = """
INSERT INTO finance.transactions (
space_id, parent_id, type, kind, category_id, comment, amount, fees, date,
is_deleted, is_done, created_by_id, tg_chat_id, tg_message_id, recurrent_id
) VALUES (
:spaceId, :parentId, :type, :kind, :categoryId, :comment, :amount, :fees, :date,
:is_deleted, :is_done, :createdById, :tgChatId, :tgMessageId, :recurrentId
)
""".trimIndent()
val batchValues = transactions.map { transaction ->
mapOf(
"spaceId" to transaction.space!!.id,
"parentId" to transaction.parent?.id,
"type" to transaction.type.name,
"kind" to transaction.kind.name,
"categoryId" to transaction.category?.id,
"comment" to transaction.comment,
"amount" to transaction.amount,
"fees" to transaction.fees,
"date" to transaction.date,
"is_deleted" to transaction.isDeleted,
"is_done" to transaction.isDone,
"createdById" to userId,
"tgChatId" to transaction.tgChatId,
"tgMessageId" to transaction.tgMessageId,
"recurrentId" to transaction.recurrentId
)
}.toTypedArray()
jdbcTemplate.batchUpdate(sql, batchValues)
}
override fun update(transaction: Transaction): Int {
// val type: TransactionType = TransactionType.EXPENSE,
// val kind: TransactionKind = TransactionKind.INSTANT,
@@ -183,7 +236,7 @@ class TransactionRepoImpl(
"id" to transaction.id,
"type" to transaction.type.name,
"kind" to transaction.kind.name,
"categoryId" to transaction.category.id,
"categoryId" to transaction.category?.id,
"comment" to transaction.comment,
"amount" to transaction.amount,
"fees" to transaction.fees,
@@ -203,4 +256,27 @@ class TransactionRepoImpl(
)
jdbcTemplate.update(sql, params)
}
override fun deleteByRecurrentId(spaceId: Int, recurrentId: Int) {
val sql = """
update finance.transactions set is_deleted = true where recurrent_id = :recurrentId
""".trimIndent()
val params = mapOf(
"recurrentId" to recurrentId,
)
jdbcTemplate.update(sql, params)
}
override fun setCategory(txId: Int, categoryId: Int) {
val sql = """
UPDATE finance.transactions
SET category_id = :categoryId
where id = :txId
""".trimIndent()
val params = mapOf(
"categoryId" to categoryId,
"txId" to txId,
)
jdbcTemplate.update(sql, params)
}
}

View File

@@ -53,12 +53,12 @@ class UserRepoImpl(
select * from finance.users u where tg_id = :tgId
""".trimIndent()
val params = mapOf("tgId" to tgId)
return jdbcTemplate.queryForObject(sql, params, userRowMapper())
return jdbcTemplate.query(sql, params, userRowMapper()).firstOrNull()
}
override fun create(user: User): User {
val sql =
"insert into finance.users(username, first_name, tg_id, tg_user_name, photo_url, password, is_active, reg_date) values (:username, :firstname, :tg_id, :tg_user_name, :photo_url :password, :isActive, :regDate) returning ID"
"insert into finance.users(username, first_name, tg_id, tg_user_name, photo_url, password, is_active, reg_date) values (:username, :firstname, :tg_id, :tg_user_name, :photo_url, :password, :isActive, :regDate) returning ID"
val params = mapOf(
"username" to user.username,
"firstname" to user.firstName,

View File

@@ -65,7 +65,7 @@ class AuthService(
fun tgAuth(tgUser: UserDTO.TelegramAuthDTO): String {
val user: User = try {
tgLogin(tgUser.id)
tgLogin(tgUser.id!!)
} catch (e: NotFoundException) {
registerTg(tgUser)
}

View File

@@ -10,4 +10,6 @@ interface RecurrentOperationService {
fun create(spaceId: Int, operation: RecurrentOperationDTO.CreateRecurrentOperationDTO): Int
fun update(spaceId: Int, operationId: Int, operation: RecurrentOperationDTO.UpdateRecurrentOperationDTO)
fun delete(spaceId: Int, id: Int)
fun createRecurrentTransactions()
}

View File

@@ -1,19 +1,33 @@
package space.luminic.finance.services
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.SupervisorJob
import kotlinx.coroutines.launch
import org.slf4j.LoggerFactory
import org.springframework.stereotype.Service
import space.luminic.finance.dtos.RecurrentOperationDTO
import space.luminic.finance.models.Category
import space.luminic.finance.models.NotFoundException
import space.luminic.finance.models.RecurrentOperation
import space.luminic.finance.models.Transaction
import space.luminic.finance.repos.RecurrentOperationRepo
import space.luminic.finance.repos.SpaceRepo
import space.luminic.finance.repos.TransactionRepo
import java.time.LocalDate
@Service
class RecurrentOperationServiceImpl(
private val authService: AuthService,
private val spaceRepo: SpaceRepo,
private val recurrentOperationRepo: RecurrentOperationRepo,
private val categoryService: CategoryService
): RecurrentOperationService {
private val categoryService: CategoryService,
private val transactionService: TransactionService,
private val transactionRepo: TransactionRepo
) : RecurrentOperationService {
private val logger = LoggerFactory.getLogger(this.javaClass)
private val serviceScope = CoroutineScope(Dispatchers.Default + SupervisorJob())
override fun findBySpaceId(spaceId: Int): List<RecurrentOperation> {
val userId = authService.getSecurityUserId()
spaceRepo.findSpaceById(spaceId, userId)
@@ -26,12 +40,14 @@ class RecurrentOperationServiceImpl(
): RecurrentOperation {
val userId = authService.getSecurityUserId()
spaceRepo.findSpaceById(spaceId, userId)
return recurrentOperationRepo.findBySpaceIdAndId(spaceId, id) ?: throw NotFoundException("Cannot find recurrent operation with id ${id}")
return recurrentOperationRepo.findBySpaceIdAndId(spaceId, id)
?: throw NotFoundException("Cannot find recurrent operation with id ${id}")
}
override fun create(spaceId: Int, operation: RecurrentOperationDTO.CreateRecurrentOperationDTO): Int {
val userId = authService.getSecurityUserId()
val space = spaceRepo.findSpaceById(spaceId, userId) ?: throw NotFoundException("Cannot find space with id ${spaceId}")
val space =
spaceRepo.findSpaceById(spaceId, userId) ?: throw NotFoundException("Cannot find space with id ${spaceId}")
val category = categoryService.getCategory(spaceId, operation.categoryId)
val creatingOperation = RecurrentOperation(
space = space,
@@ -40,14 +56,42 @@ class RecurrentOperationServiceImpl(
amount = operation.amount,
date = operation.date
)
return recurrentOperationRepo.create(creatingOperation, userId)
val createdRecurrentId = recurrentOperationRepo.create(creatingOperation, userId)
val transactionsToCreate = mutableListOf<Transaction>()
serviceScope.launch {
runCatching {
val date = LocalDate.now()
for (i in 1..12) {
transactionsToCreate.add(
Transaction(
space = space,
type = if (category.type == Category.CategoryType.EXPENSE) Transaction.TransactionType.EXPENSE else Transaction.TransactionType.INCOME,
kind = Transaction.TransactionKind.PLANNING,
category = category,
comment = creatingOperation.name,
amount = creatingOperation.amount,
date = date.plusMonths(i.toLong()),
recurrentId = createdRecurrentId
)
)
}
transactionRepo.createBatch(transactionsToCreate, userId)
// transactionService.batchCreate(spaceId, transactionsToCreate, userId)
}.onFailure {
logger.error("Error creating recurring operation", it)
}
}
return createdRecurrentId
}
override fun update(spaceId: Int, operationId: Int, operation: RecurrentOperationDTO.UpdateRecurrentOperationDTO) {
val userId = authService.getSecurityUserId()
spaceRepo.findSpaceById(spaceId, userId)
val newCategory = categoryService.getCategory(spaceId, operation.categoryId)
val existingOperation = recurrentOperationRepo.findBySpaceIdAndId(spaceId,operationId ) ?: throw NotFoundException("Cannot find operation with id $operationId")
val existingOperation = recurrentOperationRepo.findBySpaceIdAndId(spaceId, operationId)
?: throw NotFoundException("Cannot find operation with id $operationId")
val updatedOperation = existingOperation.copy(
category = newCategory,
name = operation.name,
@@ -63,4 +107,25 @@ class RecurrentOperationServiceImpl(
spaceRepo.findSpaceById(spaceId, userId)
recurrentOperationRepo.delete(id)
}
override fun createRecurrentTransactions() {
val today = LocalDate.now()
val recurrents = recurrentOperationRepo.findByDate(today.dayOfMonth)
recurrents.forEach {
transactionRepo.create(
Transaction(
space = it.space,
type = if (it.category.type == Category.CategoryType.EXPENSE) Transaction.TransactionType.EXPENSE else Transaction.TransactionType.INCOME,
kind = Transaction.TransactionKind.PLANNING,
category = it.category,
comment = it.name,
amount = it.amount,
date = today.plusMonths(13),
recurrentId = it.id
), it.createdBy?.id!!
)
}
}
}

View File

@@ -0,0 +1,20 @@
package space.luminic.finance.services
import org.slf4j.LoggerFactory
import org.springframework.scheduling.annotation.EnableScheduling
import org.springframework.scheduling.annotation.Scheduled
import org.springframework.stereotype.Service
@EnableScheduling
@Service
class Scheduler(
private val recurrentOperationService: RecurrentOperationService
) {
private val log = LoggerFactory.getLogger(Scheduler::class.java)
@Scheduled(cron = "0 0 3 * * *")
fun createRecurrentAfter13Month() {
log.info("Creating recurrent after 13 month")
recurrentOperationService.createRecurrentTransactions()
}
}

View File

@@ -7,7 +7,7 @@ interface SpaceService {
fun checkSpace(spaceId: Int): Space
fun getSpaces(): List<Space>
fun getSpace(id: Int): Space
fun getSpace(id: Int, userId: Int?): Space
fun createSpace(space: SpaceDTO.CreateSpaceDTO): Int
fun updateSpace(spaceId: Int, space: SpaceDTO.UpdateSpaceDTO): Int
fun deleteSpace(spaceId: Int)

View File

@@ -1,8 +1,5 @@
package space.luminic.finance.services
import org.springframework.cache.annotation.CacheEvict
import org.springframework.cache.annotation.Cacheable
import org.springframework.jdbc.core.JdbcTemplate
import org.springframework.stereotype.Service
import org.springframework.transaction.annotation.Transactional
import space.luminic.finance.dtos.SpaceDTO
@@ -17,7 +14,7 @@ class SpaceServiceImpl(
private val categoryService: CategoryService
) : SpaceService {
override fun checkSpace(spaceId: Int): Space {
return getSpace(spaceId)
return getSpace(spaceId, null)
}
// @Cacheable(cacheNames = ["spaces"])
@@ -27,8 +24,9 @@ class SpaceServiceImpl(
return spaces
}
override fun getSpace(id: Int): Space {
val user = authService.getSecurityUserId()
override fun getSpace(id: Int, userId: Int?): Space {
val user = userId ?: authService.getSecurityUserId()
val space = spaceRepo.findSpaceById(id, user) ?: throw NotFoundException("Space with id $id not found")
return space
@@ -56,7 +54,7 @@ class SpaceServiceImpl(
space: SpaceDTO.UpdateSpaceDTO
): Int {
val userId = authService.getSecurityUserId()
val existingSpace = getSpace(spaceId)
val existingSpace = getSpace(spaceId, null)
val updatedSpace = Space(
id = existingSpace.id,
name = space.name,

View File

@@ -11,9 +11,17 @@ interface TransactionService {
val dateTo: LocalDate? = null,
)
fun getTransactions(spaceId: Int, filter: TransactionsFilter, sortBy: String, sortDirection: String): List<Transaction>
fun getTransaction(spaceId: Int, transactionId: Int): Transaction
fun createTransaction(spaceId: Int, transaction: TransactionDTO.CreateTransactionDTO): Int
fun updateTransaction(spaceId: Int, transactionId: Int, transaction: TransactionDTO.UpdateTransactionDTO): Int
fun getTransactions(
spaceId: Int,
filter: TransactionsFilter,
sortBy: String,
sortDirection: String
): List<Transaction>
fun getTransaction(spaceId: Int, transactionId: Int): Transaction
fun createTransaction(spaceId: Int, transaction: TransactionDTO.CreateTransactionDTO): Int
fun batchCreate(spaceId: Int, transactions: List<TransactionDTO.CreateTransactionDTO>, createdById: Int?)
fun updateTransaction(spaceId: Int, transactionId: Int, transaction: TransactionDTO.UpdateTransactionDTO): Int
fun deleteTransaction(spaceId: Int, transactionId: Int)
fun deleteByRecurrentId(spaceId: Int, recurrentId: Int)
}

View File

@@ -5,6 +5,7 @@ import space.luminic.finance.dtos.TransactionDTO
import space.luminic.finance.models.NotFoundException
import space.luminic.finance.models.Transaction
import space.luminic.finance.repos.TransactionRepo
import space.luminic.finance.services.gpt.CategorizeService
@Service
class TransactionServiceImpl(
@@ -12,6 +13,7 @@ class TransactionServiceImpl(
private val categoryService: CategoryService,
private val transactionRepo: TransactionRepo,
private val authService: AuthService,
private val categorizeService: CategorizeService,
) : TransactionService {
override fun getTransactions(
spaceId: Int,
@@ -19,14 +21,15 @@ class TransactionServiceImpl(
sortBy: String,
sortDirection: String
): List<Transaction> {
return transactionRepo.findAllBySpaceId(spaceId)
val transactions = transactionRepo.findAllBySpaceId(spaceId)
return transactions
}
override fun getTransaction(
spaceId: Int,
transactionId: Int
): Transaction {
spaceService.getSpace(spaceId)
spaceService.getSpace(spaceId, null)
return transactionRepo.findBySpaceIdAndId(spaceId, transactionId)
?: throw NotFoundException("Transaction with id $transactionId not found")
}
@@ -36,8 +39,9 @@ class TransactionServiceImpl(
transaction: TransactionDTO.CreateTransactionDTO
): Int {
val userId = authService.getSecurityUserId()
val space = spaceService.getSpace(spaceId)
val category = categoryService.getCategory(spaceId, transaction.categoryId)
val space = spaceService.getSpace(spaceId, null)
val category = transaction.categoryId?.let { categoryService.getCategory(spaceId, it) }
val transaction = Transaction(
space = space,
type = transaction.type,
@@ -47,26 +51,43 @@ class TransactionServiceImpl(
amount = transaction.amount,
fees = transaction.fees,
date = transaction.date,
recurrentId = transaction.recurrentId,
)
return transactionRepo.create(transaction, userId)
}
override fun batchCreate(spaceId: Int, transactions: List<TransactionDTO.CreateTransactionDTO>, createdById: Int?) {
val userId = createdById ?: authService.getSecurityUserId()
val space = spaceService.getSpace(spaceId, userId)
val transactionsToCreate = mutableListOf<Transaction>()
transactions.forEach { transaction ->
val category = transaction.categoryId?.let { categoryService.getCategory(spaceId, it) }
transactionsToCreate.add(
Transaction(
space = space,
type = transaction.type,
kind = transaction.kind,
category = category,
comment = transaction.comment,
amount = transaction.amount,
fees = transaction.fees,
date = transaction.date,
recurrentId = transaction.recurrentId,
)
)
}
transactionRepo.createBatch(transactionsToCreate, userId)
}
override fun updateTransaction(
spaceId: Int,
transactionId: Int,
transaction: TransactionDTO.UpdateTransactionDTO
): Int {
val space = spaceService.getSpace(spaceId)
val space = spaceService.getSpace(spaceId, null)
val existingTransaction = getTransaction(space.id!!, transactionId)
val newCategory = categoryService.getCategory(spaceId, transaction.categoryId)
// val id: Int,
// val type: TransactionType = TransactionType.EXPENSE,
// val kind: TransactionKind = TransactionKind.INSTANT,
// val category: Int,
// val comment: String,
// val amount: BigDecimal,
// val fees: BigDecimal = BigDecimal.ZERO,
// val date: Instant
val newCategory = transaction.categoryId?.let { categoryService.getCategory(spaceId, it) }
val updatedTransaction = Transaction(
id = existingTransaction.id,
space = existingTransaction.space,
@@ -81,16 +102,24 @@ class TransactionServiceImpl(
isDeleted = existingTransaction.isDeleted,
isDone = transaction.isDone,
createdBy = existingTransaction.createdBy,
createdAt = existingTransaction.createdAt
createdAt = existingTransaction.createdAt,
tgChatId = existingTransaction.tgChatId,
tgMessageId = existingTransaction.tgMessageId,
)
if (existingTransaction.category == null && updatedTransaction.category != null) {
categorizeService.notifyThatCategorySelected(updatedTransaction)
}
return transactionRepo.update(updatedTransaction)
}
override fun deleteTransaction(spaceId: Int, transactionId: Int) {
val space = spaceService.getSpace(spaceId)
val space = spaceService.getSpace(spaceId, null)
getTransaction(space.id!!, transactionId)
transactionRepo.delete(transactionId)
}
override fun deleteByRecurrentId(spaceId: Int, recurrentId: Int) {
TODO("Not yet implemented")
}
}

View File

@@ -0,0 +1,26 @@
package space.luminic.finance.services.gpt
import org.springframework.scheduling.annotation.Scheduled
import org.springframework.stereotype.Service
@Service
class CategorizeBotScheduler(
private val seeder: CategoryJobSeeder,
private val picker: CategoryJobRepo,
private val service: CategorizeService
) {
@Scheduled(cron = "* * * * * *")
fun work() {
val jobs = picker.pickBatch(limit = 50)
if (jobs.isEmpty()) return
service.processBatch(jobs)
}
@Scheduled(cron = "* * * * * *")
fun createJob(){
seeder.seedMissing()
}
}

View File

@@ -0,0 +1,148 @@
package space.luminic.finance.services.gpt
import com.github.kotlintelegrambot.Bot
import com.github.kotlintelegrambot.entities.ChatId
import com.github.kotlintelegrambot.entities.InlineKeyboardMarkup
import com.github.kotlintelegrambot.entities.Message
import com.github.kotlintelegrambot.entities.MessageId
import com.github.kotlintelegrambot.entities.ParseMode
import com.github.kotlintelegrambot.entities.keyboard.InlineKeyboardButton
import com.github.kotlintelegrambot.entities.keyboard.WebAppInfo
import com.github.kotlintelegrambot.entities.reaction.ReactionType
import com.github.kotlintelegrambot.types.TelegramBotResult
import org.springframework.beans.factory.annotation.Qualifier
import org.springframework.beans.factory.annotation.Value
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate
import org.springframework.stereotype.Repository
import org.springframework.stereotype.Service
import org.springframework.transaction.annotation.Transactional
import space.luminic.finance.models.Transaction
import space.luminic.finance.repos.CategoryRepo
import space.luminic.finance.repos.TransactionRepo
enum class JobStatus { NEW, PROCESSING, DONE, FAILED }
data class CategoryResult(val categoryId: Int)
@Service
class CategorizeService(
private val transactionRepo: TransactionRepo,
@Qualifier("dsCategorizationService") private val gpt: GptClient,
@Value("\${app.categorize.parallel:4}") private val parallel: Int,
private val categoriesRepo: CategoryRepo,
private val categoryJobRepo: CategoryJobRepo,
private val bot: Bot
) {
private val exec = java.util.concurrent.Executors.newFixedThreadPool(parallel)
fun processBatch(jobs: List<CategoryJob>) {
jobs.forEach { job ->
exec.submit {
runCatching {
val tx = transactionRepo.findBySpaceIdAndId(job.spaceId, job.txId)
?: throw IllegalArgumentException("Transaction ${job.txId} not found")
val res = gpt.suggestCategory(
tx,
categoriesRepo.findBySpaceId(job.spaceId)
) // тут твой вызов GPT
var unsuccessMessage: TelegramBotResult<Message>? = null
if (res.categoryId == 0) {
if (tx.tgChatId != null && tx.tgMessageId != null) {
bot.setMessageReaction(
ChatId.fromId(tx.tgChatId),
tx.tgMessageId,
listOf(ReactionType.Emoji("💔")),
isBig = false
)
unsuccessMessage = bot.sendMessage(
ChatId.fromId(tx.tgChatId),
replyToMessageId = tx.tgMessageId,
text = "К сожалению, мы не смогли распознать категорию.\n\nПопробуйте выставить ее самостоятельно.",
replyMarkup = InlineKeyboardMarkup.create(
listOf(
InlineKeyboardButton.WebApp(
"Открыть в WebApp",
WebAppInfo("https://app.luminic.space/transactions/${tx.id}/edit")
)
)
)
)
}
} else {
transactionRepo.setCategory(job.txId, res.categoryId)
if (tx.tgChatId != null && tx.tgMessageId != null) {
bot.setMessageReaction(
ChatId.fromId(tx.tgChatId),
tx.tgMessageId,
listOf(ReactionType.Emoji("👌")),
isBig = false
)
val category = categoriesRepo.findBySpaceIdAndId(job.spaceId, res.categoryId)
if (category != null) {
bot.sendMessage(
ChatId.fromId(tx.tgChatId),
replyToMessageId = tx.tgMessageId,
text = "Определили категорию: <b>${category.name}</b>.\n\nЕсли это не так, исправьте это в WebApp.",
replyMarkup = InlineKeyboardMarkup.create(
listOf(
InlineKeyboardButton.WebApp(
"Открыть в WebApp",
WebAppInfo("https://app.luminic.space/transactions/${tx.id}/edit")
)
)
),
parseMode = ParseMode.HTML
)
}
}
}
if (unsuccessMessage != null) {
categoryJobRepo.successJob(
job.id,
unsuccessMessage.get().chat.id,
unsuccessMessage.get().messageId
)
} else {
categoryJobRepo.successJob(job.id)
}
}.onFailure { e ->
print(e.localizedMessage)
categoryJobRepo.failJob(job.id, e.localizedMessage)
}
}
}
}
fun notifyThatCategorySelected(tx: Transaction) {
val job = categoryJobRepo.getJobByTxId(tx.id!!)
job?.let {
if (tx.tgChatId != null && tx.tgMessageId != null) {
bot.setMessageReaction(
ChatId.fromId(tx.tgChatId),
tx.tgMessageId,
listOf(ReactionType.Emoji("👌"))
)
}
if (it.chatId != null && it.messageId != null) {
bot.editMessageText(
ChatId.fromId(it.chatId),
messageId = it.messageId,
text = "Выбрана: <b>${tx.category!!.name}</b>.\n\nЕсли это не так, измените это в WebApp.",
replyMarkup = InlineKeyboardMarkup.create(
listOf(
InlineKeyboardButton.WebApp(
"Открыть в WebApp",
WebAppInfo("https://app.luminic.space/transactions/${tx.id}/edit")
)
)
),
parseMode = ParseMode.HTML
)
}
}
}
}

View File

@@ -0,0 +1,97 @@
package space.luminic.finance.services.gpt
import com.github.kotlintelegrambot.Bot
import com.github.kotlintelegrambot.entities.ChatId
import com.github.kotlintelegrambot.entities.reaction.ReactionType
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate
import org.springframework.stereotype.Repository
import org.springframework.transaction.annotation.Transactional
data class CategoryJob(
val id: Long,
val spaceId: Int,
val txId: Int,
val attempts: Int,
val chatId: Long? = null,
val messageId: Long? = null
)
@Repository
class CategoryJobRepo(
private val np: NamedParameterJdbcTemplate,
private val bot: Bot
) {
@Transactional
fun pickBatch(limit: Int = 50): List<CategoryJob> {
val selectSql = """
SELECT cj.id as cj_id, cj.tx_id as cj_tx_id, cj.attempts as cj_attempts, s.id as s_id
FROM finance.category_jobs cj
JOIN finance.transactions t on cj.tx_id = t.id
JOIN finance.spaces s on t.space_id = s.id
WHERE status IN ('NEW', 'FAILED')
ORDER BY cj.created_at
FOR UPDATE SKIP LOCKED
LIMIT :limit
""".trimIndent()
val jobs = np.query(selectSql, mapOf("limit" to limit)) { rs, _ ->
CategoryJob(
id = rs.getLong("cj_id"),
spaceId = rs.getInt("s_id"),
txId = rs.getInt("cj_tx_id"),
attempts = rs.getInt("cj_attempts")
)
}
if (jobs.isNotEmpty()) {
val updateSql = """
UPDATE finance.category_jobs
SET status = 'PROCESSING',
attempts = attempts + 1,
started_at = NOW()
WHERE id IN (:ids)
""".trimIndent()
np.update(updateSql, mapOf("ids" to jobs.map { it.id }))
}
return jobs
}
@Transactional
fun successJob(id: Long, chatId: Long? = null, messageId: Long? = null) {
val sql =
"""UPDATE finance.category_jobs SET status = 'DONE', finished_at = now(), tg_chat_id = :chatId, tg_message_id = :messageId WHERE id = :id """.trimIndent()
np.update(sql, mapOf("id" to id, "chatId" to chatId, "messageId" to messageId))
}
@Transactional
fun failJob(id: Long, errorMessage: String?) {
val sql =
"""UPDATE finance.category_jobs SET status = 'FAILED', last_error = :message WHERE id = :id """.trimIndent()
np.update(sql, mapOf("id" to id, "errorMessage" to errorMessage))
}
@Transactional
fun getJobByTxId(txId: Int): CategoryJob? {
val selectSql = """
SELECT cj.id as cj_id, cj.tx_id as cj_tx_id, cj.attempts as cj_attempts, s.id as s_id, cj.tg_chat_id as cj_chat_id, cj.tg_message_id as cj_message_id
FROM finance.category_jobs cj
JOIN finance.transactions t on cj.tx_id = t.id
JOIN finance.spaces s on t.space_id = s.id
WHERE cj.tx_id = :txId
""".trimIndent()
val jobs = np.query(selectSql, mapOf("txId" to txId), { rs, _ ->
CategoryJob(
id = rs.getLong("cj_id"),
spaceId = rs.getInt("s_id"),
txId = rs.getInt("cj_tx_id"),
attempts = rs.getInt("cj_attempts"),
chatId = rs.getLong("cj_chat_id"),
messageId = rs.getLong("cj_message_id")
)
})
return jobs.firstOrNull()
}
}

View File

@@ -0,0 +1,31 @@
package space.luminic.finance.services.gpt
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate
import org.springframework.stereotype.Repository
import org.springframework.transaction.annotation.Transactional
@Repository
class CategoryJobSeeder(private val np: NamedParameterJdbcTemplate) {
/**
* Создаёт задачи для всех транзакций без категории.
* Ограничь лимит, чтобы не захлестнуть очередь.
*/
@Transactional
fun seedMissing(limit: Int = 1000) : Int {
val sql = """
INSERT INTO finance.category_jobs (tx_id)
SELECT t.id
FROM finance.transactions t
WHERE t.category_id IS NULL
AND t.is_deleted = false
AND NOT EXISTS (
SELECT 1 FROM finance.category_jobs j WHERE j.tx_id = t.id
)
ORDER BY t.date DESC
LIMIT :limit
ON CONFLICT (tx_id) DO NOTHING
""".trimIndent()
return np.update(sql, mapOf("limit" to limit))
}
}

View File

@@ -0,0 +1,80 @@
package space.luminic.finance.services.gpt
import okhttp3.*
import com.fasterxml.jackson.module.kotlin.jacksonObjectMapper
import com.fasterxml.jackson.module.kotlin.readValue
import okhttp3.MediaType.Companion.toMediaTypeOrNull
import okhttp3.RequestBody.Companion.toRequestBody
import org.slf4j.LoggerFactory
import org.springframework.beans.factory.annotation.Qualifier
import org.springframework.beans.factory.annotation.Value
import org.springframework.stereotype.Service
import space.luminic.finance.models.Category
import space.luminic.finance.models.Transaction
@Service("dsCategorizationService")
class DeepSeekCategorizationService(
@Value("\${ds.api_key}") private val apiKey: String,
) : GptClient {
private val endpoint = "https://api.deepseek.com/v1"
private val mapper = jacksonObjectMapper()
private val client = OkHttpClient()
private val logger = LoggerFactory.getLogger(javaClass)
override fun suggestCategory(tx: Transaction, categories: List<Category>): CategorySuggestion {
val catList = categories.joinToString("\n") { "- ${it.id}: ${it.name}" }
val txInfo = """
{ \"amount\": ${tx.amount}, \"comment\": \"${tx.comment}\", \"date\":${tx.date}\" }
""".trimIndent()
val prompt = """
Пользователь имеет следующие категории:
$catList
Задача:
1. Определите наиболее подходящую категорию из списка выше для транзакции пользователя.
2. Верните ответ в формате: "ID категории", например "3".
3. Если ни одна категория из списка не подходит, верните: "0".
Ответ должен быть кратким, одной строкой, без дополнительных пояснений.
""".trimIndent()
val body = mapOf(
"model" to "deepseek-chat",
"messages" to listOf(
mapOf("role" to "assistant", "content" to prompt),
mapOf("role" to "user", "content" to txInfo)
)
)
val jsonBody = mapper.writeValueAsString(body)
val requestBody = jsonBody.toRequestBody("application/json; charset=utf-8".toMediaTypeOrNull())
val request = Request.Builder()
.url("${endpoint}/chat/completions")
.addHeader("Authorization", "Bearer $apiKey")
.post(requestBody)
.build()
println(request)
logger.info(request.toString())
client.newCall(request).execute().use { response ->
if (!response.isSuccessful) error("Qwen error: ${response.code} ${response.body?.string()}")
val bodyStr = response.body?.string().orEmpty()
// Берём content из choices[0].message.content
val root = mapper.readTree(bodyStr)
val text = root["choices"]?.get(0)?.get("message")?.get("content")?.asText()
?: error("No choices[0].message.content in response")
// Парсим "ID Название (вероятность)"
// val regex = Regex("""^\s*(\d+)\s*[-]\s*(.+?)\s*\((0(?:\.\d+)?|1(?:\.0)?)\)\s*$""")
// val match = regex.find(text.trim()) ?: error("Bad format: '$text'")
// val (idStr, name, confStr) = match.destructured
val idStr = text
return CategorySuggestion(idStr.toInt(), )
}
}
}

View File

@@ -0,0 +1,10 @@
package space.luminic.finance.services.gpt
import space.luminic.finance.models.Category
import space.luminic.finance.models.Transaction
data class CategorySuggestion(val categoryId: Int, val categoryName: String? = null, val confidence: Double? = null)
interface GptClient {
fun suggestCategory(tx: Transaction, categories: List<Category>): CategorySuggestion
}

View File

@@ -0,0 +1,78 @@
package space.luminic.finance.services.gpt
import okhttp3.*
import com.fasterxml.jackson.module.kotlin.jacksonObjectMapper
import com.fasterxml.jackson.module.kotlin.readValue
import okhttp3.MediaType.Companion.toMediaTypeOrNull
import okhttp3.RequestBody.Companion.toRequestBody
import org.slf4j.LoggerFactory
import org.springframework.beans.factory.annotation.Qualifier
import org.springframework.beans.factory.annotation.Value
import org.springframework.stereotype.Service
import space.luminic.finance.models.Category
import space.luminic.finance.models.Transaction
@Service("qwenCategorizationService")
class QwenCategorizationService(
@Value("\${qwen.api_key}") private val apiKey: String,
) : GptClient {
private val endpoint = "https://dashscope-intl.aliyuncs.com/compatible-mode/v1"
private val mapper = jacksonObjectMapper()
private val client = OkHttpClient()
private val logger = LoggerFactory.getLogger(javaClass)
override fun suggestCategory(tx: Transaction, categories: List<Category>): CategorySuggestion {
val catList = categories.joinToString("\n") { "- ${it.id}: ${it.name}" }
val txInfo = """
{ \"amount\": ${tx.amount}, \"comment\": \"${tx.comment}\", \"date\":${tx.date}\" }
""".trimIndent()
val prompt = """
Пользователь имеет следующие категории:
$catList
Задача:
1. Определите наиболее подходящую категорию из списка выше для транзакции пользователя.
2. Верните ответ в формате: "ID категории имя категории (вероятность)", например "3 Продукты (0.87)".
3. Если ни одна категория из списка не подходит, верните: "0 Другое (вероятность)".
Ответ должен быть кратким, одной строкой, без дополнительных пояснений.
""".trimIndent()
val body = mapOf(
"model" to "qwen-plus",
"messages" to listOf(
mapOf("role" to "assistant", "content" to prompt),
mapOf("role" to "user", "content" to txInfo)
)
)
val jsonBody = mapper.writeValueAsString(body)
val requestBody = jsonBody.toRequestBody("application/json; charset=utf-8".toMediaTypeOrNull())
val request = Request.Builder()
.url("${endpoint}/chat/completions")
.addHeader("Authorization", "Bearer $apiKey")
.post(requestBody)
.build()
println(request)
logger.info(request.toString())
client.newCall(request).execute().use { response ->
if (!response.isSuccessful) error("Qwen error: ${response.code} ${response.body?.string()}")
val bodyStr = response.body?.string().orEmpty()
// Берём content из choices[0].message.content
val root = mapper.readTree(bodyStr)
val text = root["choices"]?.get(0)?.get("message")?.get("content")?.asText()
?: error("No choices[0].message.content in response")
// Парсим "ID Название (вероятность)"
val regex = Regex("""^\s*(\d+)\s*[-]\s*(.+?)\s*\((0(?:\.\d+)?|1(?:\.0)?)\)\s*$""")
val match = regex.find(text.trim()) ?: error("Bad format: '$text'")
val (idStr, name, confStr) = match.destructured
return CategorySuggestion(idStr.toInt(), name, confStr.toDouble())
}
}
}

View File

@@ -0,0 +1,253 @@
package space.luminic.finance.services.telegram
import com.github.kotlintelegrambot.Bot
import com.github.kotlintelegrambot.dispatch
import com.github.kotlintelegrambot.dispatcher.callbackQuery
import com.github.kotlintelegrambot.dispatcher.command
import com.github.kotlintelegrambot.dispatcher.message
import com.github.kotlintelegrambot.entities.ChatId
import com.github.kotlintelegrambot.entities.InlineKeyboardMarkup
import com.github.kotlintelegrambot.entities.ParseMode
import com.github.kotlintelegrambot.entities.keyboard.InlineKeyboardButton
import com.github.kotlintelegrambot.entities.keyboard.WebAppInfo
import com.github.kotlintelegrambot.entities.reaction.ReactionType
import com.github.kotlintelegrambot.extensions.filters.Filter
import com.github.kotlintelegrambot.logging.LogLevel
import org.springframework.beans.factory.annotation.Qualifier
import org.springframework.beans.factory.annotation.Value
import org.springframework.context.annotation.Bean
import org.springframework.stereotype.Service
import org.springframework.transaction.annotation.Transactional
import space.luminic.finance.dtos.TransactionDTO
import space.luminic.finance.models.NotFoundException
import space.luminic.finance.models.State
import space.luminic.finance.models.Transaction
import space.luminic.finance.models.User
import space.luminic.finance.repos.BotRepo
import space.luminic.finance.services.UserService
import java.time.LocalDate
@Service
class BotService(
@Value("\${telegram.bot.token}") private val botToken: String,
private val userService: UserService,
@Qualifier("spaceServiceTelegram") private val spaceService: SpaceService,
private val botRepo: BotRepo,
@Qualifier("transactionsServiceTelegram") private val transactionService: TransactionService
) {
private fun buildSpaceSelector(userId: Int): InlineKeyboardMarkup {
val spaces = spaceService.getSpaces(userId)
val keyboard = mutableListOf<List<InlineKeyboardButton>>()
val row = mutableListOf<InlineKeyboardButton>()
if (spaces.isNotEmpty()) {
for ((index, space) in spaces.withIndex()) {
val button =
InlineKeyboardButton.CallbackData(text = space.name, callbackData = "select_space_${space.id}")
row.add(button)
// Если 2 кнопки в строке — отправляем строку и очищаем
if (row.size == 2) {
keyboard.add(ArrayList(row))
row.clear()
}
}
// Если осталась 1 кнопка — добавляем последнюю строку
if (row.isNotEmpty()) {
keyboard.add(ArrayList(row))
}
} else {
row.add(InlineKeyboardButton.CallbackData("Создать пространство!", callbackData = "create_space"))
keyboard.add(ArrayList(row))
}
return InlineKeyboardMarkup.Companion.create(keyboard)
}
@Transactional
fun selectSpace(tgUserId: Long, selectedSpaceId: Int) {
val user = userService.getUserByTelegramId(tgUserId)
botRepo.setState(
user.id!!, State.StateCode.SPACE_SELECTED, mapOf(
"selected_space" to selectedSpaceId.toString(),
)
)
}
private fun buildRegister() {
}
private fun buildMenu(tgUserId: Long): InlineKeyboardMarkup {
val user = userService.getUserByTelegramId(tgUserId)
val userId = requireNotNull(user.id) { "User must have id" }
val state = botRepo.getState(tgUserId)
val spaceId = state?.data?.get("selected_space")?.toIntOrNull()
val space = spaceId?.let { id -> spaceService.getSpace(spaceId, userId) }
val keyboard = mutableListOf<List<InlineKeyboardButton>>()
// Кнопка с названием выбранного space (или плейсхолдером)
keyboard.add(
listOf(
InlineKeyboardButton.CallbackData(
text = space?.name ?: "Select space",
"select_space"
)
)
)
// Если нужен второй баттон — сделай другой смысл/текст; иначе этот блок можно убрать
keyboard.add(
listOf(
InlineKeyboardButton.WebApp(
text = "Открыть WebApp",
webApp = WebAppInfo(url = "https://app.luminic.space")
)
)
)
return InlineKeyboardMarkup.Companion.create(keyboard)
}
@Bean
fun bot(): Bot {
val bot = com.github.kotlintelegrambot.bot {
logLevel = LogLevel.None
token = botToken
dispatch {
message(Filter.Text) {
val fromId = message.from?.id ?: throw IllegalArgumentException("user is empty")
val user = userService.getUserByTelegramId(fromId)
val state = botRepo.getState(message.from?.id ?: throw IllegalArgumentException("user is empty"))
when (state?.state) {
State.StateCode.SPACE_SELECTED -> {
try {
val parts = message.text!!.trim().split(" ", limit = 2)
if (parts.isEmpty()) {
bot.sendMessage(
chatId = ChatId.fromId(message.chat.id),
text = "Введите сумму и комментарий, например: `250 обед`",
parseMode = ParseMode.MARKDOWN
)
}
val amount = parts[0].toIntOrNull()
?: throw IllegalArgumentException("Сумма транзакции не число!")
if (amount <= 0) {
throw IllegalArgumentException("Сумма не может быть меньше 1.")
}
val comment = parts.getOrNull(1)?.trim().orEmpty()
if (comment.isEmpty()) throw IllegalArgumentException("Комментарий не может быть пустым.")
// bot.sendMessage(
// chatId = ChatId.fromId(message.chat.id),
// text = "Принято: сумма = $amount, комментарий = \"$comment\""
// )
try {
transactionService.createTransaction(
state.data["selected_space"]?.toInt()
?: throw IllegalArgumentException("selected space is empty"),
user.id!!,
TransactionDTO.CreateTransactionDTO(
Transaction.TransactionType.EXPENSE,
Transaction.TransactionKind.INSTANT,
comment = comment,
amount = amount.toBigDecimal(),
date = LocalDate.now(),
),
message.chat.id,
message.messageId
)
bot.setMessageReaction(
chatId = ChatId.fromId(message.chat.id),
messageId = message.messageId,
reaction = listOf(ReactionType.Emoji("🤝")),
isBig = false
)
} catch (e: IllegalArgumentException) {
bot.sendMessage(
ChatId.Companion.fromId(message.chat.id),
text = "Кажется у вас не выбран Space",
replyMarkup = buildSpaceSelector(user.id!!)
)
}
} catch (e: IllegalArgumentException) {
bot.sendMessage(
chatId = ChatId.Companion.fromId(message.chat.id),
text = "Ошибка: ${e.message}"
)
}
}
else -> {}
}
}
callbackQuery {
if (callbackQuery.data.startsWith("select_space_")) {
val spaceId = callbackQuery.data.substringAfter("select_space_").toInt()
println(spaceId)
try {
selectSpace(callbackQuery.from.id, spaceId)
bot.editMessageText(
chatId = ChatId.Companion.fromId(callbackQuery.message!!.chat.id),
messageId = callbackQuery.message!!.messageId,
text = "Успешно!\n\nМы готовы принимать Ваши транзакции.\n\nПросто пишите их в формате:\n\n <i>сумма комментарий</i>\n\n <b>Первой обязательно должна быть сумма!</b>",
parseMode = ParseMode.HTML,
replyMarkup = buildMenu(callbackQuery.from.id)
)
} catch (e: NotFoundException) {
e.printStackTrace()
bot.sendMessage(
ChatId.Companion.fromId(callbackQuery.message!!.chat.id),
text = "Мы кажется не знакомы"
)
}
} else if (callbackQuery.data.equals("select_space", ignoreCase = true)) {
bot.editMessageText(
ChatId.Companion.fromId(callbackQuery.message!!.chat.id),
callbackQuery.message!!.messageId,
text = "Выберите новое пространство",
replyMarkup = buildSpaceSelector(
userService.getUserByTelegramId(callbackQuery.from.id).id!!
)
)
}
}
command("start") {
val user: User
try {
user = userService.getUserByTelegramId(
message.from?.id ?: throw IllegalArgumentException("User not found")
)
bot.sendMessage(
ChatId.Companion.fromId(message.chat.id),
text = "Привет!\n\nРады тебя снова видеть!\n\nНачнем с выбора пространства:",
replyMarkup = buildSpaceSelector(user.id!!)
)
} catch (e: NotFoundException) {
bot.sendMessage(ChatId.Companion.fromId(message.chat.id), text = "Кажется, мы еще не знакомы.")
bot.sendMessage(ChatId.Companion.fromId(message.chat.id), text = "Давайте зарегистрируемся? ")
bot.sendMessage(ChatId.Companion.fromId(message.chat.id), text = "")
}
}
}
}
bot.startPolling()
return bot
}
}

View File

@@ -0,0 +1,8 @@
package space.luminic.finance.services.telegram
import space.luminic.finance.models.Space
interface SpaceService {
fun getSpaces(userId: Int): List<Space>
fun getSpace(spaceId: Int, userId: Int): Space?
}

View File

@@ -0,0 +1,23 @@
package space.luminic.finance.services.telegram
import org.springframework.stereotype.Service
import space.luminic.finance.models.NotFoundException
import space.luminic.finance.models.Space
import space.luminic.finance.repos.SpaceRepo
@Service("spaceServiceTelegram")
class SpaceServiceImpl(
private val spaceRepo: SpaceRepo
) : SpaceService {
override fun getSpaces(userId: Int): List<Space> {
val spaces = spaceRepo.findSpacesAvailableForUser(userId)
return spaces
}
override fun getSpace(spaceId: Int, userId: Int): Space? {
val space =
spaceRepo.findSpaceById(spaceId, userId) ?: throw NotFoundException("Space with id $spaceId not found")
return space
}
}

View File

@@ -0,0 +1,8 @@
package space.luminic.finance.services.telegram
import space.luminic.finance.dtos.TransactionDTO
interface TransactionService {
fun createTransaction(spaceId: Int, userId: Int, transaction: TransactionDTO.CreateTransactionDTO, chatId: Long, messageId: Long ): Int
}

View File

@@ -0,0 +1,44 @@
package space.luminic.finance.services.telegram
import org.springframework.beans.factory.annotation.Qualifier
import org.springframework.stereotype.Service
import space.luminic.finance.dtos.TransactionDTO
import space.luminic.finance.models.Transaction
import space.luminic.finance.repos.TransactionRepo
import space.luminic.finance.services.CategoryServiceImpl
@Service("transactionsServiceTelegram")
class TransactionsServiceImpl(
private val transactionRepo: TransactionRepo,
@Qualifier("spaceServiceTelegram") private val spaceService: SpaceService,
private val categoryService: CategoryServiceImpl
): TransactionService {
override fun createTransaction(
spaceId: Int,
userId: Int,
transaction: TransactionDTO.CreateTransactionDTO,
chatId: Long,
messageId: Long
): Int {
val space = spaceService.getSpace(spaceId, userId)
val category = transaction.categoryId?.let { categoryService.getCategory(spaceId, it) }
val transaction = Transaction(
space = space,
type = transaction.type,
kind = transaction.kind,
category = category,
comment = transaction.comment,
amount = transaction.amount,
fees = transaction.fees,
date = transaction.date,
tgChatId = chatId,
tgMessageId = messageId,
)
print(transaction)
return transactionRepo.create(transaction, userId)
}
}

View File

@@ -9,16 +9,16 @@ logging.level.org.springframework.security = DEBUG
#logging.level.org.springframework.data.mongodb.code = DEBUG
logging.level.org.springframework.web.reactive=DEBUG
logging.level.org.mongodb.driver.protocol.command = DEBUG
logging.level.org.springframework.jdbc.core=DEBUG
logging.level.org.springframework.jdbc.core.StatementCreatorUtils=TRACE
logging.level.org.springframework.jdbc=DEBUG
logging.level.org.springframework.jdbc.datasource=DEBUG
logging.level.org.springframework.jdbc.support=DEBUG
logging.level.org.hibernate.type.descriptor.sql.BasicBinder=TRACE
logging.level.org.springframework.jdbc.core=INFO
logging.level.org.springframework.jdbc.core.StatementCreatorUtils=INFO
logging.level.org.springframework.jdbc=INFO
logging.level.org.springframework.jdbc.datasource=INFO
logging.level.org.springframework.jdbc.support=INFO
management.endpoints.web.exposure.include=*
management.endpoint.health.show-details=always
telegram.bot.token=6972242509:AAGyXuL3T-BNE4XMoo_qvtaYxw_SuiS_dDs
# vector test
telegram.bot.token=8127199836:AAEPepyKDAf8PvFpw-fpxBXUuPdx_LS20fI
nlp.address=http://127.0.0.1:8000

View File

@@ -1,5 +1,3 @@
spring.application.name=budger-app
spring.data.mongodb.uri=mongodb://budger-app:BA1q2w3e4r!@luminic.space:27017/budger-app-v2?authSource=admin&minPoolSize=10&maxPoolSize=100
logging.level.org.springframework.web=INFO
logging.level.org.springframework.data = INFO
@@ -9,15 +7,12 @@ logging.level.org.springframework.data.mongodb.code = INFO
logging.level.org.springframework.web.reactive=INFO
logging.level.org.mongodb.driver.protocol.command = INFO
#management.endpoints.web.exposure.include=*
#management.endpoint.metrics.access=read_only
telegram.bot.token = 7999296388:AAGXPE5r0yt3ZFehBoUh8FGm5FBbs9pYIks
nlp.address=https://nlp.luminic.space
spring.datasource.url=jdbc:postgresql://postgresql:5432/luminic-space-db
#spring.datasource.url=jdbc:postgresql://postgresql:5432/luminic-space-db
spring.datasource.url=jdbc:postgresql://213.226.71.138:5432/luminic-space-db
spring.datasource.username=luminicspace
spring.datasource.password=LS1q2w3e4r!

View File

@@ -17,13 +17,16 @@ spring.servlet.multipart.max-request-size=10MB
storage.location: static
spring.jackson.default-property-inclusion=non_null
# Expose prometheus, health, and info endpoints
#management.endpoints.web.exposure.include=prometheus,health,info
management.endpoints.web.exposure.include=*
management.endpoints.web.exposure.include=health,info,prometheus
# Enable Prometheus metrics export
#management.endpoint.prometheus.access=unrestricted
management.endpoint.prometheus.enabled=true
management.prometheus.metrics.export.enabled=true
management.metrics.tags.application=luminic-app
telegram.bot.username = expenses_diary_bot
spring.flyway.enabled=true
spring.flyway.locations=classpath:db/migration
@@ -31,3 +34,5 @@ spring.flyway.baseline-on-migrate= false
spring.flyway.schemas=finance
spring.jpa.properties.hibernate.default_schema=finance
spring.jpa.properties.hibernate.default_batch_fetch_size=50
qwen.api_key=sk-991942d15b424cc89513498bb2946045
ds.api_key=sk-b5949728e79747f08af0a1d65bc6a7a2

View File

@@ -0,0 +1,2 @@
ALTER TABLE finance.users
ADD CONSTRAINT uq_users_username UNIQUE (username);

View File

@@ -0,0 +1,21 @@
create table if not exists finance.bot_states
(
user_id integer primary key,
state_code varchar not null
);
create table if not exists finance.bot_states_data
(
id INTEGER GENERATED BY DEFAULT AS IDENTITY NOT NULL,
user_id integer not null,
data_code varchar(255) not null,
data_value varchar(255) not null,
CONSTRAINT pk_bot_states_data PRIMARY KEY (id)
);
ALTER TABLE finance.bot_states
ADD CONSTRAINT FK_STATE_ON_USER FOREIGN KEY (user_id) REFERENCES finance.users (id);
ALTER TABLE finance.bot_states
ADD CONSTRAINT FK_STATE_DATA_ON_USER FOREIGN KEY (user_id) REFERENCES finance.users (id);

View File

@@ -0,0 +1,3 @@
-- уникальное ограничение под ON CONFLICT (user_id, data_code)
ALTER TABLE finance.bot_states_data
ADD CONSTRAINT ux_bot_states_data_user_code UNIQUE (user_id, data_code);

View File

@@ -0,0 +1,3 @@
-- уникальное ограничение под ON CONFLICT (user_id, data_code)
ALTER TABLE finance.transactions
ALTER COLUMN category_id DROP NOT NULL;

View File

@@ -0,0 +1,19 @@
-- Очередь классификации категорий
CREATE TABLE IF NOT EXISTS finance.category_jobs (
id BIGSERIAL PRIMARY KEY,
tx_id INT NOT NULL UNIQUE, -- одна задача на транзакцию
status TEXT NOT NULL DEFAULT 'NEW', -- NEW | PROCESSING | DONE | FAILED
attempts INT NOT NULL DEFAULT 0,
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
started_at TIMESTAMPTZ,
finished_at TIMESTAMPTZ,
last_error TEXT
);
-- Быстрые выборки по статусу
CREATE INDEX IF NOT EXISTS ix_category_jobs_status ON finance.category_jobs(status);
-- (опционально) защита от «зависших» задач
CREATE INDEX IF NOT EXISTS ix_category_jobs_processing_time
ON finance.category_jobs(started_at)
WHERE status = 'PROCESSING';

View File

@@ -0,0 +1,3 @@
alter table finance.transactions
add column tg_chat_id bigint null,
add column tg_message_id bigint null;

View File

@@ -0,0 +1,3 @@
alter table finance.category_jobs
add column tg_chat_id bigint null,
add column tg_message_id bigint null;

View File

@@ -0,0 +1,5 @@
ALTER TABLE finance.recurrent_operations
ADD CONSTRAINT recurrent_operations_pk PRIMARY KEY (id);
alter table finance.transactions
add column recurrent_id integer null,
ADD CONSTRAINT FK_RECURRENTS FOREIGN KEY (recurrent_id) REFERENCES finance.recurrent_operations (id);