Search code examples
iosswiftuiimagensfilemanagernsdocumentdirectory

Swift 3 adding an Image to Documents and Retrieving


I feel like I am missing something major as I don't receive any errors or null values, but anytime I try to retrieve my image it returns "" because it doesn't exist?

This is how I am passing the image data:

URLSession.shared.dataTask(with: url! as URL, completionHandler:{ (data, response, error) in
   //if download error
   if error != nil{
   print(error!)
   return
}
guard let imageData = UIImage(data: data!) else { return }
DispatchQueue.main.async{
   self.imgPortrait.image = imageData
   ImagePortrait().saveImageDocumentDirectory(image: imageData)
}

The data is displayed on imgPortrait completely fine. Here are the rest of the functions where: /\(userUID)/\(CharacterSelection.sharedInstance.getActiveCharacterName()) is optional. When it goes to retrieve the image and when it saves it I can confirm the paths are exactly the same. I feel like it's how I am saving the image with createFile, but I am unsure.

   func saveImageDocumentDirectory(image: UIImage){
        if let userUID = Auth.auth().currentUser?.uid{
            let fileManager = FileManager.default
            let paths = (getDirectoryPath() as NSString).appendingPathComponent("/\(userUID)/\(CharacterSelection.sharedInstance.getActiveCharacterName())/characterPortrait.png")
            createDirectory(name: (getDirectoryPath() as NSString) as String)
            let imageData = UIImagePNGRepresentation(image)
            fileManager.createFile(atPath: paths as String, contents: imageData, attributes: nil)
        }
    }

    func getDirectoryPath() -> String {
        let paths = NSSearchPathForDirectoriesInDomains(.documentDirectory, .userDomainMask, true)
        let documentsDirectory = paths[0]
        return documentsDirectory
    }

    func getImage() -> String{
        if let userUID = Auth.auth().currentUser?.uid{
            let fileManager = FileManager.default
            let imagePath = (self.getDirectoryPath() as NSString).appendingPathComponent("/\(userUID)/\(CharacterSelection.sharedInstance.getActiveCharacterName())/characterPortrait.png")
            if fileManager.fileExists(atPath: imagePath){
                return imagePath
            }else{
                print("No Image")
                return ""
            }
        }
        else{
            return ""
        }
    }

    func createDirectory(name: String){
        let fileManager = FileManager.default
        let paths = (getDirectoryPath() as NSString).appendingPathComponent(name)
        if !fileManager.fileExists(atPath: paths){
            try! fileManager.createDirectory(atPath: paths, withIntermediateDirectories: true, attributes: nil)
        }else{
            print("Directory is already created.")
        }
    }

Thanks in advance!


Solution

  • Well after a few hours I figured it out. Looks like the directory wasn't being created. Solved it by:

    func saveImageDocumentDirectory(image: UIImage){
            if let userUID = Auth.auth().currentUser?.uid{
                let fileURL = try! FileManager.default.url(for: .documentDirectory, in: .userDomainMask, appropriateFor: nil, create: true).appendingPathComponent("\(userUID)_\(CharacterSelection.sharedInstance.getActiveCharacterName())_characterPortrait.png")
    
                do {
                    try UIImagePNGRepresentation(image)?.write(to: fileURL, options: .atomic)
                } catch {
                    print(error)
                }
            }
        }
    
        func getImage() -> UIImage{
            if let userUID = Auth.auth().currentUser?.uid{
                let nsDocumentDirectory = FileManager.SearchPathDirectory.documentDirectory
                let nsUserDomainMask = FileManager.SearchPathDomainMask.userDomainMask
                let paths = NSSearchPathForDirectoriesInDomains(nsDocumentDirectory, nsUserDomainMask, true)
                if let dirPath = paths.first
                {
                    let imageURL = URL(fileURLWithPath: dirPath).appendingPathComponent("\(userUID)_\(CharacterSelection.sharedInstance.getActiveCharacterName())_characterPortrait.png")
                    if let image = UIImage(contentsOfFile: imageURL.path){
                        return image
                    }
                    else{
                        print("Image isn't found.")
                        return UIImage(named: "default_portrait.png")!
                    }
    
                }
                else{
                    print("Image isn't found.")
                    return UIImage(named: "default_portrait.png")!
                }
            }
            else{
                print("Image isn't found.")
                return UIImage(named: "default_portrait.png")!
            }
        }