com.github.kondaurovdev.snippets.FileHelper.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of snippets_2.12 Show documentation

snippets

The newest version!

package com.github.kondaurovdev.snippets

import java.io._

import com.github.kondaurovdev.snippets.iface.{FileHelperIface, TryHelperIface}

import scala.io.Source

trait iFileHelperImpl extends FileHelperIface {

  def tryHelper: TryHelperIface

  def listFiles(dir: String): List[String] = {

    new File(dir).listFiles().filter(f => f.isFile ).map { f =>
      f.getName
    }.toList

  }

  def getFileNameWithoutExt(fileName: String): String = {

    val dot = fileName.lastIndexOf('.')

    if (dot != -1) {
      fileName.substring(0, dot)
    } else {
      fileName
    }

  }

  def fileExists(path: String): Boolean = {

    tryHelper.tryBlock({
      new File(path).exists()
    }, "").fold(
      _ => false,
      res => res
    )

  }

  def readFile(file: String, enc: String = "UTF-8"): Either[String, String] = {

    tryHelper.tryBlock({
      if (file.startsWith("http")) {
        Source.fromURL(file, enc).mkString
      } else {
        Source.fromFile(file, enc).mkString
      }
    }, s"Can't get content of file: $file")

  }

  def createDir(path: String): Either[String, String] = {
    tryHelper.tryBlock({
      new File(path).mkdirs()
      "Dir has been created"
    }, s"Can't create dir: $path")
  }

}

class FileHelper(
                val tryHelper: TryHelperIface
                ) extends iFileHelperImpl