import Preferences
import SharedUIComponents
import SwiftUI

#if canImport(ProHostApp)
import ProHostApp
#endif

struct PromptToCodeSettingsView: View {
    final class Settings: ObservableObject {
        @AppStorage(\.hideCommonPrecedingSpacesInPromptToCode)
        var hideCommonPrecedingSpaces
        @AppStorage(\.promptToCodeCodeFont)
        var font
        @AppStorage(\.promptToCodeGenerateDescription)
        var promptToCodeGenerateDescription
        @AppStorage(\.promptToCodeGenerateDescriptionInUserPreferredLanguage)
        var promptToCodeGenerateDescriptionInUserPreferredLanguage
        @AppStorage(\.promptToCodeChatModelId)
        var promptToCodeChatModelId
        @AppStorage(\.promptToCodeEmbeddingModelId)
        var promptToCodeEmbeddingModelId
        @AppStorage(\.wrapCodeInPromptToCode) var wrapCode
        @AppStorage(\.chatModels) var chatModels
        @AppStorage(\.embeddingModels) var embeddingModels
        init() {}
    }

    @StateObject var settings = Settings()

    var body: some View {
        VStack(alignment: .center) {
            Form {
                Picker(
                    "Chat model",
                    selection: $settings.promptToCodeChatModelId
                ) {
                    Text("Same as chat feature").tag("")

                    if !settings.chatModels
                        .contains(where: { $0.id == settings.promptToCodeChatModelId }),
                        !settings.promptToCodeChatModelId.isEmpty
                    {
                        Text(
                            (settings.chatModels.first?.name).map { "\($0) (Default)" }
                                ?? "No model found"
                        )
                        .tag(settings.promptToCodeChatModelId)
                    }

                    ForEach(settings.chatModels, id: \.id) { chatModel in
                        Text(chatModel.name).tag(chatModel.id)
                    }
                }

                Picker(
                    "Embedding model",
                    selection: $settings.promptToCodeEmbeddingModelId
                ) {
                    Text("Same as chat feature").tag("")

                    if !settings.embeddingModels
                        .contains(where: { $0.id == settings.promptToCodeEmbeddingModelId }),
                        !settings.promptToCodeEmbeddingModelId.isEmpty
                    {
                        Text(
                            (settings.embeddingModels.first?.name).map { "\($0) (Default)" }
                                ?? "No model found"
                        )
                        .tag(settings.promptToCodeEmbeddingModelId)
                    }

                    ForEach(settings.embeddingModels, id: \.id) { embeddingModel in
                        Text(embeddingModel.name).tag(embeddingModel.id)
                    }
                }

                Toggle(isOn: $settings.promptToCodeGenerateDescription) {
                    Text("Generate description")
                }

                Toggle(isOn: $settings.promptToCodeGenerateDescriptionInUserPreferredLanguage) {
                    Text("Generate description in user preferred language")
                }
            }

            SettingsDivider("UI")

            Form {
                Toggle(isOn: $settings.hideCommonPrecedingSpaces) {
                    Text("Hide common preceding spaces")
                }
                
                Toggle(isOn: $settings.wrapCode) {
                    Text("Wrap code")
                }

                #if canImport(ProHostApp)

                CodeHighlightThemePicker(scenario: .promptToCode)

                #endif
                
                FontPicker(font: $settings.font) {
                    Text("Font")
                }
            }

            ScopeForm()
        }
    }

    struct ScopeForm: View {
        class Settings: ObservableObject {
            @AppStorage(\.enableSenseScopeByDefaultInPromptToCode)
            var enableSenseScopeByDefaultInPromptToCode: Bool
            init() {}
        }

        @StateObject var settings = Settings()

        var body: some View {
            SettingsDivider("Scopes")

            VStack {
                #if canImport(ProHostApp)

                SubSection(
                    title: Text("Sense Scope (Experimental)"),
                    description: IfFeatureEnabled(\.senseScopeInChat) {
                        Text("""
                        Enable the bot to access the relevant code \
                        of the editing document in the project, third party packages and the SDK.
                        """)
                    } else: {
                        VStack(alignment: .leading) {
                            Text("""
                            Enable the bot to read the relevant code \
                            of the editing document in the SDK, and
                            """)

                            WithFeatureEnabled(\.senseScopeInChat, alignment: .inlineLeading) {
                                Text("the project and third party packages.")
                            }
                        }
                    }
                ) {
                    Form {
                        Toggle(isOn: $settings.enableSenseScopeByDefaultInPromptToCode) {
                            Text("Enable by default")
                        }
                    }
                }

                #endif
            }
        }
    }
}

#Preview {
    PromptToCodeSettingsView()
        .padding()
}