BrowserExtension.swift

import CheetahIPC

classBrowserExtensionState:JSONHandler<BrowserExtensionMessage>{
@Publishedvarmode:String?
@Publishedvarfiles=[String: String]()
@Publishedvarlogs=[String: String]()

varnavigationStart=0
varlastUpdate:Date?

publicinit(){
 super.init(respondsTo:IPCMessage.browserExtensionMessage)

 handler ={
guardlet message = $0 else{
returnnil
}

if message.navigationStart >self.navigationStart {
self.navigationStart = message.navigationStart
self.files.removeAll()
self.logs.removeAll()
}

letnewMode= message.mode
if newMode !=self.mode {
self.mode = newMode
self.files.removeAll()
self.logs.removeAll()
}

for(name, content)in message.files {
self.files[name]= content
}
for(name, content)in message.logs {
self.logs[name]= content
}

ifself.lastUpdate ==nil{
print("BrowserExtensionState: first message was received!")
}

self.lastUpdate =Date.now
returnnil
}
}

varcodeDescription:String{
if files.isEmpty {
return"N/A"
}else{
return files
.map{ name, content in"[\(name)]\n\(content)"}
.joined(separator:"\n\n")
}
}

varlogsDescription:String{
if logs.isEmpty {
return"N/A"
}else{
return logs
.map{ name, content in
letrecentLines= content.split(separator:"\n").suffix(20).joined(separator:"\n")
return"[\(name)]\n\(recentLines)"
}
.joined(separator:"\n\n")
}
}
}

structNativeMessagingManifest:Codable{
enum`Type`:String,Codable{
case stdio
}

letname:String
letdescription:String
letpath:String
lettype:`Type`
letallowedExtensions:[String]
}

func installNativeMessagingManifest()throws->Bool{
letmanifest=NativeMessagingManifest(
 name:"cheetah",
 description:"Cheetah Extension",
 path:Bundle.main.path(forAuxiliaryExecutable:"ExtensionHelper")!,
 type:.stdio,
 allowedExtensions:["cheetah@phrack.org"])

letpath=FileManager.default.homeDirectoryForCurrentUser.appending(path:"Library/Application Support/Mozilla/NativeMessagingHosts/cheetah.json").absoluteURL.path

print("Installing native messaging manifest at \(path)")

letencoder=JSONEncoder()
 encoder.keyEncodingStrategy =.convertToSnakeCase

letcontents=try encoder.encode(manifest)
returnFileManager.default.createFile(atPath: path, contents: contents)
}