Add unsplash
continuous-integration/drone/push Build is passing
Details
continuous-integration/drone/push Build is passing
Details
parent
f6e8ce25b3
commit
53caadf9d9
GPG Key ID:
1669C533B8CF6D89
@ -0,0 +1,273 @@
|
||||
package unsplash
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/base64"
|
||||
"image"
|
||||
_ "image/gif"
|
||||
"image/jpeg"
|
||||
_ "image/png"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"math/rand"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"git.fromouter.space/crunchy-rocks/clessy-ng/modules"
|
||||
"git.fromouter.space/crunchy-rocks/clessy-ng/utils"
|
||||
|
||||
"git.fromouter.space/crunchy-rocks/draw2d"
|
||||
"git.fromouter.space/crunchy-rocks/draw2d/draw2dimg"
|
||||
"git.fromouter.space/crunchy-rocks/emoji"
|
||||
"git.fromouter.space/crunchy-rocks/freetype"
|
||||
"git.fromouter.space/hamcha/tg"
|
||||
"github.com/disintegration/imaging"
|
||||
)
|
||||
|
||||
var quoteFontData draw2d.FontData
|
||||
var pics []string
|
||||
|
||||
type Module struct {
|
||||
client *tg.Telegram
|
||||
name string
|
||||
emojis emoji.Table
|
||||
}
|
||||
|
||||
func (m *Module) Initialize(options modules.ModuleOptions) error {
|
||||
m.client = options.API
|
||||
m.name = options.Name
|
||||
m.emojis = options.Emojis
|
||||
|
||||
fontfile := utils.RequireEnv("CLESSY_UNSPLASH_FONT")
|
||||
bytes, err := os.ReadFile(fontfile)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
font, err := freetype.ParseFont(bytes)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
quoteFontData = draw2d.FontData{
|
||||
Name: "gillmt",
|
||||
Family: draw2d.FontFamilySans,
|
||||
Style: draw2d.FontStyleBold,
|
||||
}
|
||||
draw2d.RegisterFont(quoteFontData, font)
|
||||
|
||||
// Read all the pictures inside a folder and save them for later
|
||||
bgpath := utils.RequireEnv("CLESSY_UNSPLASH_BG_PATH")
|
||||
files, err := ioutil.ReadDir(bgpath)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, file := range files {
|
||||
pics = append(pics, filepath.Join(bgpath, file.Name()))
|
||||
}
|
||||
|
||||
log.Printf("[unsplash] Loaded (%d available backgrounds)", len(pics))
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Module) OnUpdate(update tg.APIUpdate) {
|
||||
// Not a message? Ignore
|
||||
if update.Message == nil {
|
||||
return
|
||||
}
|
||||
message := *update.Message
|
||||
|
||||
if utils.IsCommand(message, m.name, "unsplash") {
|
||||
text := ""
|
||||
user := message.User
|
||||
|
||||
if message.ReplyTo != nil {
|
||||
switch {
|
||||
case message.ReplyTo.Text != nil:
|
||||
text = *(message.ReplyTo.Text)
|
||||
case message.ReplyTo.Caption != nil:
|
||||
text = *(message.ReplyTo.Caption)
|
||||
default:
|
||||
m.client.SendTextMessage(tg.ClientTextMessageData{
|
||||
ChatID: message.Chat.ChatID,
|
||||
Text: "Non c'e' niente di 'ispiratore' in questo..",
|
||||
ReplyID: &message.MessageID,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// For forwarded message take the original user
|
||||
if message.FwdUser != nil {
|
||||
user = *message.FwdUser
|
||||
} else {
|
||||
user = message.ReplyTo.User
|
||||
}
|
||||
} else {
|
||||
if strings.Index(*(message.Text), " ") > 0 {
|
||||
text = strings.TrimSpace(strings.SplitN(*(message.Text), " ", 2)[1])
|
||||
}
|
||||
}
|
||||
|
||||
// Cleanup chars
|
||||
text = strings.Map(stripUnreadable, text)
|
||||
|
||||
author := user.FirstName
|
||||
if user.LastName != "" {
|
||||
author = user.FirstName + " " + user.LastName
|
||||
}
|
||||
author += " (" + user.Username + ")"
|
||||
|
||||
if strings.TrimSpace(text) == "" {
|
||||
m.client.SendTextMessage(tg.ClientTextMessageData{
|
||||
ChatID: message.Chat.ChatID,
|
||||
Text: "Non c'e' niente di 'ispiratore' in questo..",
|
||||
ReplyID: &message.MessageID,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
file, err := os.Open(pics[rand.Intn(len(pics))])
|
||||
if err != nil {
|
||||
log.Printf("[unsplash] Could not open original image file: %s\n", err.Error())
|
||||
m.client.SendTextMessage(tg.ClientTextMessageData{
|
||||
ChatID: message.Chat.ChatID,
|
||||
Text: "<b>ERRORE!</b> @hamcha controlla la console!",
|
||||
ReplyID: &message.MessageID,
|
||||
})
|
||||
return
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
img, _, err := image.Decode(file)
|
||||
if err != nil {
|
||||
log.Printf("[unsplash] Image decode error: %s\n", err.Error())
|
||||
m.client.SendTextMessage(tg.ClientTextMessageData{
|
||||
ChatID: message.Chat.ChatID,
|
||||
Text: "<b>ERRORE!</b> @hamcha controlla la console!",
|
||||
ReplyID: &message.MessageID,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
m.client.SendChatAction(tg.ClientChatActionData{
|
||||
ChatID: message.Chat.ChatID,
|
||||
Action: tg.ActionUploadingPhoto,
|
||||
})
|
||||
|
||||
// Darken image
|
||||
img = imaging.AdjustBrightness(imaging.AdjustGamma(imaging.AdjustSigmoid(img, 0.5, -6.0), 0.8), -20)
|
||||
|
||||
// Create target image
|
||||
bounds := img.Bounds()
|
||||
iwidth := float64(bounds.Size().X)
|
||||
iheight := float64(bounds.Size().Y)
|
||||
|
||||
timg := image.NewRGBA(bounds)
|
||||
gc := draw2dimg.NewGraphicContext(timg)
|
||||
gc.Emojis = m.emojis
|
||||
gc.SetFontData(quoteFontData)
|
||||
gc.DrawImage(img)
|
||||
gc.SetStrokeColor(image.Black)
|
||||
gc.SetFillColor(image.White)
|
||||
|
||||
text = strings.ToUpper(strings.TrimSpace(text))
|
||||
gc.Restore()
|
||||
gc.Save()
|
||||
|
||||
// Detect appropriate font size
|
||||
scale := iheight / iwidth * (iwidth / 10) * 0.8
|
||||
gc.SetFontSize(scale)
|
||||
gc.SetLineWidth(scale / 15)
|
||||
|
||||
// Get NEW bounds
|
||||
left, top, right, bottom := gc.GetStringBounds(text)
|
||||
|
||||
width := right - left
|
||||
texts := []string{text}
|
||||
if width*1.2 > iwidth {
|
||||
// Split text
|
||||
texts = utils.SplitCenter(text)
|
||||
|
||||
// Get longest line
|
||||
longer := float64(0)
|
||||
longid := 0
|
||||
widths := make([]float64, len(texts))
|
||||
for id := range texts {
|
||||
tleft, _, tright, _ := gc.GetStringBounds(texts[id])
|
||||
widths[id] = tright - tleft
|
||||
if width > longer {
|
||||
longer = widths[id]
|
||||
longid = id
|
||||
}
|
||||
}
|
||||
|
||||
// Still too big? Decrease font size again
|
||||
iter := 0
|
||||
for width*1.2 > iwidth && iter < 10 {
|
||||
scale *= (0.9 - 0.05*float64(iter))
|
||||
gc.SetFontSize(scale)
|
||||
left, top, right, bottom = gc.GetStringBounds(texts[longid])
|
||||
width = right - left
|
||||
iter++
|
||||
}
|
||||
}
|
||||
|
||||
texts = append(texts, author)
|
||||
height := bottom - top + 20
|
||||
margin := float64(height / 50)
|
||||
txtheight := (height + margin) * float64(len(texts))
|
||||
|
||||
gc.Save()
|
||||
for id, txt := range texts {
|
||||
gc.Save()
|
||||
left, _, right, _ = gc.GetStringBounds(txt)
|
||||
width = right - left
|
||||
|
||||
x := (iwidth - width) / 2
|
||||
y := (iheight-txtheight)/2 + (height+margin*2)*float64(id+1)
|
||||
if id == len(texts)-1 {
|
||||
gc.SetFontSize(scale * 0.7)
|
||||
left, _, right, _ = gc.GetStringBounds(txt)
|
||||
width = right - left
|
||||
x = (iwidth - width) / 1.5
|
||||
y = (iheight-txtheight)/2 + (height+margin)*float64(id+1) + margin*6
|
||||
}
|
||||
|
||||
gc.Translate(x, y)
|
||||
gc.StrokeString(txt)
|
||||
gc.FillString(txt)
|
||||
gc.Restore()
|
||||
}
|
||||
|
||||
buf := new(bytes.Buffer)
|
||||
err = jpeg.Encode(buf, timg, &(jpeg.Options{Quality: 80}))
|
||||
if err != nil {
|
||||
log.Printf("[unsplash] Image encode error: %s\n", err.Error())
|
||||
m.client.SendTextMessage(tg.ClientTextMessageData{
|
||||
ChatID: message.Chat.ChatID,
|
||||
Text: "<b>ERRORE!</b> @hamcha controlla la console!",
|
||||
ReplyID: &message.MessageID,
|
||||
})
|
||||
return
|
||||
}
|
||||
m.client.SendPhoto(tg.ClientPhotoData{
|
||||
ChatID: message.Chat.ChatID,
|
||||
Bytes: base64.StdEncoding.EncodeToString(buf.Bytes()),
|
||||
Filename: "quote.jpg",
|
||||
ReplyID: &message.MessageID,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func stripUnreadable(r rune) rune {
|
||||
if r == '\n' || r == '\t' {
|
||||
return ' '
|
||||
}
|
||||
if r < 32 {
|
||||
return -1
|
||||
}
|
||||
return r
|
||||
}
|
||||
@ -0,0 +1,32 @@
|
||||
package utils
|
||||
|
||||
import "strings"
|
||||
|
||||
func abs(i int) int {
|
||||
if i < 0 {
|
||||
return -i
|
||||
}
|
||||
return i
|
||||
}
|
||||
|
||||
func SplitCenter(text string) []string {
|
||||
centerIndex := int(len(text) / 2)
|
||||
whitespaceFrontIndex := strings.IndexRune(text[centerIndex:], ' ')
|
||||
whitespaceBackIndex := strings.LastIndex(text[:centerIndex], " ")
|
||||
frontIndex := centerIndex + whitespaceFrontIndex
|
||||
frontDiff := 9999
|
||||
backDiff := 9999
|
||||
if whitespaceFrontIndex < 1 && whitespaceBackIndex < 1 {
|
||||
return []string{text}
|
||||
}
|
||||
if whitespaceFrontIndex >= 0 {
|
||||
frontDiff = len(text[:frontIndex]) - len(text[frontIndex:])
|
||||
}
|
||||
if whitespaceBackIndex >= 0 {
|
||||
backDiff = len(text[:whitespaceBackIndex]) - len(text[whitespaceBackIndex:])
|
||||
}
|
||||
if abs(frontDiff) < abs(backDiff) {
|
||||
return []string{strings.TrimSpace(text[:frontIndex]), strings.TrimSpace(text[frontIndex:])}
|
||||
}
|
||||
return []string{strings.TrimSpace(text[:whitespaceBackIndex]), strings.TrimSpace(text[whitespaceBackIndex:])}
|
||||
}
|
||||
Loading…
Reference in New Issue