Hinduja-LMS/LMS/View Controller/NewQuizViewController.swift

//
//  NewQuizViewController.swift
//  LMS
//
//  Created by Suraj Kumar Mandal on 19/10/23.
//

import UIKit
import Toast_Swift
import SideMenu

struct SubmitAnswers {
    let questionId: Int
    var answers: [String]
}

class AnswerManager {
    var submittedAnswers = [SubmitAnswers]()
    
    func addOrUpdateAnswer(for questionId: Int, with newAnswers: [String]) {
        if let existingIndex = submittedAnswers.firstIndex(where: { $0.questionId == questionId }) {
            // Update answers for existing questionId
            submittedAnswers[existingIndex].answers = newAnswers
        } else {
            // Add new SubmitAnswers
            let newSubmitAnswer = SubmitAnswers(questionId: questionId, answers: newAnswers)
            submittedAnswers.append(newSubmitAnswer)
        }
    }
}

class NewQuizViewController: UIViewController {
    
    @IBOutlet var navigationBar: UINavigationBar!
    @IBOutlet var questionNumberLabel: UILabel!
    @IBOutlet var timerLabel: UILabel!
    @IBOutlet var questionTypeLabel: UILabel!
    @IBOutlet var questionLabel: UILabel!
    @IBOutlet var answerInstructionLabel: UILabel!
    @IBOutlet var quizView: UIView!
    @IBOutlet var answerTF: UITextField!
    @IBOutlet var mcqView: UIView!
    @IBOutlet var mcqTableView: UITableView!
    @IBOutlet var trueFalseView: UIView!
    @IBOutlet var boolTableView: UITableView!
    @IBOutlet var dragView: UIView!
    @IBOutlet var option1TableView: UITableView!
    @IBOutlet var option2TableView: UITableView!
    @IBOutlet var questionNumbersButton: UIButton!
    @IBOutlet var skipButton: UIButton!
    @IBOutlet var nextButton: UIButton!
    //Quiz submit UI outlets
    @IBOutlet var quizSubmittedView: UIView!
    @IBOutlet var customSubmittedView: UIView!
    @IBOutlet var checkImageView: UIImageView!
    @IBOutlet var resultStatusLabel: UILabel!
    @IBOutlet var scoreLabel: UILabel!
    @IBOutlet var okButton: UIButton!
    @IBOutlet var viewAnswerButton: UIButton!
    //Question number UI outlets
    @IBOutlet var questionNumberView: UIView!
    @IBOutlet var questionCustomView: UIView!
    @IBOutlet var questionNumberCollectionView: UICollectionView!
    @IBOutlet var collectionViewHeightConstraint: NSLayoutConstraint!
    
    var viewModel = NewQuizViewModel()
    
    // Pre-defined values
    var assessmentId = Int()
    var assessmentName = String()
    var quesCount = Int()
    var quizTime = Int()
    var sessionId = String()
    var quesIndex = 0
    
    let userData = DBManager.sharedInstance.database.objects(UserDetailsModel.self)
    var quizModel = [QuizModel]()
    var assessmentSubmit: AssessmentSubmitModel?
    let manager = AnswerManager()
    var answer = [String]()
    var timerCount = Int()
    var isOptionSelected: Bool = false
    
    var timer = Timer()
    var alertController: UIAlertController?
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // Do any additional setup after loading the view.
        viewModel.delegate = self
        
        getQuizData()
        setupUI()
        
        // Question navigator collection view delegate and datasource
        questionNumberCollectionView.delegate = self
        questionNumberCollectionView.dataSource = self
        
        // Set unique id only once when assessment starts
        sessionId = generateUniqueID()
    }
    
    override func viewDidDisappear(_ animated: Bool) {
        timer.invalidate()
    }
    
    func setupUI() {
        // Set assessment name to navigation bar title
        navigationBar.topItem?.title = assessmentName
        // Hide other views from start
        quizSubmittedView.isHidden = true
        questionNumberView.isHidden = true
        
        // Design question number label
        questionNumberLabel.clipsToBounds = true
        questionNumberLabel.layer.cornerRadius = 10
        questionNumberLabel.layer.maskedCorners = [.layerMaxXMinYCorner, .layerMinXMinYCorner]
        questionNumbersButton.layer.cornerRadius = questionNumbersButton.frame.size.height / 2
    }
    
    // Function to generate unique id
    func generateUniqueID() -> String {
        let dateFormatter = DateFormatter()
        dateFormatter.dateFormat = "yyyyMMddHHmmssSSS"
        
        let currentTimestamp = dateFormatter.string(from: Date())
        
        return currentTimestamp
    }
    
    // Function call to get quiz data
    func getQuizData() {
        if Reachability.isConnectedToNetwork() {
            viewModel.getQuizData(assessmentId: assessmentId) {
                // Update your UI components with the responseData
                self.quizModel = self.viewModel.quizModel
                print("index: \(self.quesIndex)")
                self.setupQuizUI()
                self.setupData()
                // Assuming you've reloaded or updated the data in your UICollectionView
                self.questionNumberCollectionView.reloadData()
                // Call the method to adjust the height
                self.updateCollectionViewHeight()
                // Set quiz time and start timer
                self.timerCount = self.quizTime * 60
                self.startTimer()
            }
        } else {
            Alert.showInternetFailureAlert(on: self)
        }
    }
    
    // Setup UI for quiz
    func setupQuizUI() {
        self.questionCustomView.layer.cornerRadius = 20
        if quesIndex >= quizModel.count - 1 {
            skipButton.isHidden = true
        } else {
            skipButton.isHidden = false
        }
        if quizModel[quesIndex].questionType == AppConstant.MultipleMCQ {
            answerInstructionLabel.text = "Select [One/More Than One] Options"
            self.answerTF.isHidden = true
            self.mcqView.isHidden = false
            self.mcqTableView.delegate = self
            self.mcqTableView.dataSource = self
            self.mcqTableView.allowsMultipleSelection = true
            self.mcqTableView.allowsMultipleSelectionDuringEditing = true
            self.mcqTableView.reloadData()
            self.trueFalseView.isHidden = true
            self.dragView.isHidden = true
        } else if quizModel[quesIndex].questionType == AppConstant.SingleMCQ {
            answerInstructionLabel.text = "Select One Option"
            self.answerTF.isHidden = true
            self.mcqView.isHidden = false
            self.mcqTableView.delegate = self
            self.mcqTableView.dataSource = self
            self.mcqTableView.allowsMultipleSelection = false
            self.mcqTableView.allowsMultipleSelectionDuringEditing = false
            self.mcqTableView.reloadData()
            self.trueFalseView.isHidden = true
            self.dragView.isHidden = true
            self.isOptionSelected = false
        } else if quizModel[quesIndex].questionType == AppConstant.Fill {
            answerInstructionLabel.text = "Enter Answer In Input Box"
            self.answerTF.isHidden = false
            self.mcqView.isHidden = true
            self.trueFalseView.isHidden = true
            self.dragView.isHidden = true
        } else if quizModel[quesIndex].questionType == AppConstant.TrueFalse {
            answerInstructionLabel.text = "Choose One Option"
            self.answerTF.isHidden = true
            self.mcqView.isHidden = true
            self.trueFalseView.isHidden = false
            self.boolTableView.delegate = self
            self.boolTableView.dataSource = self
            self.boolTableView.reloadData()
            self.dragView.isHidden = true
        } else if quizModel[quesIndex].questionType == AppConstant.Match {
            answerInstructionLabel.text = "Drag And Drop Correct Match From Right Hand Side"
            self.answerTF.isHidden = true
            self.mcqView.isHidden = true
            self.trueFalseView.isHidden = true
            self.dragView.isHidden = false
            self.option1TableView.delegate = self
            self.option1TableView.dataSource = self
            self.option1TableView.reloadData()
            self.option2TableView.delegate = self
            self.option2TableView.dataSource = self
            self.option2TableView.dragDelegate = self
            self.option2TableView.dragInteractionEnabled = true
            self.option2TableView.reloadData()
        }
    }
    
    func setupData() {
        answerTF.text = ""
        answer.removeAll()
        questionNumberLabel.text = " Question No. - \(quesIndex+1)/\(quizModel.count) "
        questionTypeLabel.text = quizModel[quesIndex].questionType
        questionLabel.text = "Q. \(quizModel[quesIndex].question ?? "")"
        print(quizModel[quesIndex].question ?? "")
        print("Index: \(quesIndex)")
    }
    
    func startTimer() {
        // Create a Timer that fires every second and calls the updateTimer function
        timer = Timer.scheduledTimer(timeInterval: 1, target: self, selector: #selector(update), userInfo: nil, repeats: true)
    }
    
    @objc func update() {
        if timerCount > 0 {
            timerCount -= 1
            let minutes = timerCount / 60
            let seconds = timerCount % 60
            // Format the time as "mm:ss"
            let timeString = String(format: "%02d:%02d", minutes, seconds)
            // Update the label with the remaining time
            timerLabel.text = timeString
        } else {
            // Timer has finished, you can handle this event as needed
            timerLabel.text = "00:00" // Display 00:00 when the timer is done
            timer.invalidate()
            
            // Submit quiz in case of time over
            self.submitTotalAssessment()
        }
    }
    
    func submitAnswer() {
        if quizModel[quesIndex].questionType == AppConstant.Fill {
            if answerTF.text?.isEmpty == true {
                self.view.makeToast("Enter your answer first!")
            } else {
                answer.append(answerTF.text!)
                print(answer)
                if Reachability.isConnectedToNetwork() {
                    viewModel.submitQuizAnswer(assessmentId: "\(quizModel[quesIndex].assessmentId ?? 0)", userId: "\(userData[0].id)", questionId: quizModel[quesIndex].id ?? 0, quesType: quizModel[quesIndex].questionType ?? "", answers: self.answer, sessionId: sessionId)
                } else {
                    Alert.showInternetFailureAlert(on: self)
                }
            }
        } else {
            if Reachability.isConnectedToNetwork() {
                if isOptionSelected == true {
                    viewModel.submitQuizAnswer(assessmentId: "\(quizModel[quesIndex].assessmentId ?? 0)", userId: "\(userData[0].id)", questionId: quizModel[quesIndex].id ?? 0, quesType: quizModel[quesIndex].questionType ?? "", answers: self.answer, sessionId: sessionId)
                } else {
                    self.view.makeToast("Select an option first!")
                }
            } else {
                Alert.showInternetFailureAlert(on: self)
            }
        }
    }
    
    func submitTotalAssessment() {
        if Reachability.isConnectedToNetwork() {
            viewModel.submitAssessment(assessmentId: "\(quizModel[quesIndex].assessmentId ?? 0)", userId: userData[0].id, data: manager.submittedAnswers, sessionId: sessionId)
        } else {
            Alert.showInternetFailureAlert(on: self)
        }
    }
    
    /*
     // MARK: - Navigation
     
     // In a storyboard-based application, you will often want to do a little preparation before navigation
     override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
     // Get the new view controller using segue.destination.
     // Pass the selected object to the new view controller.
     }
     */
    
    @IBAction func sideMenuAction(_ sender: UIBarButtonItem) {
        let menu = storyboard!.instantiateViewController(withIdentifier: "SideMenuNavigationController") as! SideMenuNavigationController
        present(menu, animated: true, completion: nil)
    }
    
    @IBAction func skipAction(_ sender: Any) {
        self.quesIndex += 1
        print("index: \(self.quesIndex)")
        setupQuizUI()
        setupData()
    }
    
    @IBAction func nextAction(_ sender: Any) {
        submitAnswer()
    }
    
    @IBAction func closeAction(_ sender: Any) {
        timer.invalidate()
        let vc = self.storyboard?.instantiateViewController(withIdentifier: "NewAssessmentViewController") as! NewAssessmentViewController
        self.navigationController?.pushViewController(vc, animated: true)
    }
    
    @IBAction func viewResultAction(_ sender: Any) {
        let vc = self.storyboard?.instantiateViewController(withIdentifier: "ViewAnswersViewController") as! ViewAnswersViewController
        vc.userId = userData[0].id
        vc.assessmentId = assessmentId
        vc.sessionId = sessionId
        self.navigationController?.pushViewController(vc, animated: true)
    }
    
    @IBAction func quesNumberAction(_ sender: Any) {
        questionNumberView.isHidden = false
    }
    
    @IBAction func cancelAction(_ sender: Any) {
        questionNumberView.isHidden = true
    }
    
}

extension NewQuizViewController: NewQuizViewProtocol {
    func startLoader() {
        ActivityIndicator.start()
    }
    
    func stopLoader() {
        ActivityIndicator.stop()
    }
    
    func showError(error: String) {
        self.view.makeToast(error)
    }
    
    func answerSubmitted(model: AnswerSubmitModel) {
        if quesIndex < quizModel.count - 1 {
            // Append data to the submitted answers
            manager.addOrUpdateAnswer(for: model.questionId ?? 0, with: model.answers ?? [""])
            
            print(manager.submittedAnswers)
            self.questionNumberCollectionView.reloadData()
            
            self.quesIndex += 1
            print("index: \(self.quesIndex)")
            setupQuizUI()
            setupData()
        } else {
            // Append data to the submitted answers
            manager.addOrUpdateAnswer(for: model.questionId ?? 0, with: model.answers ?? [""])
            
            print(manager.submittedAnswers)
            self.questionNumberCollectionView.reloadData()
            
            // Create an alert controller
            alertController = UIAlertController(title: assessmentName, message: "Are you want to submit your answer?", preferredStyle: .alert)
            
            // Create a "Cancel" action
            let cancelAction = UIAlertAction(title: "Cancel", style: .cancel) { (action) in
                // Handle the "Cancel" button action
                print("Cancel Button Pressed")
            }
            
            // Create an "OK" action
            let okAction = UIAlertAction(title: "OK", style: .default) { (action) in
                // Handle the "OK" button action
                self.submitTotalAssessment()
            }
            
            // Add the actions to the alert controller
            alertController?.addAction(cancelAction)
            alertController?.addAction(okAction)
            
            // Present the alert controller
            self.present(alertController!, animated: true, completion: nil)
        }
    }
    
    func assessmentSubmitted(model: AssessmentSubmitModel) {
        self.assessmentSubmit = model
        alertController?.dismiss(animated: true)
        self.timer.invalidate()
        quizSubmittedView.isHidden = false
        resultStatusLabel.text = model.lable
        scoreLabel.text = "Score: \(model.achievedMarks ?? 0) / \(model.totalMarks ?? 0)"
        if model.achievedMarks == 0 {
            self.viewAnswerButton.isHidden = true
        } else {
            self.viewAnswerButton.isHidden = false
        }
    }
}

extension NewQuizViewController: UITableViewDelegate, UITableViewDataSource, UITableViewDragDelegate {
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        if tableView == mcqTableView {
            return quizModel[quesIndex].options?.count ?? 0
        } else if tableView == boolTableView {
            return 2
        } else if tableView == option1TableView {
            return quizModel[quesIndex].optionsOne?.count ?? 0
        } else {
            return quizModel[quesIndex].optionsTwo?.count ?? 0
        }
    }
    
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        if tableView == mcqTableView {
            guard let cell = tableView.dequeueReusableCell(withIdentifier: "MCQTableViewCell", for: indexPath) as? MCQTableViewCell else {
                return UITableViewCell()
            }
            
            cell.optionLabel.text = quizModel[quesIndex].options?[indexPath.row]
            cell.checkButton.isSelected = false // Ensure no button is selected initially
            
            // Finding the index of the object with the specific value
            if let index = manager.submittedAnswers.firstIndex(where: { $0.questionId == quizModel[quesIndex].id }) {
                // Check for answers
                let storedAnswers = manager.submittedAnswers[index].answers[0]
                if quizModel[quesIndex].options?[indexPath.row] == storedAnswers {
                    // Set the button's images for selected and deselected states
                    cell.checkButton.isSelected = true
                    self.isOptionSelected = true
                    self.answer.append(quizModel[quesIndex].options?[indexPath.row] ?? "")
                } else {
                    cell.checkButton.isSelected = false
                }
            }
            
            // Configure the cell with options and update the selected state
            cell.checkButton.tag = indexPath.row
            cell.checkButton.addTarget(self, action: #selector(checkButtonTapped), for: .touchUpInside)
            
            return cell
        } else if tableView == boolTableView {
            guard let cell = tableView.dequeueReusableCell(withIdentifier: "TrueFalseTableViewCell", for: indexPath) as? TrueFalseTableViewCell else {
                return UITableViewCell()
            }
            
            cell.optionLabel.text = AppConstant.boolOption[indexPath.row]
            
            return cell
        } else if tableView == option1TableView {
            guard let cell = tableView.dequeueReusableCell(withIdentifier: "DragTableViewCell", for: indexPath) as? DragTableViewCell else {
                return UITableViewCell()
            }
            
            cell.optionLabel.text = quizModel[quesIndex].optionsOne?[indexPath.row]
            
            return cell
        } else {
            guard let cell = tableView.dequeueReusableCell(withIdentifier: "DragTableViewCell", for: indexPath) as? DragTableViewCell else {
                return UITableViewCell()
            }
            
            cell.optionLabel.text = quizModel[quesIndex].optionsTwo?[indexPath.row]
            
            return cell
        }
    }
    
    @objc func checkButtonTapped(sender: UIButton) {
        let point = sender.convert(CGPoint.zero, to: mcqTableView)
        if let indexPath = mcqTableView.indexPathForRow(at: point) {
            if let cell = mcqTableView.cellForRow(at: indexPath) as? MCQTableViewCell {
                // Deselect all other buttons in the same section
                if let section = quizModel[quesIndex].options {
                    for i in 0..<section.count {
                        if i != indexPath.row {
                            let otherIndexPath = IndexPath(row: i, section: indexPath.section)
                            if let otherCell = mcqTableView.cellForRow(at: otherIndexPath) as? MCQTableViewCell {
                                otherCell.checkButton.isSelected = false
                                if let index = answer.firstIndex(of: quizModel[quesIndex].options?[otherIndexPath.row] ?? "") {
                                    answer.remove(at: index)
                                }
                                print(answer)
                            }
                        }
                    }
                }
                // Select the tapped button
                cell.checkButton.isSelected = true
                self.isOptionSelected = true
                answer.append(quizModel[quesIndex].options?[indexPath.row] ?? "")
                print(answer)
            }
        }
    }
    
    func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
        if tableView == mcqTableView {
            if quizModel[quesIndex].questionType == AppConstant.MultipleMCQ {
                if answer.contains(quizModel[quesIndex].options?[indexPath.row] ?? "") {
                    if let index = answer.firstIndex(of: quizModel[quesIndex].options?[indexPath.row] ?? "") {
                        answer.remove(at: index)
                    }
                    print(answer)
                } else {
                    answer.append(quizModel[quesIndex].options?[indexPath.row] ?? "")
                    print(answer)
                }
            } else if quizModel[quesIndex].questionType == AppConstant.SingleMCQ {
                // Retrieve the cell
                if let cell = tableView.cellForRow(at: indexPath) as? MCQTableViewCell {
                    // Programmatically trigger the button tap action
                    cell.checkButton.sendActions(for: .touchUpInside)
                }
            }
        } else {
            answer.removeAll()
            answer.append(AppConstant.boolOption[indexPath.row])
            print(answer)
        }
    }
    
    func tableView(_ tableView: UITableView, didDeselectRowAt indexPath: IndexPath) {
        if tableView == mcqTableView {
            if quizModel[quesIndex].questionType == AppConstant.MultipleMCQ {
                if answer.contains(quizModel[quesIndex].options?[indexPath.row] ?? "") {
                    if let index = answer.firstIndex(of: quizModel[quesIndex].options?[indexPath.row] ?? "") {
                        answer.remove(at: index)
                    }
                    print(answer)
                } else {
                    answer.append(quizModel[quesIndex].options?[indexPath.row] ?? "")
                    print(answer)
                }
            } else if quizModel[quesIndex].questionType == AppConstant.SingleMCQ {
//                answer.removeAll()
//                answer.append(quizModel[quesIndex].options?[indexPath.row] ?? "")
//                print(answer)
            }
        }
    }
    
    func tableView(_ tableView: UITableView, itemsForBeginning session: UIDragSession, at indexPath: IndexPath) -> [UIDragItem] {
        let dragItem = UIDragItem(itemProvider: NSItemProvider())
        dragItem.localObject = quizModel[quesIndex].optionsTwo?[indexPath.row]
        return [ dragItem ]
    }
    
    func tableView(_ tableView: UITableView, moveRowAt sourceIndexPath: IndexPath, to destinationIndexPath: IndexPath) {
        // Update the model
        let mover = quizModel[quesIndex].optionsTwo?.remove(at: sourceIndexPath.row) ?? ""
        quizModel[quesIndex].optionsTwo?.insert(mover, at: destinationIndexPath.row)
        
        //print("\(quizModel[quesIndex].optionsOne?[destinationIndexPath.row] ?? "") - \(quizModel[quesIndex].optionsTwo?[destinationIndexPath.row] ?? "")")
        self.answer.removeAll()
        for i in 0...quizModel[quesIndex].optionsOne!.count-1 {
            let match = "\(quizModel[quesIndex].optionsOne?[i] ?? "")-\(quizModel[quesIndex].optionsTwo?[i] ?? "")"
            answer.append(match)
        }
        print(answer)
    }
    
}

extension NewQuizViewController: UICollectionViewDelegate, UICollectionViewDataSource, UICollectionViewDelegateFlowLayout {
    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return quizModel.count
    }
    
    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
        let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "QuestionNumberCollectionViewCell", for: indexPath as IndexPath) as! QuestionNumberCollectionViewCell
        
        cell.serialNoLabel.text = "\(indexPath.row + 1)"
        
        if manager.submittedAnswers.contains(where: { $0.questionId == quizModel[indexPath.row].id }) {
            cell.customCellView.backgroundColor = #colorLiteral(red: 0.05099999905, green: 0.2980000079, blue: 0.5329999924, alpha: 1)
            cell.serialNoLabel.textColor = UIColor.white
            self.mcqTableView.reloadData()
        } else {
            cell.customCellView.backgroundColor = UIColor.systemGray5
            cell.serialNoLabel.textColor = UIColor.black
        }
        
        return cell
    }
    
    func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
        self.quesIndex = indexPath.row
        self.questionNumberView.isHidden = true
        print("index: \(self.quesIndex)")
        setupQuizUI()
        setupData()
    }
    
    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
        let cellWidth = collectionView.frame.size.width / 5
        return CGSize(width: cellWidth, height: cellWidth)
    }
    
    func updateCollectionViewHeight() {
        // Calculate the content size of the UICollectionView based on its content
        questionNumberCollectionView.layoutIfNeeded()
        let contentSize = questionNumberCollectionView.collectionViewLayout.collectionViewContentSize
        
        // Update the height constraint to match the content size
        collectionViewHeightConstraint.constant = contentSize.height
    }
}