255 lines
8.0 KiB
Rust
255 lines
8.0 KiB
Rust
|
use std::{convert::TryFrom, ffi::CStr, ffi::CString, os::raw::c_char, slice};
|
||
|
|
||
|
use anyhow::{anyhow, Result};
|
||
|
use reqwest::Url;
|
||
|
use serde::{Deserialize, Serialize};
|
||
|
|
||
|
#[cfg(not(test))]
|
||
|
use log::{error, info};
|
||
|
#[cfg(test)]
|
||
|
use std::{println as info, println as error};
|
||
|
|
||
|
use crate::{
|
||
|
cache::file_cache_dir, cache::from_file_cache, cache::update_file_cache, log_server_error,
|
||
|
result::FFIResult,
|
||
|
};
|
||
|
|
||
|
#[derive(Serialize, Deserialize, Debug)]
|
||
|
pub struct Transaction {
|
||
|
pub id: Option<u32>,
|
||
|
pub shop_id: u32,
|
||
|
pub mod_name: String,
|
||
|
pub local_form_id: u32,
|
||
|
pub is_sell: bool,
|
||
|
pub quantity: u32,
|
||
|
pub amount: u32,
|
||
|
}
|
||
|
|
||
|
impl Transaction {
|
||
|
pub fn from_game(
|
||
|
shop_id: u32,
|
||
|
mod_name: &str,
|
||
|
local_form_id: u32,
|
||
|
is_sell: bool,
|
||
|
quantity: u32,
|
||
|
amount: u32,
|
||
|
) -> Self {
|
||
|
Self {
|
||
|
id: None,
|
||
|
shop_id,
|
||
|
mod_name: mod_name.to_string(),
|
||
|
local_form_id,
|
||
|
is_sell,
|
||
|
quantity,
|
||
|
amount,
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
impl From<RawTransaction> for Transaction {
|
||
|
fn from(raw_transaction: RawTransaction) -> Self {
|
||
|
Self {
|
||
|
id: Some(raw_transaction.id),
|
||
|
shop_id: raw_transaction.shop_id,
|
||
|
mod_name: unsafe { CStr::from_ptr(raw_transaction.mod_name) }
|
||
|
.to_string_lossy()
|
||
|
.to_string(),
|
||
|
local_form_id: raw_transaction.local_form_id,
|
||
|
is_sell: raw_transaction.is_sell,
|
||
|
quantity: raw_transaction.quantity,
|
||
|
amount: raw_transaction.amount,
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[derive(Debug)]
|
||
|
#[repr(C)]
|
||
|
pub struct RawTransaction {
|
||
|
pub id: u32,
|
||
|
pub shop_id: u32,
|
||
|
pub mod_name: *const c_char,
|
||
|
pub local_form_id: u32,
|
||
|
pub is_sell: bool,
|
||
|
pub quantity: u32,
|
||
|
pub amount: u32,
|
||
|
}
|
||
|
|
||
|
impl TryFrom<Transaction> for RawTransaction {
|
||
|
type Error = anyhow::Error;
|
||
|
|
||
|
fn try_from(transaction: Transaction) -> Result<Self> {
|
||
|
if let Some(id) = transaction.id {
|
||
|
Ok(Self {
|
||
|
id,
|
||
|
shop_id: transaction.shop_id,
|
||
|
mod_name: CString::new(transaction.mod_name)
|
||
|
.unwrap_or_default()
|
||
|
.into_raw(),
|
||
|
local_form_id: transaction.local_form_id,
|
||
|
is_sell: transaction.is_sell,
|
||
|
quantity: transaction.quantity,
|
||
|
amount: transaction.amount,
|
||
|
})
|
||
|
} else {
|
||
|
Err(anyhow!("transaction.id is None"))
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[derive(Debug)]
|
||
|
#[repr(C)]
|
||
|
pub struct RawTransactionVec {
|
||
|
pub ptr: *mut RawTransaction,
|
||
|
pub len: usize,
|
||
|
pub cap: usize,
|
||
|
}
|
||
|
|
||
|
#[no_mangle]
|
||
|
pub extern "C" fn create_transaction(
|
||
|
api_url: *const c_char,
|
||
|
api_key: *const c_char,
|
||
|
shop_id: u32,
|
||
|
mod_name: *const c_char,
|
||
|
local_form_id: u32,
|
||
|
is_sell: bool,
|
||
|
quantity: u32,
|
||
|
amount: u32,
|
||
|
) -> FFIResult<RawTransaction> {
|
||
|
let api_url = unsafe { CStr::from_ptr(api_url) }.to_string_lossy();
|
||
|
let api_key = unsafe { CStr::from_ptr(api_key) }.to_string_lossy();
|
||
|
let mod_name = unsafe { CStr::from_ptr(mod_name) }.to_string_lossy();
|
||
|
let transaction =
|
||
|
Transaction::from_game(shop_id, &mod_name, local_form_id, is_sell, quantity, amount);
|
||
|
info!(
|
||
|
"create_transaction api_url: {:?}, api_key: {:?}, transaction: {:?}",
|
||
|
api_url, api_key, transaction
|
||
|
);
|
||
|
|
||
|
fn inner(api_url: &str, api_key: &str, transaction: Transaction) -> Result<Transaction> {
|
||
|
#[cfg(not(test))]
|
||
|
let url = Url::parse(api_url)?.join("v1/transactions")?;
|
||
|
#[cfg(test)]
|
||
|
let url = Url::parse(&mockito::server_url())?.join("v1/transactions")?;
|
||
|
|
||
|
let client = reqwest::blocking::Client::new();
|
||
|
let resp = client
|
||
|
.post(url)
|
||
|
.header("Api-Key", api_key)
|
||
|
.json(&transaction)
|
||
|
.send()?;
|
||
|
info!("create transaction response from api: {:?}", &resp);
|
||
|
let bytes = resp.bytes()?;
|
||
|
let json: Transaction = serde_json::from_slice(&bytes)?;
|
||
|
if let Some(id) = json.id {
|
||
|
update_file_cache(
|
||
|
&file_cache_dir(api_url)?.join(format!("transaction_{}.json", id)),
|
||
|
&bytes,
|
||
|
)?;
|
||
|
}
|
||
|
Ok(json)
|
||
|
}
|
||
|
|
||
|
match inner(&api_url, &api_key, transaction) {
|
||
|
Ok(transaction) => {
|
||
|
if let Ok(raw_transaction) = RawTransaction::try_from(transaction) {
|
||
|
FFIResult::Ok(raw_transaction)
|
||
|
} else {
|
||
|
error!("create_transaction failed. API did not return a transaction with an ID");
|
||
|
let err_string =
|
||
|
CString::new("API did not return a transaction with an ID".to_string())
|
||
|
.expect("could not create CString")
|
||
|
.into_raw();
|
||
|
// TODO: also need to drop this CString once C++ is done reading it
|
||
|
FFIResult::Err(err_string)
|
||
|
}
|
||
|
}
|
||
|
Err(err) => {
|
||
|
error!("create_transaction failed. {}", err);
|
||
|
// TODO: also need to drop this CString once C++ is done reading it
|
||
|
let err_string = CString::new(err.to_string())
|
||
|
.expect("could not create CString")
|
||
|
.into_raw();
|
||
|
FFIResult::Err(err_string)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[cfg(test)]
|
||
|
mod tests {
|
||
|
use std::ffi::CString;
|
||
|
|
||
|
use super::*;
|
||
|
use mockito::mock;
|
||
|
|
||
|
#[test]
|
||
|
fn test_create_transaction() {
|
||
|
let mock = mock("POST", "/v1/transactions")
|
||
|
.with_status(201)
|
||
|
.with_header("content-type", "application/json")
|
||
|
.with_body(
|
||
|
r#"{
|
||
|
"amount": 100,
|
||
|
"created_at": "2020-08-18T00:00:00.000",
|
||
|
"id": 1,
|
||
|
"is_sell": false,
|
||
|
"local_form_id": 1,
|
||
|
"mod_name": "Skyrim.esm",
|
||
|
"owner_id": 1,
|
||
|
"quantity": 1,
|
||
|
"shop_id": 1,
|
||
|
"updated_at": "2020-08-18T00:00:00.000"
|
||
|
}"#,
|
||
|
)
|
||
|
.create();
|
||
|
|
||
|
let api_url = CString::new("url").unwrap().into_raw();
|
||
|
let api_key = CString::new("api-key").unwrap().into_raw();
|
||
|
let mod_name = CString::new("Skyrim.esm").unwrap().into_raw();
|
||
|
let result = create_transaction(api_url, api_key, 1, mod_name, 1, false, 1, 100);
|
||
|
mock.assert();
|
||
|
match result {
|
||
|
FFIResult::Ok(raw_transaction) => {
|
||
|
assert_eq!(raw_transaction.id, 1);
|
||
|
assert_eq!(raw_transaction.shop_id, 1);
|
||
|
assert_eq!(
|
||
|
unsafe { CStr::from_ptr(raw_transaction.mod_name).to_string_lossy() },
|
||
|
"Skyrim.esm"
|
||
|
);
|
||
|
assert_eq!(raw_transaction.local_form_id, 1);
|
||
|
assert_eq!(raw_transaction.is_sell, false);
|
||
|
assert_eq!(raw_transaction.quantity, 1);
|
||
|
assert_eq!(raw_transaction.amount, 100);
|
||
|
}
|
||
|
FFIResult::Err(error) => panic!("create_transaction returned error: {:?}", unsafe {
|
||
|
CStr::from_ptr(error).to_string_lossy()
|
||
|
}),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn test_create_transaction_server_error() {
|
||
|
let mock = mock("POST", "/v1/transactions")
|
||
|
.with_status(500)
|
||
|
.with_body("Internal Server Error")
|
||
|
.create();
|
||
|
|
||
|
let api_url = CString::new("url").unwrap().into_raw();
|
||
|
let api_key = CString::new("api-key").unwrap().into_raw();
|
||
|
let mod_name = CString::new("Skyrim.esm").unwrap().into_raw();
|
||
|
let result = create_transaction(api_url, api_key, 1, mod_name, 1, false, 1, 100);
|
||
|
mock.assert();
|
||
|
match result {
|
||
|
FFIResult::Ok(raw_transaction) => panic!(
|
||
|
"create_transaction returned Ok result: {:#?}",
|
||
|
raw_transaction
|
||
|
),
|
||
|
FFIResult::Err(error) => {
|
||
|
assert_eq!(
|
||
|
unsafe { CStr::from_ptr(error).to_string_lossy() },
|
||
|
"expected value at line 1 column 1"
|
||
|
);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|