1import { getEnvConf } from '../envConfF.mts'
2import { getInCluster } from '../getReqEnvConfItemF.mts'
3import { secretsDir } from '../secretsDirF.mts'
4import { ptDir } from '../ptDirF.mts'
5import { replaceStrict } from '../../sharedF/moDashF/replaceStrictF.mts'
6import { PtErr } from '../../sharedF/throwErrorF/sharedErrF.mts'
7import { qsToUriH } from '../qsToUriF.mts'
8import { chalkRed } from '../libChalkF.mts'
9import { getReqSvcInfo } from '../clusterInfoF.mts'
10import { isDevPc } from '../../sharedF/isDevPcF.mts'
11import { disabledSecret } from '../../sharedF/disabledSecretF.mts'
12import os from 'os'
13import * as _ from 'lodash-es'
14import { runPgOnKlustCtx } from './runPgOnKlustCtxF.mts'
15import { throwIf } from '../../sharedF/throwErrorF/throwIfF.mts'
16import { assertDefined } from '../../sharedF/assertsF/assertDefinedF.mts'
17import { getPlainMappedSec } from '../secretsF/getReqJsonSecretHF.mts'
18import type { DbConnInfo } from './dbTypesF.mts'
19
20const getForkedQs = ({secretName}: {secretName: string}) => {
21  if (secretName == disabledSecret) return
22
23  const secretVal = getPlainMappedSec(secretName as any)
24  guardQsSecExist({secretName, secretVal})
25
26  return secretVal
27}
28
29export const decomposeForkedKubeQs = async ({qsName, dbFork}: {qsName: string, dbFork?: boolean | string}): Promise<DbConnInfo> => {
30  const secretsMapping = getEnvConf().secretsMapping || {}
31  const forkedQsName = secretsMapping[qsName] || qsName
32
33  const baseQs = getForkedQs({secretName: forkedQsName})
34  assertDefined(baseQs, {qsName})
35
36  throwIf(() => _.isBoolean(dbFork) && !dbFork)
37
38  const dbQs = baseQs
39
40  return {...await decomposeKubeQs(dbQs), qsName: forkedQsName}
41}
42
43export const guardQsSecExist = ({secretName, secretVal}: {secretName: string, secretVal: string | undefined}) => {
44  if (secretVal) return
45  const secretPath = replaceStrict(secretsDir, ptDir + '/', '') + '/' + secretName
46  console.trace()
47  console.log(chalkRed(`
48Missing required secret: ${secretPath}
49- If you are trying to connect to an existing database:
50    please ask the db admin to send you their ${secretPath}
51- If you are setting up a new database:
52    please make sure ${secretPath} exists and is connectable
53`))
54  throw new PtErr('missingSecret', {secretName})
55}
56
57export const decomposeKubeQs = async (dbQs: string): Promise<Omit<DbConnInfo, 'qsName'>> => {
58  assertDefined(dbQs)
59
60  const _urlObj = qsToUriH(dbQs as any)
61
62  const {password, username, svcName, host: hostWPort} = _urlObj
63
64  const svcInfo = await getReqSvcInfo(svcName)
65
66  const {cluster_name, ips, isCnPg} = svcInfo
67  let {enableSsl=false, nodePort, ptNodePortNo} = svcInfo
68  nodePort ||= ptNodePortNo
69
70  const hostname = ips[0]
71
72  const extIpPort = [hostname, nodePort].join(':')
73
74  let extQs = (() => {
75    const fromStr = `@${hostWPort}`
76    if (!runPgOnKlustCtx.getStore()) {
77      if (isDevPc || getInCluster() != cluster_name) {
78        return replaceStrict(dbQs, fromStr, `@${extIpPort}`)
79      }
80    }
81    if (isCnPg) return dbQs
82    return replaceStrict(dbQs, fromStr, `@${svcName}.default.svc.cluster.local:5432`)
83  })()
84
85  const addParams = enableSsl ? {sslmode: 'require'} : {}
86  extQs += addPgParams(addParams)
87
88  return {
89    ...svcInfo,
90    extQs, enableSsl,
91    dbQs, password, username, svcName,
92    nodePort, hostname,
93  }
94}
95
96const addPgParams = (params: Record<string, any>) => {
97  params = {
98    application_name: os.hostname(),
99    ...params,
100  }
101  return '?' + new URLSearchParams(params)
102}