import { useState } from 'react';
import { Link as RouterLink, useLocation } from 'react-router-dom';
import {
  Box,
  Container,
  Typography,
  Button,
  Grid,
  Link,
  Chip,
  Stepper,
  Step,
  StepLabel,
  StepContent,
  ToggleButton,
  ToggleButtonGroup,
  IconButton,
} from '@mui/material';
import OpenInNewIcon from '@mui/icons-material/OpenInNew';
import DownloadIcon from '@mui/icons-material/Download';
import WifiIcon from '@mui/icons-material/Wifi';
import UsbIcon from '@mui/icons-material/Usb';
import CheckCircleIcon from '@mui/icons-material/CheckCircle';
import PlayArrowIcon from '@mui/icons-material/PlayArrow';

import Layout from '../components/Layout';
import PageHero from '../components/PageHero';

// YouTube Video with styled overlay
function YouTubeEmbed({ videoId, title, version = 'wireless' }) {
  const [isPlaying, setIsPlaying] = useState(false);
  const thumbnailUrlLite = '/assets/miniature_lite.png';
  const thumbnailUrlWireless = '/assets/miniature_wireless.png';

  if (isPlaying) {
    return (
      <Box
        component="iframe"
        src={`https://www.youtube.com/embed/${videoId}?autoplay=1`}
        title={title}
        allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture"
        allowFullScreen
        sx={{
          position: 'absolute',
          top: 0,
          left: 0,
          width: '100%',
          height: '100%',
          border: 'none',
        }}
      />
    );
  }

  return (
    <Box
      onClick={() => setIsPlaying(true)}
      sx={{
        position: 'absolute',
        top: 0,
        left: 0,
        width: '100%',
        height: '100%',
        cursor: 'pointer',
        overflow: 'hidden',
        '&:hover .play-button': {
          transform: 'translate(-50%, -50%) scale(1.1)',
          boxShadow: '0 8px 40px rgba(255, 149, 0, 0.5)',
        },
        '&:hover .overlay': {
          background: 'rgba(0, 0, 0, 0.3)',
        },
      }}
    >
      {/* Thumbnail */}
      <Box
        component="img"
        src={version === 'lite' ? thumbnailUrlLite : thumbnailUrlWireless}
        alt={title}
        sx={{
          width: '100%',
          height: '100%',
          objectFit: 'cover',
        }}
      />
      
      {/* Overlay */}
      <Box
        className="overlay"
        sx={{
          position: 'absolute',
          inset: 0,
          background: 'rgba(0, 0, 0, 0.4)',
          transition: 'background 0.3s ease',
        }}
      />
      
      {/* Play Button */}
      <Box
        className="play-button"
        sx={{
          position: 'absolute',
          top: '50%',
          left: '50%',
          transform: 'translate(-50%, -50%)',
          width: 80,
          height: 80,
          borderRadius: '50%',
          background: 'linear-gradient(135deg, #FF9500 0%, #ff7b00 100%)',
          display: 'flex',
          alignItems: 'center',
          justifyContent: 'center',
          boxShadow: '0 4px 24px rgba(255, 149, 0, 0.4)',
          transition: 'all 0.3s ease',
        }}
      >
        <PlayArrowIcon sx={{ fontSize: 40, color: 'white', ml: 0.5 }} />
      </Box>
      
      {/* Watch label */}
      <Box
        sx={{
          position: 'absolute',
          bottom: 16,
          left: 16,
          display: 'flex',
          alignItems: 'center',
          gap: 1,
          px: 2,
          py: 1,
          borderRadius: 2,
          background: 'rgba(0, 0, 0, 0.6)',
          backdropFilter: 'blur(8px)',
        }}
      >
        <Typography sx={{ color: 'white', fontSize: 13, fontWeight: 600 }}>
          ▶ Watch Assembly Guide
        </Typography>
      </Box>
    </Box>
  );
}

export default function GettingStarted() {
  const location = useLocation();
  const params = new URLSearchParams(location.search);
  const urlVersion = params.get('version');
  const [version, setVersion] = useState(
    urlVersion === 'lite' ? 'lite' : 'wireless'
  );

  return (
    <Layout transparentHeader>
      <PageHero
        eyebrow="2-3 hours to get started"
        title="Welcome to Reachy Mini!"
        subtitle="Let's get your robot assembled, connected, and ready to play. No coding required — just follow these 2 simple steps."
        accentColor="#FF9500"
        stickers={[
          { src: '/assets/reachies/explorer.png', size: 240, top: 30, left: 120, rotation: -8 },
          { src: '/assets/reachies/captain.png', size: 220, bottom: 0, right: 140, rotation: 10 },
        ]}
        primitives={[
          { type: 'ring', color: '#FF9500', size: 100, top: 20, right: 200, rotation: 15 },
          { type: 'circle', color: '#f59e0b', size: 40, bottom: 60, left: 150 },
        ]}
      >
        <Chip
          icon={<CheckCircleIcon sx={{ color: 'white !important' }} />}
          label="Assemble"
          sx={{ bgcolor: 'rgba(255,255,255,0.15)', color: 'white', fontWeight: 600 }}
        />
        <Chip
          icon={<CheckCircleIcon sx={{ color: 'white !important' }} />}
          label="Connect & Play!"
          sx={{ bgcolor: 'rgba(255,255,255,0.15)', color: 'white', fontWeight: 600 }}
        />
      </PageHero>

      {/* Main Content */}
      <Container maxWidth="lg" sx={{ py: 10 }}>
        {/* Version Toggle */}
        <Box sx={{ mb: 8, textAlign: 'center' }}>
          <Typography variant="body1" color="text.secondary" sx={{ mb: 3 }}>
            Which version do you have?
          </Typography>
          <Box
            sx={{
              display: 'inline-flex',
              borderRadius: '50px',
              border: '2px solid',
              borderColor: 'divider',
              overflow: 'hidden',
            }}
          >
            <ToggleButtonGroup
              value={version}
              exclusive
              onChange={(_, v) => v && setVersion(v)}
              sx={{
                '& .MuiToggleButton-root': {
                  px: 4,
                  py: 1.5,
                  border: 'none',
                  borderRadius: 0,
                  fontWeight: 600,
                  fontSize: 15,
                  textTransform: 'none',
                  gap: 1,
                  color: 'text.secondary',
                  '&.Mui-selected': {
                    bgcolor: 'primary.main',
                    color: 'white',
                    '&:hover': {
                      bgcolor: 'primary.dark',
                    },
                  },
                  '&:not(:last-of-type)': {
                    borderRight: '1px solid',
                    borderColor: 'divider',
                  },
                },
              }}
            >
              <ToggleButton value="lite">
                <UsbIcon sx={{ fontSize: 20 }} /> Lite (USB)
              </ToggleButton>
              <ToggleButton value="wireless">
                <WifiIcon sx={{ fontSize: 20 }} /> Wireless
              </ToggleButton>
            </ToggleButtonGroup>
          </Box>
        </Box>

        {/* ======================== */}
        {/* LITE VERSION PROCEDURE */}
        {/* ======================== */}
        {version === 'lite' && (
          <>
            {/* Step 1: Assemble */}
            <Box sx={{ mb: 10, textAlign: 'center' }}>
              <Box sx={{ mb: 3 }}>
                <Typography
                  variant="overline"
                  sx={{
                    color: 'text.secondary',
                    letterSpacing: 2,
                    fontSize: 12,
                    fontWeight: 600,
                  }}
                >
                  Step 1
                </Typography>
                <Typography variant="h3" sx={{ mt: 0.5 }}>Assemble your robot</Typography>
              </Box>

              <Typography variant="body1" color="text.secondary" sx={{ mb: 4, maxWidth: 600, mx: 'auto' }}>
                Follow our visual guide to put together your Reachy Mini Lite.
                Most people finish in <strong>2-3 hours</strong> — our record is 43 minutes! 🏆
              </Typography>

              <Box
                id="assembly-video"
                sx={{
                  position: 'relative',
                  paddingBottom: '56.25%',
                  mb: 3,
                  borderRadius: 3,
                  overflow: 'hidden',
                  backgroundColor: '#000',
                  maxWidth: 1000,
                  mx: 'auto',
                  boxShadow: '0 8px 40px rgba(0, 0, 0, 0.2)',
                }}
              >
                <YouTubeEmbed videoId="PC5Yx950nMY" title="Reachy Mini Lite Assembly" version="lite" />
              </Box>

              <Button
                variant="contained"
                size="large"
                href="https://huggingface.co/spaces/pollen-robotics/Reachy_Mini_LITE_Assembly_Guide"
                target="_blank"
                endIcon={<OpenInNewIcon />}
              >
                Open the Interactive Assembly Guide ✨
              </Button>
            </Box>

            {/* Step 2: Connect & Install */}
            <Box sx={{ mb: 10 }}>
              <Box sx={{ mb: 3 }}>
                <Typography
                  variant="overline"
                  sx={{
                    color: 'text.secondary',
                    letterSpacing: 2,
                    fontSize: 12,
                    fontWeight: 600,
                  }}
                >
                  Step 2
                </Typography>
                <Typography variant="h3" sx={{ mt: 0.5 }}>Connect & install the app</Typography>
              </Box>

              <Grid container spacing={4} alignItems="center">
                <Grid size={{ xs: 12, md: 6 }}>
                  <Stepper orientation="vertical">
                    <Step active completed={false}>
                      <StepLabel>
                        <Typography fontWeight={600}>Connect via USB</Typography>
                      </StepLabel>
                      <StepContent>
                        <Typography variant="body2" color="text.secondary">
                          Use the USB cable that came with your robot to connect it to your computer.
                        </Typography>
                      </StepContent>
                    </Step>
                    <Step active completed={false}>
                      <StepLabel>
                        <Typography fontWeight={600}>Download the desktop app</Typography>
                      </StepLabel>
                      <StepContent>
                        <Typography variant="body2" color="text.secondary" sx={{ mb: 1 }}>
                          The desktop app includes everything you need to control your Lite version.
                        </Typography>
                        <Typography variant="caption" sx={{ display: 'block', mb: 2, color: 'warning.main' }}>
                          Desktop App available for macOS (Apple Silicon), Windows & Linux (beta).
                        </Typography>
                        <Button
                          variant="contained"
                          component={RouterLink}
                          to="/download"
                          startIcon={<DownloadIcon/>}
                        >
                          Download Desktop App
                        </Button>

                        <Button
                          variant="outlined"
                          href="https://huggingface.co/docs/reachy_mini/SDK/installation"
                          target="_blank"
                          startIcon={<OpenInNewIcon/>}
                        >
                          Alternative: Python SDK
                        </Button>
                        
                      </StepContent>
                    </Step>
                    <Step active completed={false}>
                      <StepLabel>
                        <Typography fontWeight={600}>Open the app & play!</Typography>
                      </StepLabel>
                      <StepContent>
                        <Typography variant="body2" color="text.secondary">
                          The app will automatically detect your robot. Click "Wake Up" to start,
                          then explore 30+ apps created by the community, from hand tracking to AI conversations! If you experience any issues, feel free to check our <a href="https://huggingface.co/docs/reachy_mini/troubleshooting" target="_blank">Troubleshooting & FAQ page</a> or contact us on <a href="https://discord.gg/HDrGY9eJHt" target="_blank">Discord</a>.
                        </Typography>
                      </StepContent>
                    </Step>
                  </Stepper>
                </Grid>


                <Grid size={{ xs: 12, md: 6 }}>
                  <Box
                    component="img"
                    src="/assets/desktop-app-screenshot--dark.png"
                    alt="Reachy Mini Control App"
                    sx={{ width: '100%', display: 'block', borderRadius: '12px' }}
                  />
                </Grid>
              </Grid>
            </Box>
          </>
        )}

        {/* ============================ */}
        {/* WIRELESS VERSION PROCEDURE */}
        {/* ============================ */}
        {version === 'wireless' && (
          <>
            {/* Step 1: Assemble */}
            <Box sx={{ mb: 10, textAlign: 'center' }}>
              <Box sx={{ mb: 3 }}>
                <Typography
                  variant="overline"
                  sx={{
                    color: 'text.secondary',
                    letterSpacing: 2,
                    fontSize: 12,
                    fontWeight: 600,
                  }}
                >
                  Step 1
                </Typography>
                <Typography variant="h3" sx={{ mt: 0.5 }}>Assemble your robot</Typography>
              </Box>

              <Typography variant="body1" color="text.secondary" sx={{ mb: 4, maxWidth: 600, mx: 'auto' }}>
                Follow our visual guide to put together your Reachy Mini.
                Most people finish in <strong>2-3 hours</strong> — our record is 43 minutes! 🏆
              </Typography>

              <Box
                id="assembly-video"
                sx={{
                  position: 'relative',
                  paddingBottom: '56.25%',
                  mb: 3,
                  borderRadius: 3,
                  overflow: 'hidden',
                  backgroundColor: '#000',
                  maxWidth: 1000,
                  mx: 'auto',
                  boxShadow: '0 8px 40px rgba(0, 0, 0, 0.2)',
                }}
              >
                <YouTubeEmbed videoId="WeKKdnuXca4" title="Reachy Mini Assembly" version="wireless" />
              </Box>

              <Button
                variant="contained"
                size="large"
                href="https://huggingface.co/spaces/pollen-robotics/Reachy_Mini_Assembly_Guide"
                target="_blank"
                endIcon={<OpenInNewIcon />}
              >
                Open the Interactive Assembly Guide ✨
              </Button>

              <Typography variant="body2" color="text.secondary" sx={{ mt: 3 }}>
                Beta version? Your guide is{' '}
                <Link href="https://huggingface.co/spaces/pollen-robotics/Reachy_Mini_BETA_Assembly_Guide" target="_blank">
                  here
                </Link>.
              </Typography>
            </Box>

            {/* Step 2: Connect & Install */}
            <Box sx={{ mb: 10 }}>
              <Box sx={{ mb: 3 }}>
                <Typography
                  variant="overline"
                  sx={{
                    color: 'text.secondary',
                    letterSpacing: 2,
                    fontSize: 12,
                    fontWeight: 600,
                  }}
                >
                  Step 2
                </Typography>
                <Typography variant="h3" sx={{ mt: 0.5 }}>Connect & install the app</Typography>
              </Box>

              <Grid container spacing={4} alignItems="center">
                <Grid size={{ xs: 12, md: 6 }}>
                  <Stepper orientation="vertical">
                    <Step active completed={false}>
                      <StepLabel>
                        <Typography fontWeight={600}>Power on your Reachy Mini</Typography>
                      </StepLabel>
                      <StepContent>
                        <Typography variant="body2" color="text.secondary">
                          Wait about 30 seconds for the robot to boot up.
                        </Typography>
                      </StepContent>
                    </Step>
                    <Step active completed={false}>
                      <StepLabel>
                        <Typography fontWeight={600}>Download the desktop app</Typography>
                      </StepLabel>
                      <StepContent>
                        <Typography variant="body2" color="text.secondary" sx={{ mb: 1 }}>
                          The desktop app includes everything you need to control your Wireless version.
                        </Typography>
                        <Typography variant="caption" sx={{ display: 'block', mb: 2, color: 'warning.main' }}>
                          Desktop App available for macOS (Apple Silicon), Windows & Linux (beta).
                        </Typography>
                        <Button
                          variant="contained"
                          component={RouterLink}
                          to="/download"
                          startIcon={<DownloadIcon/>}
                        >
                          Download Desktop App
                        </Button>

                        <Button
                          variant="outlined"
                          href="https://huggingface.co/docs/reachy_mini/SDK/installation"
                          target="_blank"
                          startIcon={<OpenInNewIcon/>}
                        >
                          Alternative: Python SDK
                        </Button>
                      </StepContent>
                    </Step>
                    <Step active completed={false}>
                      <StepLabel>
                        <Typography fontWeight={600}>Connect the robot to your Wi-Fi</Typography>
                      </StepLabel>
                      <StepContent>
                        <Typography variant="body2" color="text.secondary" sx={{ mb: 1 }}>
                          Use the desktop app to connect your robot to your Wi-Fi network.
                        </Typography>
                      </StepContent>
                    </Step>
                    <Step active completed={false}>
                      <StepLabel>
                        <Typography fontWeight={600}>Open the app & play!</Typography>
                      </StepLabel>
                      <StepContent>
                        <Typography variant="body2" color="text.secondary">
                          The app will automatically detect your robot. Click "Wake Up" to start,
                          then explore 30+ apps created by the community, from hand tracking to AI conversations! If you experience any issues, feel free to check our <a href="https://huggingface.co/docs/reachy_mini/troubleshooting" target="_blank">Troubleshooting & FAQ page</a> or contact us on <a href="https://discord.gg/HDrGY9eJHt" target="_blank">Discord</a>.
                        </Typography>
                      </StepContent>
                    </Step>
                  </Stepper>
                </Grid>

                <Grid size={{ xs: 12, md: 6 }}>
                  <Box
                    component="img"
                    src="/assets/desktop-app-screenshot--dark.png"
                    alt="Reachy Mini Control App"
                    sx={{ width: '100%', display: 'block', borderRadius: '12px' }}
                  />
                </Grid>
              </Grid>
            </Box>
          </>
        )}

        {/* CTA for Developers */}
    <Box
      sx={{
            p: 6,
            borderRadius: 4,
        background: 'linear-gradient(135deg, #1a1a2e 0%, #16213e 100%)',
        color: 'white',
        textAlign: 'center',
      }}
    >
          <Box
            component="img"
            src="/assets/reachy-how-to-create-app.svg"
            alt="Build"
            sx={{ width: 220, height: 220, mb: 3 }}
          />
          <Typography variant="h4" sx={{ mb: 2 }}>
            Want to create your own apps?
        </Typography>
          <Typography variant="body1" sx={{ mb: 4, opacity: 0.8, maxWidth: 500, mx: 'auto' }}>
            Learn how to code with the Python SDK, create custom experiences, 
            and share them with the community.
        </Typography>
          <Button
            variant="contained"
            size="large"
            component="a"
            href="https://huggingface.co/docs/reachy_mini/index#-apps--ecosystem"
            target="_blank"
            rel="noopener noreferrer"
            sx={{ px: 4 }}
          >
            Go to Create Guide →
          </Button>
        </Box>
      </Container>
    </Layout>
  );
}