summaryrefslogtreecommitdiff
path: root/lib/csv.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/csv.rb')
-rw-r--r--lib/csv.rb21
1 files changed, 20 insertions, 1 deletions
diff --git a/lib/csv.rb b/lib/csv.rb
index 143909a099..cfd1b8621e 100644
--- a/lib/csv.rb
+++ b/lib/csv.rb
@@ -1057,10 +1057,29 @@ class CSV
out_options[key] = value
end
end
+
# build input and output wrappers
- input = new(input || ARGF, **in_options)
+ input = new(input || ARGF, **in_options)
output = new(output || $stdout, **out_options)
+ # process headers
+ need_manual_header_output =
+ (in_options[:headers] and
+ out_options[:headers] == true and
+ out_options[:write_headers])
+ if need_manual_header_output
+ first_row = input.shift
+ if first_row
+ if first_row.is_a?(Row)
+ headers = first_row.headers
+ yield headers
+ output << headers
+ end
+ yield first_row
+ output << first_row
+ end
+ end
+
# read, yield, write
input.each do |row|
yield row