{-# LANGUAGE QuasiQuotes #-}

-- | Postgres DDL Source Version
--
-- Deals with catalog version - used by 'Hasura.Backends.Postgres.DDL.Source'.
module Hasura.Backends.Postgres.DDL.Source.Version
  ( latestSourceCatalogVersion,
    latestSourceCatalogVersionText,
    previousSourceCatalogVersions,
    getSourceCatalogVersion,
    setSourceCatalogVersion,
  )
where

import Database.PG.Query qualified as Q
import Hasura.Backends.Postgres.Connection
import Hasura.Prelude
import Hasura.Server.Migrate.Version

latestSourceCatalogVersion :: CatalogVersion
latestSourceCatalogVersion = CatalogVersion 2

latestSourceCatalogVersionText :: Text
latestSourceCatalogVersionText = tshow latestSourceCatalogVersion

previousSourceCatalogVersions :: [CatalogVersion]
previousSourceCatalogVersions = [CatalogVersion 0 .. pred latestSourceCatalogVersion]

setSourceCatalogVersion :: MonadTx m => m ()
setSourceCatalogVersion =
  liftTx $
    Q.unitQE
      defaultTxErrorHandler
      [Q.sql|
  INSERT INTO hdb_catalog.hdb_source_catalog_version(version, upgraded_on)
    VALUES ($1, NOW())
   ON CONFLICT ((version IS NOT NULL))
   DO UPDATE SET version = $1, upgraded_on = NOW()
  |]
      (Identity latestSourceCatalogVersionText)
      False

getSourceCatalogVersion :: MonadTx m => m Text
getSourceCatalogVersion =
  liftTx $
    runIdentity . Q.getRow
      <$> Q.withQE
        defaultTxErrorHandler
        [Q.sql| SELECT version FROM hdb_catalog.hdb_source_catalog_version |]
        ()
        False