Kharec/spasteg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: extract lookup helper, use idiomatic pipelines, remove unused imports

Browse Source
This commit is contained in:
Sandro CURY CAZZANIGA
2026-03-01 10:07:16 +01:00
parent 0de588b2d6
commit 891107c442
1 changed files with 24 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
src/storage.gleam
View File

@@ -1,6 +1,6 @@
import gleam/dict.{type Dict} import gleam/dict.{type Dict}
import gleam/erlang/process import gleam/erlang/process
import gleam/option.{type Option, None, Some} import gleam/option.{type Option}
import gleam/otp/actor import gleam/otp/actor
const max_requests_per_minute = 10 const max_requests_per_minute = 10
@@ -17,40 +17,44 @@ pub type StorageMsg {
ResetRateLimits ResetRateLimits
} }
fn lookup_paste(state: StorageState, key: String) -> Option(String) {
state.pastes
|> dict.get(key)
|> option.from_result
}
pub fn handle_message(state: StorageState, msg: StorageMsg) { pub fn handle_message(state: StorageState, msg: StorageMsg) {
case msg { case msg {
CreatePaste(key, content, reply) -> { CreatePaste(key, content, reply) -> {
let new_state = let new_pastes =
StorageState(dict.insert(state.pastes, key, content), state.rate_limits) state.pastes
|> dict.insert(key, content)
process.send(reply, True) process.send(reply, True)
actor.continue(new_state) actor.continue(StorageState(new_pastes, state.rate_limits))
} }
PeekPaste(key, reply) -> { PeekPaste(key, reply) -> {
let content = case dict.get(state.pastes, key) { process.send(reply, lookup_paste(state, key))
Ok(value) -> Some(value)
Error(_) -> None
}
process.send(reply, content)
actor.continue(state) actor.continue(state)
} }
GetPaste(key, reply) -> { GetPaste(key, reply) -> {
let content = case dict.get(state.pastes, key) { let content = lookup_paste(state, key)
Ok(value) -> Some(value) let new_pastes =
Error(_) -> None state.pastes
} |> dict.delete(key)
let new_pastes = dict.delete(state.pastes, key)
let new_state = StorageState(new_pastes, state.rate_limits)
process.send(reply, content) process.send(reply, content)
actor.continue(new_state) actor.continue(StorageState(new_pastes, state.rate_limits))
} }
CheckRateLimit(ip, reply) -> { CheckRateLimit(ip, reply) -> {
let current_count = let count =
dict.get(state.rate_limits, ip) state.rate_limits
|> dict.get(ip)
|> option.from_result |> option.from_result
|> option.unwrap(0) |> option.unwrap(0)
let allowed = current_count < max_requests_per_minute let allowed = count < max_requests_per_minute
let new_limits = case allowed { let new_limits = case allowed {
True -> dict.insert(state.rate_limits, ip, current_count + 1) True ->
state.rate_limits
|> dict.insert(ip, count + 1)
False -> state.rate_limits False -> state.rate_limits
} }
process.send(reply, allowed) process.send(reply, allowed)