nocodb/packages/nc-gui/composables/useColumn.ts

import type { ColumnType } from 'nocodb-sdk'
import { SqlUiFactory, UITypes, isVirtualCol } from 'nocodb-sdk'
import type { ComputedRef, Ref } from 'vue'
import { computed, useProject } from '#imports'

export function useColumn(column: Ref<ColumnType | undefined>) {
  const { project } = useProject()

  const uiDatatype: ComputedRef<UITypes> = computed(() => column.value?.uidt as UITypes)

  const abstractType = computed(() => {
    // kludge: CY test hack; column.value is being received NULL during attach cell delete operation
    return (column.value && isVirtualCol(column.value)) || !column.value
      ? null
      : SqlUiFactory.create(
          project.value?.bases?.[0]?.type ? { client: project.value.bases[0].type } : { client: 'mysql2' },
        ).getAbstractType(column.value)
  })

  const dataTypeLow = computed(() => column.value?.dt?.toLowerCase())
  const isBoolean = computed(() => abstractType.value === 'boolean')
  const isString = computed(() => uiDatatype.value === UITypes.SingleLineText || abstractType.value === 'string')
  const isTextArea = computed(() => uiDatatype.value === UITypes.LongText)
  const isInt = computed(() => abstractType.value === 'integer')
  const isFloat = computed(() => abstractType.value === 'float' || abstractType.value === UITypes.Number)
  const isDate = computed(() => abstractType.value === 'date' || uiDatatype.value === UITypes.Date)
  const isYear = computed(() => abstractType.value === 'year' || uiDatatype.value === UITypes.Year)
  const isTime = computed(() => abstractType.value === 'time' || uiDatatype.value === UITypes.Time)
  const isDateTime = computed(() => abstractType.value === 'datetime' || uiDatatype.value === UITypes.DateTime)
  const isJSON = computed(() => uiDatatype.value === UITypes.JSON)
  const isGeoData = computed(() => uiDatatype.value === UITypes.GeoData)
  const isEnum = computed(() => uiDatatype.value === UITypes.SingleSelect)
  const isSingleSelect = computed(() => uiDatatype.value === UITypes.SingleSelect)
  const isSet = computed(() => uiDatatype.value === UITypes.MultiSelect)
  const isMultiSelect = computed(() => uiDatatype.value === UITypes.MultiSelect)
  const isURL = computed(() => uiDatatype.value === UITypes.URL)
  const isEmail = computed(() => uiDatatype.value === UITypes.Email)
  const isAttachment = computed(() => uiDatatype.value === UITypes.Attachment)
  const isRating = computed(() => uiDatatype.value === UITypes.Rating)
  const isCurrency = computed(() => uiDatatype.value === UITypes.Currency)
  const isPhoneNumber = computed(() => uiDatatype.value === UITypes.PhoneNumber)
  const isDecimal = computed(() => uiDatatype.value === UITypes.Decimal)
  const isDuration = computed(() => uiDatatype.value === UITypes.Duration)
  const isPercent = computed(() => uiDatatype.value === UITypes.Percent)
  const isSpecificDBType = computed(() => uiDatatype.value === UITypes.SpecificDBType)
  const isAutoSaved = computed(() =>
    [
      UITypes.SingleLineText,
      UITypes.LongText,
      UITypes.PhoneNumber,
      UITypes.Email,
      UITypes.URL,
      UITypes.Number,
      UITypes.Decimal,
      UITypes.Percent,
      UITypes.Count,
      UITypes.AutoNumber,
      UITypes.SpecificDBType,
      UITypes.Geometry,
      UITypes.GeoData,
      UITypes.Duration,
    ].includes(uiDatatype.value),
  )
  const isManualSaved = computed(() => [UITypes.Currency].includes(uiDatatype.value))
  const isPrimary = computed(() => column.value?.pv)
  const isPrimaryKey = computed(() => !!column.value?.pk)

  return {
    abstractType,
    dataTypeLow,
    isPrimary,
    isBoolean,
    isString,
    isTextArea,
    isInt,
    isFloat,
    isDate,
    isYear,
    isTime,
    isDateTime,
    isJSON,
    isGeoData,
    isEnum,
    isSet,
    isURL,
    isEmail,
    isAttachment,
    isRating,
    isCurrency,
    isDecimal,
    isDuration,
    isAutoSaved,
    isManualSaved,
    isSingleSelect,
    isMultiSelect,
    isPercent,
    isPhoneNumber,
    isSpecificDBType,
    isPrimaryKey,
  }
}