package mathlearning.ui;

import mathlearning.service.EmailService;
import mathlearning.service.UserService;

import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;

public class RegisterFrame extends JFrame{
    private UserService userService;
    private EmailService emailService;
    private LoginFrame loginFrame;

    private JTextField nameField;
    private JTextField emailField;
    private JButton sendCodeButton;
    private JTextField codeField;
    private JPasswordField passwordField;
    private JPasswordField confirmPasswordField;
    private JButton registerButton;
    private JButton backButton;

    private String verificationCode;

    public RegisterFrame(UserService userService, LoginFrame loginFrame) {
        this.userService = userService;
        this.loginFrame = loginFrame;

        // 配置邮箱服务（需要替换为真实的邮箱配置）
        this.emailService = new EmailService(
                "smtp.qq.com",
                "587",
                "2793415226@qq.com",  // 替换为你的QQ邮箱
                "rmiomlakglpjddhb",  // 替换为你的授权码
                true
        );

        addWindowListener(new java.awt.event.WindowAdapter() {
            @Override
            public void windowClosing(java.awt.event.WindowEvent windowEvent) {
                loginFrame.setVisible(true);
            }
        });

        initializeUI();
    }

    private void initializeUI() {
        setTitle("数学学习软件 - 注册");
        setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE);
        setSize(450, 400);
        setLocationRelativeTo(null);
        setResizable(false);

        // 主面板
        JPanel mainPanel = new JPanel(new BorderLayout(10, 10));
        mainPanel.setBorder(BorderFactory.createEmptyBorder(20, 20, 20, 20));

        // 标题
        JLabel titleLabel = new JLabel("用户注册", JLabel.CENTER);
        titleLabel.setFont(new Font("微软雅黑", Font.BOLD, 24));
        mainPanel.add(titleLabel, BorderLayout.NORTH);

        // 表单面板
        JPanel formPanel = new JPanel(new GridLayout(5, 2, 10, 10));

        JLabel nameLabel = new JLabel("用户名:");
        nameLabel.setFont(new Font("微软雅黑", Font.PLAIN, 14));
        nameField = new JTextField();

        JLabel emailLabel = new JLabel("邮箱:");
        emailLabel.setFont(new Font("微软雅黑", Font.PLAIN, 14));
        emailField = new JTextField();

        JLabel codeLabel = new JLabel("验证码:");
        codeLabel.setFont(new Font("微软雅黑", Font.PLAIN, 14));

        JPanel codePanel = new JPanel(new BorderLayout());
        codeField = new JTextField();
        sendCodeButton = new JButton("发送验证码");
        sendCodeButton.addActionListener(new SendCodeListener());

        codePanel.add(codeField, BorderLayout.CENTER);
        codePanel.add(sendCodeButton, BorderLayout.EAST);

        JLabel passwordLabel = new JLabel("密码:");
        passwordLabel.setFont(new Font("微软雅黑", Font.PLAIN, 14));
        passwordField = new JPasswordField();

        JLabel confirmPasswordLabel = new JLabel("确认密码:");
        confirmPasswordLabel.setFont(new Font("微软雅黑", Font.PLAIN, 14));
        confirmPasswordField = new JPasswordField();

        formPanel.add(nameLabel);
        formPanel.add(nameField);
        formPanel.add(emailLabel);
        formPanel.add(emailField);
        formPanel.add(codeLabel);
        formPanel.add(codePanel);
        formPanel.add(passwordLabel);
        formPanel.add(passwordField);
        formPanel.add(confirmPasswordLabel);
        formPanel.add(confirmPasswordField);

        mainPanel.add(formPanel, BorderLayout.CENTER);

        // 按钮面板
        JPanel buttonPanel = new JPanel(new FlowLayout());

        registerButton = new JButton("注册");
        registerButton.setFont(new Font("微软雅黑", Font.PLAIN, 14));
        registerButton.addActionListener(new RegisterButtonListener());

        backButton = new JButton("返回登录");
        backButton.setFont(new Font("微软雅黑", Font.PLAIN, 14));
        backButton.addActionListener(e -> goBackToLogin());

        buttonPanel.add(registerButton);
        buttonPanel.add(backButton);

        mainPanel.add(buttonPanel, BorderLayout.SOUTH);

        add(mainPanel);
    }

    private class SendCodeListener implements ActionListener {
        @Override
        public void actionPerformed(ActionEvent e) {
            String email = emailField.getText().trim();

            if (email.isEmpty()) {
                JOptionPane.showMessageDialog(RegisterFrame.this,
                        "请输入邮箱地址", "错误", JOptionPane.ERROR_MESSAGE);
                return;
            }

            if (!userService.isValidEmail(email)) {
                JOptionPane.showMessageDialog(RegisterFrame.this,
                        "邮箱格式不正确", "错误", JOptionPane.ERROR_MESSAGE);
                return;
            }

            if (userService.userExists(email)) {
                JOptionPane.showMessageDialog(RegisterFrame.this,
                        "该邮箱已注册", "错误", JOptionPane.ERROR_MESSAGE);
                return;
            }

            // 生成6位验证码
            verificationCode = String.valueOf((int)((Math.random() * 9 + 1) * 100000));

            // 发送验证码邮件
            boolean sent = emailService.sendVerificationCode(email, verificationCode, 1);

            if (sent) {
                JOptionPane.showMessageDialog(RegisterFrame.this,
                        "验证码已发送到您的邮箱，请查收", "成功", JOptionPane.INFORMATION_MESSAGE);

                // 禁用发送按钮60秒
                sendCodeButton.setEnabled(false);
                new Thread(() -> {
                    try {
                        for (int i = 60; i > 0; i--) {
                            final int current = i;
                            SwingUtilities.invokeLater(() ->
                                    sendCodeButton.setText(current + "秒后重发"));
                            Thread.sleep(1000);
                        }
                        SwingUtilities.invokeLater(() -> {
                            sendCodeButton.setText("发送验证码");
                            sendCodeButton.setEnabled(true);
                        });
                    } catch (InterruptedException ex) {
                        ex.printStackTrace();
                    }
                }).start();

            } else {
                JOptionPane.showMessageDialog(RegisterFrame.this,
                        "验证码发送失败，请检查邮箱地址或网络连接", "错误", JOptionPane.ERROR_MESSAGE);
            }
        }
    }

    private class RegisterButtonListener implements ActionListener {
        @Override
        public void actionPerformed(ActionEvent e) {
            String username = nameField.getText().trim();
            String email = emailField.getText().trim();
            String code = codeField.getText().trim();
            String password = new String(passwordField.getPassword());
            String confirmPassword = new String(confirmPasswordField.getPassword());

            // 验证输入
            if (email.isEmpty() || code.isEmpty() || password.isEmpty() || username.isEmpty()) {
                JOptionPane.showMessageDialog(RegisterFrame.this,
                        "请填写所有字段", "错误", JOptionPane.ERROR_MESSAGE);
                return;
            }

            if (!password.equals(confirmPassword)) {
                JOptionPane.showMessageDialog(RegisterFrame.this,
                        "两次输入的密码不一致", "错误", JOptionPane.ERROR_MESSAGE);
                return;
            }

            if (!userService.validatePassword(password)) {
                JOptionPane.showMessageDialog(RegisterFrame.this,
                        "密码必须为6-10位，且包含大小写字母和数字", "错误", JOptionPane.ERROR_MESSAGE);
                return;
            }

            // 先注册用户（临时状态）
            if (userService.registerUser(email, verificationCode)) {
                // 验证用户并设置密码
                if (userService.verifyUser(username, email, code, password)) {
                    JOptionPane.showMessageDialog(RegisterFrame.this,
                            "注册成功!", "成功", JOptionPane.INFORMATION_MESSAGE);
                    goBackToLogin();
                } else {
                    JOptionPane.showMessageDialog(RegisterFrame.this,
                            "验证码错误或已过期", "错误", JOptionPane.ERROR_MESSAGE);
                }
            } else {
                JOptionPane.showMessageDialog(RegisterFrame.this,
                        "注册失败，用户可能已存在", "错误", JOptionPane.ERROR_MESSAGE);
            }
        }
    }

    private void goBackToLogin() {
        loginFrame.setVisible(true);
        this.dispose();
    }
}