forked from ebhomengo/niki
265 lines
5.4 KiB
Go
265 lines
5.4 KiB
Go
package gofakeit
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
img "image"
|
|
imgCol "image/color"
|
|
"image/jpeg"
|
|
"image/png"
|
|
"math/rand"
|
|
"strconv"
|
|
)
|
|
|
|
// ImageURL will generate a random Image Based Upon Height And Width. https://picsum.photos/
|
|
|
|
func ImageURL(width int, height int) string { return imageURL(globalFaker.Rand, width, height) }
|
|
|
|
// ImageURL will generate a random Image Based Upon Height And Width. https://picsum.photos/
|
|
|
|
func (f *Faker) ImageURL(width int, height int) string { return imageURL(f.Rand, width, height) }
|
|
|
|
func imageURL(r *rand.Rand, width int, height int) string {
|
|
|
|
return "https://picsum.photos/" + strconv.Itoa(width) + "/" + strconv.Itoa(height)
|
|
|
|
}
|
|
|
|
// Image generates a random rgba image
|
|
|
|
func Image(width int, height int) *img.RGBA { return image(globalFaker.Rand, width, height) }
|
|
|
|
// Image generates a random rgba image
|
|
|
|
func (f *Faker) Image(width int, height int) *img.RGBA { return image(f.Rand, width, height) }
|
|
|
|
func image(r *rand.Rand, width int, height int) *img.RGBA {
|
|
|
|
upLeft := img.Point{0, 0}
|
|
|
|
lowRight := img.Point{width, height}
|
|
|
|
img := img.NewRGBA(img.Rectangle{upLeft, lowRight})
|
|
|
|
// Set color for each pixel
|
|
|
|
for x := 0; x < width; x++ {
|
|
|
|
for y := 0; y < height; y++ {
|
|
|
|
img.Set(x, y, imgCol.RGBA{uint8(number(r, 0, 255)), uint8(number(r, 0, 255)), uint8(number(r, 0, 255)), 0xff})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return img
|
|
|
|
}
|
|
|
|
// ImageJpeg generates a random rgba jpeg image
|
|
|
|
func ImageJpeg(width int, height int) []byte { return imageJpeg(globalFaker.Rand, width, height) }
|
|
|
|
// ImageJpeg generates a random rgba jpeg image
|
|
|
|
func (f *Faker) ImageJpeg(width int, height int) []byte { return imageJpeg(f.Rand, width, height) }
|
|
|
|
func imageJpeg(r *rand.Rand, width int, height int) []byte {
|
|
|
|
buf := new(bytes.Buffer)
|
|
|
|
jpeg.Encode(buf, image(r, width, height), nil)
|
|
|
|
return buf.Bytes()
|
|
|
|
}
|
|
|
|
// ImagePng generates a random rgba png image
|
|
|
|
func ImagePng(width int, height int) []byte { return imagePng(globalFaker.Rand, width, height) }
|
|
|
|
// ImagePng generates a random rgba png image
|
|
|
|
func (f *Faker) ImagePng(width int, height int) []byte { return imagePng(f.Rand, width, height) }
|
|
|
|
func imagePng(r *rand.Rand, width int, height int) []byte {
|
|
|
|
buf := new(bytes.Buffer)
|
|
|
|
png.Encode(buf, image(r, width, height))
|
|
|
|
return buf.Bytes()
|
|
|
|
}
|
|
|
|
func addImageLookup() {
|
|
|
|
AddFuncLookup("imageurl", Info{
|
|
|
|
Display: "Image URL",
|
|
|
|
Category: "image",
|
|
|
|
Description: "Web address pointing to an image file that can be accessed and displayed online",
|
|
|
|
Example: "https://picsum.photos/500/500",
|
|
|
|
Output: "string",
|
|
|
|
Params: []Param{
|
|
|
|
{Field: "width", Display: "Width", Type: "int", Default: "500", Description: "Image width in px"},
|
|
|
|
{Field: "height", Display: "Height", Type: "int", Default: "500", Description: "Image height in px"},
|
|
},
|
|
|
|
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
|
|
|
|
width, err := info.GetInt(m, "width")
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if width < 10 || width >= 1000 {
|
|
|
|
return nil, errors.New("invalid image width, must be greater than 10, less than 1000")
|
|
|
|
}
|
|
|
|
height, err := info.GetInt(m, "height")
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if height < 10 || height >= 1000 {
|
|
|
|
return nil, errors.New("invalid image height, must be greater than 10, less than 1000")
|
|
|
|
}
|
|
|
|
return imageURL(r, width, height), nil
|
|
|
|
},
|
|
})
|
|
|
|
AddFuncLookup("imagejpeg", Info{
|
|
|
|
Display: "Image JPEG",
|
|
|
|
Category: "image",
|
|
|
|
Description: "Image file format known for its efficient compression and compatibility",
|
|
|
|
Example: "file.jpeg - bytes",
|
|
|
|
Output: "[]byte",
|
|
|
|
ContentType: "image/jpeg",
|
|
|
|
Params: []Param{
|
|
|
|
{Field: "width", Display: "Width", Type: "int", Default: "500", Description: "Image width in px"},
|
|
|
|
{Field: "height", Display: "Height", Type: "int", Default: "500", Description: "Image height in px"},
|
|
},
|
|
|
|
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
|
|
|
|
width, err := info.GetInt(m, "width")
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if width < 10 || width >= 1000 {
|
|
|
|
return nil, errors.New("invalid image width, must be greater than 10, less than 1000")
|
|
|
|
}
|
|
|
|
height, err := info.GetInt(m, "height")
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if height < 10 || height >= 1000 {
|
|
|
|
return nil, errors.New("invalid image height, must be greater than 10, less than 1000")
|
|
|
|
}
|
|
|
|
return imageJpeg(r, width, height), nil
|
|
|
|
},
|
|
})
|
|
|
|
AddFuncLookup("imagepng", Info{
|
|
|
|
Display: "Image PNG",
|
|
|
|
Category: "image",
|
|
|
|
Description: "Image file format known for its lossless compression and support for transparency",
|
|
|
|
Example: "file.png - bytes",
|
|
|
|
Output: "[]byte",
|
|
|
|
ContentType: "image/png",
|
|
|
|
Params: []Param{
|
|
|
|
{Field: "width", Display: "Width", Type: "int", Default: "500", Description: "Image width in px"},
|
|
|
|
{Field: "height", Display: "Height", Type: "int", Default: "500", Description: "Image height in px"},
|
|
},
|
|
|
|
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
|
|
|
|
width, err := info.GetInt(m, "width")
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if width < 10 || width >= 1000 {
|
|
|
|
return nil, errors.New("invalid image width, must be greater than 10, less than 1000")
|
|
|
|
}
|
|
|
|
height, err := info.GetInt(m, "height")
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if height < 10 || height >= 1000 {
|
|
|
|
return nil, errors.New("invalid image height, must be greater than 10, less than 1000")
|
|
|
|
}
|
|
|
|
return imagePng(r, width, height), nil
|
|
|
|
},
|
|
})
|
|
|
|
}
|