????
Your IP : 216.73.216.188
/////////////////////////////////////////////////////////////
//
// pgAdmin 4 - PostgreSQL Tools
//
// Copyright (C) 2013 - 2024, The pgAdmin Development Team
// This software is released under the PostgreSQL Licence
//
//////////////////////////////////////////////////////////////
import { makeStyles } from '@mui/styles';
import React from 'react';
import SchemaView from '../../../../../../static/js/SchemaView';
import BaseUISchema from '../../../../../../static/js/SchemaView/base_schema.ui';
import gettext from 'sources/gettext';
import { QueryToolContext } from '../QueryToolComponent';
import url_for from 'sources/url_for';
import _ from 'lodash';
import PropTypes from 'prop-types';
class MacrosCollection extends BaseUISchema {
constructor(keyOptions) {
super();
this.keyOptions = keyOptions;
}
get idAttribute() {
return 'mid';
}
get baseFields() {
let obj = this;
return [
{
id: 'id', label: gettext('Key'), cell: 'select', noEmpty: true,
width: 100, options: obj.keyOptions, optionsReloadBasis: obj.keyOptions.length,
controlProps: {
allowClear: false,
}
},
{
id: 'name', label: gettext('Name'), cell: 'text', noEmpty: true,
width: 100,
},
{
id: 'sql', label: gettext('SQL'), cell: 'sql', noEmpty: true,
width: 300, controlProps: {
options: {
foldGutter: false,
lineNumbers: false,
gutters: [],
readOnly: true,
lineWrapping: true,
},
}
},
];
}
}
class MacrosSchema extends BaseUISchema {
constructor(keyOptions) {
super();
this.macrosCollObj = new MacrosCollection(keyOptions);
}
get baseFields() {
let obj = this;
return [
{
id: 'macro', label: '', type: 'collection', schema: obj.macrosCollObj,
canAdd: true, canDelete: true, isFullTab: true, group: 'temp',
},
];
}
validate(state, setError) {
let allKeys = state.macro.map((m)=>m.id.toString());
if(allKeys.length != new Set(allKeys).size) {
setError('macro', gettext('Key must be unique.'));
return true;
}
return false;
}
}
const useStyles = makeStyles((theme)=>({
root: {
...theme.mixins.tabPanel,
padding: 0,
},
}));
function getChangedMacros(macrosData, changeData) {
/* For backend, added, removed is changed. Convert all added, removed to changed. */
let changed = [];
for (const m of (changeData.macro.changed || [])) {
let newM = {...m};
if('id' in m) {
/* if key changed, clear prev and add new */
changed.push({id: m.mid, name: null, sql: null});
let em = _.find(macrosData, (d)=>d.mid==m.mid);
newM = {name: em.name, sql: em.sql, ...m};
} else {
newM.id = m.mid;
}
delete newM.mid;
changed.push(newM);
}
for (const m of (changeData.macro.deleted || [])) {
changed.push({id: m.id, name: null, sql: null});
}
for (const m of (changeData.macro.added || [])) {
changed.push(m);
}
return changed;
}
export default function MacrosDialog({onClose, onSave}) {
const classes = useStyles();
const queryToolCtx = React.useContext(QueryToolContext);
const [macrosData, setMacrosData] = React.useState([]);
const [macrosErr, setMacrosErr] = React.useState(null);
React.useEffect(async ()=>{
try {
let {data: respData} = await queryToolCtx.api.get(url_for('sqleditor.get_macros', {
'trans_id': queryToolCtx.params.trans_id,
}));
/* Copying id to mid to track key id changes */
setMacrosData(respData.macro.map((m)=>({...m, mid: m.id})));
} catch (error) {
setMacrosErr(error);
}
}, []);
const onSaveClick = (_isNew, changeData)=>{
return new Promise((resolve, reject)=>{
const setMacros = async ()=>{
try {
let changed = getChangedMacros(macrosData, changeData);
let {data: respData} = await queryToolCtx.api.put(url_for('sqleditor.set_macros', {
'trans_id': queryToolCtx.params.trans_id,
}), {changed: changed});
resolve();
onSave(respData.macro?.filter((m)=>Boolean(m.name)));
onClose();
} catch (error) {
reject(error);
}
};
setMacros();
});
};
const keyOptions = macrosData.map((m)=>({
label: m.key_label,
value: m.id,
}));
if(keyOptions.length <= 0) {
return <></>;
}
return (
<SchemaView
formType={'dialog'}
getInitData={()=>{
if(macrosErr) {
return Promise.reject(macrosErr);
}
return Promise.resolve({macro: macrosData.filter((m)=>Boolean(m.name))});
}}
schema={new MacrosSchema(keyOptions)}
viewHelperProps={{
mode: 'edit',
}}
onSave={onSaveClick}
onClose={onClose}
hasSQL={false}
disableSqlHelp={true}
disableDialogHelp={true}
isTabView={false}
formClassName={classes.root}
/>
);
}
MacrosDialog.propTypes = {
onClose: PropTypes.func,
onSave: PropTypes.func,
};