ref: c9faa5d093ef6a890e00fb92880ab1a2edf9801c
dir: /browser/browser.go/
package browser
import (
"9fans.net/go/draw"
"errors"
"fmt"
"golang.org/x/net/html"
"golang.org/x/net/publicsuffix"
"image"
"io/ioutil"
"net/http"
"net/http/cookiejar"
"net/url"
"opossum"
"opossum/domino"
"opossum/img"
"opossum/logger"
"opossum/nodes"
"opossum/style"
"strconv"
"strings"
"unicode"
"github.com/mjl-/duit"
_ "image/gif"
_ "image/jpeg"
_ "image/png"
)
const debugPrintHtml = false
const stashElements = true
const experimentalUseSlicedDrawing = false
var DebugDumpCSS *bool
var ExperimentalJsInsecure *bool
var browser *Browser // TODO: limit global objects;
// at least put them in separate pkgs
// with good choiced private/public
// p
var Style = style.Map{}
var dui *duit.DUI
var colorCache = make(map[draw.Color]*draw.Image)
var cache = make(map[string]struct {
opossum.ContentType
buf []byte
})
var numElements int64
var log *logger.Logger
var scroller *duit.Scroll
var display *draw.Display
func SetLogger(l *logger.Logger) {
log = l
}
type ColoredLabel struct {
*duit.Label
Map style.Map
}
func (ui *ColoredLabel) Draw(dui *duit.DUI, self *duit.Kid, img *draw.Image, orig image.Point, m draw.Mouse, force bool) {
// TODO: hacky function, might lead to crashes and memory leaks
c := ui.Map.Color()
i, ok := colorCache[c]
if !ok {
var err error
i, err = dui.Display.AllocImage(image.Rect(0, 0, 10, 10), draw.ARGB32, true, c)
if err != nil {
panic(err.Error())
}
colorCache[c] = i
}
var swap *draw.Image = dui.Regular.Normal.Text
dui.Regular.Normal.Text = i
ui.Label.Draw(dui, self, img, orig, m, force)
dui.Regular.Normal.Text = swap
}
type CodeView struct {
duit.UI
}
func NewCodeView(s string, n style.Map) (cv *CodeView) {
log.Printf("NewCodeView(%+v)", s)
cv = &CodeView{}
edit := &duit.Edit{}
edit.Keys = func(k rune, m draw.Mouse) (e duit.Event) {
//log.Printf("k=%v (c %v p %v)", k, unicode.IsControl(k), unicode.IsPrint(k))
if unicode.IsPrint(k) {
e.Consumed = true
}
return
}
formatted := ""
lines := strings.Split(s, "\n")
for _, line := range lines {
formatted += strings.TrimSpace(line) + "\n"
}
log.Printf("formatted=%+v", formatted)
edit.Append([]byte(formatted))
cv.UI = &duit.Box{
Kids: duit.NewKids(edit),
Height: (int(n.FontSize()) + 4) * len(lines),
}
return
}
func (cv *CodeView) Mouse(dui *duit.DUI, self *duit.Kid, m draw.Mouse, origM draw.Mouse, orig image.Point) (r duit.Result) {
//log.Printf("m=%+v",m.Buttons)
if m.Buttons == 8 || m.Buttons == 16 {
//r.Consumed = true
return
}
return cv.UI.Mouse(dui, self, m, origM, orig)
}
type Image struct {
*duit.Image
src string
}
func NewImage(n nodes.Node) duit.UI {
img, err := newImage(n)
if err != nil {
log.Errorf("could not load image: %v", err)
return &duit.Label{}
}
return img
}
func newImage(n nodes.Node) (ui duit.UI, err error) {
if display == nil {
// probably called from a unit test
return nil, fmt.Errorf("display nil")
}
src := attr(*n.DomSubtree, "src")
if src == "" {
return nil, fmt.Errorf("no src in %+v", n.Attr)
}
var w int
var h int
wStr, ok := n.Declarations["width"]
if ok {
w, _ = strconv.Atoi(strings.TrimSuffix(wStr.Value, "px"))
}
hStr, ok := n.Declarations["height"]
if ok {
h, _ = strconv.Atoi(strings.TrimSuffix(hStr.Value, "px"))
}
r, err := img.Load(browser, src, w, h)
if err != nil {
return nil, fmt.Errorf("load draw image: %w", err)
}
log.Printf("Read %v...", src)
img, err := duit.ReadImage(display, r)
if err != nil {
return nil, fmt.Errorf("duit read image: %w", err)
}
log.Printf("Done reading %v", src)
return &Element{
UI: &Image{
Image: &duit.Image{
Image: img,
},
src: src,
},
}, nil
}
type Element struct {
duit.UI
CS style.Map
IsLink bool
Click func() duit.Event
}
func NewElement(ui duit.UI, cs style.Map) *Element {
if ui == nil {
return nil
}
if cs.IsDisplayNone() {
return nil
}
if stashElements {
existingEl, ok := ui.(*Element)
if ok && existingEl != nil {
// TODO: check is cs and existingEl shouldn't be vice-versa
ccs := cs.ApplyChildStyle(existingEl.CS)
return &Element{
UI: existingEl.UI,
CS: ccs,
}
}
}
return &Element{
UI: ui,
CS: cs,
}
}
func NewBoxElement(ui duit.UI, cs style.Map) *Element {
if ui == nil {
return nil
}
if cs.IsDisplayNone() {
return nil
}
var w int
var h int
wStr, ok := cs.Declarations["width"]
if ok {
w, _ = strconv.Atoi(strings.TrimSuffix(wStr.Value, "px"))
}
hStr, ok := cs.Declarations["height"]
if ok {
h, _ = strconv.Atoi(strings.TrimSuffix(hStr.Value, "px"))
}
var i *draw.Image
var err error
if w == 0 && h == 0 {
return NewElement(ui, cs)
}
if i, err = cs.BoxBackground(); err != nil {
log.Printf("box background: %f", err)
}
box := &duit.Box{
Kids: duit.NewKids(ui),
Width: w,
Height: h,
Background: i,
}
el := NewElement(box, cs)
return el
}
func (el *Element) Draw(dui *duit.DUI, self *duit.Kid, img *draw.Image, orig image.Point, m draw.Mouse, force bool) {
if el == nil {
return
}
if el.slicedDraw(dui, self, img, orig, m, force) {
return
}
box, ok := el.UI.(*duit.Box)
if ok && box.Width > 0 && box.Height > 0 {
uiSize := image.Point{X: box.Width, Y: box.Height}
duit.KidsDraw(dui, self, box.Kids, uiSize, box.Background, img, orig, m, force)
} else {
el.UI.Draw(dui, self, img, orig, m, force)
}
}
func (el *Element) Layout(dui *duit.DUI, self *duit.Kid, sizeAvail image.Point, force bool) {
if el == nil {
return
}
box, ok := el.UI.(*duit.Box)
if ok && box.Width > 0 && box.Height > 0 {
//dui.debugLayout(self)
//if ui.Image == nil {
// self.R = image.ZR
//} else {
// self.R = rect(ui.Image.R.Size())
//}
//duit.KidsLayout(dui, self, box.Kids, true)
el.UI.Layout(dui, self, sizeAvail, force)
self.R = image.Rect(0, 0, box.Width, box.Height)
} else {
el.UI.Layout(dui, self, sizeAvail, force)
}
return
}
func (el *Element) Mark(self *duit.Kid, o duit.UI, forLayout bool) (marked bool) {
if el != nil {
return el.UI.Mark(self, o, forLayout)
}
return
}
func NewSubmitButton(b *Browser, n *nodes.Node) *Element {
t := attr(*n.DomSubtree, "value")
if t == "" {
t = "Submit"
}
btn := &duit.Button{
Text: t,
Font: n.Font(),
Click: func() (r duit.Event) {
b.submit(n.ParentForm().DomSubtree)
//r.Consumed = true
return duit.Event{
Consumed: true,
NeedLayout: true,
NeedDraw: true,
}
//return
},
}
return &Element{
UI: btn,
CS: n.Map,
}
}
func NewInputField(n *nodes.Node) *Element {
t := attr(*n.DomSubtree, "type")
return &Element{
UI: &duit.Box{
Kids: duit.NewKids(&duit.Field{
Font: n.Font(),
Placeholder: attr(*n.DomSubtree, "placeholder"),
Password: t == "password",
Text: attr(*n.DomSubtree, "value"),
Changed: func(t string) (r duit.Event) {
setAttr(n.DomSubtree, "value", t)
r.Consumed = true
return
},
}),
MaxWidth: 200,
},
CS: n.Map,
}
}
func (el *Element) Mouse(dui *duit.DUI, self *duit.Kid, m draw.Mouse, origM draw.Mouse, orig image.Point) (r duit.Result) {
if el == nil {
return
}
if m.Buttons == 1 {
if el.Click != nil {
el.Click()
}
}
x := m.Point.X
y := m.Point.Y
maxX := self.R.Dx()
maxY := self.R.Dy()
if 5 <= x && x <= (maxX-5) && 5 <= y && y <= (maxY-5) {
//log.Printf("Mouse %v (m ~ %v); Kid.R.Dx/Dy=%v/%v\n", el.UI, m.Point, self.R.Dx(), self.R.Dy())
if el.IsLink {
yolo := [2 * 16]uint8{255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 90, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}
dui.Display.SetCursor(&draw.Cursor{
Set: yolo,
})
if m.Buttons == 0 {
r.Consumed = true
return r
}
}
} else {
if el.IsLink {
dui.Display.SetCursor(nil)
} else {
dui.Display.SetCursor(nil)
}
}
return el.UI.Mouse(dui, self, m, origM, orig)
}
func (el *Element) onClickLeafes(f func() duit.Event) {
PrintTree(el) // todo: debug remove me
TraverseTree(el, func(ui duit.UI) {
el, ok := ui.(*Element)
if ok && el != nil {
el.IsLink = true
el.Click = f
return
}
l, ok := ui.(*duit.Label)
if ok && l != nil {
l.Click = f
return
}
})
}
func attr(n html.Node, key string) (val string) {
for _, a := range n.Attr {
if a.Key == key {
return a.Val
}
}
return
}
func hasAttr(n html.Node, key string) bool {
for _, a := range n.Attr {
if a.Key == key {
return true
}
}
return false
}
func setAttr(n *html.Node, key, val string) {
newAttr := html.Attribute{
Key: key,
Val: val,
}
for i, a := range n.Attr {
if a.Key == key {
n.Attr[i] = newAttr
return
}
}
n.Attr = append(n.Attr, newAttr)
}
func formData(n html.Node) (data url.Values) {
data = make(url.Values)
if n.Data == "input" {
if k := attr(n, "name"); k != "" {
data.Set(k, attr(n, "value"))
}
}
for c := n.FirstChild; c != nil; c = c.NextSibling {
for k, vs := range formData(*c) {
data.Set(k, vs[0]) // TODO: what aboot the rest?
}
}
return
}
func (b *Browser) submit(form *html.Node) {
var err error
method := "GET" // TODO
if m := attr(*form, "method"); m != "" {
method = strings.ToUpper(m)
}
uri := b.URL()
log.Printf("form = %+v", form)
if action := attr(*form, "action"); action != "" {
uri, err = b.LinkedUrl(action)
if err != nil {
log.Printf("error parsing %v", action)
return
}
}
var buf []byte
var contentType opossum.ContentType
if method == "GET" {
q := uri.Query()
for k, vs := range formData(*form) {
log.Printf("add query info %v => %v", k, vs[0])
q.Set(k, vs[0]) // TODO: what is with the rest?
}
uri.RawQuery = q.Encode()
log.Printf("uri raw query=%v", uri.RawQuery)
buf, contentType, err = b.get(uri, true)
log.Printf("uri=%v", uri.String())
} else {
buf, contentType, err = b.PostForm(uri, formData(*form))
}
if err == nil {
if contentType.IsHTML() {
b.render(buf)
} else {
log.Fatalf("post: unexpected %v", contentType)
}
} else {
log.Printf("post form: %v", err)
}
}
func Arrange(cs style.Map, elements ...*Element) *Element {
if cs.IsFlex() {
if cs.IsFlexDirectionRow() {
return NewElement(horizontalSequenceOf(true, elements), cs)
} else {
return NewElement(verticalSequenceOf(elements), cs)
}
}
rows := make([][]*Element, 0, 10)
currentRow := make([]*Element, 0, 10)
flushCurrentRow := func() {
if len(currentRow) > 0 {
rows = append(rows, currentRow)
currentRow = make([]*Element, 0, 10)
}
}
for _, e := range elements {
if !e.CS.IsInline() {
flushCurrentRow()
}
currentRow = append(currentRow, e)
if !e.CS.IsInline() {
flushCurrentRow()
}
}
flushCurrentRow()
if len(rows) == 0 {
return nil
} else if len(rows) == 1 {
if len(rows[0]) == 0 {
return nil
} else if len(rows[0]) == 1 {
return rows[0][0]
}
numElements++
return NewElement(horizontalSequenceOf(true, rows[0]), cs)
} else {
seqs := make([]*Element, 0, len(rows))
for _, row := range rows {
seq := horizontalSequenceOf(true, row)
numElements++
seqs = append(seqs, NewElement(seq, cs))
}
numElements++
return NewElement(verticalSequenceOf(seqs), cs)
}
}
func horizontalSequenceOf(wrap bool, es []*Element) duit.UI {
if len(es) == 0 {
return nil
} else if len(es) == 1 {
return es[0]
}
halign := make([]duit.Halign, 0, len(es))
valign := make([]duit.Valign, 0, len(es))
for i := 0; i < len(es); i++ {
halign = append(halign, duit.HalignLeft)
valign = append(valign, duit.ValignTop)
}
uis := make([]duit.UI, 0, len(es))
for _, e := range es {
uis = append(uis, e)
}
if wrap {
log.Printf("wrap")
finalUis := make([]duit.UI, 0, len(uis))
for _, ui := range uis {
log.Printf("wrap, tree:")
PrintTree(ui)
el, ok := ui.(*Element)
if ok {
label, isLabel := el.UI.(*duit.Label)
if isLabel {
tts := strings.Split(label.Text, " ")
for _, t := range tts {
finalUis = append(finalUis, NewElement(&duit.Label{
Text: t,
Font: label.Font,
}, el.CS))
}
} else {
finalUis = append(finalUis, ui)
}
} else {
finalUis = append(finalUis, ui)
}
}
return &duit.Box{
Padding: duit.SpaceXY(6, 4),
Margin: image.Pt(6, 4),
Kids: duit.NewKids(finalUis...),
}
} else {
return &duit.Grid{
Columns: len(es),
Padding: duit.NSpace(len(es), duit.SpaceXY(0, 3)),
Halign: halign,
Valign: valign,
Kids: duit.NewKids(uis...),
}
}
}
func verticalSequenceOf(es []*Element) duit.UI {
if len(es) == 0 {
return nil
} else if len(es) == 1 {
return es[0]
}
uis := make([]duit.UI, 0, len(es))
for _, e := range es {
uis = append(uis, e)
}
return &duit.Grid{
Columns: 1,
Padding: duit.NSpace(1, duit.SpaceXY(0, 3)),
Halign: []duit.Halign{duit.HalignLeft},
Valign: []duit.Valign{duit.ValignTop},
Kids: duit.NewKids(uis...),
}
}
func check(err error, msg string) {
if err != nil {
log.Fatalf("%s: %s\n", msg, err)
}
}
func RichInnerContentFrom(r int, b *Browser, n *nodes.Node) *Element {
childrenAsEls := make([]*Element, 0, 1)
for _, c := range n.Children {
tmp := NodeToBox(r+1, b, c)
if tmp != nil {
numElements++
el := NewElement(tmp, c.Map.ApplyChildStyle(style.TextNode))
childrenAsEls = append(childrenAsEls, el)
}
}
if len(childrenAsEls) == 0 {
return nil
} else if len(childrenAsEls) == 1 {
return childrenAsEls[0]
}
res := Arrange(n.Map, childrenAsEls...)
return res
}
type Table struct {
rows []*TableRow
}
func NewTable(n *nodes.Node) (t *Table) {
t = &Table{
rows: make([]*TableRow, 0, 10),
}
if n.Text != "" || n.DomSubtree.Data != "table" {
log.Printf("invalid table root")
return nil
}
trContainers := make([]*nodes.Node, 0, 2)
for _, c := range n.Children {
if c.DomSubtree.Data == "tbody" || c.DomSubtree.Data == "thead" {
trContainers = append(trContainers, c)
}
}
if len(trContainers) == 0 {
trContainers = []*nodes.Node{n}
}
for _, tc := range trContainers {
for _, c := range tc.Children {
if txt := c.Text; txt != "" && strings.TrimSpace(txt) == "" {
continue
}
if c.DomSubtree.Data == "tr" {
row := NewTableRow(c)
t.rows = append(t.rows, row)
} else {
log.Printf("unexpected row element '%v' (%v)", c.DomSubtree.Data, c.DomSubtree.Type)
}
}
}
return
}
func (t *Table) numColsMin() (min int) {
min = t.numColsMax()
for _, r := range t.rows {
if l := len(r.columns); l < min {
min = l
}
}
return
}
func (t *Table) numColsMax() (max int) {
for _, r := range t.rows {
if l := len(r.columns); l > max {
max = l
}
}
return
}
func (t *Table) Element(r int, b *Browser, cs style.Map) *Element {
numRows := len(t.rows)
numCols := t.numColsMax()
useOneGrid := t.numColsMin() == t.numColsMax()
if useOneGrid {
uis := make([]duit.UI, 0, numRows*numCols)
for _, row := range t.rows {
for _, td := range row.columns {
uis = append(uis, NodeToBox(r+1, b, td))
}
}
log.Printf("use on grid")
halign := make([]duit.Halign, 0, len(uis))
valign := make([]duit.Valign, 0, len(uis))
for i := 0; i < numCols; i++ {
halign = append(halign, duit.HalignLeft)
valign = append(valign, duit.ValignTop)
}
return NewElement(
&duit.Grid{
Columns: numCols,
Padding: duit.NSpace(numCols, duit.SpaceXY(0, 3)),
Halign: halign,
Valign: valign,
Kids: duit.NewKids(uis...),
},
cs,
)
} else {
log.Printf("combine")
seqs := make([]*Element, 0, len(t.rows))
for _, row := range t.rows {
rowEls := make([]*Element, 0, len(row.columns))
for _, col := range row.columns {
ui := NodeToBox(r+1, b, col)
if ui != nil {
el := NewElement(ui, col.Map)
rowEls = append(rowEls, el)
}
}
log.Printf("len rowsEls=%v", len(rowEls))
if len(rowEls) > 0 {
seq := horizontalSequenceOf(false, rowEls)
numElements++
seqs = append(seqs, NewElement(seq, cs))
}
}
numElements++
return NewElement(verticalSequenceOf(seqs), cs)
}
}
type TableRow struct {
columns []*nodes.Node
}
func NewTableRow(n *nodes.Node) (tr *TableRow) {
tr = &TableRow{
columns: make([]*nodes.Node, 0, 5),
}
if n.Type() != html.ElementNode || n.Data() != "tr" {
log.Printf("invalid tr root")
return nil
}
for _, c := range n.Children {
if c.Type() == html.TextNode && strings.TrimSpace(c.Data()) == "" {
continue
}
if c.DomSubtree.Data == "td" || c.DomSubtree.Data == "th" {
tr.columns = append(tr.columns, c)
} else {
log.Printf("unexpected row element '%v' (%v)", c.Data(), c.Type())
}
}
return tr
}
func grepBody(n *html.Node) *html.Node {
var body *html.Node
if n.Type == html.ElementNode {
if n.Data == "body" {
return n
}
}
for c := n.FirstChild; c != nil; c = c.NextSibling {
res := grepBody(c)
if res != nil {
body = res
}
}
return body
}
func NodeToBox(r int, b *Browser, n *nodes.Node) *Element {
if attr(*n.DomSubtree, "aria-hidden") == "true" || hasAttr(*n.DomSubtree, "hidden") {
return nil
}
if n.IsDisplayNone() {
return nil
}
if n.Type() == html.ElementNode {
switch n.Data() {
case "style", "script", "svg", "template":
return nil
case "input":
numElements++
t := attr(*n.DomSubtree, "type")
if isPw := t == "password"; t == "text" || t == "" || t == "search" || isPw {
return NewInputField(n)
} else if t == "submit" {
return NewSubmitButton(b, n)
} else {
return nil
}
case "button":
numElements++
if t := attr(*n.DomSubtree, "type"); t == "" || t == "submit" {
return NewSubmitButton(b, n)
} else {
btn := &duit.Button{
Text: nodes.ContentFrom(*n),
Font: n.Font(),
}
return NewElement(
btn,
n.Map,
)
}
case "table":
numElements++
return NewTable(n).Element(r+1, b, n.Map)
case "noscript":
if *ExperimentalJsInsecure {
return nil
}
fallthrough
case "body", "p", "h1", "center", "nav", "article", "header", "div", "td":
var innerContent duit.UI
if nodes.IsPureTextContent(*n) {
t := strings.TrimSpace(nodes.ContentFrom(*n))
innerContent = &ColoredLabel{
Label: &duit.Label{
Text: t,
Font: n.Font(),
},
Map: n.Map.ApplyChildStyle(style.TextNode),
}
} else {
innerContent = RichInnerContentFrom(r+1, b, n)
}
numElements++
return NewBoxElement(
innerContent,
n.Map.ApplyChildStyle(style.TextNode),
)
case "img":
numElements++
return NewElement(
NewImage(*n),
n.Map,
)
case "pre":
numElements++
return NewElement(
NewCodeView(nodes.ContentFrom(*n), n.Map),
n.Map,
)
case "li":
var innerContent duit.UI
if nodes.IsPureTextContent(*n) {
t := nodes.ContentFrom(*n)
if s, ok := n.Map.Declarations["list-style"]; !ok || s.Value != "none" {
t = "• " + t
}
innerContent = &ColoredLabel{
Label: &duit.Label{
Text: t,
Font: n.Font(),
},
Map: n.Map,
}
} else {
innerContent = RichInnerContentFrom(r+1, b, n)
}
numElements++
return NewElement(
innerContent,
n.Map,
)
case "a":
var href string
for _, a := range n.Attr {
if a.Key == "href" {
href = a.Val
}
}
var innerContent duit.UI
if nodes.IsPureTextContent(*n) {
innerContent = &ColoredLabel{
Label: &duit.Label{
Text: nodes.ContentFrom(*n),
Font: n.Font(),
Click: browser.SetAndLoadUrl(href),
},
Map: n.Map, //.ApplyIsLink(),
}
} else {
// TODO: make blue borders and different
// mouse cursor and actually clickable
innerContent = RichInnerContentFrom(r+1, b, n)
}
numElements++
if innerContent == nil {
return nil
}
el := NewElement(
innerContent,
n.Map,
)
el.IsLink = true // TODO make this cascading
// also a way to bubble up
// will be needed eventually
el.onClickLeafes(browser.SetAndLoadUrl(href))
return el
default:
// Internal node object
els := make([]*Element, 0, 10)
for _, c := range n.Children {
el := NodeToBox(r+1, b, c)
if el != nil && !c.IsDisplayNone() {
els = append(els, el)
}
}
if len(els) == 0 {
return nil
} else if len(els) == 1 {
return els[0]
} else {
for _, e := range els {
_ = e
}
return Arrange(n.Map, els...)
}
}
} else if n.Type() == html.TextNode {
// Leaf text object
if text := strings.TrimSpace(nodes.ContentFrom(*n)); text != "" {
text = strings.ReplaceAll(text, "\n", "")
text = strings.ReplaceAll(text, "\t", "")
l := strings.Split(text, " ")
nn := make([]string, 0, len(l))
for _, w := range l {
if w != "" {
nn = append(nn, w)
}
}
text = strings.Join(nn, " ")
ui := &duit.Label{
Text: text,
Font: n.Font(),
}
numElements++
return NewElement(
ui,
style.TextNode,
)
} else {
return nil
}
} else {
return nil
}
}
type Website struct {
duit.UI
}
func TraverseTree(ui duit.UI, f func(ui duit.UI)) {
traverseTree(0, ui, f)
}
func traverseTree(r int, ui duit.UI, f func(ui duit.UI)) {
if ui == nil {
panic("null")
return
}
f(ui)
switch v := ui.(type) {
case nil:
panic("null")
case *duit.Scroll:
traverseTree(r+1, v.Kid.UI, f)
case *duit.Box:
for _, kid := range v.Kids {
traverseTree(r+1, kid.UI, f)
}
case *Element:
if v == nil {
// TODO: repair?!
//panic("null element")
return
}
traverseTree(r+1, v.UI, f)
case *duit.Grid:
for _, kid := range v.Kids {
traverseTree(r+1, kid.UI, f)
}
case *duit.Image:
case *duit.Label:
case *ColoredLabel:
traverseTree(r+1, v.Label, f)
case *duit.Button:
case *Image:
traverseTree(r+1, v.Image, f)
case *duit.Field:
case *CodeView:
default:
panic(fmt.Sprintf("unknown: %+v", v))
}
}
func PrintTree(ui duit.UI) {
if log.Debug {
printTree(0, ui)
}
}
func printTree(r int, ui duit.UI) {
for i := 0; i < r; i++ {
fmt.Printf(" ")
}
if ui == nil {
fmt.Printf("ui=nil\n")
return
}
switch v := ui.(type) {
case nil:
fmt.Printf("v=nil\n")
return
case *duit.Scroll:
fmt.Printf("duit.Scroll\n")
printTree(r+1, v.Kid.UI)
case *duit.Box:
fmt.Printf("duit.Box\n")
for _, kid := range v.Kids {
printTree(r+1, kid.UI)
}
case *Element:
if v == nil {
fmt.Printf("v:*Element=nil\n")
return
}
fmt.Printf("Element\n")
printTree(r+1, v.UI)
case *duit.Grid:
fmt.Printf("duit.Grid %vx%v\n", len(v.Kids)/v.Columns, v.Columns)
for _, kid := range v.Kids {
printTree(r+1, kid.UI)
}
case *duit.Image:
fmt.Printf("Image %v\n", v)
case *duit.Label:
t := v.Text
if len(t) > 20 {
t = t[:15] + "..."
}
fmt.Printf("Label %v\n", t)
case *ColoredLabel:
t := v.Text
if len(t) > 20 {
t = t[:15] + "..."
}
fmt.Printf("ColoredLabel %v\n", t)
default:
fmt.Printf("default :-) %+v\n", v)
}
}
type History struct {
urls []*url.URL
}
func (h History) URL() *url.URL {
return h.urls[len(h.urls)-1]
}
func (h *History) Push(u *url.URL) {
h.urls = append(h.urls, u)
}
func (h *History) Back() {
if len(h.urls) > 1 {
h.urls = h.urls[:len(h.urls)-1]
}
}
func (h *History) String() string {
addrs := make([]string, len(h.urls))
for i, u := range h.urls {
addrs[i] = u.String()
}
return strings.Join(addrs, ", ")
}
type Browser struct {
History
dui *duit.DUI
html string
Website *Website
StatusBar *duit.Label
LocationField *duit.Field
client *http.Client
}
func NewBrowser(_dui *duit.DUI, initUrl string) (b *Browser) {
jar, err := cookiejar.New(&cookiejar.Options{PublicSuffixList: publicsuffix.List})
if err != nil {
log.Fatal(err)
}
b = &Browser{
client: &http.Client{
Jar: jar,
},
dui: _dui,
StatusBar: &duit.Label{
Text: "",
},
Website: &Website{},
}
b.LocationField = &duit.Field{
Text: initUrl,
Font: Style.Font(),
}
u, err := url.Parse(initUrl)
if err != nil {
log.Fatalf("parse: %v", err)
}
b.History.Push(u)
buf, _, err := b.Get(u)
if err != nil {
log.Fatalf("get: %v", err)
}
b.html = string(buf)
browser = b
style.SetFetcher(b)
dui = _dui
display = dui.Display
b.layoutWebsite()
return
}
func (b *Browser) LinkedUrl(addr string) (a *url.URL, err error) {
log.Printf("LinkedUrl: addr=%v, b.URL=%v", addr, b.URL())
if strings.HasPrefix(addr, "//") {
addr = b.URL().Scheme + ":" + addr
} else if strings.HasPrefix(addr, "/") {
addr = b.URL().Scheme + "://" + b.URL().Host + addr
} else if !strings.HasPrefix(addr, "http") {
if strings.HasSuffix(b.URL().Path, "/") {
log.Printf("A")
addr = "/" + b.URL().Path + "/" + addr
} else {
log.Printf("B")
m := strings.LastIndex(b.URL().Path, "/")
if m > 0 {
log.Printf("B.>")
folder := b.URL().Path[0:m]
addr = "/" + folder + "/" + addr
} else {
log.Printf("B.<=")
addr = "/" + addr
}
}
addr = strings.ReplaceAll(addr, "//", "/")
addr = b.URL().Scheme + "://" + b.URL().Host + addr
}
return url.Parse(addr)
}
func (b *Browser) Back() (e duit.Event) {
if len(b.History.urls) > 0 {
b.History.Back()
b.LocationField.Text = b.History.URL().String()
b.LoadUrl()
}
e.Consumed = true
return
}
func (b *Browser) SetAndLoadUrl(addr string) func() duit.Event {
a := addr
return func() duit.Event {
log.Printf("SetAndLoadUrl::callback: addr=%v", addr)
log.Printf(" b.URL=%v", b.URL())
url, err := b.LinkedUrl(addr)
if err == nil {
b.LocationField.Text = url.String()
b.LoadUrl()
} else {
log.Printf("parse url %v: %v", a, err)
}
return duit.Event{
Consumed: true,
}
}
}
func (b *Browser) showBodyMessage(msg string) {
b.Website.UI = &duit.Label{
Text: msg,
Font: style.Map{}.Font(),
}
dui.MarkLayout(dui.Top.UI)
dui.MarkDraw(dui.Top.UI)
dui.Render()
}
func (b *Browser) LoadUrl() (e duit.Event) {
addr := b.LocationField.Text
if !strings.HasPrefix(addr, "http") {
addr = "https://" + addr
}
log.Printf("Getting %v...", addr)
url, err := url.Parse(addr)
if err != nil {
log.Errorf("load url: error parsing %v", addr)
return
}
buf, contentType, err := b.get(url, true)
if err != nil {
log.Errorf("error loading %v: %v", addr, err)
err = errors.Unwrap(err)
if strings.Contains(err.Error(), "HTTP response to HTTPS client") {
b.LocationField.Text = strings.Replace(url.String(), "https://", "http://", 1)
return b.LoadUrl()
}
b.showBodyMessage(err.Error())
return
}
if contentType.IsHTML() || contentType.IsPlain() {
b.render(buf)
} else {
log.Errorf("unhandled content type: %v", contentType)
}
return duit.Event{
Consumed: true,
NeedLayout: true,
NeedDraw: true,
}
}
func (b *Browser) render(buf []byte) {
b.html = string(buf) // TODO: correctly interpret UTF8
b.layoutWebsite()
dui.MarkLayout(dui.Top.UI)
dui.MarkDraw(dui.Top.UI)
TraverseTree(b.Website.UI, func(ui duit.UI) {
// just checking
if ui == nil {
panic("nil")
}
})
PrintTree(b.Website.UI)
//log.Printf("Empty image cache...")
//cache = make(map[string][]byte)
log.Printf("Render...")
dui.Render()
log.Printf("Rendering done")
}
func (b *Browser) Get(uri *url.URL) (buf []byte, contentType opossum.ContentType, err error) {
c, ok := cache[uri.String()]
if ok {
log.Printf("use %v from cache", uri)
} else {
c.buf, c.ContentType, err = b.get(uri, false)
if err == nil {
cache[uri.String()] = c
}
}
return c.buf, c.ContentType, err
}
func (b *Browser) statusBarMsg(msg string, emptyBody bool) {
if dui == nil || dui.Top.UI == nil {
return
}
b.StatusBar.Text = msg
if emptyBody {
b.Website.UI = &duit.Label{}
}
dui.MarkLayout(dui.Top.UI)
dui.MarkDraw(dui.Top.UI)
dui.Render()
}
func (b *Browser) get(uri *url.URL, isNewOrigin bool) (buf []byte, contentType opossum.ContentType, err error) {
msg := fmt.Sprintf("Get %v", uri.String())
log.Printf(msg)
b.statusBarMsg(msg, true)
defer func() {
b.statusBarMsg("", true)
}()
req, err := http.NewRequest("GET", uri.String(), nil)
if err != nil {
return
}
req.Header.Add("User-Agent", "opossum")
resp, err := b.client.Do(req)
if err != nil {
return nil, opossum.ContentType{}, fmt.Errorf("error loading %v: %w", uri, err)
}
defer resp.Body.Close()
buf, err = ioutil.ReadAll(resp.Body)
if err != nil {
return nil, opossum.ContentType{}, fmt.Errorf("error reading")
}
contentType, err = opossum.NewContentType(resp.Header.Get("Content-Type"))
log.Printf("%v\n", resp.Header)
if err == nil && (contentType.IsHTML() || contentType.IsCSS() || contentType.IsPlain()) {
buf = contentType.Utf8(buf)
}
if isNewOrigin {
b.History.Push(resp.Request.URL)
log.Printf("b.History is now %s", b.History.String())
b.LocationField.Text = b.URL().String()
}
return
}
func (b *Browser) PostForm(uri *url.URL, data url.Values) (buf []byte, contentType opossum.ContentType, err error) {
b.Website.UI = &duit.Label{Text: "Posting..."}
dui.MarkLayout(dui.Top.UI)
dui.MarkDraw(dui.Top.UI)
dui.Render()
req, err := http.NewRequest("POST", uri.String(), strings.NewReader(data.Encode()))
if err != nil {
return
}
req.Header.Add("User-Agent", "opossum")
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
resp, err := b.client.Do(req)
if err != nil {
return nil, opossum.ContentType{}, fmt.Errorf("error loading %v: %w", uri, err)
}
defer resp.Body.Close()
b.History.Push(resp.Request.URL)
buf, err = ioutil.ReadAll(resp.Body)
if err != nil {
return nil, opossum.ContentType{}, fmt.Errorf("error reading")
}
contentType, err = opossum.NewContentType(resp.Header.Get("Content-Type"))
return
}
func (b *Browser) layoutWebsite() {
pass := func(htm string, csss ...string) (*html.Node, map[*html.Node]style.Map) {
if debugPrintHtml {
log.Printf("%v\n", htm)
}
var doc *html.Node
var err error
doc, err = html.ParseWithOptions(
strings.NewReader(htm),
html.ParseOptionEnableScripting(*ExperimentalJsInsecure),
)
if err != nil {
panic(err.Error())
}
log.Printf("Retrieving CSS Rules...")
var cssSize int
nodeMap := make(map[*html.Node]style.Map)
for i, css := range csss {
log.Printf("CSS size %v kB", cssSize/1024)
nm, err := style.FetchNodeMap(doc, css, 1280)
if err == nil {
log.Printf("[%v/%v] Fetch CSS Rules successful!", i+1, len(csss))
if debugPrintHtml {
log.Printf("%v", nm)
}
style.MergeNodeMaps(nodeMap, nm)
} else {
log.Errorf("Fetch CSS Rules failed: %v", err)
if *DebugDumpCSS {
ioutil.WriteFile("info.css", []byte(css), 0644)
}
}
}
return doc, nodeMap
}
log.Printf("1st pass")
doc, _ := pass(b.html)
log.Printf("2nd pass")
log.Printf("Download style...")
cssHrefs := style.Hrefs(doc)
csss := make([]string, 0, len(cssHrefs))
for _, href := range cssHrefs {
url, err := b.LinkedUrl(href)
if err != nil {
log.Printf("error parsing %v", href)
continue
}
log.Printf("Download %v", url)
buf, contentType, err := b.Get(url)
if err != nil {
log.Printf("error downloading %v", url)
continue
}
if contentType.IsCSS() {
csss = append(csss, string(buf))
} else {
log.Printf("css: unexpected %v", contentType)
}
}
csss = append([]string{ /*string(revertCSS), */ style.AddOnCSS}, csss...)
doc, nodeMap := pass(b.html, csss...)
if *ExperimentalJsInsecure {
log.Printf("3rd pass")
nt := nodes.NewNodeTree(doc, style.Map{}, nodeMap, nil)
jsSrcs := domino.Srcs(nt)
downloads := make(map[string]string)
for _, src := range jsSrcs {
url, err := b.LinkedUrl(src)
if err != nil {
log.Printf("error parsing %v", src)
continue
}
log.Printf("Download %v", url)
buf, _, err := b.Get(url)
if err != nil {
log.Printf("error downloading %v", url)
continue
}
downloads[src] = string(buf)
}
codes := domino.Scripts(nt, downloads)
log.Infof("JS pipeline start")
jsProcessed, err := processJS2(b.html, nt, codes)
if err == nil {
if b.html != jsProcessed {
log.Infof("html changed")
}
b.html = jsProcessed
if debugPrintHtml {
log.Printf("%v\n", jsProcessed)
}
doc, nodeMap = pass(b.html, csss...)
} else {
log.Errorf("JS error: %v", err)
}
log.Infof("JS pipeline end")
}
var countHtmlNodes func(*html.Node) int
countHtmlNodes = func(n *html.Node) (num int) {
num++
for c := n.FirstChild; c != nil; c = c.NextSibling {
num += countHtmlNodes(c)
}
return
}
log.Printf("%v html nodes found...", countHtmlNodes(doc))
body := grepBody(doc)
log.Printf("Layout website...")
numElements = 0
scroller = duit.NewScroll(
NodeToBox(0, b, nodes.NewNodeTree(body, style.Map{}, nodeMap, nil)),
)
b.Website.UI = scroller
log.Printf("Layouting done (%v elements created)", numElements)
if numElements < 10 {
log.Errorf("Less than 10 elements layouted, seems css processing failed. Will layout without css")
scroller = duit.NewScroll(
NodeToBox(0, b, nodes.NewNodeTree(body, style.Map{}, make(map[*html.Node]style.Map), nil)),
)
b.Website.UI = scroller
}
log.Flush()
}