82 lines
2.5 KiB
TypeScript
82 lines
2.5 KiB
TypeScript
import { useEffect, useState } from 'react'
|
|
|
|
import {
|
|
SetVarNameModal,
|
|
createSetVarNameModal,
|
|
} from '@src/components/SetVarNameModal'
|
|
import { useModelingContext } from '@src/hooks/useModelingContext'
|
|
import { useKclContext } from '@src/lang/KclProvider'
|
|
import { moveValueIntoNewVariable } from '@src/lang/modifyAst'
|
|
import { isNodeSafeToReplace } from '@src/lang/queryAst'
|
|
import type { PathToNode, SourceRange } from '@src/lang/wasm'
|
|
import { recast } from '@src/lang/wasm'
|
|
import { codeManager, editorManager, kclManager } from '@src/lib/singletons'
|
|
import { err, reportRejection, trap } from '@src/lib/trap'
|
|
import { toSync } from '@src/lib/utils'
|
|
|
|
export const getVarNameModal = createSetVarNameModal(SetVarNameModal)
|
|
|
|
export function useConvertToVariable(range?: SourceRange) {
|
|
const { ast } = useKclContext()
|
|
const { context } = useModelingContext()
|
|
const [enable, setEnabled] = useState(false)
|
|
|
|
useEffect(() => {
|
|
editorManager.convertToVariableEnabled = enable
|
|
}, [enable])
|
|
|
|
useEffect(() => {
|
|
// Return early if there are no selection ranges for whatever reason
|
|
if (!context.selectionRanges) return
|
|
const parsed = ast
|
|
|
|
const meta = isNodeSafeToReplace(
|
|
parsed,
|
|
range ||
|
|
context.selectionRanges.graphSelections?.[0]?.codeRef?.range ||
|
|
[]
|
|
)
|
|
if (trap(meta)) return
|
|
|
|
const { isSafe, value } = meta
|
|
const canReplace = isSafe && value.type !== 'Name'
|
|
const isOnlyOneSelection =
|
|
!!range || context.selectionRanges.graphSelections.length === 1
|
|
|
|
setEnabled(canReplace && isOnlyOneSelection)
|
|
}, [context.selectionRanges])
|
|
|
|
const handleClick = async (
|
|
valueName?: string
|
|
): Promise<PathToNode | undefined> => {
|
|
try {
|
|
const { variableName } = await getVarNameModal({
|
|
valueName: valueName || 'var',
|
|
selectionRanges: context.selectionRanges,
|
|
})
|
|
|
|
const { modifiedAst: _modifiedAst, pathToReplacedNode } =
|
|
moveValueIntoNewVariable(
|
|
ast,
|
|
kclManager.variables,
|
|
range || context.selectionRanges.graphSelections[0]?.codeRef?.range,
|
|
variableName
|
|
)
|
|
|
|
await kclManager.updateAst(_modifiedAst, true)
|
|
|
|
const newCode = recast(_modifiedAst)
|
|
if (err(newCode)) return
|
|
codeManager.updateCodeEditor(newCode)
|
|
|
|
return pathToReplacedNode
|
|
} catch (e) {
|
|
console.log('error', e)
|
|
}
|
|
}
|
|
|
|
editorManager.convertToVariableCallback = toSync(handleClick, reportRejection)
|
|
|
|
return { enable, handleClick }
|
|
}
|