refact

Author: shynome

.gitignore

@@ -0,0 +1,23 @@
+# If you prefer the allow list template instead of the deny list, see community template:
+# https://github.com/github/gitignore/blob/main/community/Golang/Go.AllowList.gitignore
+#
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/
+
+# Go workspace file
+go.work
+
+/lcode

LICENSE

@@ -0,0 +1,24 @@
+# 简介
+
+起因是某位朋友抱怨 vim 太难用了, 想在服务器上使用本地的 vscode 进行编辑,
+但 vscode remote ssh 太吃内存了, 简单编辑用 webdav 文件协议编辑也许可行?
+
+也许会加上命令行支持?
+
+# 进度
+
+- [x] webdav
+- [ ] edit with local vscode
+
+# 设计概述
+
+```sh
+# open vscode for opening lcode plugin, lcode plugin will start server
+code
+ssh -R 4349:127.0.0.1:4349 root@your_host
+# defualt connect server addr is 127.0.0.1:4349
+lcode -c 127.0.0.1:4349
+# log vscode link, you can click the link to open vscode
+# if vscode lcode plugin is active will auto open
+-> vscode://lcode-plugin/uuid-key
+```

README.md

@@ -0,0 +1,8 @@
+module github.com/shynome/lcode
+
+go 1.18
+
+require (
+	github.com/shynome/httprelay-go v0.0.2
+	golang.org/x/net v0.0.0-20210614182718-04defd469f4e
+)

go.mod

@@ -0,0 +1,4 @@
+github.com/shynome/httprelay-go v0.0.2 h1:Z+Zvh7w9OIeBluZTg7Ef52EImmu2XbkhdS7tvnxQ2Dk=
+github.com/shynome/httprelay-go v0.0.2/go.mod h1:CjDFSTBc/hqGTOfWZ6RmRKbIBzuD/qigX6AMYXZiLr0=
+golang.org/x/net v0.0.0-20210614182718-04defd469f4e h1:XpT3nA5TvE525Ne3hInMh6+GETgn27Zfm9dxsThnX2Q=
+golang.org/x/net v0.0.0-20210614182718-04defd469f4e/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=

go.sum

@@ -0,0 +1,41 @@
+package main
+
+import (
+	"log"
+	"os"
+	"path/filepath"
+
+	"github.com/shynome/httprelay-go"
+	"github.com/shynome/lcode/table"
+)
+
+var LCODE_CONNECT string = "http://127.0.0.1:4349"
+
+var db *table.Table
+
+func init() {
+	c := os.Getenv("LCODE_CONNECT")
+	if c != "" {
+		LCODE_CONNECT = c
+	}
+
+	lcodeTmpdir := filepath.Join(os.TempDir(), "lcode-nuts")
+	db = table.New(lcodeTmpdir)
+	err := db.Open()
+	if err != nil {
+		panic(err)
+	}
+}
+
+func initProxy(proxy *httprelay.Proxy) {
+	id, err := makeUniqueID()
+	if err != nil {
+		log.Fatal(err)
+	}
+	proxy.Auth.ID = id
+	proxy.Auth.Secret = id
+	if proxy.Parallel < 4 {
+		proxy.Parallel = 4
+	}
+	proxy.Println = func(i ...interface{}) {}
+}

init.go

@@ -0,0 +1,39 @@
+package main
+
+import (
+	"fmt"
+	"net/http"
+	"os"
+	"os/signal"
+	"path/filepath"
+
+	"github.com/shynome/httprelay-go"
+)
+
+func main() {
+	defer db.Dispose()
+
+	proxy := httprelay.NewProxy(LCODE_CONNECT)
+	initProxy(proxy)
+	initWebdav(http.DefaultServeMux)
+
+	pwd, _ := os.Getwd()
+	var codedir = "."
+	if len(os.Args) >= 2 {
+		codedir = os.Args[1]
+	}
+	codedir = filepath.Join(pwd, codedir)
+
+	db.Allow(codedir)
+	defer db.Deny(codedir)
+
+	vscodeLink := genVscodeLink(proxy.Auth.ID, codedir)
+
+	fmt.Println(vscodeLink)
+	// go reqOpen(vscodeLink)
+	go proxy.Serve(nil)
+
+	c := make(chan os.Signal, 1)
+	signal.Notify(c, os.Interrupt)
+	<-c
+}

lcode.go

@@ -0,0 +1,7 @@
+package main
+
+import "testing"
+
+func TestRunMain(t *testing.T) {
+	main()
+}

lcode_test.go

@@ -0,0 +1,31 @@
+package table_test
+
+import (
+	"os"
+	"testing"
+	"time"
+
+	"github.com/shynome/lcode/table"
+)
+
+func TestTable(t *testing.T) {
+	pwd, _ := os.Getwd()
+	table := table.New(pwd)
+	table.EmptyRulesFile()
+	err := table.Open()
+	if err != nil {
+		t.Error(err)
+		return
+	}
+	table.Allow("/a/b")
+	time.Sleep(time.Second)
+	if table.IsDeny("/a/b") {
+		t.Error("should be allowed")
+	}
+	table.Deny("/a/b")
+	time.Sleep(time.Second)
+	if !table.IsDeny("/a/b") {
+		t.Error("should be deny")
+	}
+	return
+}

table/lib_test.go

@@ -0,0 +1,170 @@
+package table
+
+import (
+	"bufio"
+	"io"
+	"io/fs"
+	"os"
+	"path/filepath"
+	"strings"
+	"sync"
+	"time"
+)
+
+type AllowedDirs map[string]bool
+
+type Table struct {
+	lock        *sync.RWMutex
+	allowedDirs AllowedDirs
+	rulesFile   string
+	stopWatch   chan bool
+}
+
+func New(dir string) *Table {
+	table := &Table{
+		lock:        &sync.RWMutex{},
+		allowedDirs: AllowedDirs{},
+		stopWatch:   make(chan bool),
+		rulesFile:   filepath.Join(dir, "/allowed-dirs.table"),
+	}
+	return table
+}
+
+func (t *Table) Open() (err error) {
+	err = t.LoadTable()
+	go t.Watch()
+	return
+}
+
+func (t *Table) Allow(dir string) {
+	t.writeToFile("+", dir)
+
+	t.lock.Lock()
+	defer t.lock.Unlock()
+	t.allowedDirs[dir] = true
+}
+
+func (t *Table) Deny(dir string) {
+	t.writeToFile("-", dir)
+
+	t.lock.Lock()
+	defer t.lock.Unlock()
+	delete(t.allowedDirs, dir)
+}
+
+func (t *Table) writeToFile(flag, dir string) {
+	f, err := os.OpenFile(t.rulesFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
+	if err != nil {
+		return
+	}
+	defer f.Close()
+	rule := flag + dir + "\n"
+	if _, err := f.WriteString(rule); err != nil {
+		panic(err)
+	}
+}
+func (t *Table) EmptyRulesFile() error {
+	return os.Remove(t.rulesFile)
+}
+
+func (t *Table) IsDeny(dir string) (deny bool) {
+	t.lock.RLock()
+	defer t.lock.RUnlock()
+
+	deny = true
+	for d := range t.allowedDirs {
+		if strings.HasPrefix(dir, d) {
+			return false
+		}
+	}
+	return true
+}
+
+func (t *Table) GetAllowedDirs() map[string]bool {
+	t.lock.RLock()
+	defer t.lock.RUnlock()
+
+	return t.allowedDirs
+}
+
+func (t *Table) Dispose() (err error) {
+	t.stopWatch <- true
+
+	t.LoadTable()
+
+	var count int = 0
+	for range t.allowedDirs {
+		count++
+	}
+	if count == 0 {
+		if err = t.EmptyRulesFile(); err != nil {
+			return
+		}
+	}
+	return
+}
+
+func (t *Table) Watch() {
+	ticker := time.NewTicker(time.Millisecond * 500)
+	lastMtime := t.getRulesFileStat().ModTime()
+	for {
+		select {
+		case <-t.stopWatch:
+			ticker.Stop()
+			return
+		case <-ticker.C:
+			stat := t.getRulesFileStat()
+			mtime := stat.ModTime()
+			if lastMtime.Equal(mtime) {
+				continue
+			}
+			lastMtime = mtime
+			if err := t.LoadTable(); err != nil {
+				panic(err)
+			}
+		}
+	}
+}
+func (t *Table) getRulesFileStat() fs.FileInfo {
+	stat, err := os.Stat(t.rulesFile)
+	if err != nil {
+		panic(err)
+	}
+	return stat
+}
+
+func (t *Table) LoadTable() (err error) {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+
+	f, err := os.OpenFile(t.rulesFile, os.O_CREATE|os.O_RDONLY, 0644)
+	if err != nil {
+		return
+	}
+	defer f.Close()
+	m := LoadTable(f)
+
+	t.allowedDirs = m
+	return
+}
+
+func LoadTable(r io.Reader) AllowedDirs {
+	m := AllowedDirs{}
+
+	scanner := bufio.NewScanner(r)
+	for scanner.Scan() {
+		line := scanner.Text()
+		if strings.HasPrefix(line, "+") {
+			dir := strings.TrimPrefix(line, "+")
+			m[dir] = true
+		}
+		if strings.HasPrefix(line, "-") {
+			dir := strings.TrimPrefix(line, "-")
+			_, ok := m[dir]
+			if ok {
+				delete(m, dir)
+			}
+		}
+	}
+	return m
+}