rbcsv 0.1.6 → 0.1.8

data/ext/rbcsv/src/ruby_api.rs

@@ -0,0 +1,111 @@
+use magnus::{Error as MagnusError, Ruby};
+use crate::parser::{parse_csv_core, parse_csv_file, write_csv_file};
+
+/// CSV文字列をパースする（通常版）
+///
+/// # Arguments
+/// * `ruby` - Ruby VMの参照
+/// * `s` - パースするCSV文字列
+///
+/// # Returns
+/// * `Result<Vec<Vec<String>>, MagnusError>` - パース結果またはエラー
+pub fn parse(ruby: &Ruby, s: String) -> Result<Vec<Vec<String>>, MagnusError> {
+    parse_csv_core(&s, csv::Trim::None)
+        .map_err(|e| MagnusError::new(ruby.exception_runtime_error(), e.to_string()))
+}
+
+/// CSV文字列をパースする（trim版）
+///
+/// # Arguments
+/// * `ruby` - Ruby VMの参照
+/// * `s` - パースするCSV文字列
+///
+/// # Returns
+/// * `Result<Vec<Vec<String>>, MagnusError>` - パース結果またはエラー
+pub fn parse_trim(ruby: &Ruby, s: String) -> Result<Vec<Vec<String>>, MagnusError> {
+    parse_csv_core(&s, csv::Trim::All)
+        .map_err(|e| MagnusError::new(ruby.exception_runtime_error(), e.to_string()))
+}
+
+/// CSVファイルを読み込む（通常版）
+///
+/// # Arguments
+/// * `ruby` - Ruby VMの参照
+/// * `file_path` - 読み込むCSVファイルのパス
+///
+/// # Returns
+/// * `Result<Vec<Vec<String>>, MagnusError>` - パース結果またはエラー
+pub fn read(ruby: &Ruby, file_path: String) -> Result<Vec<Vec<String>>, MagnusError> {
+    parse_csv_file(&file_path, csv::Trim::None)
+        .map_err(|e| MagnusError::new(ruby.exception_runtime_error(), e.to_string()))
+}
+
+/// CSVファイルを読み込む（trim版）
+///
+/// # Arguments
+/// * `ruby` - Ruby VMの参照
+/// * `file_path` - 読み込むCSVファイルのパス
+///
+/// # Returns
+/// * `Result<Vec<Vec<String>>, MagnusError>` - パース結果またはエラー
+pub fn read_trim(ruby: &Ruby, file_path: String) -> Result<Vec<Vec<String>>, MagnusError> {
+    parse_csv_file(&file_path, csv::Trim::All)
+        .map_err(|e| MagnusError::new(ruby.exception_runtime_error(), e.to_string()))
+}
+
+/// CSVファイルに書き込む
+///
+/// # Arguments
+/// * `ruby` - Ruby VMの参照
+/// * `file_path` - 書き込み先ファイルのパス
+/// * `data` - 書き込むCSVデータ（2次元配列）
+///
+/// # Returns
+/// * `Result<(), MagnusError>` - 成功時は空、失敗時はエラー
+pub fn write(ruby: &Ruby, file_path: String, data: Vec<Vec<String>>) -> Result<(), MagnusError> {
+    write_csv_file(&file_path, &data)
+        .map_err(|e| MagnusError::new(ruby.exception_runtime_error(), e.to_string()))
+}
+
+
+#[cfg(test)]
+mod tests {
+
+    #[test]
+    fn test_parse_basic() {
+        let csv_data = "a,b,c\n1,2,3";
+        let result = crate::parser::parse_csv_core(csv_data, csv::Trim::None);
+
+        assert!(result.is_ok());
+        let records = result.unwrap();
+        assert_eq!(records.len(), 2);
+        assert_eq!(records[0], vec!["a", "b", "c"]);
+        assert_eq!(records[1], vec!["1", "2", "3"]);
+    }
+
+    #[test]
+    fn test_parse_with_trim_enabled() {
+        let csv_data = " a , b , c \n 1 , 2 , 3 ";
+        let result = crate::parser::parse_csv_core(csv_data, csv::Trim::All);
+
+        assert!(result.is_ok());
+        let records = result.unwrap();
+        assert_eq!(records[0], vec!["a", "b", "c"]);
+        assert_eq!(records[1], vec!["1", "2", "3"]);
+    }
+
+    #[test]
+    fn test_parse_with_trim_disabled() {
+        let csv_data = " a , b , c \n 1 , 2 , 3 ";
+        let result = crate::parser::parse_csv_core(csv_data, csv::Trim::None);
+
+        assert!(result.is_ok());
+        let records = result.unwrap();
+        assert_eq!(records[0], vec![" a ", " b ", " c "]);
+        assert_eq!(records[1], vec![" 1 ", " 2 ", " 3 "]);
+    }
+
+    // Note: Ruby API functions that return MagnusError cannot be tested
+    // in unit tests because they require a Ruby VM context.
+    // File reading functionality is tested in the parser module.
+}

data/lib/rbcsv/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RbCsv
-  VERSION = "0.1.6"
+  VERSION = "0.1.8"
 end

data/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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rbcsv
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.8
 platform: ruby
 authors:
 - fujitani sora
@@ -33,6 +33,7 @@ extensions:
 extra_rdoc_files: []
 files:
 - ".ruby-version"
+- ".rust-analyzer.json"
 - ".serena/.gitignore"
 - ".serena/memories/code_style_conventions.md"
 - ".serena/memories/project_overview.md"
@@ -50,7 +51,10 @@ files:
 - benchmark.rb
 - ext/rbcsv/Cargo.toml
 - ext/rbcsv/extconf.rb
+- ext/rbcsv/src/error.rs
 - ext/rbcsv/src/lib.rs
+- ext/rbcsv/src/parser.rs
+- ext/rbcsv/src/ruby_api.rs
 - lib/rbcsv.rb
 - lib/rbcsv/version.rb
 - output_comparison.rb
@@ -60,6 +64,7 @@ files:
 - test.rb
 - test_fixed.rb
 - test_install.rb
+- test_write_functionality.rb
 homepage: https://github.com/fs0414/rbcsv
 licenses:
 - MIT