file.go 809 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. package filesystem
  2. import (
  3. "io"
  4. "os"
  5. "v2ray.com/core/common/buf"
  6. "v2ray.com/core/common/platform"
  7. )
  8. type FileReaderFunc func(path string) (io.ReadCloser, error)
  9. var NewFileReader FileReaderFunc = func(path string) (io.ReadCloser, error) {
  10. return os.Open(path)
  11. }
  12. func ReadFile(path string) ([]byte, error) {
  13. reader, err := NewFileReader(path)
  14. if err != nil {
  15. return nil, err
  16. }
  17. defer reader.Close()
  18. return buf.ReadAllToBytes(reader)
  19. }
  20. func ReadAsset(file string) ([]byte, error) {
  21. return ReadFile(platform.GetAssetLocation(file))
  22. }
  23. func CopyFile(dst string, src string) error {
  24. bytes, err := ReadFile(src)
  25. if err != nil {
  26. return err
  27. }
  28. f, err := os.OpenFile(dst, os.O_CREATE|os.O_WRONLY, 0644)
  29. if err != nil {
  30. return err
  31. }
  32. defer f.Close()
  33. _, err = f.Write(bytes)
  34. return err
  35. }