rbcsv 0.1.7 → 0.2.0

data/examples/features/test_typed_functionality.rb

@@ -0,0 +1,109 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require_relative '../../lib/rbcsv'
+
+puts "=== RbCsv 型認識機能テスト ==="
+puts
+
+# テストデータ
+csv_data = <<~CSV
+  name,age,score,rating
+  Alice,25,85.5,A
+  Bob,30,92,B+
+  Charlie,0,100.0,S
+CSV
+
+puts "元のCSVデータ:"
+puts csv_data
+puts
+
+# 通常のparseテスト（すべて文字列）
+puts "1. RbCsv.parse (すべて文字列):"
+result = RbCsv.parse(csv_data)
+result.each_with_index do |row, i|
+  puts "Row #{i}: #{row.inspect}"
+  if i > 0  # ヘッダー以外
+    puts "  age (#{row[1].class}): #{row[1]}"
+    puts "  score (#{row[2].class}): #{row[2]}"
+  end
+end
+puts
+
+# 型認識parseテスト
+puts "2. RbCsv.parse_typed (数値は数値型):"
+result_typed = RbCsv.parse_typed(csv_data)
+result_typed.each_with_index do |row, i|
+  puts "Row #{i}: #{row.inspect}"
+  if i > 0  # ヘッダー以外
+    puts "  age (#{row[1].class}): #{row[1]}"
+    puts "  score (#{row[2].class}): #{row[2]}"
+    puts "  計算可能: age * 2 = #{row[1] * 2}"
+  end
+end
+puts
+
+# エッジケースのテスト
+edge_case_csv = <<~CSV
+  type,value
+  integer,123
+  negative,-456
+  float,45.6
+  scientific,1.23e-4
+  empty,
+  text,hello world
+  mixed,123abc
+CSV
+
+puts "3. エッジケーステスト:"
+puts "CSVデータ:"
+puts edge_case_csv
+puts
+
+puts "RbCsv.parse_typed の結果:"
+result_edge = RbCsv.parse_typed(edge_case_csv)
+result_edge.each_with_index do |row, i|
+  if i > 0  # ヘッダー以外
+    value = row[1]
+    type_name = value.class.name
+    puts "#{row[0]}: #{value.inspect} (#{type_name})"
+  end
+end
+puts
+
+# trim版のテスト
+csv_with_spaces = "  name  ,  age  ,  score  \n  Alice  ,  25  ,  85.5  "
+
+puts "4. RbCsv.parse_typed! (trim + 型認識):"
+puts "CSVデータ（空白付き）: #{csv_with_spaces.inspect}"
+result_trim = RbCsv.parse_typed!(csv_with_spaces)
+result_trim.each do |row|
+  puts "Row: #{row.inspect}"
+end
+puts
+
+# ファイル書き込み→型認識読み込みテスト
+test_file = '/tmp/test_typed.csv'
+write_data = [
+  ['product', 'price', 'quantity', 'in_stock'],
+  ['Apple', '100', '50', 'true'],
+  ['Orange', '80.5', '30', 'false'],
+  ['Banana', '60.25', '0', 'yes']
+]
+
+puts "5. ファイル書き込み→型認識読み込みテスト:"
+RbCsv.write(test_file, write_data)
+puts "書き込み完了: #{test_file}"
+
+read_typed = RbCsv.read_typed(test_file)
+puts "RbCsv.read_typed の結果:"
+read_typed.each_with_index do |row, i|
+  puts "Row #{i}: #{row.inspect}"
+  if i > 0
+    puts "  price (#{row[1].class}): #{row[1]}"
+    puts "  quantity (#{row[2].class}): #{row[2]}"
+  end
+end
+puts
+
+puts "=== テスト完了 ==="

data/examples/features/test_write_functionality.rb

@@ -0,0 +1,225 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# RbCsv.write() 機能のテストスクリプト
+#
+# 実行方法:
+#   ruby test_write_functionality.rb
+#
+# 前提条件:
+#   - bundle exec rake compile でライブラリがビルド済みであること
+
+require_relative '../../lib/rbcsv'
+require 'fileutils'
+
+class RbCsvWriteTest
+  def initialize
+    @test_dir = '/tmp/rbcsv_write_tests'
+    @success_count = 0
+    @total_count = 0
+    setup_test_directory
+  end
+
+  def run_all_tests
+    puts "=" * 60
+    puts "RbCsv.write() 機能テスト開始"
+    puts "=" * 60
+    puts
+
+    test_basic_write
+    test_roundtrip_write_read
+    test_file_overwrite
+    test_empty_data_error
+    test_field_count_mismatch_error
+    test_single_row_write
+    test_unicode_content
+    test_special_characters
+
+    puts
+    puts "=" * 60
+    puts "テスト結果: #{@success_count}/#{@total_count} 成功"
+    puts "=" * 60
+
+    cleanup_test_directory
+
+    if @success_count == @total_count
+      puts "✅ すべてのテストが成功しました！"
+      exit 0
+    else
+      puts "❌ 一部のテストが失敗しました"
+      exit 1
+    end
+  end
+
+  private
+
+  def setup_test_directory
+    FileUtils.mkdir_p(@test_dir)
+    puts "テストディレクトリを作成: #{@test_dir}"
+  end
+
+  def cleanup_test_directory
+    FileUtils.rm_rf(@test_dir)
+    puts "テストディレクトリを削除: #{@test_dir}"
+  end
+
+  def run_test(name)
+    @total_count += 1
+    print "#{name}... "
+
+    begin
+      yield
+      @success_count += 1
+      puts "✅ 成功"
+    rescue => e
+      puts "❌ 失敗: #{e.message}"
+      puts "   #{e.backtrace.first}"
+    end
+  end
+
+  def test_basic_write
+    run_test("基本的なCSV書き込み") do
+      file_path = File.join(@test_dir, 'basic.csv')
+      data = [
+        ['name', 'age', 'city'],
+        ['Alice', '25', 'Tokyo'],
+        ['Bob', '30', 'Osaka']
+      ]
+
+      RbCsv.write(file_path, data)
+
+      content = File.read(file_path)
+      expected = "name,age,city\nAlice,25,Tokyo\nBob,30,Osaka\n"
+
+      raise "書き込み内容が期待値と異なります" unless content == expected
+    end
+  end
+
+  def test_roundtrip_write_read
+    run_test("書き込み→読み込みの往復テスト") do
+      file_path = File.join(@test_dir, 'roundtrip.csv')
+      original_data = [
+        ['product', 'price', 'category'],
+        ['Apple', '100', 'Fruit'],
+        ['Carrot', '50', 'Vegetable']
+      ]
+
+      RbCsv.write(file_path, original_data)
+      read_data = RbCsv.read(file_path)
+
+      raise "往復テストで元データと異なります" unless original_data == read_data
+    end
+  end
+
+  def test_file_overwrite
+    run_test("ファイル上書きテスト") do
+      file_path = File.join(@test_dir, 'overwrite.csv')
+
+      # 最初のデータ
+      first_data = [['old'], ['data']]
+      RbCsv.write(file_path, first_data)
+      first_content = File.read(file_path)
+
+      # 上書き
+      second_data = [['new', 'header'], ['updated', 'content']]
+      RbCsv.write(file_path, second_data)
+      second_content = File.read(file_path)
+
+      expected_first = "old\ndata\n"
+      expected_second = "new,header\nupdated,content\n"
+
+      raise "最初の書き込み内容が不正" unless first_content == expected_first
+      raise "上書き後の内容が不正" unless second_content == expected_second
+    end
+  end
+
+  def test_empty_data_error
+    run_test("空データエラーテスト") do
+      file_path = File.join(@test_dir, 'empty.csv')
+
+      error_raised = false
+      begin
+        RbCsv.write(file_path, [])
+      rescue RuntimeError => e
+        error_raised = true
+        raise "エラーメッセージが期待値と異なります" unless e.message.include?("CSV data is empty")
+      end
+
+      raise "空データでエラーが発生しませんでした" unless error_raised
+    end
+  end
+
+  def test_field_count_mismatch_error
+    run_test("フィールド数不一致エラーテスト") do
+      file_path = File.join(@test_dir, 'mismatch.csv')
+      inconsistent_data = [
+        ['name', 'age'],
+        ['Alice', '25', 'Tokyo']  # 3フィールド（期待は2フィールド）
+      ]
+
+      error_raised = false
+      begin
+        RbCsv.write(file_path, inconsistent_data)
+      rescue RuntimeError => e
+        error_raised = true
+        unless e.message.include?("Field count mismatch") && e.message.include?("line 2")
+          raise "エラーメッセージが期待値と異なります: #{e.message}"
+        end
+      end
+
+      raise "フィールド数不一致でエラーが発生しませんでした" unless error_raised
+    end
+  end
+
+  def test_single_row_write
+    run_test("単一行書き込みテスト") do
+      file_path = File.join(@test_dir, 'single.csv')
+      data = [['single', 'row', 'test']]
+
+      RbCsv.write(file_path, data)
+      content = File.read(file_path)
+      expected = "single,row,test\n"
+
+      raise "単一行の書き込み内容が不正" unless content == expected
+    end
+  end
+
+  def test_unicode_content
+    run_test("Unicode文字テスト") do
+      file_path = File.join(@test_dir, 'unicode.csv')
+      data = [
+        ['名前', '年齢', '都市'],
+        ['田中太郎', '30', '東京'],
+        ['山田花子', '25', '大阪'],
+        ['🎉', '😀', '🌸']
+      ]
+
+      RbCsv.write(file_path, data)
+      read_data = RbCsv.read(file_path)
+
+      raise "Unicode文字の往復テストが失敗" unless data == read_data
+    end
+  end
+
+  def test_special_characters
+    run_test("特殊文字テスト") do
+      file_path = File.join(@test_dir, 'special.csv')
+      data = [
+        ['field1', 'field2', 'field3'],
+        ['comma,test', 'quote"test', 'newline\ntest'],
+        ['tab\ttest', 'backslash\\test', 'normal']
+      ]
+
+      RbCsv.write(file_path, data)
+      read_data = RbCsv.read(file_path)
+
+      raise "特殊文字の往復テストが失敗" unless data == read_data
+    end
+  end
+end
+
+# テスト実行
+if __FILE__ == $0
+  tester = RbCsvWriteTest.new
+  tester.run_all_tests
+end

data/ext/rbcsv/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "rbcsv"
-version = "0.1.0"
+version = "0.2.0"
 edition = "2021"
 authors = ["fujitani sora <fujitanisora0414@gmail.com>"]
 license = "MIT"

data/ext/rbcsv/src/error.rs

@@ -19,6 +19,10 @@ pub enum ErrorKind {
     FieldCountMismatch,
     // 空のCSVデータ
     EmptyData,
+    // 書き込み権限エラー
+    WritePermission,
+    // 無効なデータエラー
+    InvalidData,
     // その他のエラー
     #[allow(dead_code)]
     Other,
@@ -32,6 +36,8 @@ impl fmt::Display for CsvError {
             ErrorKind::Encoding => write!(f, "Encoding Error: {}", self.message),
             ErrorKind::FieldCountMismatch => write!(f, "Field Count Mismatch: {}", self.message),
             ErrorKind::EmptyData => write!(f, "Empty Data: {}", self.message),
+            ErrorKind::WritePermission => write!(f, "Write Permission Error: {}", self.message),
+            ErrorKind::InvalidData => write!(f, "Invalid Data Error: {}", self.message),
             ErrorKind::Other => write!(f, "Error: {}", self.message),
         }
     }
@@ -70,9 +76,17 @@ impl CsvError {
     pub fn empty_data() -> Self {
         Self::new(ErrorKind::EmptyData, "CSV data is empty")
     }
+
+    pub fn write_permission(message: impl Into<String>) -> Self {
+        Self::new(ErrorKind::WritePermission, message)
+    }
+
+    pub fn invalid_data(message: impl Into<String>) -> Self {
+        Self::new(ErrorKind::InvalidData, message)
+    }
 }
 
-// csv crateのエラーからの変換
+// csv crate error to CsvError conversion
 impl From<csv::Error> for CsvError {
     fn from(err: csv::Error) -> Self {
         match err.kind() {
@@ -84,4 +98,4 @@ impl From<csv::Error> for CsvError {
             _ => CsvError::parse(err.to_string()),
         }
     }
-}
+}

data/ext/rbcsv/src/lib.rs

@@ -1,9 +1,10 @@
 mod error;
 mod parser;
 mod ruby_api;
+mod value;
 
 use magnus::{Object, Ruby};
-use ruby_api::{parse, parse_trim, read, read_trim};
+use ruby_api::{parse, parse_trim, read, read_trim, write, parse_typed, parse_typed_trim, read_typed, read_typed_trim};
 
 #[magnus::init]
 fn init(ruby: &Ruby) -> Result<(), magnus::Error> {
@@ -13,6 +14,13 @@ fn init(ruby: &Ruby) -> Result<(), magnus::Error> {
     module.define_singleton_method("parse!", magnus::function!(parse_trim, 1))?;
     module.define_singleton_method("read", magnus::function!(read, 1))?;
     module.define_singleton_method("read!", magnus::function!(read_trim, 1))?;
+    module.define_singleton_method("write", magnus::function!(write, 2))?;
+
+    // typed variants
+    module.define_singleton_method("parse_typed", magnus::function!(parse_typed, 1))?;
+    module.define_singleton_method("parse_typed!", magnus::function!(parse_typed_trim, 1))?;
+    module.define_singleton_method("read_typed", magnus::function!(read_typed, 1))?;
+    module.define_singleton_method("read_typed!", magnus::function!(read_typed_trim, 1))?;
 
     Ok(())
 }