summaryrefslogtreecommitdiffstats
path: root/internals/src/bind/encoded_word/impls.rs
blob: f7051da8a55108429d334060a1eaa982fe09d77a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
use soft_ascii_string::{ SoftAsciiString, SoftAsciiChar, SoftAsciiStr };

use vec1::Vec1;
use grammar::encoded_word::{ MAX_ECW_LEN, ECW_SEP_OVERHEAD };
use ::encoder::EncodingWriter;
use super::{ EncodedWordWriter, EncodedWordEncoding as Encoding };

pub struct VecWriter<'a> {
    data: Vec1<SoftAsciiString >,
    charset: &'a SoftAsciiStr,
    encoding: Encoding
}

impl<'a> VecWriter<'a> {
    pub fn new(charset: &'a SoftAsciiStr, encoding: Encoding) -> Self {
        let data = Vec1::new( SoftAsciiString::new() );
        VecWriter { data, charset, encoding }
    }

    pub fn data( &self ) -> &[SoftAsciiString] {
        &*self.data
    }
}

impl<'a> Into<Vec1<SoftAsciiString>> for VecWriter<'a> {
    fn into(self) -> Vec1<SoftAsciiString> {
        self.data
    }
}

impl<'a> EncodedWordWriter for VecWriter<'a> {

    fn encoding( &self ) -> Encoding {
        self.encoding
    }

    fn write_char( &mut self, ch: SoftAsciiChar ) {
        self.data.last_mut().push( ch );
    }

    fn write_charset( &mut self ) {
        self.data.last_mut().extend( self.charset.chars() )
    }

    fn write_ecw_seperator( &mut self ) {
        self.data.push( SoftAsciiString::new() )
    }

    fn max_payload_len( &self ) -> usize {
        MAX_ECW_LEN - ECW_SEP_OVERHEAD - self.charset.len() - 1
    }
}

pub struct WriterWrapper<'a, 'b: 'a>{
    charset: &'a SoftAsciiStr,
    encoding: Encoding,
    encoder_handle: &'a mut EncodingWriter<'b>
}

impl<'a, 'b: 'a> WriterWrapper<'a, 'b> {
    pub fn new_with_charset(charset: &'a SoftAsciiStr,
                            encoding: Encoding,
                            encoder: &'a mut EncodingWriter<'b> ) -> Self
    {
        WriterWrapper { charset, encoding, encoder_handle: encoder }
    }

    pub fn new(encoding: Encoding,
               encoder: &'a mut EncodingWriter<'b> ) -> Self
    {
        Self::new_with_charset(SoftAsciiStr::from_unchecked("utf8"), encoding, encoder)
    }
}

impl<'a, 'b: 'a> EncodedWordWriter for WriterWrapper<'a, 'b> {

    fn encoding( &self ) -> Encoding {
        self.encoding
    }

    fn write_charset( &mut self ) {
        //TODO fix
        let _ = self.encoder_handle.write_str( self.charset );
    }

    fn write_ecw_seperator( &mut self ) {
        self.encoder_handle.write_fws();
    }

    fn write_char( &mut self, ch: SoftAsciiChar ) {
        //TODO fix
        let _ = self.encoder_handle.write_char( ch );
    }

    fn max_payload_len( &self ) -> usize {
        MAX_ECW_LEN - ECW_SEP_OVERHEAD - self.charset.len() - 1
    }
}