// CLI output formatting: multi-line user-facing messages where embedded \n

// produces the intended layout. Splitting into logger.log("") + logger.log(...)

// pairs is the canonical rewrite but doesnt preserve the visual flow for these

// specific outputs.

/* oxlint-disable-next-line socket/no-file-scope-oxlint-disable -- legitimate file-scope: domain-grouped layout or test fixture; per-call would produce many redundant disables. */

/* oxlint-disable socket/no-logger-newline-literal -- intended layout */

import { existsSync } from 'node:fs'

import { isErrnoException } from '@socketsecurity/lib-stable/errors/predicates'

import { debug, debugDir } from '@socketsecurity/lib-stable/debug/output'

import { getDefaultLogger } from '@socketsecurity/lib-stable/logger/default'

import { confirm } from '@socketsecurity/lib-stable/stdio/prompts'

import { addSocketWrapper } from './add-socket-wrapper.mts'

import { checkSocketWrapperSetup } from './check-socket-wrapper-setup.mts'

import { getBashRcPath, getZshRcPath } from '../../constants/paths.mts'

import { getBashrcDetails } from '../../util/cli/completion.mts'

import { FileSystemError, getErrorCause } from '../../util/error/errors.mjs'

import { updateInstalledTabCompletionScript } from '../install/setup-tab-completion.mts'

const logger = getDefaultLogger()

export async function postinstallWrapper() {

const bashRcPath = getBashRcPath()

const zshRcPath = getZshRcPath()

const socketWrapperEnabled =

(existsSync(bashRcPath) && checkSocketWrapperSetup(bashRcPath)) ||

(existsSync(zshRcPath) && checkSocketWrapperSetup(zshRcPath))

if (!socketWrapperEnabled) {

await setupSocketWrapper(

`

The Socket CLI is now successfully installed! 🎉

To better protect yourself against supply-chain attacks, our Socket npm wrapper can warn you about malicious packages whenever you run 'npm install'.

Do you want to install the Socket npm wrapper (this will create an alias to the \`socket npm\` command)?

`.trim(),

)

}

// Attempt to update the existing tab completion

let updatedTabCompletion = false

try {

const details = getBashrcDetails('') // Note: command is not relevant, we just want the config path

if (details.ok) {

if (existsSync(details.data.targetPath)) {

// Replace the file with the one from this installation

const result = updateInstalledTabCompletionScript(

details.data.targetPath,

)

if (result.ok) {

// This will work no matter what alias(es) were registered since that

// is controlled by bashrc and they all share the same tab script.

logger.success('Updated the installed Socket tab completion script')

updatedTabCompletion = true

}

}

}

} catch (e) {

debug('Tab completion setup failed (non-fatal)')

debugDir(e)

// Ignore. Skip tab completion setup.

}

if (!updatedTabCompletion) {

// Setting up tab completion requires bashrc modification. I'm not sure if

// it's cool to just do that from an `npm install`...

logger.log('Run `socket install completion` to setup bash tab completion')

}

}

export async function setupSocketWrapper(query: string): Promise<void> {

logger.log(`

_____ _ _

| __|___ ___| |_ ___| |_

|__ | . | _| '_| -_| _|

|_____|___|___|_,_|___|_|

`)

if (

await confirm({

message: query,

default: true,

})

) {

const bashRcPath = getBashRcPath()

const zshRcPath = getZshRcPath()

try {

if (existsSync(bashRcPath)) {

await addSocketWrapper(bashRcPath)

}

if (existsSync(zshRcPath)) {

await addSocketWrapper(zshRcPath)

}

} catch (e) {

throw new FileSystemError(

`failed to add socket aliases to ${bashRcPath} / ${zshRcPath} (${getErrorCause(e)}); check that your shell rc files exist and are writable`,

{ code: isErrnoException(e) ? e.code : undefined },

)

}

}

}