use bitcoin::hashes::Hash;
use bitcoin::{bech32::u5, util::address::Payload, Address, Network};
use byteorder::{BigEndian, ByteOrder};
use chrono::{offset::Local, DateTime, Duration};
use lightning_invoice::{Currency, Fallback, Invoice, InvoiceDescription, RouteHop};
use serde::{Deserialize, Serialize};
const WRONG_CID: &'static str = "incorrect short channel ID HRF format";
pub fn parse_short_channel_id(cid: &str) -> u64 {
let mut split = cid.split("x");
let blocknum: u64 = split.next().expect(WRONG_CID).parse().expect(WRONG_CID);
if blocknum & 0xFFFFFF != blocknum {
panic!(WRONG_CID);
}
let txnum: u64 = split.next().expect(WRONG_CID).parse().expect(WRONG_CID);
if txnum & 0xFFFFFF != txnum {
panic!(WRONG_CID);
}
let outnum: u64 = split.next().expect(WRONG_CID).parse().expect(WRONG_CID);
if outnum & 0xFFFF != outnum {
panic!(WRONG_CID);
}
blocknum << 40 | txnum << 16 | outnum
}
pub fn fmt_short_channel_id(cid: u64) -> String {
let blocknum = cid >> 40;
let txnum = cid >> 16 & 0x00FFFFFF;
let outnum = cid & 0xFFFF;
format!("{}x{}x{}", blocknum, txnum, outnum)
}
#[derive(Clone, PartialEq, Eq, Debug, Deserialize, Serialize)]
pub struct RouteHopInfo {
pub pubkey: ::HexBytes,
pub short_channel_id: u64,
pub short_channel_id_hex: ::HexBytes,
pub short_channel_id_hrf: String,
pub fee_base_msat: u32,
pub fee_proportional_millionths: u32,
pub cltv_expiry_delta: u16,
}
impl ::GetInfo<RouteHopInfo> for RouteHop {
fn get_info(&self, _network: Network) -> RouteHopInfo {
let ssid_hex = &self.short_channel_id[..];
let ssid = BigEndian::read_u64(&ssid_hex);
RouteHopInfo {
pubkey: self.pubkey.serialize()[..].into(),
short_channel_id: ssid,
short_channel_id_hex: ssid_hex.into(),
short_channel_id_hrf: fmt_short_channel_id(ssid),
fee_base_msat: self.fee_base_msat,
fee_proportional_millionths: self.fee_proportional_millionths,
cltv_expiry_delta: self.cltv_expiry_delta,
}
}
}
#[derive(Clone, PartialEq, Eq, Debug, Deserialize, Serialize)]
pub struct InvoiceInfo {
pub timestamp: DateTime<Local>,
pub payment_hash: String, pub description: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub payee_pub_key: Option<::HexBytes>,
#[serde(skip_serializing_if = "Option::is_none")]
pub expiry_time: Option<DateTime<Local>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub min_final_cltv_expiry: Option<u64>,
#[serde(skip_serializing_if = "Vec::is_empty")]
pub fallback_addresses: Vec<Address>,
#[serde(skip_serializing_if = "Vec::is_empty")]
pub routes: Vec<Vec<RouteHopInfo>>,
pub currency: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub amount_pico_btc: Option<u64>,
}
impl ::GetInfo<InvoiceInfo> for Invoice {
fn get_info(&self, network: Network) -> InvoiceInfo {
InvoiceInfo {
timestamp: self.timestamp().clone().into(),
payment_hash: format!("{:?}", self.payment_hash()),
description: match self.description() {
InvoiceDescription::Direct(s) => s.clone().into_inner(),
InvoiceDescription::Hash(h) => format!("{:?}", h),
},
payee_pub_key: self.payee_pub_key().map(|pk| pk.serialize()[..].into()),
expiry_time: self.expiry_time().map(|e| {
let duration = Duration::from_std(*e.as_duration()).expect("invalid expiry");
Local::now() + duration
}),
min_final_cltv_expiry: self.min_final_cltv_expiry().map(|e| e.0),
fallback_addresses: self
.fallbacks()
.iter()
.map(|f| {
Address {
payload: match f {
Fallback::PubKeyHash(pkh) => {
Payload::PubkeyHash(Hash::from_slice(&pkh[..]).expect("wrong hash"))
}
Fallback::ScriptHash(sh) => {
Payload::ScriptHash(Hash::from_slice(&sh[..]).expect("wrong hash"))
}
Fallback::SegWitProgram {
version: v,
program: p,
} => Payload::WitnessProgram {
version: u5::try_from_u8(v.to_u8())
.expect("invalid segwit version"),
program: p.to_vec(),
},
},
network: network,
}
})
.collect(),
routes: self
.routes()
.iter()
.map(|r| r.iter().map(|h| ::GetInfo::get_info(h, network)).collect())
.collect(),
currency: match self.currency() {
Currency::Bitcoin => "bitcoin".to_owned(),
Currency::BitcoinTestnet => "bitcoin-testnet".to_owned(),
},
amount_pico_btc: self.amount_pico_btc(),
}
}
}