@@ -2,8 +2,10 @@ package main
22
33import (
44 "fmt"
5+ "image"
6+ "image/color"
7+ "image/png"
58 "log"
6- "math/rand"
79 "net/http"
810 "os"
911 "os/signal"
@@ -29,12 +31,7 @@ func main() {
2931 Value : 8088 ,
3032 Aliases : []string {"p" },
3133 },
32- & cli.StringFlag {
33- Name : FILE_URL ,
34- Usage : "Specifies the url to serve the file at." ,
35- Value : "/test.txt" ,
36- Aliases : []string {"f" },
37- },
34+
3835 & cli.UintFlag {
3936 Name : FILE_LENGTH ,
4037 Usage : "Specifies the length of the file to serve." ,
@@ -44,17 +41,22 @@ func main() {
4441 },
4542 Action : func (c * cli.Context ) error {
4643 const (
47- fileName = "test.txt "
44+ fileName = "test.png "
4845 )
4946
5047 fileContent := generateFileData (c .Int (FILE_LENGTH ))
5148 filePath , cleanup := setupFile (fileName , fileContent )
5249 defer cleanup ()
5350
54- http .HandleFunc (c . String ( FILE_URL ) , func (w http.ResponseWriter , r * http.Request ) {
51+ http .HandleFunc ("/" , func (w http.ResponseWriter , r * http.Request ) {
5552 // Set the Content-Disposition header to suggest a filename
5653 w .Header ().Set ("Content-Disposition" , fmt .Sprintf ("attachment; filename=\" %s\" " , fileName ))
5754
55+ // Add CORS headers to allow all origins (*).
56+ w .Header ().Set ("Access-Control-Allow-Origin" , "*" )
57+ w .Header ().Set ("Access-Control-Allow-Headers" , "*" )
58+ w .Header ().Set ("Access-Control-Expose-Headers" , "*" )
59+
5860 http .ServeFile (w , r , filePath )
5961 })
6062
@@ -81,7 +83,7 @@ func waitForKillSignal() {
8183}
8284
8385// setupFile creates a file with the given name and content, and returns a cleanup function
84- func setupFile (fileName string , fileContent string ) (string , func ()) {
86+ func setupFile (fileName string , fileContent * image. RGBA ) (string , func ()) {
8587 dataFolder , err := os .MkdirTemp ("" , "sample-file-server-*" )
8688 if err != nil {
8789 panic (err )
@@ -94,11 +96,12 @@ func setupFile(fileName string, fileContent string) (string, func()) {
9496 }
9597 defer file .Close ()
9698
97- _ , err = file . WriteString ( fileContent )
98- if err != nil {
99+ if err := png . Encode ( file , fileContent ); err != nil {
100+ fmt . Println ( "Failed to encode image:" , err )
99101 os .Remove (filePath )
100102 panic (err )
101103 }
104+
102105 return filePath , func () {
103106 err := os .Remove (fileName )
104107 if err != nil {
@@ -108,22 +111,26 @@ func setupFile(fileName string, fileContent string) (string, func()) {
108111}
109112
110113// generateFileData generates a string of the given length composed of random words
111- func generateFileData (length int ) (fileData string ) {
112- if length < 10 {
113- panic ("file length must be at least 10" )
114- }
115- wordSelection := []string {
116- "Alpha" , "Bravo" , "Charlie" , "Delta" , "Echo" , "Foxtrot" , "Golf" , "Hotel" ,
117- "India" , "Juliet" , "Kilo" , "Lima" , "Mike" , "November" , "Oscar" , "Papa" ,
118- "Quebec" , "Romeo" , "Sierra" , "Tango" , "Uniform" , "Victor" , "Whiskey" ,
119- "X-ray" , "Yankee" , "Zulu" ,
120- }
121- fileData = "START"
122- // Continue adding words until we reach the desired length or beyond
123- for len (fileData ) < length {
124- randomIndex := rand .Intn (len (wordSelection ))
125- fileData = fileData + " " + wordSelection [randomIndex ]
114+ func generateFileData (length int ) (img * image.RGBA ) {
115+ // Define image dimensions
116+ width , height := length , length
117+
118+ // Create an empty RGBA image
119+ img = image .NewRGBA (image .Rect (0 , 0 , width , height ))
120+
121+ // Fill the image with a gradient
122+ for y := 0 ; y < height ; y ++ {
123+ for x := 0 ; x < width ; x ++ {
124+ // Gradient: Horizontal red, Vertical blue
125+ c := color.RGBA {
126+ R : uint8 (x * 255 / width ),
127+ B : uint8 (y * 255 / height ),
128+ G : 0 ,
129+ A : 255 , // Fully opaque
130+ }
131+ img .Set (x , y , c )
132+ }
126133 }
127134
128- return fileData [: length - 3 ] + "END"
135+ return img
129136}
0 commit comments