mirror of
https://github.com/bellingcat/auto-archiver.git
synced 2026-06-10 20:28:28 +03:00
204 lines
6.5 KiB
TypeScript
204 lines
6.5 KiB
TypeScript
import { useState } from "react";
|
|
import { useSortable } from "@dnd-kit/sortable";
|
|
import ReactMarkdown from 'react-markdown';
|
|
|
|
import { CSS } from "@dnd-kit/utilities";
|
|
|
|
import {
|
|
Card,
|
|
CardContent,
|
|
CardActions,
|
|
CardHeader,
|
|
Button,
|
|
Dialog,
|
|
DialogTitle,
|
|
DialogContent,
|
|
Box,
|
|
IconButton,
|
|
Checkbox,
|
|
Select,
|
|
MenuItem,
|
|
FormControl,
|
|
FormControlLabel,
|
|
Textarea,
|
|
FormHelperText,
|
|
TextField,
|
|
Stack,
|
|
Typography,
|
|
} from '@mui/material';
|
|
import Grid from '@mui/material/Grid2';
|
|
import DragIndicatorIcon from '@mui/icons-material/DragIndicator';
|
|
import { set } from "yaml/dist/schema/yaml-1.1/set";
|
|
|
|
|
|
Object.defineProperty(String.prototype, 'capitalize', {
|
|
value: function() {
|
|
return this.charAt(0).toUpperCase() + this.slice(1);
|
|
},
|
|
enumerable: false
|
|
});
|
|
|
|
const StepCard = ({
|
|
type,
|
|
module,
|
|
toggleModule,
|
|
enabledModules,
|
|
configValues
|
|
}: {
|
|
type: string,
|
|
module: object,
|
|
toggleModule: any,
|
|
enabledModules: any,
|
|
configValues: any
|
|
}) => {
|
|
const {
|
|
attributes,
|
|
listeners,
|
|
setNodeRef,
|
|
transform,
|
|
transition,
|
|
isDragging
|
|
} = useSortable({ id: module.name });
|
|
|
|
|
|
const style = {
|
|
transform: CSS.Transform.toString(transform),
|
|
transition,
|
|
zIndex: isDragging ? "100" : "auto",
|
|
opacity: isDragging ? 0.3 : 1
|
|
};
|
|
|
|
let name = module.name;
|
|
const [helpOpen, setHelpOpen] = useState(false);
|
|
const [configOpen, setConfigOpen] = useState(false);
|
|
const enabled = enabledModules[type].find((m: any) => m[0] === name)[1];
|
|
|
|
return (
|
|
<Grid ref={setNodeRef} size={{ xs: 6, sm: 4, md: 3 }} style={style}>
|
|
<Card>
|
|
<CardHeader
|
|
title={
|
|
<FormControlLabel
|
|
control={<Checkbox id={name} onClick={toggleModule} checked={enabled} />}
|
|
label={module.display_name} />
|
|
}
|
|
action ={
|
|
<IconButton size="small" {...listeners} {...attributes}>
|
|
<DragIndicatorIcon />
|
|
</IconButton>
|
|
}
|
|
/>
|
|
<CardActions>
|
|
<Button size="small" onClick={() => setHelpOpen(true)}>Info</Button>
|
|
{enabled && module.configs && name != 'cli_feeder' ? (
|
|
<Button size="small" onClick={() => setConfigOpen(true)}>Configure</Button>
|
|
) : null}
|
|
|
|
</CardActions>
|
|
</Card>
|
|
<Dialog
|
|
open={helpOpen}
|
|
onClose={() => setHelpOpen(false)}
|
|
maxWidth="lg"
|
|
>
|
|
<DialogTitle>
|
|
{module.display_name}
|
|
</DialogTitle>
|
|
<DialogContent>
|
|
<ReactMarkdown>
|
|
{module.manifest.description.split("\n").map((line: string) => line.trim()).join("\n")}
|
|
</ReactMarkdown>
|
|
</DialogContent>
|
|
</Dialog>
|
|
{module.configs && name != 'cli_feeder' && <ConfigPanel module={module} open={configOpen} setOpen={setConfigOpen} configValues={configValues} />}
|
|
</Grid>
|
|
)
|
|
}
|
|
|
|
|
|
function ConfigPanel({ module, open, setOpen, configValues }: { module: any, open: boolean, setOpen: any, configValues: any }) {
|
|
|
|
function setConfigValue(config: any, value: any) {
|
|
configValues[module.name][config] = value;
|
|
}
|
|
return (
|
|
<>
|
|
<Dialog
|
|
key={module}
|
|
open={open}
|
|
onClose={() => setOpen(false)}
|
|
maxWidth="lg"
|
|
>
|
|
<DialogTitle>
|
|
{module.display_name}
|
|
</DialogTitle>
|
|
<DialogContent>
|
|
<Stack key={module} direction="column" spacing={1}>
|
|
{Object.keys(module.configs).map((config_value: any) => {
|
|
const config_args = module.configs[config_value];
|
|
const config_name = config_value.replace(/_/g," ");
|
|
const config_display_name = config_name.capitalize();
|
|
const value = configValues[module.name][config_value] || config_args.default;
|
|
return (
|
|
<Box key={config_value}>
|
|
<Typography variant='body1' style={{fontWeight : 'bold'}}>{config_display_name}</Typography>
|
|
<FormControl size="small">
|
|
{ config_args.type === 'bool' ?
|
|
<FormControlLabel control={
|
|
<Checkbox defaultChecked={value} size="small" id={`${module}.${config_value}`}
|
|
onChange={(e) => {
|
|
setConfigValue(config_value, e.target.checked);
|
|
}}
|
|
/>} label={config_args.help}
|
|
/>
|
|
:
|
|
(
|
|
config_args.choices !== undefined ?
|
|
<Select size="small" id={`${module}.${config_value}`}
|
|
defaultValue={value}
|
|
onChange={(e) => {
|
|
setConfigValue(config_value, e.target.value);
|
|
}}
|
|
>
|
|
{config_args.choices.map((choice: any) => {
|
|
return (
|
|
<MenuItem key={`${module}.${config_value}.${choice}`}
|
|
value={choice}>{choice}</MenuItem>
|
|
);
|
|
})}
|
|
</Select>
|
|
:
|
|
( config_args.type === 'json_loader' ?
|
|
<TextField multiline size="small" id={`${module}.${config_value}`} defaultValue={JSON.stringify(value, null, 2)} rows={6} onChange={
|
|
(e) => {
|
|
try {
|
|
let val = JSON.parse(e.target.value);
|
|
setConfigValue(config_value, val);
|
|
} catch (e) {
|
|
console.log(e);
|
|
}
|
|
}
|
|
} />
|
|
:
|
|
<TextField size="small" id={`${module}.${config_value}`} defaultValue={value} type={config_args.type === 'int' ? 'number' : 'text'}
|
|
onChange={(e) => {
|
|
setConfigValue(config_value, e.target.value);
|
|
}} />
|
|
)
|
|
)
|
|
}
|
|
{config_args.type !== 'bool' && (
|
|
<FormHelperText >{config_args.help}</FormHelperText>
|
|
)}
|
|
</FormControl>
|
|
</Box>
|
|
);
|
|
})}
|
|
</Stack>
|
|
</DialogContent>
|
|
</Dialog>
|
|
</>
|
|
);
|
|
}
|
|
|
|
export default StepCard; |