class Google::Cloud::Bigquery::External::BigtableSource::ColumnFamily

def self.from_gapi gapi
  new_fam = new
  new_fam.instance_variable_set :@gapi, gapi
  columns = Array(gapi.columns).map { |col_gapi| BigtableSource::Column.from_gapi col_gapi }
  new_fam.instance_variable_set :@columns, columns
  new_fam
end

def add_boolean qualifier, as: nil
  col = add_column qualifier, as: as, type: "BOOLEAN"
  yield col if block_given?
  col
end

def add_bytes qualifier, as: nil
  col = add_column qualifier, as: as, type: "BYTES"
  yield col if block_given?
  col
end

def add_column qualifier, as: nil, type: nil
  frozen_check!
  col = BigtableSource::Column.new
  col.qualifier = qualifier
  col.field_name = as if as
  col.type = type if type
  yield col if block_given?
  @columns << col
  col
end

def add_float qualifier, as: nil
  col = add_column qualifier, as: as, type: "FLOAT"
  yield col if block_given?
  col
end

def add_integer qualifier, as: nil
  col = add_column qualifier, as: as, type: "INTEGER"
  yield col if block_given?
  col
end

def add_string qualifier, as: nil
  col = add_column qualifier, as: as, type: "STRING"
  yield col if block_given?
  col
end

def columns
  @columns
end

def encoding
  @gapi.encoding
end

def encoding= new_encoding
  frozen_check!
  @gapi.encoding = new_encoding
end

def family_id
  @gapi.family_id
end

def family_id= new_family_id
  frozen_check!
  @gapi.family_id = new_family_id
end

def freeze
  @columns.map(&:freeze!)
  @columns.freeze!
  super
end

def frozen_check!
  return unless frozen?
  raise ArgumentError, "Cannot modify external data source when frozen"
end

def initialize
  @gapi = Google::Apis::BigqueryV2::BigtableColumnFamily.new
  @columns = []
end

def latest
  @gapi.only_read_latest
end

def latest= new_latest
  frozen_check!
  @gapi.only_read_latest = new_latest
end

def to_gapi
  @gapi.columns = @columns.map(&:to_gapi)
  @gapi
end

def type
  @gapi.type
end

def type= new_type
  frozen_check!
  @gapi.type = new_type
end