X-Git-Url: https://uglyman.kremlin.cc/gitweb/gitweb.cgi?a=blobdiff_plain;f=httpd.go;h=29b6ca9d36e37f4d66cdda3ee3b6a62ba6987011;hb=80d578694684c8a8fd3cebb2a77dbda114a22e91;hp=856d7dfe26909b38ee55c929c67bb4176e44b491;hpb=53d46aa3f7b18ebe0bdaf0061bc0218aa45afa4b;p=k_transpose.git diff --git a/httpd.go b/httpd.go index 856d7df..29b6ca9 100644 --- a/httpd.go +++ b/httpd.go @@ -1,14 +1,366 @@ package main -import "fmt" -import "net/http" +import( + "fmt" + "os" + + "io" + "io/ioutil" + "bufio" + + "strings" + "strconv" + + "net/http" + "net/url" + + "github.com/wsxiaoys/terminal/color" +) + +/* holds our palette, or 16 ANSI colors (8 normal colors + bright complements) + and two foreground/background colors. colors are 3 byte arrays (RGB) */ +type ktPalette struct { + + black [3]byte + bblack [3]byte + + red [3]byte + bred [3]byte + + green [3]byte + bgreen [3]byte + + yellow [3]byte + byellow [3]byte + + blue [3]byte + bblue [3]byte + + purple [3]byte + bpurple [3]byte + + cyan [3]byte + bcyan [3]byte + + white [3]byte + bwhite [3]byte + + fg [3]byte + bg [3]byte +} + +/* the default "control" ANSI color set (boring) */ +var ansiColors = ktPalette { + + black : [3]byte {0,0,0}, + bblack : [3]byte {128,128,128}, + + red : [3]byte {128,0,0}, + bred : [3]byte {255,0,0}, + + green : [3]byte {0,128,0}, + bgreen : [3]byte {0,255,0}, + + yellow : [3]byte {128,128,0}, + byellow : [3]byte {255,255,0}, + + blue : [3]byte {0,0,128}, + bblue : [3]byte {0,0,255}, + + purple : [3]byte {128,0,128}, + bpurple : [3]byte {255,0,255}, + + cyan : [3]byte {0,128,128}, + bcyan : [3]byte {0,255,255}, + + white : [3]byte {128,128,128}, + bwhite : [3]byte {255,255,255}, + + fg : [3]byte {0,0,0}, + bg : [3]byte {255,255,255}, +} + +/* lets get our money's worth out of this utf-8 crap */ +var checkM = "✓" +var xM = "✗" + +/* parses a colorfile, returns palette struct. given a nil file pointer, + returns standard ANSI color set (our "control") */ +func parseColors(colorfile *os.File) (pal ktPalette, err error) { + + if colorfile == nil { return ansiColors, nil } + + ret := ktPalette {} + var e errConsumer + + scanner := bufio.NewScanner(colorfile) + for scanner.Scan() { + + if strings.Contains(scanner.Text(), "color") && + !strings.Contains(scanner.Text(), "!") { + + /* i am so sorry */ + cur := strings.Replace(scanner.Text(), " ", "", -1) + split := strings.Split(cur, ":") + hexColor := strings.Replace(split[1], "#", "", -1) + coNumStr := strings.Replace(split[0], "URxvt*color", "", -1) + + r, err := strconv.ParseUint(hexColor[0:2], 16, 8); e.Consume(err) + g, err := strconv.ParseUint(hexColor[2:4], 16, 8); e.Consume(err) + b, err := strconv.ParseUint(hexColor[4:], 16, 8); e.Consume(err) + + colorNo, err := strconv.Atoi(coNumStr); e.Consume(err) + + if e.err != nil { + return ktPalette{}, e.err + } + + switch(colorNo) { + case 0: ret.black[0] = byte(r); ret.black[1] = byte(g); ret.black[2] = byte(b) + case 1: ret.red[0] = byte(r); ret.red[1] = byte(g); ret.red[2] = byte(b) + case 2: ret.green[0] = byte(r); ret.green[1] = byte(g); ret.green[2] = byte(b) + case 3: ret.yellow[0] = byte(r); ret.yellow[1] = byte(g); ret.yellow[2] = byte(b) + case 4: ret.blue[0] = byte(r); ret.blue[1] = byte(g); ret.blue[2] = byte(b) + case 5: ret.purple[0] = byte(r); ret.purple[1] = byte(g); ret.purple[2] = byte(b) + case 6: ret.cyan[0] = byte(r); ret.cyan[1] = byte(g); ret.cyan[2] = byte(b) + case 7: ret.white[0] = byte(r); ret.white[1] = byte(g); ret.white[2] = byte(b) + case 8: ret.bblack[0] = byte(r); ret.bblack[1] = byte(g); ret.bblack[2] = byte(b) + case 9: ret.bred[0] = byte(r); ret.bred[1] = byte(g); ret.bred[2] = byte(b) + case 10: ret.bgreen[0] = byte(r); ret.bgreen[1] = byte(g); ret.bgreen[2] = byte(b) + case 11: ret.byellow[0] = byte(r); ret.byellow[1] = byte(g); ret.byellow[2] = byte(b) + case 12: ret.bblue[0] = byte(r); ret.bblue[1] = byte(g); ret.bblue[2] = byte(b) + case 13: ret.bpurple[0] = byte(r); ret.bpurple[1] = byte(g); ret.bpurple[2] = byte(b) + case 14: ret.bcyan[0] = byte(r); ret.bcyan[1] = byte(g); ret.bcyan[2] = byte(b) + case 15: ret.bwhite[0] = byte(r); ret.bwhite[1] = byte(g); ret.bwhite[2] = byte(b) + } + + } else if strings.Contains(scanner.Text(), "background") { + + hex := strings.Split(scanner.Text(), "#") + het := hex[1] + + r, err := strconv.ParseUint(het[0:2], 16, 8); e.Consume(err) + g, err := strconv.ParseUint(het[2:4], 16, 8); e.Consume(err) + b, err := strconv.ParseUint(het[4:], 16, 8); e.Consume(err) + + if e.err != nil { + return ktPalette{}, e.err + } + + ret.bg[0] = byte(r) + ret.bg[1] = byte(g) + ret.bg[2] = byte(b) + + } else if strings.Contains(scanner.Text(), "foreground") { + + hex := strings.Split(scanner.Text(), "#") + het := hex[1] + + r, err := strconv.ParseUint(het[0:2], 16, 8); e.Consume(err) + g, err := strconv.ParseUint(het[2:4], 16, 8); e.Consume(err) + b, err := strconv.ParseUint(het[4:], 16, 8); e.Consume(err) + + if e.err != nil { + return ktPalette{}, e.err + } + + ret.fg[0] = byte(r) + ret.fg[1] = byte(g) + ret.fg[2] = byte(b) + } + } + + return ret, e.err +} + +var httpdStatus bool + +func ktInit(dirPrepend string, port int, colorfilePath string) error { + + httpdStatus = false + + color.Print("@yparsing colorfile :: @{|}") + file, err := os.Open(colorfilePath) + if err != nil { + color.Printf("@r[%s]@{|} - bad colorfile path\n", xM) + return fmt.Errorf("%s\n", "bad colorfile path") + } + + pal, err := parseColors(file) + + if err != nil { + color.Printf("@r[%s]@{|} - malformed colorfile [%s]\n", xM, err) + return fmt.Errorf("%s\n", "malformed colorfile") + } + + color.Printf("@g[%s]@{|}\n", checkM) + + if pal == ansiColors {} + + color.Printf("@yverifying & preprocessing colorsets@{|} :: @y[SKIP]\n") + color.Printf("@ygenerating transpositional colorspace@{|} :: @y[SKIP]\n") + + color.Printf("@ystarting httpd on port @b%d@{|} :: ", port) + http.HandleFunc("/kt/", transposePage) + + var portString string + fmt.Sprintf(portString, ":%d", port) + err = http.ListenAndServe(portString, nil) + if err != nil { + + color.Printf("@r[%s]@{-} - run me as root!\n", xM) + return fmt.Errorf("%s\n", "failed to start httpd") + } + + color.Printf("@g[%s]@{-}\n", checkM) + + return nil +} + +func transposePage(writer http.ResponseWriter, req *http.Request) { + + if !httpdStatus { + + httpdStatus = true + color.Printf("@g[%s]@{|}\n", checkM) + } + + if req.URL.Path == "/kt/" { + writer.Write([]byte("wtf")) + return + } + + fqdn := req.URL.Path[4:] + targetURL := fmt.Sprintf("http://%s", fqdn) + + resp, err := http.Get(targetURL) + + if err != nil || resp.StatusCode != 200 { + + io.WriteString(writer, "failed to get that page! -kt\n") + io.WriteString(writer, targetURL + "\n") + + io.WriteString(writer, resp.Status) + return + } + + conType := resp.Header.Get("Content-Type") + + switch conType[0:strings.Index(conType, ";")] { + + case "text/html": + writer.Write(transposeHTML(bufio.NewScanner(resp.Body), fqdn)) + + case "text/css": + writer.Write(transposeCSS(bufio.NewScanner(resp.Body), fqdn)) + + default: + page, _ := ioutil.ReadAll(resp.Body) + writer.Write(page) + } + + resp.Body.Close() +} + +/* swap href="" & src="" */ +func transposeHTML(scan *bufio.Scanner, fqdn string) []byte { + + var ret []byte + var i int + + scan.Split(bufio.ScanWords) + for scan.Scan() { + + i++ + cur := scan.Text() + + //fmt.Printf("%s\n", cur) + + if len(cur) < 7 { + + + } else if(cur[0:6] == "href=\\") { + + urlStr := cur[7:strings.Index(cur[7:], "\\") + 7] + + u, err := url.Parse(urlStr) + if err != nil { + fmt.Printf("malformed URL: %s\n", urlStr) + } + + if u.Host == "" { + + u.Host = fmt.Sprintf("localhost/kt/%s", fqdn) +// cur = append(cur[0:6], + } + + fmt.Printf("[F] URL: %s // PATH: %s\n", u.Host, u.Path) + if u == u {} + + + } else if(cur[0:5] == "href=") { + + urlStr := cur[6:strings.Index(cur[6:], "\"") + 6] + + u, err := url.Parse(urlStr) + if err != nil { + fmt.Printf("malformed URL: %s\n", urlStr) + } + + if u == u {} + fmt.Printf("URL: %s // PATH: %s\n", u.Host, u.Path) + + } else if(cur[0:5] == "src=\"") { + + //fmt.Printf("%s\n", cur) + urlStr := cur[5:strings.Index(cur[5:], "\"") + 5] + + u, err := url.Parse(urlStr) + if err != nil { + fmt.Printf("malformed URL: %s\n", urlStr) + } + + if u.Host == "" { + + u.Host = fmt.Sprintf("localhost/kt/%s", fqdn) +// cur = append(cur[0:6], + + fmt.Printf("[S] URL: %s // PATH: %s\n", u.Host, u.Path) + if u == u {} + } + + } + + ret = append(ret, byte(' ')) + ret = append(ret, cur...) + } + + fmt.Printf("%d\n", i) + + return ret +} + +func transposeCSS(scan *bufio.Scanner, fqdn string) []byte { + + var ret []byte + + return ret +} func main() { - /* make sure to close() anything you need to (you need to) */ - fmt.Println("welp"); - resp, err := http.Get("http://kremlin.cc") + err := ktInit("kolors", 999, "/home/kremlin/.Xresources") if err != nil {} - fmt.Println(resp) } + +type errConsumer struct { + err error +} + +func (e *errConsumer) Consume(err error) { + if e.err == nil && err != nil { + e.err = err + } +} +