file.go 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. package filesystem
  2. import (
  3. "io"
  4. "os"
  5. "path/filepath"
  6. "github.com/v2fly/v2ray-core/v5/common/buf"
  7. "github.com/v2fly/v2ray-core/v5/common/platform"
  8. "github.com/v2fly/v2ray-core/v5/common/platform/filesystem/fsifce"
  9. )
  10. var NewFileSeeker fsifce.FileSeekerFunc = func(path string) (io.ReadSeekCloser, error) {
  11. return os.Open(path)
  12. }
  13. var NewFileReader fsifce.FileReaderFunc = func(path string) (io.ReadCloser, error) {
  14. return os.Open(path)
  15. }
  16. var NewFileWriter fsifce.FileWriterFunc = func(path string) (io.WriteCloser, error) {
  17. basePath := filepath.Dir(path)
  18. if err := os.MkdirAll(basePath, 0o700); err != nil {
  19. return nil, err
  20. }
  21. return os.Create(path)
  22. }
  23. var NewFileRemover fsifce.FileRemoveFunc = os.Remove
  24. var NewFileReadDir fsifce.FileReadDirFunc = os.ReadDir
  25. func ReadFile(path string) ([]byte, error) {
  26. reader, err := NewFileReader(path)
  27. if err != nil {
  28. return nil, err
  29. }
  30. defer reader.Close()
  31. return buf.ReadAllToBytes(reader)
  32. }
  33. func WriteFile(path string, payload []byte) error {
  34. writer, err := NewFileWriter(path)
  35. if err != nil {
  36. return err
  37. }
  38. defer writer.Close()
  39. return buf.WriteAllBytes(writer, payload)
  40. }
  41. func ReadAsset(file string) ([]byte, error) {
  42. return ReadFile(platform.GetAssetLocation(file))
  43. }
  44. func CopyFile(dst string, src string, perm os.FileMode) error {
  45. bytes, err := ReadFile(src)
  46. if err != nil {
  47. return err
  48. }
  49. f, err := os.OpenFile(dst, os.O_CREATE|os.O_WRONLY, perm)
  50. if err != nil {
  51. return err
  52. }
  53. defer f.Close()
  54. _, err = f.Write(bytes)
  55. return err
  56. }