module DeltaLake

def try_get_deltatable(table_uri, storage_options)
  Table.new(table_uri, storage_options: storage_options)
rescue TableNotFoundError
  nil
end

def try_get_table_and_table_uri(table_or_uri, storage_options)
  if !table_or_uri.is_a?(String) && !table_or_uri.is_a?(Table)
    raise ArgumentError, "table_or_uri must be a String or Table"
  end
  if table_or_uri.is_a?(String)
    table = try_get_deltatable(table_or_uri, storage_options)
    table_uri = table_or_uri.to_s
  else
    table = table_or_uri
    table_uri = table._table.table_uri
  end
  [table, table_uri]
end

def write(
  table_or_uri,
  data,
  partition_by: nil,
  mode: "error",
  name: nil,
  description: nil,
  configuration: nil,
  schema_mode: nil,
  storage_options: nil,
  predicate: nil,
  target_file_size: nil,
  writer_properties: nil,
  commit_properties: nil,
  post_commithook_properties: nil
)
  table, table_uri = try_get_table_and_table_uri(table_or_uri, storage_options)
  if partition_by.is_a?(String)
    partition_by = [partition_by]
  end
  if !table.nil? && mode == "ignore"
    return
  end
  data = Utils.convert_data(data)
  write_deltalake_rust(
    table_uri,
    data,
    mode,
    table&._table,
    schema_mode,
    partition_by,
    predicate,
    target_file_size,
    name,
    description,
    configuration,
    storage_options,
    writer_properties,
    commit_properties,
    post_commithook_properties
  )
  if table
    table.update_incremental
  end
end