shard-crystal-sqlite3/src/sqlite3/statement2.cr

class SQLite3::Statement2 < DB::Statement
  def initialize(connection, sql)
    super(connection)
    check LibSQLite3.prepare_v2(@connection, sql, sql.bytesize + 1, out @stmt, nil)
  end

  protected def begin_parameters
    LibSQLite3.reset(self)
  end

  protected def on_close
    check LibSQLite3.finalize(self)
  end

  protected def add_parameter(index : Int32, value)
    bind_arg(index + 1, value)
  end

  protected def add_parameter(name : String, value)
    converted_name = ":#{name}"
    index = LibSQLite3.bind_parameter_index(self, converted_name)
    raise "Unknown parameter: #{name}" if index == 0
    bind_arg(index, value)
  end

  protected def perform
    ResultSet2.new(self)
  end

  private def bind_arg(index, value : Nil)
    check LibSQLite3.bind_null(self, index)
  end

  private def bind_arg(index, value : Int32)
    check LibSQLite3.bind_int(self, index, value)
  end

  private def bind_arg(index, value : Int64)
    check LibSQLite3.bind_int64(self, index, value)
  end

  private def bind_arg(index, value : Float32)
    check LibSQLite3.bind_double(self, index, value.to_f64)
  end

  private def bind_arg(index, value : Float64)
    check LibSQLite3.bind_double(self, index, value)
  end

  private def bind_arg(index, value : String)
    check LibSQLite3.bind_text(self, index, value, value.bytesize, nil)
  end

  private def bind_arg(index, value : Slice(UInt8))
    check LibSQLite3.bind_blob(self, index, value, value.size, nil)
  end

  private def check(code)
    raise Exception.new(@connection) unless code == 0
  end

  def to_unsafe
    @stmt
  end
end