import React from 'react';

import { Trans } from '@lingui/macro';
import Button from '@mui/material/Button';
import Grid from '@mui/material/Grid';
import Stack from '@mui/material/Stack';
import MenuItem from '@mui/material/MenuItem';
import Typography from '@mui/material/Typography';

import Dialog from './Dialog';
import Select from '../Select';
import Video from '../coders/settings/Video';
import Audio from '../coders/settings/Audio';

const Stream = function (props) {
	const handleChange = (what) => (event) => {
		const value = event.target.value;

		let stream = {
			...props.stream,
		};

		if (what === 'type') {
			if (value === 'audio') {
				stream.codec = 'aac';
				if (stream.sampling_hz === 0) {
					stream.sampling_hz = 44100;
				}
				if (stream.layout === '') {
					stream.layout = 'stereo';
					stream.channels = 2;
				}
			} else {
				stream.codec = 'h264';
				if (stream.width === 0) {
					stream.width = 1920;
					stream.height = 1080;
				}

				if (stream.pix_fmt === '') {
					stream.pix_fmt = 'yuv240p';
				}
			}
			stream.type = value;
		} else if (what === 'size') {
			const [width, height] = value.split('x');

			stream.width = width;
			stream.height = height;
		} else {
			stream[what] = value;
		}

		props.onChange(stream);
	};

	return (
		<Grid container spacing={1}>
			<Grid item xs={6}>
				<Select label={<Trans>Type</Trans>} value={props.stream.type} onChange={handleChange('type')}>
					<MenuItem value="audio">Audio</MenuItem>
					<MenuItem value="video">Video</MenuItem>
				</Select>
			</Grid>
			{props.stream.type === 'audio' ? (
				<React.Fragment>
					<Grid item xs={6}>
						<Select label={<Trans>Codec</Trans>} value={props.stream.codec} onChange={handleChange('codec')}>
							<MenuItem value="aac">AAC</MenuItem>
							<MenuItem value="mp3">MP3</MenuItem>
						</Select>
					</Grid>
					<Grid item xs={12}>
						<Audio.Sampling value={props.stream.sampling_hz} onChange={handleChange('sampling_hz')} allowCustom />
					</Grid>
					<Grid item xs={12}>
						<Audio.Layout value={props.stream.layout} onChange={handleChange('layout')} allowCustom />
					</Grid>
				</React.Fragment>
			) : (
				<React.Fragment>
					<Grid item xs={6}>
						<Select label={<Trans>Codec</Trans>} value={props.stream.codec} onChange={handleChange('codec')}>
							<MenuItem value="h264">H264</MenuItem>
							<MenuItem value="hevc">HEVC</MenuItem>
							<MenuItem value="vp9">VP9</MenuItem>
							<MenuItem value="av1">AV1</MenuItem>
							<MenuItem value="vp8">VP8</MenuItem>
						</Select>
					</Grid>
					<Grid item xs={12}>
						<Video.Size value={props.stream.width + 'x' + props.stream.height} onChange={handleChange('size')} allowCustom />
					</Grid>
					<Grid item xs={12}>
						<Video.PixFormat value={props.stream.pix_fmt} onChange={handleChange('pix_fmt')} allowCustom />
					</Grid>
				</React.Fragment>
			)}
		</Grid>
	);
};

Stream.defaultProps = {
	stream: {},
	onChange: () => {},
};

const Streams = function (props) {
	const handleChange = (index) => (stream) => {
		const streams = props.streams.slice();

		streams[index] = stream;

		props.onChange(streams);
	};

	const handleAddStream = () => {
		const streams = props.streams.slice();

		streams.push({
			index: props.type === 'video' ? 0 : 1,
			stream: streams.length,
			type: 'audio',
			codec: 'aac',
			width: 0,
			height: 0,
			sampling_hz: 44100,
			layout: 'stereo',
			channels: 2,
		});

		props.onChange(streams);
	};

	const handleRemoveStream = (index) => () => {
		const streams = props.streams.toSpliced(index, 1);

		props.onChange(streams);
	};

	return (
		<Grid container spacing={1}>
			{props.streams.map((stream, index) => (
				<Grid key={stream.index + ':' + stream.stream} item xs={12}>
					<Stack spacing={1}>
						<Typography>Stream {stream.stream}</Typography>
						<Stream stream={stream} onChange={handleChange(index)} />
						{index > 0 && (
							<Button variant="outlined" color="secondary" onClick={handleRemoveStream(index)}>
								<Trans>Remove Stream</Trans>
							</Button>
						)}
					</Stack>
				</Grid>
			))}
			<Grid item xs={12}>
				<Button variant="outlined" color="default" onClick={handleAddStream}>
					<Trans>Add Stream</Trans>
				</Button>
			</Grid>
		</Grid>
	);
};

Streams.defaultProps = {
	streams: [],
	type: '',
	onChange: () => {},
};

const Component = function (props) {
	return (
		<Dialog
			open={props.open}
			onClose={props.onClose}
			title={props.title}
			buttonsLeft={
				<Button variant="outlined" color="secondary" onClick={props.onClose}>
					<Trans>Close</Trans>
				</Button>
			}
			buttonsRight={
				<Button variant="outlined" color="default" onClick={props.onDone}>
					<Trans>Save</Trans>
				</Button>
			}
		>
			<Streams type={props.type} streams={props.streams} onChange={props.onChange} />
		</Dialog>
	);
};

export default Component;

Component.defaultProps = {
	open: false,
	title: '',
	streams: [],
	type: '',
	onClose: null,
	onDone: () => {},
	onHelp: null,
};