mirror of
https://github.com/idanoo/gomatrixbot
synced 2025-07-01 08:02:21 +00:00
Fixed?
This commit is contained in:
parent
be94d805cf
commit
b492111b82
4 changed files with 77 additions and 59 deletions
|
@ -26,14 +26,19 @@ func (db *Database) DeleteAllPics() {
|
|||
}
|
||||
|
||||
// StorePic
|
||||
func (db *Database) StorePic(filename string) error {
|
||||
func (db *Database) StorePic(filename string) (int64, error) {
|
||||
min := db.getPicMinViewCount()
|
||||
_, err := db.db.Exec(
|
||||
exec, err := db.db.Exec(
|
||||
"INSERT INTO `pics` (`filename`, `viewed`) VALUES (?,?)",
|
||||
filename,
|
||||
min,
|
||||
)
|
||||
return err
|
||||
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return exec.LastInsertId()
|
||||
}
|
||||
|
||||
// DeletePic
|
||||
|
@ -48,9 +53,9 @@ func (db *Database) ResetPics() {
|
|||
}
|
||||
|
||||
// GetAllPics
|
||||
func (db *Database) GetAllPics() ([]string, error) {
|
||||
var pics []string
|
||||
row, err := db.db.Query("SELECT filename FROM pics")
|
||||
func (db *Database) GetAllPics() (map[int64]string, error) {
|
||||
pics := make(map[int64]string)
|
||||
row, err := db.db.Query("SELECT id, filename FROM pics")
|
||||
if err != nil {
|
||||
return pics, err
|
||||
}
|
||||
|
@ -58,8 +63,9 @@ func (db *Database) GetAllPics() ([]string, error) {
|
|||
defer row.Close()
|
||||
for row.Next() {
|
||||
var pic string
|
||||
row.Scan(&pic)
|
||||
pics = append(pics, pic)
|
||||
var id int64
|
||||
row.Scan(&id, &pic)
|
||||
pics[id] = pic
|
||||
}
|
||||
|
||||
return pics, nil
|
||||
|
|
|
@ -58,51 +58,50 @@ func (mtrx *MtrxClient) postBirb(ctx context.Context, evt *event.Event) {
|
|||
})
|
||||
}
|
||||
|
||||
func (mtrx *MtrxClient) postBirbWithText(ctx context.Context, evt *event.Event, text string, attempt int, filename string) {
|
||||
idd, birb, err := mtrx.db.GetLastUsedPic()
|
||||
if err != nil {
|
||||
mtrx.c.Log.Err(err).Msg("Failed to get birb")
|
||||
return
|
||||
}
|
||||
func (mtrx *MtrxClient) postBirbWithText(ctx context.Context, evt *event.Event, text string, filename string) {
|
||||
var reader *os.File
|
||||
var err error
|
||||
var birb string
|
||||
|
||||
if filename != "" {
|
||||
birb = filename
|
||||
}
|
||||
|
||||
reader, err := os.Open(BIRB_PATH + "/" + birb)
|
||||
if err != nil {
|
||||
mtrx.sendMessage(ctx, evt.RoomID, "Failed to load image "+err.Error()+". "+birb)
|
||||
mtrx.db.DeletePic(idd)
|
||||
if attempt > 3 {
|
||||
mtrx.db.DeletePic(idd)
|
||||
if filename == "" {
|
||||
// Do a lookup
|
||||
dbId, birb, err := mtrx.db.GetLastUsedPic()
|
||||
if err != nil {
|
||||
mtrx.sendMessage(ctx, evt.RoomID, "Failed to get an image "+err.Error()+". "+birb)
|
||||
return
|
||||
}
|
||||
reader, err = os.Open(BIRB_PATH + "/" + birb)
|
||||
if err != nil {
|
||||
mtrx.sendMessage(ctx, evt.RoomID, "Failed to load image "+err.Error()+". "+birb)
|
||||
mtrx.db.DeletePic(dbId)
|
||||
delete(mtrx.pics, dbId)
|
||||
return
|
||||
}
|
||||
} else {
|
||||
birb = filename
|
||||
reader, err = os.Open(BIRB_PATH + "/" + birb)
|
||||
if err != nil {
|
||||
mtrx.sendMessage(ctx, evt.RoomID, "Failed to load image "+err.Error()+". "+birb)
|
||||
var id int64
|
||||
for k, v := range mtrx.pics {
|
||||
if v == birb {
|
||||
id = k
|
||||
break
|
||||
}
|
||||
}
|
||||
mtrx.db.DeletePic(id)
|
||||
delete(mtrx.pics, id)
|
||||
return
|
||||
}
|
||||
|
||||
// Retry
|
||||
mtrx.postBirbWithText(ctx, evt, text, attempt+1, "")
|
||||
return
|
||||
}
|
||||
|
||||
// do the swaps
|
||||
img, _, err := image.Decode(reader)
|
||||
if err != nil {
|
||||
mtrx.db.DeletePic(idd)
|
||||
if attempt > 3 {
|
||||
mtrx.db.DeletePic(idd)
|
||||
mtrx.sendMessage(ctx, evt.RoomID, "I can't read this image. I give up. "+err.Error()+". "+birb)
|
||||
return
|
||||
}
|
||||
|
||||
// Retry
|
||||
mtrx.postBirbWithText(ctx, evt, text, attempt+1, "")
|
||||
mtrx.sendMessage(ctx, evt.RoomID, "I can't read this image. I give up. "+err.Error()+". "+birb)
|
||||
return
|
||||
|
||||
}
|
||||
|
||||
// r := img.Bounds()
|
||||
// w := r.Dx() // w
|
||||
// h := r.Dy() // h
|
||||
|
||||
imgNew := resize.Resize(1024, 1024, img, resize.Lanczos2)
|
||||
m := gg.NewContextForImage(imgNew)
|
||||
|
||||
|
@ -192,20 +191,31 @@ func (mtrx *MtrxClient) UpdatePics() error {
|
|||
}
|
||||
|
||||
for _, file := range files {
|
||||
found := false
|
||||
|
||||
if file.IsDir() || strings.HasPrefix(file.Name(), ".") {
|
||||
continue
|
||||
}
|
||||
|
||||
if contains(mtrx.pics, file.Name()) {
|
||||
// check if in our map
|
||||
for _, v := range mtrx.pics {
|
||||
if v == file.Name() {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if found {
|
||||
continue
|
||||
}
|
||||
|
||||
err = mtrx.db.StorePic(file.Name())
|
||||
idd, err := mtrx.db.StorePic(file.Name())
|
||||
if err != nil {
|
||||
mtrx.c.Log.Err(err).Msg("Failed to store pic")
|
||||
return err
|
||||
}
|
||||
|
||||
mtrx.sendMessage(context.Background(), mtrx.adminRoom, "New pic added: "+file.Name())
|
||||
mtrx.pics = append(mtrx.pics, file.Name())
|
||||
mtrx.pics[idd] = file.Name()
|
||||
}
|
||||
|
||||
return nil
|
||||
|
|
|
@ -162,9 +162,9 @@ func (mtrx *MtrxClient) handleCommand(ctx context.Context, evt *event.Event) {
|
|||
return
|
||||
case ".rb":
|
||||
if len(args) < 2 {
|
||||
mtrx.postBirbWithText(ctx, evt, mtrx.getAnyRandomQuote(true), 0, "")
|
||||
mtrx.postBirbWithText(ctx, evt, mtrx.getAnyRandomQuote(true), "")
|
||||
} else {
|
||||
mtrx.postBirbWithText(ctx, evt, mtrx.getAnyRandomQuoteSearch(true, strings.Join(args[1:], " ")), 0, "")
|
||||
mtrx.postBirbWithText(ctx, evt, mtrx.getAnyRandomQuoteSearch(true, strings.Join(args[1:], " ")), "")
|
||||
}
|
||||
return
|
||||
case ".delquote":
|
||||
|
@ -262,17 +262,19 @@ func (mtrx *MtrxClient) handleCommand(ctx context.Context, evt *event.Event) {
|
|||
filesearch := strings.ReplaceAll(args[0], ".", "")
|
||||
|
||||
// lookup filename from quote, shuffle each time
|
||||
tmpPics := make([]string, len(mtrx.pics))
|
||||
perm := rand.Perm(len(mtrx.pics))
|
||||
for i, v := range perm {
|
||||
tmpPics[v] = mtrx.pics[i]
|
||||
matches := make(map[int64]string)
|
||||
keys := make([]int64, 0)
|
||||
for k, v := range mtrx.pics {
|
||||
if strings.Contains(strings.ToLower(strings.Split(v, ".")[0]), strings.ToLower(filesearch)) {
|
||||
matches[k] = v
|
||||
keys = append(keys, k)
|
||||
}
|
||||
}
|
||||
|
||||
for _, v := range tmpPics {
|
||||
if strings.Contains(strings.ToLower(strings.Split(v, ".")[0]), strings.ToLower(filesearch)) {
|
||||
mtrx.postBirbWithText(ctx, evt, mtrx.getAnyRandomQuoteSearch(true, strings.Join(args[1:], " ")), 0, v)
|
||||
return
|
||||
}
|
||||
// Shuffle key
|
||||
if len(matches) > 0 {
|
||||
filename := matches[keys[rand.Intn(len(keys)-1)]]
|
||||
mtrx.postBirbWithText(ctx, evt, mtrx.getAnyRandomQuoteSearch(true, strings.Join(args[1:], " ")), filename)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -44,7 +44,7 @@ type MtrxClient struct {
|
|||
savedFiles map[id.EventID]string
|
||||
events map[id.EventID]string
|
||||
recentMessages map[id.RoomID]map[int64]string
|
||||
pics []string
|
||||
pics map[int64]string
|
||||
|
||||
// This is gonna be a mem leak... To fix I guess
|
||||
quoteCache map[id.EventID]*Quote
|
||||
|
@ -64,7 +64,7 @@ func Run(db *database.Database) {
|
|||
pics, err := mtrx.db.GetAllPics()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
mtrx.pics = make([]string, 0)
|
||||
mtrx.pics = make(map[int64]string, 0)
|
||||
} else {
|
||||
mtrx.pics = pics
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue