@dpkit/core/resource/infer.ts

Fast TypeScript data management framework built on top of the Data Package standard and Polars DataFrames

import {
  getFilename,
  getFormat,
  getName,
  getProtocol,
} from "../general/index.ts"
import type { Resource } from "./Resource.ts"

export function inferResourceName(resource: Partial<Resource>) {
  let name = resource.name

  if (!name) {
    const path = Array.isArray(resource.path) ? resource.path[0] : resource.path
    if (path) {
      const filename = getFilename(path)
      name = getName(filename)
    }
  }

  return name ?? "resource"
}

export function inferResourceFormat(resource: Partial<Resource>) {
  let format = resource.format

  if (!format) {
    if (resource.path) {
      const path = Array.isArray(resource.path)
        ? resource.path[0]
        : resource.path

      if (path) {
        const protocol = getProtocol(path)

        if (DATABASE_PROTOCOLS.includes(protocol)) {
          format = protocol
        } else {
          const filename = getFilename(path)
          format = getFormat(filename)
        }
      }
    }
  }

  return format
}

const DATABASE_PROTOCOLS = ["postgresql", "mysql", "sqlite"]