Source code
Revision control
Copy as Markdown
Other Tools
/* -*- Mode: rust; rust-indent-offset: 2 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
extern crate uritemplate;
use uritemplate::UriTemplate;
use std::{ptr, str};
use nserror::{nsresult, NS_ERROR_INVALID_ARG, NS_OK};
extern crate nsstring;
use nsstring::nsACString;
use xpcom::{AtomicRefcnt, RefCounted, RefPtr};
pub struct UriTemplateWrapper {
builder: UriTemplate,
refcnt: AtomicRefcnt,
}
impl UriTemplateWrapper {
fn new(input: &nsACString) -> Result<RefPtr<UriTemplateWrapper>, nsresult> {
let template = str::from_utf8(input).map_err(|_| NS_ERROR_INVALID_ARG)?;
let builder: *mut UriTemplateWrapper = Box::into_raw(Box::new(Self {
builder: UriTemplate::new(template),
refcnt: unsafe { AtomicRefcnt::new() },
}));
unsafe { Ok(RefPtr::from_raw(builder).unwrap()) }
}
}
#[no_mangle]
pub unsafe extern "C" fn uri_template_addref(builder: &UriTemplateWrapper) {
builder.refcnt.inc();
}
#[no_mangle]
pub unsafe extern "C" fn uri_template_release(builder: &UriTemplateWrapper) {
let rc = builder.refcnt.dec();
if rc == 0 {
drop(Box::from_raw(ptr::from_ref(builder).cast_mut()));
}
}
// xpcom::RefPtr support
unsafe impl RefCounted for UriTemplateWrapper {
unsafe fn addref(&self) {
uri_template_addref(self);
}
unsafe fn release(&self) {
uri_template_release(self);
}
}
#[no_mangle]
pub extern "C" fn uri_template_new(input: &nsACString, result: &mut *const UriTemplateWrapper) {
*result = ptr::null_mut();
if let Ok(builder) = UriTemplateWrapper::new(input) {
builder.forget(result);
}
}
#[no_mangle]
pub extern "C" fn uri_template_set(
builder: &mut UriTemplateWrapper,
name: &nsACString,
value: &nsACString,
) -> nsresult {
let Ok(name_conv) = str::from_utf8(name) else {
return NS_ERROR_INVALID_ARG;
};
let Ok(value_conv) = str::from_utf8(value) else {
return NS_ERROR_INVALID_ARG;
};
builder.builder.set(name_conv, value_conv);
NS_OK
}
#[no_mangle]
pub extern "C" fn uri_template_set_int(
builder: &mut UriTemplateWrapper,
name: &nsACString,
value: i32,
) -> nsresult {
let Ok(name_conv) = str::from_utf8(name) else {
return NS_ERROR_INVALID_ARG;
};
builder.builder.set(name_conv, value.to_string());
NS_OK
}
#[no_mangle]
pub extern "C" fn uri_template_build(builder: &mut UriTemplateWrapper, result: &mut nsACString) {
let res = builder.builder.build();
result.assign(&res);
}