c58ce6c53b
This uses [napi-rs](https://napi.rs/) to allow for automatic generation of node bindings for the native code. I also changed the `isolatedModules` TS flag to false to allow for `static enum` to be shared across modules. It doesn't seem to be necessary for the build system that CK uses. Currently this method does not work with ID generators with longer IDs. Likely the best solution is to add another key in the database. Some benchmarks for 1 million conversions: ``` node, x1_000_000: 2.847s rust, x1_000_000: 1.265s ``` There are still optimizations that can be made, but I think this is a good starting point and a good way to bring rust into the CK stack. Co-authored-by: s1idewhist1e <trombonedude05@gmail.com> Reviewed-on: https://codeberg.org/calckey/calckey/pulls/9786 Co-authored-by: s1idewhist1e <s1idewhist1e@noreply.codeberg.org> Co-committed-by: s1idewhist1e <s1idewhist1e@noreply.codeberg.org>
70 lines
1.5 KiB
Rust
70 lines
1.5 KiB
Rust
use napi::{bindgen_prelude::*, Error, Status};
|
|
use napi_derive::napi;
|
|
|
|
static CHAR_COLLECTION: &str = "0123456789abcdefghijklmnopqrstuvwxyz";
|
|
|
|
// -- NAPI exports --
|
|
|
|
#[napi]
|
|
pub enum IdConvertType {
|
|
MastodonId,
|
|
CalckeyId,
|
|
}
|
|
|
|
#[napi]
|
|
pub fn convert_id(in_id: String, id_convert_type: IdConvertType) -> napi::Result<String> {
|
|
use IdConvertType::*;
|
|
match id_convert_type {
|
|
MastodonId => {
|
|
let mut out: i64 = 0;
|
|
for (i, c) in in_id.to_lowercase().chars().rev().enumerate() {
|
|
out += num_from_char(c)? as i64 * 36_i64.pow(i as u32);
|
|
}
|
|
|
|
Ok(out.to_string())
|
|
}
|
|
CalckeyId => {
|
|
let mut input: i64 = match in_id.parse() {
|
|
Ok(s) => s,
|
|
Err(_) => {
|
|
return Err(Error::new(
|
|
Status::InvalidArg,
|
|
"Unable to parse ID as MasstodonId",
|
|
))
|
|
}
|
|
};
|
|
let mut out = String::new();
|
|
|
|
while input != 0 {
|
|
out.insert(0, char_from_num((input % 36) as u8)?);
|
|
input /= 36;
|
|
}
|
|
|
|
Ok(out)
|
|
}
|
|
}
|
|
}
|
|
|
|
// -- end --
|
|
|
|
#[inline(always)]
|
|
fn num_from_char(character: char) -> napi::Result<u8> {
|
|
for (i, c) in CHAR_COLLECTION.chars().enumerate() {
|
|
if c == character {
|
|
return Ok(i as u8);
|
|
}
|
|
}
|
|
|
|
Err(Error::new(
|
|
Status::InvalidArg,
|
|
"Invalid character in parsed base36 id",
|
|
))
|
|
}
|
|
|
|
#[inline(always)]
|
|
fn char_from_num(number: u8) -> napi::Result<char> {
|
|
CHAR_COLLECTION
|
|
.chars()
|
|
.nth(number as usize)
|
|
.ok_or(Error::from_status(Status::Unknown))
|
|
}
|