~michaelforrest/use-case-mapper/trunk

« back to all changes in this revision

Viewing changes to vendor/rails/actionmailer/lib/action_mailer/vendor/tmail-1.2.3/tmail/stringio.rb

  • Committer: Michael Forrest
  • Date: 2010-10-15 16:28:50 UTC
  • Revision ID: michael.forrest@canonical.com-20101015162850-tj2vchanv0kr0dun
refrozeĀ gems

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# encoding: utf-8
 
2
=begin rdoc
 
3
 
 
4
= String handling class
 
5
 
 
6
=end
 
7
#--
 
8
# Copyright (c) 1998-2003 Minero Aoki <aamine@loveruby.net>
 
9
#
 
10
# Permission is hereby granted, free of charge, to any person obtaining
 
11
# a copy of this software and associated documentation files (the
 
12
# "Software"), to deal in the Software without restriction, including
 
13
# without limitation the rights to use, copy, modify, merge, publish,
 
14
# distribute, sublicense, and/or sell copies of the Software, and to
 
15
# permit persons to whom the Software is furnished to do so, subject to
 
16
# the following conditions:
 
17
#
 
18
# The above copyright notice and this permission notice shall be
 
19
# included in all copies or substantial portions of the Software.
 
20
#
 
21
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 
22
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 
23
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 
24
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
 
25
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
 
26
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 
27
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
28
#
 
29
# Note: Originally licensed under LGPL v2+. Using MIT license for Rails
 
30
# with permission of Minero Aoki.
 
31
#++
 
32
 
 
33
class StringInput#:nodoc:
 
34
 
 
35
  include Enumerable
 
36
 
 
37
  class << self
 
38
 
 
39
    def new( str )
 
40
      if block_given?
 
41
        begin
 
42
          f = super
 
43
          yield f
 
44
        ensure
 
45
          f.close if f
 
46
        end
 
47
      else
 
48
        super
 
49
      end
 
50
    end
 
51
 
 
52
    alias open new
 
53
  
 
54
  end
 
55
 
 
56
  def initialize( str )
 
57
    @src = str
 
58
    @pos = 0
 
59
    @closed = false
 
60
    @lineno = 0
 
61
  end
 
62
 
 
63
  attr_reader :lineno
 
64
 
 
65
  def string
 
66
    @src
 
67
  end
 
68
 
 
69
  def inspect
 
70
    "#<#{self.class}:#{@closed ? 'closed' : 'open'},src=#{@src[0,30].inspect}>"
 
71
  end
 
72
 
 
73
  def close
 
74
    stream_check!
 
75
    @pos = nil
 
76
    @closed = true
 
77
  end
 
78
 
 
79
  def closed?
 
80
    @closed
 
81
  end
 
82
 
 
83
  def pos
 
84
    stream_check!
 
85
    [@pos, @src.size].min
 
86
  end
 
87
 
 
88
  alias tell pos
 
89
 
 
90
  def seek( offset, whence = IO::SEEK_SET )
 
91
    stream_check!
 
92
    case whence
 
93
    when IO::SEEK_SET
 
94
      @pos = offset
 
95
    when IO::SEEK_CUR
 
96
      @pos += offset
 
97
    when IO::SEEK_END
 
98
      @pos = @src.size - offset
 
99
    else
 
100
      raise ArgumentError, "unknown seek flag: #{whence}"
 
101
    end
 
102
    @pos = 0 if @pos < 0
 
103
    @pos = [@pos, @src.size + 1].min
 
104
    offset
 
105
  end
 
106
 
 
107
  def rewind
 
108
    stream_check!
 
109
    @pos = 0
 
110
  end
 
111
 
 
112
  def eof?
 
113
    stream_check!
 
114
    @pos > @src.size
 
115
  end
 
116
 
 
117
  def each( &block )
 
118
    stream_check!
 
119
    begin
 
120
      @src.each(&block)
 
121
    ensure
 
122
      @pos = 0
 
123
    end
 
124
  end
 
125
 
 
126
  def gets
 
127
    stream_check!
 
128
    if idx = @src.index(?\n, @pos)
 
129
      idx += 1  # "\n".size
 
130
      line = @src[ @pos ... idx ]
 
131
      @pos = idx
 
132
      @pos += 1 if @pos == @src.size
 
133
    else
 
134
      line = @src[ @pos .. -1 ]
 
135
      @pos = @src.size + 1
 
136
    end
 
137
    @lineno += 1
 
138
 
 
139
    line
 
140
  end
 
141
 
 
142
  def getc
 
143
    stream_check!
 
144
    ch = @src[@pos]
 
145
    @pos += 1
 
146
    @pos += 1 if @pos == @src.size
 
147
    ch
 
148
  end
 
149
 
 
150
  def read( len = nil )
 
151
    stream_check!
 
152
    return read_all unless len
 
153
    str = @src[@pos, len]
 
154
    @pos += len
 
155
    @pos += 1 if @pos == @src.size
 
156
    str
 
157
  end
 
158
 
 
159
  alias sysread read
 
160
 
 
161
  def read_all
 
162
    stream_check!
 
163
    return nil if eof?
 
164
    rest = @src[@pos ... @src.size]
 
165
    @pos = @src.size + 1
 
166
    rest
 
167
  end
 
168
 
 
169
  def stream_check!
 
170
    @closed and raise IOError, 'closed stream'
 
171
  end
 
172
 
 
173
end
 
174
 
 
175
 
 
176
class StringOutput#:nodoc:
 
177
 
 
178
  class << self
 
179
 
 
180
    def new( str = '' )
 
181
      if block_given?
 
182
        begin
 
183
          f = super
 
184
          yield f
 
185
        ensure
 
186
          f.close if f
 
187
        end
 
188
      else
 
189
        super
 
190
      end
 
191
    end
 
192
 
 
193
    alias open new
 
194
  
 
195
  end
 
196
 
 
197
  def initialize( str = '' )
 
198
    @dest = str
 
199
    @closed = false
 
200
  end
 
201
 
 
202
  def close
 
203
    @closed = true
 
204
  end
 
205
 
 
206
  def closed?
 
207
    @closed
 
208
  end
 
209
 
 
210
  def string
 
211
    @dest
 
212
  end
 
213
 
 
214
  alias value string
 
215
  alias to_str string
 
216
 
 
217
  def size
 
218
    @dest.size
 
219
  end
 
220
 
 
221
  alias pos size
 
222
 
 
223
  def inspect
 
224
    "#<#{self.class}:#{@dest ? 'open' : 'closed'},#{object_id}>"
 
225
  end
 
226
 
 
227
  def print( *args )
 
228
    stream_check!
 
229
    raise ArgumentError, 'wrong # of argument (0 for >1)' if args.empty?
 
230
    args.each do |s|
 
231
      raise ArgumentError, 'nil not allowed' if s.nil?
 
232
      @dest << s.to_s
 
233
    end
 
234
    nil
 
235
  end
 
236
 
 
237
  def puts( *args )
 
238
    stream_check!
 
239
    args.each do |str|
 
240
      @dest << (s = str.to_s)
 
241
      @dest << "\n" unless s[-1] == ?\n
 
242
    end
 
243
    @dest << "\n" if args.empty?
 
244
    nil
 
245
  end
 
246
 
 
247
  def putc( ch )
 
248
    stream_check!
 
249
    @dest << ch.chr
 
250
    nil
 
251
  end
 
252
 
 
253
  def printf( *args )
 
254
    stream_check!
 
255
    @dest << sprintf(*args)
 
256
    nil
 
257
  end
 
258
 
 
259
  def write( str )
 
260
    stream_check!
 
261
    s = str.to_s
 
262
    @dest << s
 
263
    s.size
 
264
  end
 
265
 
 
266
  alias syswrite write
 
267
 
 
268
  def <<( str )
 
269
    stream_check!
 
270
    @dest << str.to_s
 
271
    self
 
272
  end
 
273
 
 
274
  private
 
275
 
 
276
  def stream_check!
 
277
    @closed and raise IOError, 'closed stream'
 
278
  end
 
279
 
 
280
end