use crate::{error::TextError, Result};
use std::collections::HashMap;
use std::io::{BufRead, Read};
#[derive(Default)]
pub struct Po {
header: Option<HashMap<String, String>>,
entities: Entities,
contexts: HashMap<String, Entities>,
}
type Entities = HashMap<String, String>;
impl Po {
pub fn parse<R: Read>(reader: R) -> Result<Self> {
let mut reader = std::io::BufReader::new(reader);
let mut entities = HashMap::new();
let mut contexts: HashMap<String, Entities> = HashMap::new();
let mut line = String::new();
enum State {
None,
Context(String),
Msgid {
id: String,
ctx: Option<String>,
},
Entity {
msgid: String,
msgstr: String,
ctx: Option<String>,
},
}
let mut state = State::None;
loop {
line.clear();
let eof = reader.read_line(&mut line)?;
if eof == 0 {
match state {
State::Msgid { .. } | State::Context(..) => return Err(TextError::FormatError),
State::Entity { msgid, msgstr, ctx } => match ctx {
Some(ctx) => {
contexts.entry(ctx).or_default().insert(msgid, msgstr);
}
None => {
entities.insert(msgid, msgstr);
}
},
_ => (),
}
break;
}
let is_empty_line = line.is_empty() || line.trim().is_empty();
let is_comment = line.starts_with("#");
if is_empty_line || is_comment {
continue;
}
if line.starts_with("msgctxt") {
if let State::Entity { msgid, msgstr, ctx } = state {
match ctx {
Some(ctx) => {
contexts.entry(ctx).or_default().insert(msgid, msgstr);
}
None => {
entities.insert(msgid, msgstr);
}
}
}
let s: &str = line[7..].trim();
let context = unqoute(s).map(|s| s.to_owned())?;
state = State::Context(context);
continue;
}
if line.starts_with("msgid") {
let s: &str = line[5..].trim();
let id = unqoute(s).map(|s| s.to_owned())?;
match state {
State::Entity { msgid, msgstr, ctx } => match ctx {
Some(ctx) => {
contexts.entry(ctx).or_default().insert(msgid, msgstr);
state = State::Msgid { id, ctx: None };
}
None => {
entities.insert(msgid, msgstr);
state = State::Msgid { id, ctx: None };
}
},
State::Context(ctx) => {
state = State::Msgid { id, ctx: Some(ctx) };
}
_ => {
state = State::Msgid { id, ctx: None };
}
}
continue;
}
match state {
State::Msgid { id, ctx } if line.starts_with("msgstr") => {
let s: &str = line[6..].trim();
let msgstr = unqoute(s).map(|s| s.to_owned())?;
state = State::Entity {
msgid: id.clone(),
ctx,
msgstr,
};
continue;
}
State::Entity { ref mut msgstr, .. } if unqoute(line.trim()).is_ok() => {
let s = unqoute(line.trim()).unwrap();
msgstr.push_str(&s);
continue;
}
_ => return Err(TextError::FormatError),
}
}
let header = entities.get("").and_then(|s| Some(parse_header(s)));
Ok(Self {
entities,
contexts,
header,
})
}
pub fn get(&self, id: &str) -> Option<&str> {
self.entities.get(id).and_then(|s| Some(s.as_str()))
}
pub fn getc(&self, context: &str, id: &str) -> Option<&str> {
self.contexts
.get(context)
.and_then(|entities| entities.get(id))
.and_then(|s| Some(s.as_str()))
}
pub fn header(&self) -> Option<&HashMap<String, String>> {
self.header.as_ref()
}
}
fn parse_header(s: &str) -> HashMap<String, String> {
s.split("\\n")
.map(|line| {
line.find(":")
.and_then(|pos| Some((&line[..pos], &line[pos + 1..])))
})
.flatten()
.map(|(key, value)| (key.to_owned(), value.to_owned()))
.collect()
}
fn unqoute<'a>(s: &'a str) -> Result<&'a str> {
if !s.starts_with("\"") || !s.ends_with("\"") {
return Err(TextError::FormatError);
}
Ok(&s[1..s.len() - 1])
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn parse_po_file() {
let file = "msgid \"ask_location_menu.next_button\"\n\
msgstr \"Next\"\n";
let po = Po::parse(file.as_bytes()).unwrap();
assert_eq!(po.get("ask_location_menu.next_button"), Some("Next"));
}
#[test]
fn parse_po_file_messy() {
let file = "msgid \"ask_location_menu.next_button\" \n\
msgstr \"Next\" \n ";
let po = Po::parse(file.as_bytes()).unwrap();
assert_eq!(po.get("ask_location_menu.next_button"), Some("Next"));
}
#[test]
fn parse_po_file_emptylines() {
let file = r#"
msgid "ask_location_menu.next_button"
msgstr "Next""#;
let po = Po::parse(file.as_bytes()).unwrap();
assert_eq!(po.get("ask_location_menu.next_button"), Some("Next"));
}
#[test]
fn parse_po_file_comments() {
let file = r#"
# translator-comments
#. extracted-comments
#: reference…
#, flag…
msgid "ask_location_menu.next_button"
msgstr "Next""#;
let po = Po::parse(file.as_bytes()).unwrap();
assert_eq!(po.get("ask_location_menu.next_button"), Some("Next"));
}
#[test]
fn parse_po_file_empty() {
let file = "";
let po = Po::parse(file.as_bytes());
assert!(po.is_ok());
}
#[test]
fn parse_po_file_multi_entities() {
let file = "msgid \"id1\"\n\
msgstr \"v1\"\n\
msgid \"id2\"\n\
msgstr \"v2\"\n";
let po = Po::parse(file.as_bytes()).unwrap();
assert_eq!(po.get("id1"), Some("v1"));
assert_eq!(po.get("id2"), Some("v2"));
let file = "msgid \"id1\"\n\
msgstr \"v1\"\n\
\n\
msgid \"id2\"\n\
msgstr \"v2\"\n";
let po = Po::parse(file.as_bytes()).unwrap();
assert_eq!(po.get("id1"), Some("v1"));
assert_eq!(po.get("id2"), Some("v2"));
}
#[test]
fn parse_po_file_multiline() {
let file = "msgid \"id\"\n\
msgstr \"1\"\n\
\"2\"\n\
\"3\"\n";
let po = Po::parse(file.as_bytes()).unwrap();
assert_eq!(po.get("id"), Some("123"));
}
#[test]
fn parse_po_file_context() {
let file = "msgctxt \"default\"\n\
msgid \"id\"\n\
msgstr \"1\"\n";
let po = Po::parse(file.as_bytes()).unwrap();
assert_eq!(po.getc("default", "id"), Some("1"));
assert_eq!(po.get("id"), None);
}
}