X-Git-Url: https://uglyman.kremlin.cc/gitweb/gitweb.cgi?p=k_transpose.git;a=blobdiff_plain;f=httpd.go;h=74c4acac0b955724ab68eb0aa7ce3e2ecc6ad563;hp=d50afdcd50f23770a924dc834dba48b70ca94a18;hb=HEAD;hpb=08ff4e7b600e41ea59750f3d002677c9b459b3e0 diff --git a/httpd.go b/httpd.go index d50afdc..74c4aca 100644 --- a/httpd.go +++ b/httpd.go @@ -1,217 +1,427 @@ package main -import( - "fmt" - "os" +import ( + "fmt" + "os" - "strings" - "strconv" - "bufio" + "bufio" + "io" + "io/ioutil" - "net/http" + "strconv" + "strings" - "github.com/wsxiaoys/terminal/color" + "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 - black [3]byte - bblack [3]byte - - red [3]byte - bred [3]byte + red [3]byte + bred [3]byte - green [3]byte - bgreen [3]byte + green [3]byte + bgreen [3]byte - yellow [3]byte - byellow [3]byte + yellow [3]byte + byellow [3]byte - blue [3]byte - bblue [3]byte + blue [3]byte + bblue [3]byte - purple [3]byte - bpurple [3]byte + purple [3]byte + bpurple [3]byte - cyan [3]byte - bcyan [3]byte + cyan [3]byte + bcyan [3]byte - white [3]byte - bwhite [3]byte + white [3]byte + bwhite [3]byte - fg [3]byte - bg [3]byte + fg [3]byte + bg [3]byte } /* the default "control" ANSI color set (boring) */ -var ansiColors = ktPalette { +var ansiColors = ktPalette{ - black : [3]byte {0,0,0}, - bblack : [3]byte {128,128,128}, + black: [3]byte{0, 0, 0}, + bblack: [3]byte{128, 128, 128}, - red : [3]byte {128,0,0}, - bred : [3]byte {255,0,0}, + red: [3]byte{128, 0, 0}, + bred: [3]byte{255, 0, 0}, - green : [3]byte {0,128,0}, - bgreen : [3]byte {0,255,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}, + yellow: [3]byte{128, 128, 0}, + byellow: [3]byte{255, 255, 0}, - blue : [3]byte {0,0,128}, - bblue : [3]byte {0,0,255}, + blue: [3]byte{0, 0, 128}, + bblue: [3]byte{0, 0, 255}, - purple : [3]byte {128,0,128}, - bpurple : [3]byte {255,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}, + cyan: [3]byte{0, 128, 128}, + bcyan: [3]byte{0, 255, 255}, - white : [3]byte {128,128,128}, - bwhite : [3]byte {255,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}, + 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 = "✗" +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 + 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 { - 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") - } + 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) - pal, err := parseColors(file) + if pal == ansiColors { + } - if err != nil { - color.Printf("@r[%s]@{|} - malformed colorfile [%s]\n", xM, err) - return fmt.Errorf("%s\n", "malformed colorfile") - } + color.Printf("@yverifying & preprocessing colorsets@{|} :: @y[SKIP]\n") + color.Printf("@ygenerating transpositional colorspace@{|} :: @y[SKIP]\n") - color.Printf("@g[%s]@{|}\n", checkM) + color.Printf("@ystarting httpd on port @b%d@{|} :: ", port) + http.HandleFunc("/kt/", transposePage) - if pal == ansiColors {} + var portString string + fmt.Sprintf(portString, ":%d", port) + err = http.ListenAndServe(portString, nil) + if err != nil { - color.Printf("@ystarting httpd on port @b%d@{|} :: ", port) + 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 + return nil } -func main() { +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) - err := ktInit("kolors", 999, "/home/kremlin/.Xresources") + if len(cur) < 7 { - /* make sure to close() anything you need to (you need to) */ - _, err = http.Get("http://kremlin.cc") + } else if cur[0:6] == "href=\\" { - if err != nil {} + 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() { + + err := ktInit("kolors", 999, "/home/kremlin/.Xresources") + + if err != nil { + } } type errConsumer struct { - err error + err error } func (e *errConsumer) Consume(err error) { - if e.err == nil && err != nil { - e.err = err - } + if e.err == nil && err != nil { + e.err = err + } } -