from pymongo import MongoClient
import os
from dotenv import load_dotenv

load_dotenv()

cluster = MongoClient(os.getenv("MONGO_URI"))

db = cluster["traveloka"]

dataAcc = db["account"]
dataFlight = db["flight"]
dataHotel = db["hotel"]
dataTransaction = db["transaction"]

from Controller import auth_controller
import os

class MenuUtama:
    def __init__(self):
        self.auth = auth_controller.User()

    def run(self):
        try:
            os.system('cls')
            while True:
                print('=================================')
                print('|            Welcome            |')
                print('|           Traveloka           |')
                print('=================================')
                print('|>>>>> Silahkan pilih opsi <<<<<|')
                print('|                               |')
                print('|   1. Registrasi Akun          |')
                print('|   2. Login Akun               |')
                print('|   3. Lupa Password            |')
                print('|   4. Keluar                   |')
                print('|                               |')
                print('=================================')

                opsi = str(input('Tentukan opsi anda (1/2/3/4): '))

                if opsi == '1':
                    self.auth.register()
                elif opsi == '2':
                    self.auth.login()
                elif opsi == '3':
                    self.auth.forgot_password()
                elif opsi == '4':
                    print('Terima kasih telah menggunakan layanan kami!')
                    exit()
                else:
                    print('Opsi tidak tersedia!')
                    os.system('cls')

 except KeyboardInterrupt:
            print('Terjadi kesalahan!')
            exit()

from Controller import flight_controller as fc
from Controller import auth_controller as auth
from Controller import user_controller as uc
from View import main_view as main
import os

class UserView:

    def __init__(self):
        self.flight = fc.LinkedList()
        self.auth = auth.User()
        self.user = uc.UserController()

    def menu_user(self):
        try:
            os.system('cls')
            while True:
                print("==================================")
                print("|            M E N U             |")
                print("==================================")
                print("|-----> Menu yang tersedia <-----|")
                print("|                                |")
                print("|    1. Cari Tiket Pesawat       |")
                print("|    2. Lihat Tiket Pesawat      |")
                print("|    3. Riwayat Pembelian        |")
                print("|    4. Isi Saldo                |")
                print("|    5. Cek Profil               |")
                print("|    6. Urutkan Data Pesawat     |")
                print("|    7. Sign Out                 |")
                print("|                                |")
                print("==================================")
                opsi = str(input("Tentukan opsi anda (1/2/3/4/5): "))

                if opsi == '1':
                    os.system('cls')
                    self.user.buyTicket()
                elif opsi == '2':
                    os.system('cls')
                    self.flight.display()
                elif opsi == '3':
                    os.system('cls')
                    self.user.checkHistory()
                elif opsi == '4':
                    os.system('cls')
                    self.user.addBalance()
                elif opsi == '5':
                    os.system('cls')
                    self.auth.profile()
                elif opsi == '6':
                    self.flight.sort_flights()
                elif opsi == '7':
                    self.auth.logout()
                    main.MenuUtama().run()
                else:
                    print("Opsi tidak tersedia!!")

        except KeyboardInterrupt:
            print("\nTerjadi Kesalahan!")
            exit()

from View import main_view as main
from Controller import flight_controller as fc
from Controller import auth_controller as auth
import os

def menu_admin(self):
        try:
            os.system('cls')
            while True:
                print("==================================")
                print("|            M E N U             |")
                print("==================================")
                print("|-----> Menu yang tersedia <-----|")
                print("|                                |")
                print("|    1. Tambah Pesawat           |")
                print("|    2. Lihat Pesawat            |")
                print("|    3. Edit Pesawat             |")
                print("|    4. Hapus Pesawat            |")
                print("|    5. Sign Out                 |")
                print("|                                |")
                print("==================================")
                opsi = str(input("Tentukan opsi anda (1/2/3/4/5): "))

                if opsi == '1':
                    os.system('cls')
                    self.flight.addFlight()
                elif opsi == '2':
                    os.system('cls')
                    self.flight.display()
                elif opsi == '3':
                    os.system('cls')
                    self.flight.updateFlight()
                elif opsi == '4':
                    os.system('cls')
                    self.flight.deleteFlight()
                elif opsi == '5':
                    self.auth.logout()
                    main.MenuUtama().run()
                else:
                    print("Opsi tidak tersedia!")

from Model import database as db
from View import admin_view
from View import user_view
import main
import pwinput
import os
import re

class User:

    user_session = []

def register(self):
        try:
            os.system('cls')
            print("----------------------- REGISTRASI ----------------------")
            name = str.capitalize(input("Masukkan username: "))
            if db.dataAcc.find_one({"name": name}):
                print("Username sudah digunakan!")
                return False
            if not name:
                print("Nama pengguna tidak boleh kosong!")
                return False
            elif re.search("[^A-Za-z0-9_]", name):
                print("Nama pengguna hanya boleh terdiri dari huruf, angka, dan underscore (_)!")
                return False
            else:
                password = str.lower(pwinput.pwinput("Masukkan password: "))
                if not password:
                    print("Kata sandi tidak boleh kosong!")
                    return False
                elif re.search("[^A-Za-z0-9!@#$%^&*()_+-=]", password):
                    print("Kata sandi hanya boleh terdiri dari huruf, angka, dan karakter simbol seperti !, @, #, $, %, ^, &, *, (, ), _, +, -, =!")
                    return False

else:
    password = str.lower(pwinput.pwinput("Masukkan password: "))
    if not password:
        print("Kata sandi tidak boleh kosong!")
        return False
    elif re.search("[^A-Za-z0-9!@#$%^&*()_+-=]", password):
        print("Kata sandi hanya boleh terdiri dari huruf, angka, dan karakter simbol seperti !, @, #, $, %, ^, &, *, (, ), _, +, -, =!")
        return False

else:
     saldo = int(input("Masukkan saldo awal: "))
     if saldo < 0:
         print("Saldo tidak boleh kurang dari 0!")
         return False
     if saldo > 10000000:
         print("Saldo tidak boleh lebih dari 10.000.000!")
         return False
     else:
         email = str(input("Masukkan email: "))
         if not email:
             print("Email tidak boleh kosong!")
             return False
         elif email.count("@") > 1:
             print("Email hanya boleh satu simbol @!")
             return False
         elif email == db.dataAcc.find_one({"email": email}):
             print("Email sudah digunakan!")
             return False
         elif re.search("[^A-Za-z0-9@.]", email):
             print("Email hanya boleh terdiri dari huruf, angka, dan karakter simbol seperti @ dan .!")
             return False

db.dataAcc.insert_one(
    {"name": name, "password": password, "saldo": saldo, "email": email, "privilege": "Reguler",
    "role": "user"})
print("\n--------------- DATA TELAH DIKONFRIMASI ---------------")
print("> Username: ", name)
print("> Password: ", password)
print("> Saldo: ", saldo)
print("> Email: ", email)
print("---------------------------------------------------------")
confirm = input("Tekan [ENTER] jika sesuai dengan anda input!")
if confirm == '':
    print("\nRegistrasi Berhasil!")
    return True
else:
    return True

def login(self):
        os.system('cls')
        print("----------------------- LOGIN ---------------------------")
        count = 3
        while count >= 0:
            count -= 1
            name = str.capitalize(input("> Masukkan username anda: "))
            password = str.lower(pwinput.pwinput("> Masukkan password anda: "))
            user = db.dataAcc.find_one({"name": name, "password": password})
            if user and user["name"] == name and user["password"] == password:
                if user:
                    if user.get("role") == "admin":
                        print("Login berhasil!")
                        self.logged_in = True
                        self.username = user["name"]
                        admin_view.AdminView().menu_admin()
                        return True
                    elif user.get("role") == "user":
                        print("Login berhasil!")
                        self.logged_in = True
                        self.username = user["name"]
                        session_data = {"username": self.username}
                        User.user_session.append(session_data)
                        user_view.UserView().menu_user()
                        return True

from Controller import flight_controller as fc
from Controller import auth_controller as auth
from Controller import email_controller as email
from Model import database as db
from datetime import datetime
from prettytable import PrettyTable

class Node:
    def __init__(self, data=None, time=None):
        self.data = data
        self.time = time
        self.next = None

class LinkedList:
    def __init__(self):
        self.head = None

    def append(self, data, time):
        new_node = Node(data, time) 
        if not self.head:
            self.head = new_node
            return
        curr_node = self.head
        while curr_node.next:
            curr_node = curr_node.next
        curr_node.next = new_node

    def display(self):
        curr_node = self.head
        while curr_node:
            print (f"{curr_node.time}: {curr_node.data}") 
            curr_node = curr_node.next

class UserController:
    def __init__(self):
        self.flight = fc.LinkedList()
        self.db = db.dataFlight
        self.history = LinkedList()

    def buyTicket(self):
        try:
            user = db.dataAcc.find_one({"name": auth.User.user_session[0]['username']})
            self.flight.display()
            idFlight = str.upper(input("Masukkan ID Flight: "))
            self.flight.search(idFlight)
            if self.flight.search(idFlight):
                for d in self.db.find({"idFlight": idFlight}):
                    price = d['price']
                    total = price
                    if user['saldo'] >= total:
                        db.dataAcc.update_one({"name": user["name"]}, {"$inc": {"saldo": -total}})
                        print(f"Transaksi berhasil! Sisa saldo anda adalah: {user['saldo']}")
                        email.send_email(user['email'], d['idFlight'], user['name'], d['origin'], d['destination'], d['airline'], d['dateTime'], d['departureTime'], d['arrivalTime'], d['price'])

                        transaction = f"Beli tiket {total} untuk penerbangan {idFlight} dengan total harga {total}"
                        self.history.append(transaction, datetime.now().strftime('%Y-%m-%d %H:%M:%S'))

                    else:
                        print("Saldo tidak cukup!")
            else:
                print("Tiket tidak ditemukan")
                
        except KeyboardInterrupt:
            print("Terjadi Kesalahan!")

def addBalance(self):
        try:
            user = db.dataAcc.find_one({"name": auth.User.user_session[0]['username']})
            print("Saldo anda adalah: ", user['saldo'])

            add = int(input("Masukkan jumlah saldo yang ingin ditambahkan: "))
            if add < 0:
                print("Jumlah saldo tidak boleh kurang dari 0")
                return
            if add > 10000000:
                print("Jumlah saldo tidak boleh lebih dari 10.000.000")
                return
            else:
                user['saldo'] += add
                db.dataAcc.update_one({"name": user["name"]}, {"$set": {"saldo": user['saldo']}})
                print(f"Saldo berhasil ditambahkan! Saldo sekarang adalah: {user['saldo']}")

                transaction = f"Menambahkan saldo sebesar {add}"
                self.history.append(transaction, datetime.now().strftime('%Y-%m-%d %H:%M:%S'))

        except ValueError:
            print("Saldo harus berupa angka!")
        except KeyboardInterrupt:
            print("Terjadi Kesalahan!")

def checkHistory(self):
        try:
            print("Transaction History:")
            table = PrettyTable()
            table.field_names = ["Tanggal", "Transaksi"]
            curr_node = self.history.head
            while curr_node:
                table.add_row([curr_node.time, curr_node.data])
                curr_node = curr_node.next
            
            print(table)

        except KeyboardInterrupt:
            print("Terjadi Kesalahan!")

from __future__ import print_function
import sib_api_v3_sdk
from sib_api_v3_sdk.rest import ApiException
from pprint import pprint
from dotenv import load_dotenv
import random
import os

load_dotenv()

configuration = sib_api_v3_sdk.Configuration()
configuration.api_key['api-key'] = os.getenv('SENDINBLUE_API_KEY')

def send_email(email, id_flight, name, asal, tujuan, pesawat, tanggal, waktu_keberangkatan, waktu_kedatangan, harga):
    api_instance = sib_api_v3_sdk.TransactionalEmailsApi(sib_api_v3_sdk.ApiClient(configuration))
    receipt = str(random.randint(100000, 999999))
    ticket_code = str(random.randint(100000, 999999))
    to = [{"email": email, "name": name}]
    params = {
        "RECEIPT":receipt,
        "ID_FLIGHT": id_flight,
        "NAMA": name,
        "ASAL": asal,
        "TUJUAN": tujuan,
        "NAMA_PESAWAT": pesawat,
        "TANGGAL": tanggal,
        "WAKTU_KEBERANGKATAN": waktu_keberangkatan,
        "WAKTU_KEDATANGAN": waktu_kedatangan,
        "KODE_TIKET": ticket_code,
        "HARGA": harga
    }
    send_smtp_email = sib_api_v3_sdk.SendSmtpEmail(to=to, params=params, template_id=1)

    try:
        api_response = api_instance.send_transac_email(send_smtp_email)
        pprint(api_response)
    except ApiException as e:
        print("Exception when calling SMTPApi->send_transac_email: %s\n" % e)

from Model import database as db
from prettytable import PrettyTable
import random
import os
import re

def append(self, data):
        new_node = Node(data)
        if self.head is None:
            self.head = new_node
            return
        last_node = self.head
        while last_node.next:
            last_node = last_node.next
        last_node.next = new_node

 def display(self):
        data = []
        for d in self.db.find({}):
            data.append(d)

        if not data:
                print("List kosong")
            else:
                table = PrettyTable(
                    ['ID Flight', 'Pesawat', 'Asal', 'Tujuan', 'Waktu Keberangkatan', 'Waktu Kedatangan', 'Tanggal Keberangkatan',
                    'Harga'])
                for d in data:
                    table.add_row(
                        [d['idFlight'], d['airline'], d['origin'], d['destination'], d['departureTime'], d['arrivalTime'],
                        d['dateTime'], d['price']])
                print(table)

def search(self, key):
        data = []
        for d in self.db.find({}):
            data.append(d)

        if not data:
            print("List kosong")
            return

        n = len(data)
        step = int(n ** 0.5)
        prev = 0
        while prev < n and data[prev]['idFlight'] < key:
            prev += step
        prev -= step
        while prev < n:
            if data[prev]['idFlight'] == key:
                return Node(data[prev])
            prev += 1
        return None

 def addFlight(self):
        try:
            print("=====> Masukkan data penerbangan baru <=====")
            airline = str.capitalize(input("> Nama Pesawat: "))
            if not airline:
                print("Nama pesawat tidak boleh kosong!\n")
                return
            
            origin = str.capitalize(input("> Kota Asal: "))
            if not origin:
                print("Kota asal tidak boleh kosong!\n")
                return
            
            destination = str.capitalize(input("> Kota Tujuan: "))
            if not destination:
                print("Kota tujuan tidak boleh kosong!\n")
                return
            
            departureTime = str(input("> Waktu Keberangkatan (hh:mm):"))
            if not departureTime:
                print("Waktu keberangkatan tidak boleh kosong!\n")
                return
            elif re.match(r"([01]?[0-9]|2[0-3]):[0-5][0-9]", departureTime) is None:
                print("Format waktu keberangkatan salah!\n")
                return
            
            arrivalTime = str(input("> Waktu Kedatangan (hh:mm): "))
            if not arrivalTime:
                print("Waktu kedatangan tidak boleh kosong!\n")
                return
            elif re.match(r"([01]?[0-9]|2[0-3]):[0-5][0-9]", arrivalTime) is None:
                print("Format waktu kedatangan salah!\n")
                return
            
            dateTime = str(input("> Tanggal Keberangkatan (yyyy-mm-dd): "))
            if not dateTime:
                print("Tanggal keberangkatan tidak boleh kosong!\n")
                return
            elif len(dateTime) != 10:
                print("Format tanggal salah!\n")
                return
            elif re.match(r"([12]\d{3}-(0[1-9]|1[0-2])-(0[1-9]|[12]\d|3[01]))", dateTime) is None:
                print("Format tanggal salah!\n")
                return
            
            price = int(input("> Harga tiket: "))
            if not price:
                print("Harga tiket tidak boleh kosong!\n")
                return
            elif price < 0:
                print("Harga tiket tidak boleh kurang dari 0!\n")
                return

def idFlight():
       if "garuda indonesia" in airline.lower():
           if "GA" + str(random.randint(100, 999)) == self.search("GA" + str(random.randint(100, 999))):
               return "GA" + str(random.randint(100, 999))
           else:
               return "GA" + str(random.randint(100, 999))

       elif "lion air" in airline.lower():
           if "JT" + str(random.randint(100, 999)) == self.search("JT" + str(random.randint(100, 999))):
               return "JT" + str(random.randint(100, 999))
           else:
               return "JT" + str(random.randint(100, 999))

       elif "sriwijaya air" in airline.lower():
           if "SJ" + str(random.randint(100, 999)) == self.search("SJ" + str(random.randint(100, 999))):
               return "SJ" + str(random.randint(100, 999))
           else:
               return "SJ" + str(random.randint(100, 999))

       elif "citilink" in airline.lower():
           if "QG" + str(random.randint(100, 999)) == self.search("QG" + str(random.randint(100, 999))):
               return "QG" + str(random.randint(100, 999))
           else:
               return "QG" + str(random.randint(100, 999))

       elif "air asia" in airline.lower():
           if "QZ" + str(random.randint(100, 999)) == self.search("QZ" + str(random.randint(100, 999))):
               return "QZ" + str(random.randint(100, 999))
           else:
               return "QZ" + str(random.randint(100, 999))

       elif "batik air" in airline.lower():
           if "ID" + str(random.randint(100, 999)) == self.search("ID" + str(random.randint(100, 999))):
               return "ID" + str(random.randint(100, 999))
           else:
               return "ID" + str(random.randint(100, 999))

       else:
           if "XX" + str(random.randint(100, 999)) == self.search("XX" + str(random.randint(100, 999))):
               return "XX" + str(random.randint(100, 999))
           else:
               return "XX" + str(random.randint(100, 999))

new_flight = {
                "idFlight": idFlight(),
                "airline": airline,
                "origin": origin,
                "destination": destination,
                "departureTime": departureTime,
                "arrivalTime": arrivalTime,
                "dateTime": dateTime,
                "price": price
            }

            new_node = Node(new_flight)

            if not self.head:
                self.head = new_node
                self.tail = new_node
            else:
                self.tail.next = new_node
                self.tail = new_node

            self.db.insert_one(new_flight)
            print("Pesawat berhasil ditambahkan!\n")
        except ValueError:
            print("Masukkan data dengan benar!\n")
        except KeyboardInterrupt:
            print("Terjadi Kesalahan!\n")

def deleteFlight(self):
        try:
            print("Hapus Pesawat\n")
            self.display()
            idFlight = str(input("Masukkan ID Pesawat: "))
            self.search(idFlight)
            if self.search(idFlight):
                self.db.delete_one({"idFlight": idFlight})
                print("Pesawat berhasil dihapus!\n")
            elif not self.search(idFlight):
                print("Pesawat tidak ditemukan!\n")
            else:
                print("Pesawat tidak ditemukan!\n")

        except ValueError:
            print("Masukkan data dengan benar!\n")
        except KeyboardInterrupt:
            print("Terjadi Kesalahan!\n")

  def updateFlight(self):
        print("Edit Pesawat\n")
        self.display()
        idFlight = str(input("Masukkan ID Pesawat yang ingin di update: "))
        self.search(idFlight)
        if self.search(idFlight):
            print('=================================')
            print('|   Apa yang ingin di update?   |')
            print('=================================')
            print('|>>>>> Silahkan pilih opsi <<<<<|')
            print('|                               |')
            print('|   1. Kota Asal                |')
            print('|   2. Kota Tujuan              |')
            print('|   3. Waktu Keberangkatan      |')
            print('|   4. Waktu Kedatangan         |')
            print('|   5. Tanggal Keberangkatan    |')
            print('|   6. Harga                    |')
            print('|   7. Kembali                  |')
            print('|                               |')
            print('=================================')
            update = str(input('Pilih data yang ingin di update: '))

            if update == '1':
                newData = str.capitalize(input("> Masukkan kota asal baru: "))
                if not newData:
                    print("Data tidak boleh kosong!\n")
                else:
                    self.db.update_one({"idFlight": idFlight}, {"$set": {"origin": newData}})
                    print("Data berhasil di update!\n")

            elif update == '2':
                newData = str.capitalize(input("> Masukkan kota tujuan baru: "))
                if not newData:
                    print("Data tidak boleh kosong!\n")
                else:
                    self.db.update_one({"idFlight": idFlight}, {"$set": {"destination": newData}})
                    print("Data berhasil di update!\n")

            elif update == '3':
                newData = str(input("> Masukkan waktu keberangkatan baru (hh:mm): "))
                if not newData:
                    print("Data tidak boleh kosong!\n")
                else:
                    self.db.update_one({"idFlight": idFlight}, {"$set": {"departureTime": newData}})
                    print("Data berhasil di update!\n")

            elif update == '4':
                newData = str(input("> Masukkan waktu kedatangan baru (hh:mm): "))
                if not newData:
                    print("Data tidak boleh kosong!\n")
                else:
                    self.db.update_one({"idFlight": idFlight}, {"$set": {"arrivalTime": newData}})
                    print("Data berhasil di update!\n")

            elif update == '5':
                newData = str(input("> Masukkan tanggal keberangkatan baru (yyyy-mm-dd): "))
                if not newData:
                    print("Data tidak boleh kosong!\n")
                else:
                    self.db.update_one({"idFlight": idFlight}, {"$set": {"dateTime": newData}})
                    print("Data berhasil di update!\n")

            elif update == '6':
                newData = int(input("> Masukkan harga baru: "))
                if newData < 0:
                    print("Harga tidak boleh kurang dari nol!\n")
                else:   
                    self.db.update_one({"idFlight": idFlight}, {"$set": {"price": newData}})
                    print("Data berhasil di update!\n")

            elif update == '7':
                return os.system('cls')

            else:
                print("Pilihan tidak tersedia!\n") 

        elif not self.search(idFlight):
            print("Pesawat tidak ditemukan!\n")
        else:
            print("Pesawat tidak ditemukan!\n")

 def quick_sort(self, data):
        if len(data) <= 1:
            return data

        pivot = data[0]
        left = []
        right = []

        for i in range(1, len(data)):
            if data[i]['airline'] < pivot['airline']:
                left.append(data[i])
            elif data[i]['airline'] > pivot['airline']:
                right.append(data[i])
            else:
                if data[i]['price'] < pivot['price']:
                    left.append(data[i])
                else:
                    right.append(data[i])

        return self.quick_sort(left) + [pivot] + self quick_sort(right)

 def sort_flights(self):
        data = []
        for i in self.db.find():
            data.append(i)

        sorted_data = self.quick_sort(data)
        
        table = PrettyTable()
        table.title = "Data sortir berdasarkan nama pesawat"
        table.field_names = ['ID Flight', 'Pesawat', 'Asal', 'Tujuan', 'Waktu Keberangkatan', 'Waktu Kedatangan', 'Tanggal Keberangkatan',
                 'Harga']

        for flight in sorted_data:
            table.add_row(
                    [flight['idFlight'], flight['airline'], flight['origin'], flight['destination'], flight['departureTime'], flight['arrivalTime'],
                     flight['dateTime'], flight['price']])

        print(table)