//
//  JavaClass.swift
//  textToCode
//
//  Created by Michael Smith on 4/21/19.
//  Copyright © 2019 Michael Smith. All rights reserved.
//

import Foundation
import UIKit

enum ItemVisability: String{
    case publicItem =  "public"
    case protectedItem = "protected"
    case defaultItem = ""
    case privateItem = "private"
}

class JavaClass{
    var name: String;
    var visability: ItemVisability;
    var classVariables: [JavaVariable] = [];
    var methods: [JavaMethod] = [];
    
    init(className: String, vis: String) {
        self.name = className;
        self.visability = ItemVisability(rawValue: vis) ?? ItemVisability.defaultItem;
    }
    
    func addMethod(methodName: String, vis: String, returnType: String){
        methods.append(JavaMethod.init(name: methodName, vis: vis, returnType: returnType));
    }
    
    func addVar(varName: String, vis: String, type: String){
        classVariables.append(JavaVariable.init(name: varName, vis: vis, type: type));
    }
    
    func findMethod(_ methodName: String) -> JavaMethod?{
        return methods.first(where: { (m:JavaMethod) -> Bool in
             return m.getName() == methodName
        })
    }
    
    func copy() -> JavaClass{
        let copy = JavaClass(className: name, vis: visability.rawValue)
        for xvar in classVariables{
            copy.classVariables.append(xvar.copy())
        }
        for xmethod in methods{
            copy.methods.append(xmethod.copy())
        }
        return copy
    }
    
    func toString() -> String{
        var outputStr: String = "";
        outputStr += "\(visability.rawValue) class \(name)(){"
        outputStr += "\n"
        for classVar in classVariables{
            outputStr += "    \(classVar.toString()) \n";
        }
        for method in methods{
            let methodLines = method.toString().components(separatedBy: "\n");
            for line in methodLines{
                outputStr += "    \(line) \n";
            }
            
        }
        outputStr += "}";
        return outputStr;
    }
    
    func toFormattedString() -> NSMutableAttributedString{
        var output = NSMutableAttributedString()
        let classHead = NSMutableAttributedString(string: "\(visability.rawValue) class \(name)(){ \n");
        output.append(classHead);
        for classVar in classVariables{
            let formattedVar = NSMutableAttributedString(string: "    \(classVar.toString()) \n");
            output.append(formattedVar)
        }
        for xmethod in methods{
            output.append(xmethod.toFormattedString())
        }
        
        let classEnd = NSMutableAttributedString(string: "} \n ");
        output.append(classEnd);
        
        if let xclass = SpeechProcessor.state.currentClass{
            if xclass.name == self.name && SpeechProcessor.state.currentMethod == nil{
                output = NSMutableAttributedString(string: self.toString());
                output.addAttributes([NSAttributedString.Key.foregroundColor: UIColor.init(red: 0.7, green: 0.0, blue: 1.0, alpha: 1.0)] as [NSAttributedString.Key: Any], range: NSRange(location: 0, length: self.toString().count));
            }
        }
        
        return output
    }
    
}