import { useState, useEffect } from "react";
import { useNavigate, Link, useLocation } from "react-router-dom";
import { aethexUserService } from "@/lib/aethex-database-adapter";
import { useAuth } from "@/contexts/AuthContext";
import { useAethexToast } from "@/hooks/use-aethex-toast";
import Layout from "@/components/Layout";
import SEO from "@/components/SEO";
import { Button } from "@/components/ui/button";
import {
  Card,
  CardContent,
  CardDescription,
  CardHeader,
  CardTitle,
} from "@/components/ui/card";
import { Input } from "@/components/ui/input";
import { Label } from "@/components/ui/label";
import { Badge } from "@/components/ui/badge";
import { Alert, AlertDescription, AlertTitle } from "@/components/ui/alert";
import LoadingScreen from "@/components/LoadingScreen";
import {
  LogIn,
  ArrowRight,
  Shield,
  Sparkles,
  Github,
  Mail,
  Lock,
  User,
  Info,
  Wallet,
} from "lucide-react";

const DiscordIcon = () => (
  <svg
    className="h-4 w-4"
    viewBox="0 0 24 24"
    fill="currentColor"
    xmlns="http://www.w3.org/2000/svg"
  >
    <path d="M20.317 4.3671a19.8062 19.8062 0 0 0-4.8851-1.5152.074.074 0 0 0-.0784.0371c-.211.3754-.444.8635-.607 1.2491-1.798-.2704-3.5915-.2704-5.3719 0-.163-.3856-.405-.8737-.62-1.2491a.077.077 0 0 0-.0784-.037 19.7363 19.7363 0 0 0-4.888 1.5152.07.07 0 0 0-.0325.0277C1.618 8.443.134 12.4693 1.981 16.4267a.0842.0842 0 0 0 .0313.0355c1.555.8679 3.064 1.3975 4.555 1.7031a.083.083 0 0 0 .09-.0395c.23-.4354.435-.8888.607-1.3518a.083.083 0 0 0-.046-.1159c-.606-.2324-1.184-.5255-1.738-.8614a.084.084 0 0 1-.008-.1404c.117-.0877.234-.1783.346-.2716a.083.083 0 0 1 .088-.0105c3.646 1.6956 7.596 1.6956 11.182 0a.083.083 0 0 1 .088.009c.112.0933.23.1839.347.2717a.083.083 0 0 1-.006.1404c-.557.3359-1.135.6291-1.742.8615a.084.084 0 0 0-.046.1159c.173.4647.377.9189.607 1.3518a.083.083 0 0 0 .09.0395c1.494-.3066 3.003-.8352 4.555-1.7031a.083.083 0 0 0 .035-.0355c2.0037-4.0016.6248-8.0511-2.607-11.3586a.06.06 0 0 0-.031-.0277ZM8.02 13.3328c-.983 0-1.79-.9015-1.79-2.0074 0-1.1059.795-2.0074 1.79-2.0074 1.001 0 1.799.9039 1.79 2.0074 0 1.1059-.795 2.0074-1.79 2.0074Zm7.975 0c-.984 0-1.79-.9015-1.79-2.0074 0-1.1059.795-2.0074 1.79-2.0074.999 0 1.799.9039 1.789 2.0074 0 1.1059-.79 2.0074-1.789 2.0074Z" />
  </svg>
);
import {
  Dialog,
  DialogTrigger,
  DialogContent,
  DialogHeader,
  DialogTitle,
  DialogDescription,
  DialogFooter,
  DialogClose,
} from "@/components/ui/dialog";

function OrgLogin() {
  const [email, setEmail] = useState("");
  const [sending, setSending] = useState(false);
  const [sent, setSent] = useState<string | null>(null);
  const [error, setError] = useState<string | null>(null);
  const isValid = /@aethex\.dev$/i.test(email);
  return (
    <div className="space-y-3 p-3 rounded border border-border/40 bg-background/50">
      <div className="flex items-center justify-between">
        <div className="text-sm font-medium">Aethex Login (org)</div>
        <Badge variant="outline" className="uppercase">
          @aethex.dev
        </Badge>
      </div>
      {sent ? (
        <Alert className="border-aethex-400/30 bg-aethex-500/10 text-foreground">
          <AlertTitle>Check your inbox</AlertTitle>
          <AlertDescription>
            We sent a magic link to {email}. If email isn’t configured, a manual
            link is shown below.
          </AlertDescription>
          {sent.startsWith("http") && (
            <p className="mt-2 break-all rounded bg-background/60 px-3 py-2 font-mono text-xs text-foreground/90">
              {sent}
            </p>
          )}
        </Alert>
      ) : null}
      {error ? (
        <Alert className="border-red-400/30 bg-red-500/10 text-foreground">
          <AlertTitle>Request failed</AlertTitle>
          <AlertDescription>{error}</AlertDescription>
        </Alert>
      ) : null}
      <div className="grid gap-2 md:grid-cols-3">
        <div className="md:col-span-2">
          <Input
            type="email"
            placeholder="name@aethex.dev"
            value={email}
            onChange={(e) => setEmail(e.target.value)}
          />
        </div>
        <Button
          type="button"
          disabled={!isValid || sending}
          onClick={async () => {
            setSending(true);
            setError(null);
            setSent(null);
            try {
              const r = await fetch("/api/auth/send-org-link", {
                method: "POST",
                headers: { "content-type": "application/json" },
                body: JSON.stringify({
                  email,
                  redirectTo: window.location.origin + "/dashboard",
                }),
              });
              if (!r.ok) {
                const msg = await r.text().catch(() => "");
                throw new Error(msg || String(r.status));
              }
              const data = await r.json().catch(() => ({}));
              setSent(data?.verificationUrl ?? "sent");
            } catch (e: any) {
              setError(e?.message || "Unexpected error");
            } finally {
              setSending(false);
            }
          }}
        >
          {sending ? "Sending…" : "Send magic link"}
        </Button>
      </div>
    </div>
  );
}

export default function Login() {
  const [isLoading, setIsLoading] = useState(false);
  const [isSignUp, setIsSignUp] = useState(false);
  const [email, setEmail] = useState("");
  const [password, setPassword] = useState("");
  const [fullName, setFullName] = useState("");
  const [manualVerificationLink, setManualVerificationLink] = useState<
    string | null
  >(null);
  const [showReset, setShowReset] = useState(false);
  const [resetEmail, setResetEmail] = useState("");
  const [errorFromUrl, setErrorFromUrl] = useState<string | null>(null);
  const navigate = useNavigate();
  const location = useLocation();
  const {
    signIn,
    signUp,
    signInWithOAuth,
    user,
    loading,
    profileComplete,
    requestPasswordReset,
  } = useAuth();
  const { info: toastInfo, error: toastError } = useAethexToast();

  // Check for error messages from URL query parameters (e.g., from OAuth callbacks)
  useEffect(() => {
    const params = new URLSearchParams(location.search);
    const errorType = params.get("error");
    const errorMessage = params.get("message");

    if (errorType && errorMessage) {
      setErrorFromUrl(decodeURIComponent(errorMessage));
      // Show in toast as well
      if (errorType === "account_exists") {
        toastError({
          title: "Account Already Exists",
          description: errorMessage,
        });
      } else if (errorType === "auth_create") {
        toastError({
          title: "Authentication Error",
          description: errorMessage,
        });
      }
    }
  }, [location.search, toastError]);

  // After auth resolves and a user exists, navigate to next path or dashboard
  useEffect(() => {
    if (!loading && user) {
      const params = new URLSearchParams(location.search);
      const next = params.get("next");
      const safeNext = next && next.startsWith("/") ? next : null;
      navigate(safeNext || (profileComplete ? "/dashboard" : "/onboarding"), {
        replace: true,
      });
    }
  }, [user, loading, profileComplete, navigate, location.search]);

  const handleSubmit = async (e: React.FormEvent) => {
    e.preventDefault();
    setIsLoading(true);

    try {
      if (isSignUp) {
        const result = await signUp(email, password, {
          id: "",
          full_name: fullName,
          user_type: "game_developer",
          username: email.split("@")[0],
        });

        if (result?.emailSent) {
          setManualVerificationLink(null);
        } else if (result?.verificationUrl) {
          setManualVerificationLink(result.verificationUrl);
          try {
            if (
              typeof navigator !== "undefined" &&
              navigator.clipboard?.writeText
            ) {
              await navigator.clipboard.writeText(result.verificationUrl);
              toastInfo({
                title: "Verification link copied",
                description:
                  "We copied the manual verification link to your clipboard. Paste it into your browser to finish signup.",
              });
            } else {
              throw new Error("clipboard unsupported");
            }
          } catch {
            toastInfo({
              title: "Manual verification required",
              description:
                "Copy the link shown in the banner to verify your account.",
            });
          }
        }

        setIsSignUp(false);
      } else {
        await signIn(email, password);
        // Do not navigate immediately; wait for auth state to update
      }
    } catch (error: any) {
      console.error("Authentication error:", error);
      toastError({
        title: "Authentication failed",
        description:
          error?.message || "Something went wrong. Please try again.",
      });
    } finally {
      setIsLoading(false);
    }
  };

  const handleSocialLogin = async (provider: "github" | "google") => {
    setIsLoading(true);
    try {
      await signInWithOAuth(provider);
    } catch (error: any) {
      console.error(`${provider} authentication error:`, error);
    } finally {
      setIsLoading(false);
    }
  };

  const handleWeb3Login = async () => {
    setIsLoading(true);
    try {
      navigate("/web3-callback");
    } catch (error: any) {
      console.error("Web3 navigation error:", error);
      setIsLoading(false);
    }
  };

  // Show loading screen only during form submission, not during auth context loading
  if (isLoading && !loading) {
    return (
      <LoadingScreen
        message="Authenticating your account..."
        showProgress={true}
        duration={2000}
      />
    );
  }

  // If auth context is still loading, show a different loading state
  if (loading) {
    return (
      <LoadingScreen
        message="Initializing AeThex OS..."
        showProgress={true}
        duration={3000}
      />
    );
  }

  return (
    <>
      <SEO
        pageTitle="Login"
        description="Sign in to your AeThex account to access the dashboard and community."
        canonical={
          typeof window !== "undefined"
            ? window.location.href
            : (undefined as any)
        }
      />
      <Layout>
        <div className="min-h-screen bg-aethex-gradient py-12 flex items-center justify-center">
          <div className="container mx-auto px-4 max-w-md">
            {/* Floating particles effect */}
            <div className="absolute inset-0 pointer-events-none overflow-hidden opacity-10">
              {[...Array(20)].map((_, i) => (
                <div
                  key={i}
                  className="absolute w-1 h-1 bg-aethex-400 rounded-full animate-float"
                  style={{
                    left: `${Math.random() * 100}%`,
                    top: `${Math.random() * 100}%`,
                    animationDelay: `${Math.random() * 3}s`,
                    animationDuration: `${3 + Math.random() * 2}s`,
                  }}
                />
              ))}
            </div>

            <Card className="bg-card/50 backdrop-blur-sm border border-border/50 shadow-2xl animate-scale-in relative z-10">
              <CardHeader className="text-center space-y-4">
                <div className="flex justify-center">
                  <div className="p-4 rounded-full bg-gradient-to-r from-aethex-500/20 to-neon-blue/20 border border-aethex-400/20">
                    <Shield className="h-8 w-8 text-aethex-400 animate-pulse-glow" />
                  </div>
                </div>
                <div className="space-y-2">
                  <CardTitle className="text-2xl text-gradient-purple">
                    {isSignUp ? "Create Account" : "Welcome Back"}
                  </CardTitle>
                  <CardDescription>
                    {isSignUp
                      ? "Create your AeThex account to get started"
                      : "Sign in to your AeThex account to access the dashboard"}
                  </CardDescription>
                </div>
                <Badge
                  variant="outline"
                  className="border-aethex-400/50 text-aethex-400"
                >
                  <Sparkles className="h-3 w-3 mr-1" />
                  Secure Login
                </Badge>
              </CardHeader>

              <CardContent className="space-y-6">
                {manualVerificationLink ? (
                  <Alert className="border-aethex-400/30 bg-aethex-500/10 text-foreground">
                    <Info className="h-4 w-4 text-aethex-300" />
                    <AlertTitle>Manual verification required</AlertTitle>
                    <AlertDescription>
                      <p>
                        We couldn't send the verification email automatically.
                        Use the link below to confirm your account:
                      </p>
                      <p className="mt-2 break-all rounded bg-background/60 px-3 py-2 font-mono text-xs text-foreground/90">
                        {manualVerificationLink}
                      </p>
                      <Button
                        type="button"
                        size="sm"
                        variant="outline"
                        className="mt-3 border-aethex-400/40"
                        onClick={() =>
                          window.open(
                            manualVerificationLink,
                            "_blank",
                            "noopener",
                          )
                        }
                      >
                        Open verification link
                      </Button>
                    </AlertDescription>
                  </Alert>
                ) : null}
                {/* Social Login Buttons */}
                <div className="space-y-3">
                  <Button
                    variant="outline"
                    className="w-full hover-lift interactive-scale"
                    onClick={() => handleSocialLogin("github")}
                  >
                    <Github className="h-4 w-4 mr-2" />
                    Continue with GitHub
                  </Button>
                  <Button
                    variant="outline"
                    className="w-full hover-lift interactive-scale"
                    onClick={() => handleSocialLogin("google")}
                  >
                    <Mail className="h-4 w-4 mr-2" />
                    Continue with Google
                  </Button>
                  <Button
                    variant="outline"
                    className="w-full hover-lift interactive-scale"
                    onClick={handleWeb3Login}
                  >
                    <Wallet className="h-4 w-4 mr-2" />
                    Connect Ethereum Wallet
                  </Button>
                  <Button
                    variant="outline"
                    className="w-full hover-lift interactive-scale"
                    onClick={() => {
                      const apiBase =
                        (import.meta as any)?.env?.VITE_API_BASE ||
                        window.location.origin;
                      const u = new URL("/api/roblox/oauth/start", apiBase);
                      const next = new URLSearchParams(
                        window.location.search,
                      ).get("next");
                      if (next && next.startsWith("/"))
                        u.searchParams.set("state", next);
                      window.location.href = u.toString();
                    }}
                  >
                    <Sparkles className="h-4 w-4 mr-2" />
                    Continue with Roblox
                  </Button>
                  <Button
                    variant="outline"
                    className="w-full hover-lift interactive-scale"
                    onClick={() => {
                      const apiBase =
                        (import.meta as any)?.env?.VITE_API_BASE ||
                        window.location.origin;
                      const u = new URL("/api/discord/oauth/start", apiBase);
                      const next = new URLSearchParams(
                        window.location.search,
                      ).get("next");
                      if (next && next.startsWith("/"))
                        u.searchParams.set("state", next);
                      window.location.href = u.toString();
                    }}
                  >
                    <DiscordIcon />
                    <span className="ml-2">Continue with Discord</span>
                  </Button>
                </div>

                <div className="relative">
                  <div className="absolute inset-0 flex items-center">
                    <div className="w-full border-t border-border/50" />
                  </div>
                  <div className="relative flex justify-center text-xs uppercase">
                    <span className="bg-background px-2 text-muted-foreground">
                      Or continue with email
                    </span>
                  </div>
                </div>

                {/* Aethex Org Login (Magic Link) */}
                <OrgLogin />

                {/* Email/Password Form */}
                <form onSubmit={handleSubmit} className="space-y-4">
                  {isSignUp && (
                    <div className="space-y-2">
                      <Label htmlFor="fullName" className="text-sm font-medium">
                        Full Name
                      </Label>
                      <div className="relative">
                        <User className="absolute left-3 top-1/2 transform -translate-y-1/2 h-4 w-4 text-muted-foreground" />
                        <Input
                          id="fullName"
                          type="text"
                          value={fullName}
                          onChange={(e) => setFullName(e.target.value)}
                          placeholder="Enter your full name"
                          className="pl-10 bg-background/50 border-border/50 focus:border-aethex-400"
                          required={isSignUp}
                        />
                      </div>
                    </div>
                  )}

                  <div className="space-y-2">
                    <Label htmlFor="email" className="text-sm font-medium">
                      Email Address
                    </Label>
                    <div className="relative">
                      <Mail className="absolute left-3 top-1/2 transform -translate-y-1/2 h-4 w-4 text-muted-foreground" />
                      <Input
                        id="email"
                        type="email"
                        value={email}
                        onChange={(e) => setEmail(e.target.value)}
                        placeholder="Enter your email"
                        className="pl-10 bg-background/50 border-border/50 focus:border-aethex-400"
                        required
                      />
                    </div>
                  </div>

                  <div className="space-y-2">
                    <Label htmlFor="password" className="text-sm font-medium">
                      Password
                    </Label>
                    <div className="relative">
                      <Lock className="absolute left-3 top-1/2 transform -translate-y-1/2 h-4 w-4 text-muted-foreground" />
                      <Input
                        id="password"
                        type="password"
                        value={password}
                        onChange={(e) => setPassword(e.target.value)}
                        placeholder={
                          isSignUp ? "Create a password" : "Enter your password"
                        }
                        className="pl-10 bg-background/50 border-border/50 focus:border-aethex-400"
                        required
                        minLength={isSignUp ? 6 : undefined}
                      />
                    </div>
                    {isSignUp && (
                      <p className="text-xs text-muted-foreground">
                        Password must be at least 6 characters long
                      </p>
                    )}
                  </div>

                  {!isSignUp && (
                    <div className="flex items-center justify-between text-sm">
                      <label className="flex items-center space-x-2 cursor-pointer">
                        <input
                          type="checkbox"
                          className="rounded border-border/50"
                        />
                        <span className="text-muted-foreground">
                          Remember me
                        </span>
                      </label>
                      <button
                        type="button"
                        className="text-aethex-400 hover:underline"
                        onClick={() => {
                          setResetEmail(email || "");
                          setShowReset(true);
                        }}
                      >
                        Forgot password?
                      </button>
                    </div>
                  )}

                  <Button
                    type="submit"
                    className="w-full bg-gradient-to-r from-aethex-500 to-neon-blue hover:from-aethex-600 hover:to-neon-blue/90 hover-lift interactive-scale glow-blue"
                    disabled={
                      !email ||
                      !password ||
                      (isSignUp && !fullName) ||
                      isLoading
                    }
                  >
                    <LogIn className="h-4 w-4 mr-2" />
                    {isSignUp ? "Create Account" : "Sign In"}
                    <ArrowRight className="h-4 w-4 ml-2" />
                  </Button>
                </form>

                <div className="text-center pt-4">
                  <p className="text-sm text-muted-foreground">
                    {isSignUp
                      ? "Already have an account?"
                      : "Don't have an account?"}{" "}
                    <button
                      onClick={() => {
                        setIsSignUp((prev) => !prev);
                        setManualVerificationLink(null);
                      }}
                      className="text-aethex-400 hover:underline font-medium"
                    >
                      {isSignUp ? "Sign In" : "Join AeThex"}
                    </button>
                  </p>
                </div>
              </CardContent>
            </Card>

            {/* Security Notice */}
            <div className="mt-6 text-center animate-fade-in">
              <p className="text-xs text-muted-foreground">
                🔒 Your data is protected with enterprise-grade security
              </p>
            </div>
          </div>
        </div>

        <Dialog open={showReset} onOpenChange={setShowReset}>
          <DialogContent>
            <DialogHeader>
              <DialogTitle>Reset your password</DialogTitle>
              <DialogDescription>
                Enter the email associated with your account. We'll send a reset
                link.
              </DialogDescription>
            </DialogHeader>
            <div className="space-y-3 py-2">
              <Label htmlFor="resetEmail" className="text-sm font-medium">
                Email Address
              </Label>
              <Input
                id="resetEmail"
                type="email"
                value={resetEmail}
                onChange={(e) => setResetEmail(e.target.value)}
                placeholder="you@example.com"
              />
            </div>
            <DialogFooter className="sm:justify-end">
              <DialogClose asChild>
                <Button variant="outline">Cancel</Button>
              </DialogClose>
              <Button
                onClick={async () => {
                  if (!resetEmail) return;
                  setIsLoading(true);
                  try {
                    await requestPasswordReset(resetEmail);
                    setShowReset(false);
                  } catch {}
                  setIsLoading(false);
                }}
                disabled={!resetEmail || isLoading}
              >
                Send reset link
              </Button>
            </DialogFooter>
          </DialogContent>
        </Dialog>
      </Layout>
    </>
  );
}