60 lines
1.5 KiB
Rust
60 lines
1.5 KiB
Rust
|
use std::fs;
|
||
|
|
|
||
|
|
use rocket::get;
|
||
|
|
use rocket::State;
|
||
|
|
use rocket::http::Status;
|
||
|
|
use rocket::response::{content, status};
|
||
|
|
use rocket::serde::json::Json;
|
||
|
|
use rusqlite::params;
|
||
|
|
|
||
|
|
use crate::database::WordDb;
|
||
|
|
|
||
|
|
#[get("/frontend")]
|
||
|
|
pub fn frontend() -> Option<content::Html<String>> {
|
||
|
|
match fs::read_to_string("static/index.html") {
|
||
|
|
Ok(file) => Some(content::Html(file)),
|
||
|
|
Err(_) => None
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[get("/langs/<lang>/words/<word>")]
|
||
|
|
pub fn get_word(db: &State<WordDb>, lang: &str, word: &str) -> Option<content::Json<String>> {
|
||
|
|
let connection = db.connect();
|
||
|
|
|
||
|
|
let word = connection.query_row(&format!(
|
||
|
|
"SELECT content FROM {}_words WHERE word = ?",
|
||
|
|
lang),
|
||
|
|
[word],
|
||
|
|
|row| row.get(0)
|
||
|
|
);
|
||
|
|
|
||
|
|
match word {
|
||
|
|
Ok(w) => Some(content::Json(w)),
|
||
|
|
Err(_) => None
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[get("/langs/<lang>/words?<like>&<limit>&<offset>")]
|
||
|
|
pub fn get_word_like(db: &State<WordDb>, lang: &str, like: &str, limit: usize, offset: usize) -> Json<Vec<String>> {
|
||
|
|
let connection = db.connect();
|
||
|
|
|
||
|
|
let mut statement = connection.prepare(&format!(
|
||
|
|
"SELECT word
|
||
|
|
FROM {}_words
|
||
|
|
WHERE word LIKE ?
|
||
|
|
ORDER BY length(word) ASC
|
||
|
|
LIMIT ?
|
||
|
|
OFFSET ?",
|
||
|
|
lang)
|
||
|
|
).unwrap();
|
||
|
|
|
||
|
|
let mut rows = statement.query(params![format!("%{}%", like), limit, offset]).unwrap();
|
||
|
|
|
||
|
|
let mut words = Vec::new();
|
||
|
|
while let Some(row) = rows.next().unwrap() {
|
||
|
|
words.push(row.get(0).unwrap());
|
||
|
|
}
|
||
|
|
|
||
|
|
Json(words)
|
||
|
|
}
|