re
/
pkger
forked from mirror/pkger
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

when you wake up feeling old

This commit is contained in:
Mark Bates 2019-09-20 10:56:26 -04:00
parent bb509c2487
commit 830ed679a8
9 changed files with 303 additions and 113 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

151
pkging/mem/file.go
View File

@ -2,7 +2,6 @@ package mem
import ( import (
"bytes" "bytes"
"encoding/json"
"fmt" "fmt"
"io" "io"
"net/http" "net/http"
@ -111,25 +110,25 @@ func (f File) String() string {
return f.Path().String() return f.Path().String()
} }
func (f File) Format(st fmt.State, verb rune) { // func (f File) Format(st fmt.State, verb rune) {
switch verb { // switch verb {
case 'v': // case 'v':
if st.Flag('+') { // if st.Flag('+') {
b, err := json.MarshalIndent(f, "", " ") // b, err := json.MarshalIndent(f, "", " ")
if err != nil { // if err != nil {
fmt.Fprint(os.Stderr, err) // fmt.Fprint(os.Stderr, err)
return // return
} // }
fmt.Fprint(st, string(b)) // fmt.Fprint(st, string(b))
return // return
} // }
fmt.Fprint(st, f.String()) // fmt.Fprint(st, f.String())
case 'q': // case 'q':
fmt.Fprintf(st, "%q", f.String()) // fmt.Fprintf(st, "%q", f.String())
default: // default:
fmt.Fprint(st, f.String()) // fmt.Fprint(st, f.String())
} // }
} // }
func (f *File) Readdir(count int) ([]os.FileInfo, error) { func (f *File) Readdir(count int) ([]os.FileInfo, error) {
var infos []os.FileInfo var infos []os.FileInfo
@ -206,60 +205,60 @@ func (f *File) Open(name string) (http.File, error) {
return di, nil return di, nil
} }
func (f File) MarshalJSON() ([]byte, error) { // func (f File) MarshalJSON() ([]byte, error) {
m := map[string]interface{}{ // m := map[string]interface{}{
"info": f.info, // "info": f.info,
"her": f.her, // "her": f.her,
"path": f.path, // "path": f.path,
"data": f.data, // "data": f.data,
"parent": f.parent, // "parent": f.parent,
} // }
return json.Marshal(m) // return json.Marshal(m)
} // }
func (f *File) UnmarshalJSON(b []byte) error { // func (f *File) UnmarshalJSON(b []byte) error {
m := map[string]json.RawMessage{} // m := map[string]json.RawMessage{}
if err := json.Unmarshal(b, &m); err != nil { // if err := json.Unmarshal(b, &m); err != nil {
return err // return err
} // }
//
info, ok := m["info"] // info, ok := m["info"]
if !ok { // if !ok {
return fmt.Errorf("missing info") // return fmt.Errorf("missing info")
} // }
//
f.info = &pkging.FileInfo{} // f.info = &pkging.FileInfo{}
if err := json.Unmarshal(info, f.info); err != nil { // if err := json.Unmarshal(info, f.info); err != nil {
return err // return err
} // }
//
her, ok := m["her"] // her, ok := m["her"]
if !ok { // if !ok {
return fmt.Errorf("missing her") // return fmt.Errorf("missing her")
} // }
if err := json.Unmarshal(her, &f.her); err != nil { // if err := json.Unmarshal(her, &f.her); err != nil {
return err // return err
} // }
//
path, ok := m["path"] // path, ok := m["path"]
if !ok { // if !ok {
return fmt.Errorf("missing path") // return fmt.Errorf("missing path")
} // }
if err := json.Unmarshal(path, &f.path); err != nil { // if err := json.Unmarshal(path, &f.path); err != nil {
return err // return err
} // }
//
parent, ok := m["parent"] // parent, ok := m["parent"]
if !ok { // if !ok {
return fmt.Errorf("missing parent") // return fmt.Errorf("missing parent")
} // }
if err := json.Unmarshal(parent, &f.parent); err != nil { // if err := json.Unmarshal(parent, &f.parent); err != nil {
return err // return err
} // }
//
if err := json.Unmarshal(m["data"], &f.data); err != nil { // if err := json.Unmarshal(m["data"], &f.data); err != nil {
return err // return err
} // }
//
return nil // return nil
} // }

52
pkging/mem/file_test.go Normal file
View File

@ -0,0 +1,52 @@
package mem
import (
"io/ioutil"
"testing"
"github.com/markbates/pkger/here"
"github.com/stretchr/testify/require"
)
func Test_File_Seek(t *testing.T) {
r := require.New(t)
info, err := here.Current()
r.NoError(err)
pkg, err := New(info)
r.NoError(err)
f, err := pkg.Create(":/wilco.band")
r.NoError(err)
data := []byte("a shot in the arm")
f.Write(data)
r.NoError(f.Close())
f, err = pkg.Open(":/wilco.band")
r.NoError(err)
b, err := ioutil.ReadAll(f)
r.NoError(err)
r.Equal(data, b)
_, err = f.Seek(0, 0)
r.NoError(err)
b, err = ioutil.ReadAll(f)
r.NoError(err)
r.Equal(data, b)
b, err = ioutil.ReadAll(f)
r.NoError(err)
r.NotEqual(data, b)
_, err = f.Seek(10, 0)
r.NoError(err)
b, err = ioutil.ReadAll(f)
r.NoError(err)
r.NotEqual(data, b)
r.Equal([]byte("the arm"), b)
}

6
pkging/mem/mem.go
View File

@ -29,6 +29,8 @@ func New(info here.Info) (*Pkger, error) {
files: &maps.Files{}, files: &maps.Files{},
current: info, current: info,
} }
f.infos.Store(info.ImportPath, info)
f.MkdirAll("/", 0755)
return f, nil return f, nil
} }
@ -105,6 +107,7 @@ func (fx *Pkger) RemoveAll(name string) error {
} }
func (fx *Pkger) Create(name string) (pkging.File, error) { func (fx *Pkger) Create(name string) (pkging.File, error) {
fx.MkdirAll("/", 0755)
pt, err := fx.Parse(name) pt, err := fx.Parse(name)
if err != nil { if err != nil {
return nil, err return nil, err
@ -115,7 +118,8 @@ func (fx *Pkger) Create(name string) (pkging.File, error) {
return nil, err return nil, err
} }
if _, err := fx.Stat(filepath.Dir(pt.Name)); err != nil { dir := filepath.Dir(pt.Name)
if _, err := fx.Stat(dir); err != nil {
return nil, err return nil, err
} }

5
pkging/mem/mem_test.go
View File

@ -15,13 +15,12 @@ func Test_Pkger(t *testing.T) {
return nil, err return nil, err
} }
wh, err := New(info) pkg, err := New(info)
if err != nil { if err != nil {
return nil, err return nil, err
} }
WithInfo(wh, info) return pkg, nil
return wh, nil
}) })
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)

9
pkging/pkgtest/http.go
View File

@ -12,24 +12,15 @@ import (
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
) )
// examples/app
// ├── Dockerfile
// ├── Makefile
// ├── go.mod
// ├── go.sum
// ├── main.go // ├── main.go
// ├── public // ├── public
// │   ├── images // │   ├── images
// │   │   ├── mark-small.png
// │   │   ├── mark.png // │   │   ├── mark.png
// │   │   ├── mark_250px.png
// │   │   └── mark_400px.png
// │   └── index.html // │   └── index.html
// └── templates // └── templates
// ├── a.txt // ├── a.txt
// └── b // └── b
// └── b.txt // └── b.txt
func (s Suite) LoadFolder(pkg pkging.Pkger) error { func (s Suite) LoadFolder(pkg pkging.Pkger) error {
files := []string{ files := []string{
"/main.go", "/main.go",

143
pkging/pkgtest/suite.go
View File

@ -67,7 +67,7 @@ func (s Suite) Test(t *testing.T) {
// } // }
// } // }
// if ns, ok := s.Pkger.(WithRootable); ok { // if ns, ok := s.Pkger.(WithRootable); ok {
// dir, err := ioutil.TempDir("", "") // dir, err := ioutil.TempDir("")
// if err != nil { // if err != nil {
// return s, err // return s, err
// } // }
@ -467,15 +467,134 @@ func (s Suite) Test_Stat_File(t *testing.T) {
} }
} }
// func (s Suite) Test_Walk(t *testing.T) { func (s Suite) Test_Walk(t *testing.T) {
// panic("not implemented") r := require.New(t)
// }
// pkg, err := s.Make()
// func (s Suite) Test_Remove(t *testing.T) { r.NoError(err)
// panic("not implemented") r.NoError(s.LoadFolder(pkg))
// }
// cur, err := pkg.Current()
// func (s Suite) Test_RemoveAll(t *testing.T) { r.NoError(err)
// panic("not implemented")
// } ip := cur.ImportPath
table := []struct {
in string
}{
{in: ip},
{in: "/"},
{in: ":/"},
{in: ip + ":/"},
}
for _, tt := range table {
s.Run(t, tt.in, func(st *testing.T) {
r := require.New(st)
var act []string
err := pkg.Walk(tt.in, func(path string, info os.FileInfo, err error) error {
if err != nil {
return err
}
act = append(act, path)
return nil
})
r.NoError(err)
exp := []string{
"github.com/markbates/pkger:/",
"github.com/markbates/pkger:/main.go",
"github.com/markbates/pkger:/public",
"github.com/markbates/pkger:/public/images",
"github.com/markbates/pkger:/public/images/mark.png",
"github.com/markbates/pkger:/public/index.html",
"github.com/markbates/pkger:/templates",
"github.com/markbates/pkger:/templates/a.txt",
"github.com/markbates/pkger:/templates/b",
"github.com/markbates/pkger:/templates/b/b.txt",
}
r.Equal(exp, act)
})
}
}
func (s Suite) Test_Remove(t *testing.T) {
r := require.New(t)
pkg, err := s.Make()
r.NoError(err)
cur, err := pkg.Current()
r.NoError(err)
ip := cur.ImportPath
table := []struct {
in string
}{
{in: "/public/images/mark.png"},
{in: ":/public/images/mark.png"},
{in: ip + ":/public/images/mark.png"},
}
for _, tt := range table {
s.Run(t, tt.in, func(st *testing.T) {
r := require.New(st)
pkg, err := s.Make()
r.NoError(err)
r.NoError(s.LoadFolder(pkg))
_, err = pkg.Stat(tt.in)
r.NoError(err)
r.NoError(pkg.Remove(tt.in))
_, err = pkg.Stat(tt.in)
r.Error(err)
r.Error(pkg.Remove("unknown"))
})
}
}
func (s Suite) Test_RemoveAll(t *testing.T) {
r := require.New(t)
pkg, err := s.Make()
r.NoError(err)
cur, err := pkg.Current()
r.NoError(err)
ip := cur.ImportPath
table := []struct {
in string
}{
{in: "/public"},
{in: ":/public"},
{in: ip + ":/public"},
}
for _, tt := range table {
s.Run(t, tt.in, func(st *testing.T) {
r := require.New(st)
pkg, err := s.Make()
r.NoError(err)
r.NoError(s.LoadFolder(pkg))
_, err = pkg.Stat(tt.in)
r.NoError(err)
r.NoError(pkg.RemoveAll(tt.in))
_, err = pkg.Stat(tt.in)
r.Error(err)
})
}
}

15
pkging/stdos/file.go
View File

@ -20,11 +20,11 @@ type File struct {
} }
func NewFile(fx pkging.Pkger, osf *os.File) (*File, error) { func NewFile(fx pkging.Pkger, osf *os.File) (*File, error) {
pt, err := fx.Parse(osf.Name()) name := osf.Name()
pt, err := fx.Parse(name)
if err != nil { if err != nil {
return nil, err return nil, err
} }
info, err := osf.Stat() info, err := osf.Stat()
if err != nil { if err != nil {
return nil, err return nil, err
@ -97,15 +97,10 @@ func (f *File) Stat() (os.FileInfo, error) {
return f.info, nil return f.info, nil
} }
abs, err := f.Abs() nf, err := NewFile(f.pkging, f.File)
if err != nil { if err != nil {
return nil, err return nil, err
} }
(*f) = *nf
info, err := os.Stat(abs) return f.info, nil
if err != nil {
return nil, err
}
f.info = pkging.NewFileInfo(info)
return info, nil
} }

29
pkging/stdos/file_test.go Normal file
View File

@ -0,0 +1,29 @@
package stdos
import (
"testing"
"github.com/stretchr/testify/require"
)
func Test_File_Stat_No_Info(t *testing.T) {
r := require.New(t)
pkg, err := New()
r.NoError(err)
f, err := pkg.Open(":/pkging/stdos/file_test.go")
r.NoError(err)
defer f.Close()
sf, ok := f.(*File)
r.True(ok)
oi := sf.info
sf.info = nil
info, err := sf.Stat()
r.NoError(err)
r.Equal(oi.Name(), info.Name())
// r.Equal("", f.Name())
}

6
pkging/stdos/stdos.go
View File

@ -43,13 +43,15 @@ func New() (*Pkger, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
return &Pkger{ p := &Pkger{
infos: &maps.Infos{}, infos: &maps.Infos{},
paths: &maps.Paths{ paths: &maps.Paths{
Current: info, Current: info,
}, },
current: info, current: info,
}, nil }
p.infos.Store(info.ImportPath, info)
return p, nil
} }
func (fx *Pkger) Create(name string) (pkging.File, error) { func (fx *Pkger) Create(name string) (pkging.File, error) {