| OLD | NEW |
| 1 package main | 1 package main |
| 2 | 2 |
| 3 import ( | 3 import ( |
| 4 "bytes" | 4 "bytes" |
| 5 "crypto/md5" | 5 "crypto/md5" |
| 6 "database/sql" | 6 "database/sql" |
| 7 "encoding/base64" | 7 "encoding/base64" |
| 8 "encoding/binary" | 8 "encoding/binary" |
| 9 "encoding/json" | 9 "encoding/json" |
| 10 "flag" | 10 "flag" |
| (...skipping 237 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 248 )` | 248 )` |
| 249 _, err = db.Exec(sql) | 249 _, err = db.Exec(sql) |
| 250 if err != nil { | 250 if err != nil { |
| 251 log.Printf("Info: status creating sqlite table for sourc
es: %q\n", err) | 251 log.Printf("Info: status creating sqlite table for sourc
es: %q\n", err) |
| 252 } | 252 } |
| 253 | 253 |
| 254 sql = `CREATE TABLE IF NOT EXISTS webtry ( | 254 sql = `CREATE TABLE IF NOT EXISTS webtry ( |
| 255 code TEXT DEFAULT '' NOT NULL, | 255 code TEXT DEFAULT '' NOT NULL, |
| 256 create_ts TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL, | 256 create_ts TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL, |
| 257 hash CHAR(64) DEFAULT '' NOT NULL, | 257 hash CHAR(64) DEFAULT '' NOT NULL, |
| 258 width INTEGER DEFAULT 256 NOT NULL, | 258 width »» » INTEGER DEFAULT 256 NOT
NULL, |
| 259 height INTEGER DEFAULT 256 NOT NULL, | 259 height »» » INTEGER DEFAULT 256 NOT
NULL, |
| 260 gpu BOOL DEFAULT 0 NOT NULL, | |
| 261 source_image_id INTEGER DEFAULT 0 NOT NULL, | 260 source_image_id INTEGER DEFAULT 0 NOT NULL, |
| 262 | 261 |
| 263 PRIMARY KEY(hash) | 262 PRIMARY KEY(hash) |
| 264 )` | 263 )` |
| 265 _, err = db.Exec(sql) | 264 _, err = db.Exec(sql) |
| 266 if err != nil { | 265 if err != nil { |
| 267 log.Printf("Info: status creating sqlite table for webtr
y: %q\n", err) | 266 log.Printf("Info: status creating sqlite table for webtr
y: %q\n", err) |
| 268 } | 267 } |
| 269 | 268 |
| 270 sql = `CREATE TABLE IF NOT EXISTS workspace ( | 269 sql = `CREATE TABLE IF NOT EXISTS workspace ( |
| 271 name CHAR(64) DEFAULT '' NOT NULL, | 270 name CHAR(64) DEFAULT '' NOT NULL, |
| 272 create_ts TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL, | 271 create_ts TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL, |
| 273 PRIMARY KEY(name) | 272 PRIMARY KEY(name) |
| 274 )` | 273 )` |
| 275 _, err = db.Exec(sql) | 274 _, err = db.Exec(sql) |
| 276 if err != nil { | 275 if err != nil { |
| 277 log.Printf("Info: status creating sqlite table for works
pace: %q\n", err) | 276 log.Printf("Info: status creating sqlite table for works
pace: %q\n", err) |
| 278 } | 277 } |
| 279 | 278 |
| 280 sql = `CREATE TABLE IF NOT EXISTS workspacetry ( | 279 sql = `CREATE TABLE IF NOT EXISTS workspacetry ( |
| 281 name CHAR(64) DEFAULT '' NOT NULL, | 280 name CHAR(64) DEFAULT '' NOT NULL, |
| 282 create_ts TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL, | 281 create_ts TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL, |
| 283 hash CHAR(64) DEFAULT '' NOT NULL, | 282 hash CHAR(64) DEFAULT '' NOT NULL, |
| 284 width INTEGER DEFAULT 256 NOT NULL, | 283 width » » INTEGER DEFAULT 256 NOT NULL, |
| 285 height INTEGER DEFAULT 256 NOT NULL, | 284 height » » INTEGER DEFAULT 256 NOT NULL, |
| 286 gpu BOOL DEFAULT 0 NOT NULL, | |
| 287 hidden INTEGER DEFAULT 0 NOT NULL, | 285 hidden INTEGER DEFAULT 0 NOT NULL, |
| 288 source_image_id INTEGER DEFAULT 0 NOT NULL, | 286 source_image_id INTEGER DEFAULT 0 NOT NULL, |
| 289 | 287 |
| 290 FOREIGN KEY (name) REFERENCES workspace(name) | 288 FOREIGN KEY (name) REFERENCES workspace(name) |
| 291 )` | 289 )` |
| 292 _, err = db.Exec(sql) | 290 _, err = db.Exec(sql) |
| 293 if err != nil { | 291 if err != nil { |
| 294 log.Printf("Info: status creating sqlite table for works
pace try: %q\n", err) | 292 log.Printf("Info: status creating sqlite table for works
pace try: %q\n", err) |
| 295 } | 293 } |
| 296 } | 294 } |
| (...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 349 GitHash string | 347 GitHash string |
| 350 GitInfo string | 348 GitInfo string |
| 351 } | 349 } |
| 352 | 350 |
| 353 // userCode is used in template expansion. | 351 // userCode is used in template expansion. |
| 354 type userCode struct { | 352 type userCode struct { |
| 355 Code string | 353 Code string |
| 356 Hash string | 354 Hash string |
| 357 Width int | 355 Width int |
| 358 Height int | 356 Height int |
| 359 GPU bool | |
| 360 Source int | 357 Source int |
| 361 Titlebar Titlebar | 358 Titlebar Titlebar |
| 362 } | 359 } |
| 363 | 360 |
| 364 // writeTemplate creates a given output file and writes the template | 361 // writeTemplate creates a given output file and writes the template |
| 365 // result there. | 362 // result there. |
| 366 func writeTemplate(filename string, t *template.Template, context interface{}) e
rror { | 363 func writeTemplate(filename string, t *template.Template, context interface{}) e
rror { |
| 367 f, err := os.Create(filename) | 364 f, err := os.Create(filename) |
| 368 if err != nil { | 365 if err != nil { |
| 369 return err | 366 return err |
| 370 } | 367 } |
| 371 defer f.Close() | 368 defer f.Close() |
| 372 return t.Execute(f, context) | 369 return t.Execute(f, context) |
| 373 } | 370 } |
| 374 | 371 |
| 375 // expandToFile expands the template and writes the result to the file. | 372 // expandToFile expands the template and writes the result to the file. |
| 376 func expandToFile(filename string, code string, t *template.Template) error { | 373 func expandToFile(filename string, code string, t *template.Template) error { |
| 377 return writeTemplate(filename, t, userCode{ | 374 return writeTemplate(filename, t, userCode{ |
| 378 Code: code, | 375 Code: code, |
| 379 Titlebar: Titlebar{GitHash: gitHash, GitInfo: gitInfo}, | 376 Titlebar: Titlebar{GitHash: gitHash, GitInfo: gitInfo}, |
| 380 }) | 377 }) |
| 381 } | 378 } |
| 382 | 379 |
| 383 // expandCode expands the template into a file and calculates the MD5 hash. | 380 // expandCode expands the template into a file and calculates the MD5 hash. |
| 384 // We include the width and height here so that a single hash can capture | 381 // We include the width and height here so that a single hash can capture |
| 385 // both the code and the supplied width/height parameters. | 382 // both the code and the supplied width/height parameters. |
| 386 func expandCode(code string, source int, width, height int, gpu bool) (string, e
rror) { | 383 func expandCode(code string, source int, width, height int) (string, error) { |
| 387 // in order to support fonts in the chroot jail, we need to make sure | 384 // in order to support fonts in the chroot jail, we need to make sure |
| 388 // we're using portable typefaces. | 385 // we're using portable typefaces. |
| 389 // TODO(humper): Make this more robust, supporting things like setTypef
ace | 386 // TODO(humper): Make this more robust, supporting things like setTypef
ace |
| 390 | 387 |
| 391 inputCodeLines := strings.Split(code, "\n") | 388 inputCodeLines := strings.Split(code, "\n") |
| 392 » outputCodeLines := []string{ | 389 » outputCodeLines := []string{"DECLARE_bool(portableFonts);", fmt.Sprintf(
"// WxH: %d, %d", width, height)} |
| 393 » » "DECLARE_bool(portableFonts);", | |
| 394 » » fmt.Sprintf("// WxH: %d, %d", width, height), | |
| 395 » » fmt.Sprintf("// GPU: %v", gpu), | |
| 396 » } | |
| 397 for _, line := range inputCodeLines { | 390 for _, line := range inputCodeLines { |
| 398 outputCodeLines = append(outputCodeLines, line) | 391 outputCodeLines = append(outputCodeLines, line) |
| 399 if strings.HasPrefix(strings.TrimSpace(line), "SkPaint p") { | 392 if strings.HasPrefix(strings.TrimSpace(line), "SkPaint p") { |
| 400 outputCodeLines = append(outputCodeLines, "FLAGS_portabl
eFonts = true;") | 393 outputCodeLines = append(outputCodeLines, "FLAGS_portabl
eFonts = true;") |
| 401 outputCodeLines = append(outputCodeLines, "sk_tool_utils
::set_portable_typeface(&p, \"Helvetica\", SkTypeface::kNormal);") | 394 outputCodeLines = append(outputCodeLines, "sk_tool_utils
::set_portable_typeface(&p, \"Helvetica\", SkTypeface::kNormal);") |
| 402 } | 395 } |
| 403 } | 396 } |
| 404 | 397 |
| 405 fontFriendlyCode := strings.Join(outputCodeLines, "\n") | 398 fontFriendlyCode := strings.Join(outputCodeLines, "\n") |
| 406 | 399 |
| (...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 465 log.Printf("Error: %s\n%s", message, err.Error()) | 458 log.Printf("Error: %s\n%s", message, err.Error()) |
| 466 resp, err := json.Marshal(m) | 459 resp, err := json.Marshal(m) |
| 467 if err != nil { | 460 if err != nil { |
| 468 http.Error(w, "Failed to serialize a response", 500) | 461 http.Error(w, "Failed to serialize a response", 500) |
| 469 return | 462 return |
| 470 } | 463 } |
| 471 w.Header().Set("Content-Type", "text/plain") | 464 w.Header().Set("Content-Type", "text/plain") |
| 472 w.Write(resp) | 465 w.Write(resp) |
| 473 } | 466 } |
| 474 | 467 |
| 475 func writeToDatabase(hash string, code string, workspaceName string, source int,
width, height int, gpu bool) { | 468 func writeToDatabase(hash string, code string, workspaceName string, source int,
width, height int) { |
| 476 if db == nil { | 469 if db == nil { |
| 477 return | 470 return |
| 478 } | 471 } |
| 479 » if _, err := db.Exec("INSERT INTO webtry (code, hash, width, height, gpu
, source_image_id) VALUES(?, ?, ?, ?, ?, ?)", code, hash, width, height, gpu, so
urce); err != nil { | 472 » if _, err := db.Exec("INSERT INTO webtry (code, hash, width, height, sou
rce_image_id) VALUES(?, ?, ?, ?, ?)", code, hash, width, height, source); err !=
nil { |
| 480 log.Printf("ERROR: Failed to insert code into database: %q\n", e
rr) | 473 log.Printf("ERROR: Failed to insert code into database: %q\n", e
rr) |
| 481 } | 474 } |
| 482 if workspaceName != "" { | 475 if workspaceName != "" { |
| 483 » » if _, err := db.Exec("INSERT INTO workspacetry (name, hash, widt
h, height, gpu, source_image_id) VALUES(?, ?, ?, ?, ?, ?)", workspaceName, hash,
width, height, gpu, source); err != nil { | 476 » » if _, err := db.Exec("INSERT INTO workspacetry (name, hash, widt
h, height, source_image_id) VALUES(?, ?, ?, ?, ?)", workspaceName, hash, width,
height, source); err != nil { |
| 484 log.Printf("ERROR: Failed to insert into workspacetry ta
ble: %q\n", err) | 477 log.Printf("ERROR: Failed to insert into workspacetry ta
ble: %q\n", err) |
| 485 } | 478 } |
| 486 } | 479 } |
| 487 } | 480 } |
| 488 | 481 |
| 489 type Sources struct { | 482 type Sources struct { |
| 490 Id int `json:"id"` | 483 Id int `json:"id"` |
| 491 } | 484 } |
| 492 | 485 |
| 493 // sourcesHandler serves up the PNG of a specific try. | 486 // sourcesHandler serves up the PNG of a specific try. |
| (...skipping 122 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 616 } | 609 } |
| 617 } | 610 } |
| 618 | 611 |
| 619 type Workspace struct { | 612 type Workspace struct { |
| 620 Name string | 613 Name string |
| 621 Code string | 614 Code string |
| 622 Hash string | 615 Hash string |
| 623 Width int | 616 Width int |
| 624 Height int | 617 Height int |
| 625 Source int | 618 Source int |
| 626 GPU bool | |
| 627 Tries []Try | 619 Tries []Try |
| 628 Titlebar Titlebar | 620 Titlebar Titlebar |
| 629 } | 621 } |
| 630 | 622 |
| 631 // newWorkspace generates a new random workspace name and stores it in the datab
ase. | 623 // newWorkspace generates a new random workspace name and stores it in the datab
ase. |
| 632 func newWorkspace() (string, error) { | 624 func newWorkspace() (string, error) { |
| 633 for i := 0; i < 10; i++ { | 625 for i := 0; i < 10; i++ { |
| 634 adj := workspaceNameAdj[rand.Intn(len(workspaceNameAdj))] | 626 adj := workspaceNameAdj[rand.Intn(len(workspaceNameAdj))] |
| 635 noun := workspaceNameNoun[rand.Intn(len(workspaceNameNoun))] | 627 noun := workspaceNameNoun[rand.Intn(len(workspaceNameNoun))] |
| 636 suffix := rand.Intn(1000) | 628 suffix := rand.Intn(1000) |
| 637 name := fmt.Sprintf("%s-%s-%d", adj, noun, suffix) | 629 name := fmt.Sprintf("%s-%s-%d", adj, noun, suffix) |
| 638 if _, err := db.Exec("INSERT INTO workspace (name) VALUES(?)", n
ame); err == nil { | 630 if _, err := db.Exec("INSERT INTO workspace (name) VALUES(?)", n
ame); err == nil { |
| 639 return name, nil | 631 return name, nil |
| 640 } else { | 632 } else { |
| 641 log.Printf("ERROR: Failed to insert workspace into datab
ase: %q\n", err) | 633 log.Printf("ERROR: Failed to insert workspace into datab
ase: %q\n", err) |
| 642 } | 634 } |
| 643 } | 635 } |
| 644 return "", fmt.Errorf("Failed to create a new workspace") | 636 return "", fmt.Errorf("Failed to create a new workspace") |
| 645 } | 637 } |
| 646 | 638 |
| 647 // getCode returns the code for a given hash, or the empty string if not found. | 639 // getCode returns the code for a given hash, or the empty string if not found. |
| 648 func getCode(hash string) (string, int, int, int, bool, error) { | 640 func getCode(hash string) (string, int, int, int, error) { |
| 649 code := "" | 641 code := "" |
| 650 width := 0 | 642 width := 0 |
| 651 height := 0 | 643 height := 0 |
| 652 source := 0 | 644 source := 0 |
| 653 » gpu := false | 645 » if err := db.QueryRow("SELECT code, width, height, source_image_id FROM
webtry WHERE hash=?", hash).Scan(&code, &width, &height, &source); err != nil { |
| 654 » if err := db.QueryRow("SELECT code, width, height, gpu, source_image_id
FROM webtry WHERE hash=?", hash).Scan(&code, &width, &height, &gpu, &source); er
r != nil { | |
| 655 log.Printf("ERROR: Code for hash is missing: %q\n", err) | 646 log.Printf("ERROR: Code for hash is missing: %q\n", err) |
| 656 » » return code, width, height, source, gpu, err | 647 » » return code, width, height, source, err |
| 657 } | 648 } |
| 658 » return code, width, height, source, gpu, nil | 649 » return code, width, height, source, nil |
| 659 } | 650 } |
| 660 | 651 |
| 661 func workspaceHandler(w http.ResponseWriter, r *http.Request) { | 652 func workspaceHandler(w http.ResponseWriter, r *http.Request) { |
| 662 log.Printf("Workspace Handler: %q\n", r.URL.Path) | 653 log.Printf("Workspace Handler: %q\n", r.URL.Path) |
| 663 if r.Method == "GET" { | 654 if r.Method == "GET" { |
| 664 tries := []Try{} | 655 tries := []Try{} |
| 665 match := workspaceLink.FindStringSubmatch(r.URL.Path) | 656 match := workspaceLink.FindStringSubmatch(r.URL.Path) |
| 666 name := "" | 657 name := "" |
| 667 if len(match) == 2 { | 658 if len(match) == 2 { |
| 668 name = match[1] | 659 name = match[1] |
| (...skipping 11 matching lines...) Expand all Loading... |
| 680 continue | 671 continue |
| 681 } | 672 } |
| 682 tries = append(tries, Try{Hash: hash, Source: so
urce, CreateTS: create_ts.Format("2006-02-01")}) | 673 tries = append(tries, Try{Hash: hash, Source: so
urce, CreateTS: create_ts.Format("2006-02-01")}) |
| 683 } | 674 } |
| 684 } | 675 } |
| 685 var code string | 676 var code string |
| 686 var hash string | 677 var hash string |
| 687 var width int | 678 var width int |
| 688 var height int | 679 var height int |
| 689 source := 0 | 680 source := 0 |
| 690 gpu := false | |
| 691 if len(tries) == 0 { | 681 if len(tries) == 0 { |
| 692 code = DEFAULT_SAMPLE | 682 code = DEFAULT_SAMPLE |
| 693 width = 256 | 683 width = 256 |
| 694 height = 256 | 684 height = 256 |
| 695 } else { | 685 } else { |
| 696 hash = tries[len(tries)-1].Hash | 686 hash = tries[len(tries)-1].Hash |
| 697 » » » code, width, height, source, gpu, _ = getCode(hash) | 687 » » » code, width, height, source, _ = getCode(hash) |
| 698 } | 688 } |
| 699 w.Header().Set("Content-Type", "text/html") | 689 w.Header().Set("Content-Type", "text/html") |
| 700 » » if err := workspaceTemplate.Execute(w, Workspace{Tries: tries, C
ode: code, Name: name, Hash: hash, Width: width, Height: height, GPU: gpu, Sourc
e: source, Titlebar: Titlebar{GitHash: gitHash, GitInfo: gitInfo}}); err != nil
{ | 690 » » if err := workspaceTemplate.Execute(w, Workspace{Tries: tries, C
ode: code, Name: name, Hash: hash, Width: width, Height: height, Source: source,
Titlebar: Titlebar{GitHash: gitHash, GitInfo: gitInfo}}); err != nil { |
| 701 log.Printf("ERROR: Failed to expand template: %q\n", err
) | 691 log.Printf("ERROR: Failed to expand template: %q\n", err
) |
| 702 } | 692 } |
| 703 } else if r.Method == "POST" { | 693 } else if r.Method == "POST" { |
| 704 name, err := newWorkspace() | 694 name, err := newWorkspace() |
| 705 if err != nil { | 695 if err != nil { |
| 706 http.Error(w, "Failed to create a new workspace.", 500) | 696 http.Error(w, "Failed to create a new workspace.", 500) |
| 707 return | 697 return |
| 708 } | 698 } |
| 709 http.Redirect(w, r, "/w/"+name, 302) | 699 http.Redirect(w, r, "/w/"+name, 302) |
| 710 } | 700 } |
| 711 } | 701 } |
| 712 | 702 |
| 713 // hasPreProcessor returns true if any line in the code begins with a # char. | 703 // hasPreProcessor returns true if any line in the code begins with a # char. |
| 714 func hasPreProcessor(code string) bool { | 704 func hasPreProcessor(code string) bool { |
| 715 lines := strings.Split(code, "\n") | 705 lines := strings.Split(code, "\n") |
| 716 for _, s := range lines { | 706 for _, s := range lines { |
| 717 if strings.HasPrefix(strings.TrimSpace(s), "#") { | 707 if strings.HasPrefix(strings.TrimSpace(s), "#") { |
| 718 return true | 708 return true |
| 719 } | 709 } |
| 720 } | 710 } |
| 721 return false | 711 return false |
| 722 } | 712 } |
| 723 | 713 |
| 724 type TryRequest struct { | 714 type TryRequest struct { |
| 725 Code string `json:"code"` | 715 Code string `json:"code"` |
| 726 Width int `json:"width"` | 716 Width int `json:"width"` |
| 727 Height int `json:"height"` | 717 Height int `json:"height"` |
| 728 GPU bool `json:"gpu"` | |
| 729 Name string `json:"name"` // Optional name of the workspace the code
is in. | 718 Name string `json:"name"` // Optional name of the workspace the code
is in. |
| 730 Source int `json:"source"` // ID of the source image, 0 if none. | 719 Source int `json:"source"` // ID of the source image, 0 if none. |
| 731 } | 720 } |
| 732 | 721 |
| 733 // iframeHandler handles the GET and POST of the main page. | 722 // iframeHandler handles the GET and POST of the main page. |
| 734 func iframeHandler(w http.ResponseWriter, r *http.Request) { | 723 func iframeHandler(w http.ResponseWriter, r *http.Request) { |
| 735 log.Printf("IFrame Handler: %q\n", r.URL.Path) | 724 log.Printf("IFrame Handler: %q\n", r.URL.Path) |
| 736 if r.Method != "GET" { | 725 if r.Method != "GET" { |
| 737 http.NotFound(w, r) | 726 http.NotFound(w, r) |
| 738 return | 727 return |
| 739 } | 728 } |
| 740 match := iframeLink.FindStringSubmatch(r.URL.Path) | 729 match := iframeLink.FindStringSubmatch(r.URL.Path) |
| 741 if len(match) != 2 { | 730 if len(match) != 2 { |
| 742 http.NotFound(w, r) | 731 http.NotFound(w, r) |
| 743 return | 732 return |
| 744 } | 733 } |
| 745 hash := match[1] | 734 hash := match[1] |
| 746 if db == nil { | 735 if db == nil { |
| 747 http.NotFound(w, r) | 736 http.NotFound(w, r) |
| 748 return | 737 return |
| 749 } | 738 } |
| 750 var code string | 739 var code string |
| 751 » code, width, height, source, gpu, err := getCode(hash) | 740 » code, width, height, source, err := getCode(hash) |
| 752 if err != nil { | 741 if err != nil { |
| 753 http.NotFound(w, r) | 742 http.NotFound(w, r) |
| 754 return | 743 return |
| 755 } | 744 } |
| 756 // Expand the template. | 745 // Expand the template. |
| 757 w.Header().Set("Content-Type", "text/html") | 746 w.Header().Set("Content-Type", "text/html") |
| 758 » if err := iframeTemplate.Execute(w, userCode{Code: code, Width: width, H
eight: height, GPU: gpu, Hash: hash, Source: source}); err != nil { | 747 » if err := iframeTemplate.Execute(w, userCode{Code: code, Width: width, H
eight: height, Hash: hash, Source: source}); err != nil { |
| 759 log.Printf("ERROR: Failed to expand template: %q\n", err) | 748 log.Printf("ERROR: Failed to expand template: %q\n", err) |
| 760 } | 749 } |
| 761 } | 750 } |
| 762 | 751 |
| 763 type TryInfo struct { | 752 type TryInfo struct { |
| 764 Hash string `json:"hash"` | 753 Hash string `json:"hash"` |
| 765 Code string `json:"code"` | 754 Code string `json:"code"` |
| 766 Width int `json:"width"` | 755 Width int `json:"width"` |
| 767 Height int `json:"height"` | 756 Height int `json:"height"` |
| 768 GPU bool `json:"gpu"` | |
| 769 Source int `json:"source"` | 757 Source int `json:"source"` |
| 770 } | 758 } |
| 771 | 759 |
| 772 // tryInfoHandler returns information about a specific try. | 760 // tryInfoHandler returns information about a specific try. |
| 773 func tryInfoHandler(w http.ResponseWriter, r *http.Request) { | 761 func tryInfoHandler(w http.ResponseWriter, r *http.Request) { |
| 774 log.Printf("Try Info Handler: %q\n", r.URL.Path) | 762 log.Printf("Try Info Handler: %q\n", r.URL.Path) |
| 775 if r.Method != "GET" { | 763 if r.Method != "GET" { |
| 776 http.NotFound(w, r) | 764 http.NotFound(w, r) |
| 777 return | 765 return |
| 778 } | 766 } |
| 779 match := tryInfoLink.FindStringSubmatch(r.URL.Path) | 767 match := tryInfoLink.FindStringSubmatch(r.URL.Path) |
| 780 if len(match) != 2 { | 768 if len(match) != 2 { |
| 781 http.NotFound(w, r) | 769 http.NotFound(w, r) |
| 782 return | 770 return |
| 783 } | 771 } |
| 784 hash := match[1] | 772 hash := match[1] |
| 785 » code, width, height, source, gpu, err := getCode(hash) | 773 » code, width, height, source, err := getCode(hash) |
| 786 if err != nil { | 774 if err != nil { |
| 787 http.NotFound(w, r) | 775 http.NotFound(w, r) |
| 788 return | 776 return |
| 789 } | 777 } |
| 790 m := TryInfo{ | 778 m := TryInfo{ |
| 791 Hash: hash, | 779 Hash: hash, |
| 792 Code: code, | 780 Code: code, |
| 793 Width: width, | 781 Width: width, |
| 794 Height: height, | 782 Height: height, |
| 795 GPU: gpu, | |
| 796 Source: source, | 783 Source: source, |
| 797 } | 784 } |
| 798 resp, err := json.Marshal(m) | 785 resp, err := json.Marshal(m) |
| 799 if err != nil { | 786 if err != nil { |
| 800 reportError(w, r, err, "Failed to serialize a response.") | 787 reportError(w, r, err, "Failed to serialize a response.") |
| 801 return | 788 return |
| 802 } | 789 } |
| 803 w.Header().Set("Content-Type", "application/json") | 790 w.Header().Set("Content-Type", "application/json") |
| 804 w.Write(resp) | 791 w.Write(resp) |
| 805 } | 792 } |
| 806 | 793 |
| 807 func cleanCompileOutput(s, hash string) string { | 794 func cleanCompileOutput(s, hash string) string { |
| 808 old := "../../../cache/src/" + hash + ".cpp:" | 795 old := "../../../cache/src/" + hash + ".cpp:" |
| 809 log.Printf("INFO: replacing %q\n", old) | 796 log.Printf("INFO: replacing %q\n", old) |
| 810 return strings.Replace(s, old, "usercode.cpp:", -1) | 797 return strings.Replace(s, old, "usercode.cpp:", -1) |
| 811 } | 798 } |
| 812 | 799 |
| 813 // mainHandler handles the GET and POST of the main page. | 800 // mainHandler handles the GET and POST of the main page. |
| 814 func mainHandler(w http.ResponseWriter, r *http.Request) { | 801 func mainHandler(w http.ResponseWriter, r *http.Request) { |
| 815 log.Printf("Main Handler: %q\n", r.URL.Path) | 802 log.Printf("Main Handler: %q\n", r.URL.Path) |
| 816 requestsCounter.Inc(1) | 803 requestsCounter.Inc(1) |
| 817 if r.Method == "GET" { | 804 if r.Method == "GET" { |
| 818 code := DEFAULT_SAMPLE | 805 code := DEFAULT_SAMPLE |
| 819 source := 0 | 806 source := 0 |
| 820 width := 256 | 807 width := 256 |
| 821 height := 256 | 808 height := 256 |
| 822 gpu := false | |
| 823 match := directLink.FindStringSubmatch(r.URL.Path) | 809 match := directLink.FindStringSubmatch(r.URL.Path) |
| 824 var hash string | 810 var hash string |
| 825 if len(match) == 2 && r.URL.Path != "/" { | 811 if len(match) == 2 && r.URL.Path != "/" { |
| 826 hash = match[1] | 812 hash = match[1] |
| 827 if db == nil { | 813 if db == nil { |
| 828 http.NotFound(w, r) | 814 http.NotFound(w, r) |
| 829 return | 815 return |
| 830 } | 816 } |
| 831 // Update 'code' with the code found in the database. | 817 // Update 'code' with the code found in the database. |
| 832 » » » if err := db.QueryRow("SELECT code, width, height, gpu,
source_image_id FROM webtry WHERE hash=?", hash).Scan(&code, &width, &height, &g
pu, &source); err != nil { | 818 » » » if err := db.QueryRow("SELECT code, width, height, sourc
e_image_id FROM webtry WHERE hash=?", hash).Scan(&code, &width, &height, &source
); err != nil { |
| 833 http.NotFound(w, r) | 819 http.NotFound(w, r) |
| 834 return | 820 return |
| 835 } | 821 } |
| 836 } | 822 } |
| 837 // Expand the template. | 823 // Expand the template. |
| 838 w.Header().Set("Content-Type", "text/html") | 824 w.Header().Set("Content-Type", "text/html") |
| 839 » » if err := indexTemplate.Execute(w, userCode{Code: code, Hash: ha
sh, Source: source, Width: width, Height: height, GPU: gpu, Titlebar: Titlebar{G
itHash: gitHash, GitInfo: gitInfo}}); err != nil { | 825 » » if err := indexTemplate.Execute(w, userCode{Code: code, Hash: ha
sh, Source: source, Width: width, Height: height, Titlebar: Titlebar{GitHash: gi
tHash, GitInfo: gitInfo}}); err != nil { |
| 840 log.Printf("ERROR: Failed to expand template: %q\n", err
) | 826 log.Printf("ERROR: Failed to expand template: %q\n", err
) |
| 841 } | 827 } |
| 842 } else if r.Method == "POST" { | 828 } else if r.Method == "POST" { |
| 843 w.Header().Set("Content-Type", "application/json") | 829 w.Header().Set("Content-Type", "application/json") |
| 844 buf := bytes.NewBuffer(make([]byte, 0, MAX_TRY_SIZE)) | 830 buf := bytes.NewBuffer(make([]byte, 0, MAX_TRY_SIZE)) |
| 845 n, err := buf.ReadFrom(r.Body) | 831 n, err := buf.ReadFrom(r.Body) |
| 846 if err != nil { | 832 if err != nil { |
| 847 reportTryError(w, r, err, "Failed to read a request body
.", "") | 833 reportTryError(w, r, err, "Failed to read a request body
.", "") |
| 848 return | 834 return |
| 849 } | 835 } |
| 850 if n == MAX_TRY_SIZE { | 836 if n == MAX_TRY_SIZE { |
| 851 err := fmt.Errorf("Code length equal to, or exceeded, %d
", MAX_TRY_SIZE) | 837 err := fmt.Errorf("Code length equal to, or exceeded, %d
", MAX_TRY_SIZE) |
| 852 reportTryError(w, r, err, "Code too large.", "") | 838 reportTryError(w, r, err, "Code too large.", "") |
| 853 return | 839 return |
| 854 } | 840 } |
| 855 request := TryRequest{} | 841 request := TryRequest{} |
| 856 if err := json.Unmarshal(buf.Bytes(), &request); err != nil { | 842 if err := json.Unmarshal(buf.Bytes(), &request); err != nil { |
| 857 reportTryError(w, r, err, "Coulnd't decode JSON.", "") | 843 reportTryError(w, r, err, "Coulnd't decode JSON.", "") |
| 858 return | 844 return |
| 859 } | 845 } |
| 860 if hasPreProcessor(request.Code) { | 846 if hasPreProcessor(request.Code) { |
| 861 err := fmt.Errorf("Found preprocessor macro in code.") | 847 err := fmt.Errorf("Found preprocessor macro in code.") |
| 862 reportTryError(w, r, err, "Preprocessor macros aren't al
lowed.", "") | 848 reportTryError(w, r, err, "Preprocessor macros aren't al
lowed.", "") |
| 863 return | 849 return |
| 864 } | 850 } |
| 865 » » hash, err := expandCode(LineNumbers(request.Code), request.Sourc
e, request.Width, request.Height, request.GPU) | 851 » » hash, err := expandCode(LineNumbers(request.Code), request.Sourc
e, request.Width, request.Height) |
| 866 if err != nil { | 852 if err != nil { |
| 867 reportTryError(w, r, err, "Failed to write the code to c
ompile.", hash) | 853 reportTryError(w, r, err, "Failed to write the code to c
ompile.", hash) |
| 868 return | 854 return |
| 869 } | 855 } |
| 870 » » writeToDatabase(hash, request.Code, request.Name, request.Source
, request.Width, request.Height, request.GPU) | 856 » » writeToDatabase(hash, request.Code, request.Name, request.Source
, request.Width, request.Height) |
| 871 err = expandGyp(hash) | 857 err = expandGyp(hash) |
| 872 if err != nil { | 858 if err != nil { |
| 873 reportTryError(w, r, err, "Failed to write the gyp file.
", hash) | 859 reportTryError(w, r, err, "Failed to write the gyp file.
", hash) |
| 874 return | 860 return |
| 875 } | 861 } |
| 876 cmd := fmt.Sprintf("scripts/fiddle_wrapper %s --width %d --heigh
t %d", hash, request.Width, request.Height) | 862 cmd := fmt.Sprintf("scripts/fiddle_wrapper %s --width %d --heigh
t %d", hash, request.Width, request.Height) |
| 877 if request.GPU { | |
| 878 cmd += " --gpu" | |
| 879 } | |
| 880 if *useChroot { | 863 if *useChroot { |
| 881 cmd = "schroot -c webtry --directory=/ -- /skia_build/sk
ia/experimental/webtry/" + cmd | 864 cmd = "schroot -c webtry --directory=/ -- /skia_build/sk
ia/experimental/webtry/" + cmd |
| 882 } | 865 } |
| 883 if request.Source > 0 { | 866 if request.Source > 0 { |
| 884 cmd += fmt.Sprintf(" --source image-%d.png", request.Sou
rce) | 867 cmd += fmt.Sprintf(" --source image-%d.png", request.Sou
rce) |
| 885 } | 868 } |
| 886 | 869 |
| 887 message, err := doCmd(cmd) | 870 message, err := doCmd(cmd) |
| 888 if err != nil { | 871 if err != nil { |
| 889 reportTryError(w, r, err, "Failed to run the code:\n"+me
ssage, hash) | 872 reportTryError(w, r, err, "Failed to run the code:\n"+me
ssage, hash) |
| (...skipping 30 matching lines...) Expand all Loading... |
| 920 http.HandleFunc("/sources/", autogzip.HandleFunc(sourcesHandler)) | 903 http.HandleFunc("/sources/", autogzip.HandleFunc(sourcesHandler)) |
| 921 | 904 |
| 922 // Resources are served directly | 905 // Resources are served directly |
| 923 // TODO add support for caching/etags/gzip | 906 // TODO add support for caching/etags/gzip |
| 924 http.Handle("/res/", autogzip.Handle(http.FileServer(http.Dir("./")))) | 907 http.Handle("/res/", autogzip.Handle(http.FileServer(http.Dir("./")))) |
| 925 | 908 |
| 926 // TODO Break out /c/ as it's own handler. | 909 // TODO Break out /c/ as it's own handler. |
| 927 http.HandleFunc("/", autogzip.HandleFunc(mainHandler)) | 910 http.HandleFunc("/", autogzip.HandleFunc(mainHandler)) |
| 928 log.Fatal(http.ListenAndServe(*port, nil)) | 911 log.Fatal(http.ListenAndServe(*port, nil)) |
| 929 } | 912 } |
| OLD | NEW |