just/src/cooked_string.rs

use common::*;

#[derive(PartialEq, Debug)]
pub struct CookedString<'a> {
  pub raw:    &'a str,
  pub cooked: String,
}

impl<'a> CookedString<'a> {
  pub fn new(token: &Token<'a>) -> CompilationResult<'a, CookedString<'a>> {
    let raw = &token.lexeme[1..token.lexeme.len()-1];

    if let TokenKind::RawString = token.kind {
      Ok(CookedString{cooked: raw.to_string(), raw})
    } else if let TokenKind::StringToken = token.kind {
      let mut cooked = String::new();
      let mut escape = false;
      for c in raw.chars() {
        if escape {
          match c {
            'n'   => cooked.push('\n'),
            'r'   => cooked.push('\r'),
            't'   => cooked.push('\t'),
            '\\'  => cooked.push('\\'),
            '"'   => cooked.push('"'),
            other => return Err(token.error(CompilationErrorKind::InvalidEscapeSequence {
              character: other,
            })),
          }
          escape = false;
          continue;
        }
        if c == '\\' {
          escape = true;
          continue;
        }
        cooked.push(c);
      }
      Ok(CookedString{raw, cooked})
    } else {
      Err(token.error(CompilationErrorKind::Internal {
        message: "cook_string() called on non-string token".to_string()
      }))
    }
  }
}