ShhShell/ShhShell/Views/Terminal/ShellTabView.swift

//
//  ShellTabView.swift
//  ShhShell
//
//  Created by neon443 on 03/07/2025.
//

import SwiftUI

struct ShellTabView: View {
	@State var handler: SSHHandler?
	@ObservedObject var hostsManager: HostsManager
	
	@ObservedObject var container = TerminalViewContainer.shared
	@State var selectedID: UUID?
	var selectedHandler: SSHHandler {
		container.sessions[selectedID ?? UUID()]?.handler ?? handler ?? SSHHandler(host: Host.blank, keyManager: nil)
	}
	
	@State var showSnippetPicker: Bool = false
	
	@Environment(\.dismiss) var dismiss
	
	var foreground: Color {
		let selectedTheme = hostsManager.selectedTheme
		let foreground = selectedTheme.foreground
		let background = selectedTheme.background
		
		if selectedTheme.ansi[hostsManager.selectedAnsi].luminance > 0.5 {
			if foreground.luminance > 0.5 {
				return background.suiColor
			} else {
				return foreground.suiColor
			}
		} else {
			if foreground.luminance > 0.5 {
				return foreground.suiColor
			} else {
				return background.suiColor
			}
		}
	}
	var background: Color { hostsManager.selectedTheme.background.suiColor }
	
	var body: some View {
		ZStack {
			background
				.ignoresSafeArea(.all)
			VStack(spacing: 0) {
				let oneTabWidth = max(100, (UIScreen.main.bounds.width)/CGFloat(container.sessionIDs.count))
				
				HStack(alignment: .center, spacing: 10) {
					Button() {
						for session in container.sessions.values {
							session.handler.disconnect()
						}
						dismiss()
					} label: {
						TrafficLightRed()
					}
					Button() {
						dismiss()
					} label: {
						TrafficLightYellow()
					}
					Spacer()
					VStack {
						Text(selectedHandler.title)
							.bold()
							.foregroundStyle(foreground)
							.monospaced()
							.contentTransition(.numericText())
						if container.sessionIDs.count == 1 {
							Text(selectedHandler.host.description)
								.bold()
								.foregroundStyle(foreground)
								.monospaced()
								.font(.caption2)
						}
					}
					Spacer()
					Button() {
						showSnippetPicker.toggle()
					} label: {
						Image(systemName: "paperclip")
							.resizable().scaledToFit()
							.frame(width: 20, height: 20)
					}
					.foregroundStyle(foreground)
					.popover(isPresented: $showSnippetPicker) {
						SnippetPicker(hostsManager: hostsManager) {
							selectedHandler.writeToChannel($0.content)
						}
						.frame(minWidth: 200, minHeight: 300)
						.modifier(presentationCompactPopover())
					}
					Menu {
						Button() {
							UIPasteboard.general.string = selectedHandler.scrollback.joined()
							Haptic.success.trigger()
						} label: {
							Label("Copy Scrollback", systemImage: "document.on.document")
						}
					} label: {
						Image(systemName: "ellipsis")
							.resizable().scaledToFit()
							.frame(width: 20, height: 20)
					}
					.foregroundStyle(foreground)
				}
				.padding(.horizontal, 10)
				.padding(.vertical, 10)
				.background(hostsManager.tint, ignoresSafeAreaEdges: .all)
				.frame(height: 40)
				
				if container.sessionIDs.count > 1 {
					ScrollView(.horizontal, showsIndicators: false) {
						HStack(spacing: 0) {
							ForEach(container.sessionIDs, id: \.self) { id in
								let selected: Bool = selectedID == id
								ZStack {
									Rectangle()
										.fill(selected ? hostsManager.tint : background)
									HStack {
										Spacer()
										VStack {
											if !selected {
												Text(container.sessions[id]!.handler.title)
													.monospaced()
													.foregroundStyle(selected ? foreground : hostsManager.tint)
													.opacity(0.7)
													.font(.callout)
											}
											Text(container.sessions[id]!.handler.host.description)
												.foregroundStyle(selected ? foreground : hostsManager.tint)
												.opacity(selected ? 1 : 0.7)
												.monospaced()
												.bold(selected)
												.font(.caption2)
										}
										Spacer()
									}
								}
								.frame(width: oneTabWidth)
								.onTapGesture {
									withAnimation { selectedID = id }
								}
							}
						}
					}
					.frame(height: 30)
					.onAppear {
						if selectedID == nil {
							if let handler {
								selectedID = handler.sessionID
							} else {
								dismiss()
							}
						}
					}
				}
				
				//the acc terminal lol
				if let selectedID,
				   let session = container.sessions[selectedID] {
					ShellView(
						handler: session.handler,
						hostsManager: hostsManager
					)
					.onAppear {
						UIApplication.shared.isIdleTimerDisabled = hostsManager.settings.caffeinate
						if hostsManager.settings.locationPersist {
							Backgrounder.shared.startBgTracking()
						}
					}
					.onDisappear {
						if !checkShell(session.handler.state) {
							if let lastSession = container.sessionIDs.last {
								withAnimation { self.selectedID = lastSession }
							} else {
								dismiss()
							}
						}
						UIApplication.shared.isIdleTimerDisabled = false
						if container.sessions.isEmpty {
							Backgrounder.shared.stopBgTracking()
						}
					}
					.id(selectedID)
					.transition(.opacity)
				} else {
					if let handler {
						ShellView(
							handler: handler,
							hostsManager: hostsManager
						)
						.onAppear {
							if selectedID == nil {
								selectedID = handler.sessionID
							}
						}
					} else {
						Text("No Session")
					}
				}
			}
		}
	}
}

#Preview {
	ShellTabView(
		handler: SSHHandler(host: Host.blank, keyManager: nil),
		hostsManager: HostsManager()
	)
}