18 "github.com/chzyer/readline"
19 "github.com/docker/pinata/v1/fs/p9p/new"
20 "golang.org/x/net/context"
26 flag.StringVar(&addr, "addr", ":5640", "addr of 9p service")
31 log.Println(http.ListenAndServe("localhost:6060", nil))
34 ctx := context.Background()
39 if strings.HasPrefix(addr, "unix:") {
44 log.Println("dialing", addr)
45 conn, err := net.Dial(proto, addr)
50 csession, err := p9pnew.NewSession(ctx, conn)
55 commander := &fsCommander{
56 ctx: context.Background(),
63 completer := readline.NewPrefixCompleter(
64 readline.PcItem("ls"),
65 // readline.PcItem("find"),
66 readline.PcItem("stat"),
67 readline.PcItem("cat"),
68 readline.PcItem("cd"),
69 readline.PcItem("pwd"),
72 rl, err := readline.NewEx(&readline.Config{
73 HistoryFile: ".history",
74 AutoComplete: completer,
79 commander.readline = rl
81 msize, version := commander.session.Version()
85 log.Println("9p version", version, msize)
89 if _, err := commander.session.Attach(commander.ctx, commander.nextfid, p9pnew.NOFID, "anyone", "/"); err != nil {
92 commander.rootfid = commander.nextfid
95 // clone the pwd fid so we can clunk it
96 if _, err := commander.session.Walk(commander.ctx, commander.rootfid, commander.nextfid); err != nil {
99 commander.pwdfid = commander.nextfid
103 commander.readline.SetPrompt(fmt.Sprintf("%s 🐳 > ", commander.pwd))
105 line, err := rl.Readline()
107 log.Fatalln("error: ", err)
114 args := strings.Fields(line)
117 var cmd func(ctx context.Context, args ...string) error
121 cmd = commander.cmdls
123 cmd = commander.cmdcd
125 cmd = commander.cmdpwd
127 cmd = commander.cmdcat
129 cmd = commander.cmdstat
131 cmd = func(ctx context.Context, args ...string) error {
132 return fmt.Errorf("command not implemented")
136 ctx, _ = context.WithTimeout(commander.ctx, 5*time.Second)
137 if err := cmd(ctx, args[1:]...); err != nil {
138 log.Printf("👹 %s: %v", name, err)
143 type fsCommander struct {
145 session p9pnew.Session
152 readline *readline.Instance
157 func (c *fsCommander) cmdls(ctx context.Context, args ...string) error {
158 ps := []string{c.pwd}
163 wr := tabwriter.NewWriter(c.stdout, 0, 8, 8, ' ', 0)
165 for _, p := range ps {
166 // create a header if have more than one path.
168 fmt.Fprintln(wr, p+":")
172 p = path.Join(c.pwd, p)
175 targetfid := c.nextfid
177 components := strings.Split(strings.Trim(p, "/"), "/")
178 if _, err := c.session.Walk(ctx, c.rootfid, targetfid, components...); err != nil {
181 defer c.session.Clunk(ctx, targetfid)
183 _, iounit, err := c.session.Open(ctx, targetfid, p9pnew.OREAD)
189 msize, _ := c.session.Version()
190 iounit = uint32(msize - 24) // size of message max minus fcall io header (Rread)
193 p := make([]byte, iounit)
195 n, err := c.session.Read(ctx, targetfid, p, 0)
200 rd := bytes.NewReader(p[:n])
201 codec := p9pnew.NewCodec() // TODO(stevvooe): Need way to resolve codec based on session.
204 if err := p9pnew.DecodeDir(codec, rd, &d); err != nil {
212 fmt.Fprintf(wr, "%v\t%v\t%v\t%s\n", os.FileMode(d.Mode), d.Length, d.ModTime, d.Name)
220 // all output is dumped only after success.
224 func (c *fsCommander) cmdcd(ctx context.Context, args ...string) error {
232 return fmt.Errorf("cd: invalid args: %v", args)
236 p = path.Join(c.pwd, p)
239 targetfid := c.nextfid
241 components := strings.Split(strings.TrimSpace(strings.Trim(p, "/")), "/")
242 if _, err := c.session.Walk(c.ctx, c.rootfid, targetfid, components...); err != nil {
245 defer c.session.Clunk(c.ctx, c.pwdfid)
247 log.Println("cd", p, targetfid)
254 func (c *fsCommander) cmdstat(ctx context.Context, args ...string) error {
255 ps := []string{c.pwd}
260 wr := tabwriter.NewWriter(c.stdout, 0, 8, 8, ' ', 0)
262 for _, p := range ps {
263 targetfid := c.nextfid
265 components := strings.Split(strings.Trim(p, "/"), "/")
266 if _, err := c.session.Walk(ctx, c.rootfid, targetfid, components...); err != nil {
269 defer c.session.Clunk(ctx, targetfid)
271 d, err := c.session.Stat(ctx, targetfid)
276 fmt.Fprintf(wr, "%v\t%v\t%v\t%s\n", os.FileMode(d.Mode), d.Length, d.ModTime, d.Name)
282 func (c *fsCommander) cmdpwd(ctx context.Context, args ...string) error {
284 return fmt.Errorf("pwd takes no arguments")
291 func (c *fsCommander) cmdcat(ctx context.Context, args ...string) error {
299 return fmt.Errorf("cd: invalid args: %v", args)
303 p = path.Join(c.pwd, p)
306 targetfid := c.nextfid
308 components := strings.Split(strings.TrimSpace(strings.Trim(p, "/")), "/")
309 if _, err := c.session.Walk(ctx, c.rootfid, targetfid, components...); err != nil {
312 defer c.session.Clunk(ctx, c.pwdfid)
314 _, iounit, err := c.session.Open(ctx, targetfid, p9pnew.OREAD)
320 msize, _ := c.session.Version()
321 iounit = uint32(msize - 24) // size of message max minus fcall io header (Rread)
324 b := make([]byte, iounit)
326 n, err := c.session.Read(ctx, targetfid, b, 0)
331 if _, err := os.Stdout.Write(b[:n]); err != nil {
335 os.Stdout.Write([]byte("\n"))