Models.swift

import CopilotModel
import Foundation

public struct EditorContent: Codable {
    public struct Selection: Codable {
        public var start: CursorPosition
        public var end: CursorPosition

        public init(start: CursorPosition, end: CursorPosition) {
            self.start = start
            self.end = end
        }
    }

    public init(
        content: String,
        lines: [String],
        uti: String,
        cursorPosition: CursorPosition,
        selections: [Selection],
        tabSize: Int,
        indentSize: Int,
        usesTabsForIndentation: Bool
    ) {
        self.content = content
        self.lines = lines
        self.uti = uti
        self.cursorPosition = cursorPosition
        self.selections = selections
        self.tabSize = tabSize
        self.indentSize = indentSize
        self.usesTabsForIndentation = usesTabsForIndentation
    }

    public var content: String
    public var lines: [String]
    public var uti: String
    public var cursorPosition: CursorPosition
    public var selections: [Selection]
    public var tabSize: Int
    public var indentSize: Int
    public var usesTabsForIndentation: Bool

    public func selectedCode(in selection: Selection) -> String {
        let startPosition = selection.start
        let endPosition = selection.end

        guard startPosition.line >= 0, startPosition.line < lines.count else { return "" }
        guard endPosition.line >= 0, endPosition.line < lines.count else { return "" }
        guard endPosition.line >= startPosition.line else { return "" }

        var code = ""
        if startPosition.line == endPosition.line {
            let line = lines[startPosition.line]
            let startIndex = line.index(line.startIndex, offsetBy: startPosition.character)
            let endIndex = line.index(line.startIndex, offsetBy: endPosition.character)
            code = String(line[startIndex...endIndex])
        } else {
            let startLine = lines[startPosition.line]
            let startIndex = startLine.index(
                startLine.startIndex,
                offsetBy: startPosition.character
            )
            code += String(startLine[startIndex...])

            if startPosition.line + 1 < endPosition.line {
                for line in lines[startPosition.line + 1...endPosition.line - 1] {
                    code += line
                }
            }

            let endLine = lines[endPosition.line]
            let endIndex = endLine.index(endLine.startIndex, offsetBy: endPosition.character)
            code += String(endLine[...endIndex])
        }
        
        return code
    }
}

public struct UpdatedContent: Codable {
    public init(content: String, newCursor: CursorPosition? = nil, modifications: [Modification]) {
        self.content = content
        self.newCursor = newCursor
        self.modifications = modifications
    }

    public var content: String
    public var newCursor: CursorPosition?
    public var modifications: [Modification]
}