Skip to content

Commit ed0837a

Browse files
kraukrau
committed
refactor: replace logger usage with common.Log for consistent logging
1 parent 65fee89 commit ed0837a

27 files changed

+152
-158
lines changed

‎bot/bot.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@ import (
1111
"github.com/glebarez/sqlite"
1212
"github.com/gotd/td/telegram/dcs"
1313
"github.com/gotd/td/tg"
14+
"github.com/krau/SaveAny-Bot/common"
1415
"github.com/krau/SaveAny-Bot/config"
15-
"github.com/krau/SaveAny-Bot/logger"
1616
"golang.org/x/net/proxy"
1717
)
1818

@@ -27,7 +27,7 @@ func newProxyDialer(proxyUrl string) (proxy.Dialer, error) {
2727
}
2828

2929
func Init() {
30-
logger.L.Info("初始化 Telegram 客户端...")
30+
common.Log.Info("初始化 Telegram 客户端...")
3131
ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
3232
defer cancel()
3333
resultChan := make(chan struct {
@@ -87,15 +87,15 @@ func Init() {
8787

8888
select {
8989
case <-ctx.Done():
90-
logger.L.Fatal("初始化客户端失败: 超时")
90+
common.Log.Fatal("初始化客户端失败: 超时")
9191
os.Exit(1)
9292
case result := <-resultChan:
9393
if result.err != nil {
94-
logger.L.Fatalf("初始化客户端失败: %s", result.err)
94+
common.Log.Fatalf("初始化客户端失败: %s", result.err)
9595
os.Exit(1)
9696
}
9797
Client = result.client
9898
RegisterHandlers(Client.Dispatcher)
99-
logger.L.Info("客户端初始化完成")
99+
common.Log.Info("客户端初始化完成")
100100
}
101101
}

‎bot/handle_add_task.go

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,9 @@ import (
1313
"github.com/gotd/td/telegram/message/entity"
1414
"github.com/gotd/td/telegram/message/styling"
1515
"github.com/gotd/td/tg"
16+
"github.com/krau/SaveAny-Bot/common"
1617
"github.com/krau/SaveAny-Bot/config"
1718
"github.com/krau/SaveAny-Bot/dao"
18-
"github.com/krau/SaveAny-Bot/logger"
1919
"github.com/krau/SaveAny-Bot/queue"
2020
"github.com/krau/SaveAny-Bot/types"
2121
"gorm.io/gorm"
@@ -37,7 +37,7 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
3737
cbDataId, _ := strconv.Atoi(args[1])
3838
cbData, err := dao.GetCallbackData(uint(cbDataId))
3939
if err != nil {
40-
logger.L.Errorf("获取回调数据失败: %s", err)
40+
common.Log.Errorf("获取回调数据失败: %s", err)
4141
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
4242
QueryID: update.CallbackQuery.QueryID,
4343
Alert: true,
@@ -56,7 +56,7 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
5656

5757
user, err := dao.GetUserByChatID(update.CallbackQuery.UserID)
5858
if err != nil {
59-
logger.L.Errorf("获取用户失败: %s", err)
59+
common.Log.Errorf("获取用户失败: %s", err)
6060
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
6161
QueryID: update.CallbackQuery.QueryID,
6262
Alert: true,
@@ -69,7 +69,7 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
6969
if !addToDir {
7070
dirs, err := dao.GetDirsByUserIDAndStorageName(user.ID, storageName)
7171
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
72-
logger.L.Errorf("获取路径失败: %s", err)
72+
common.Log.Errorf("获取路径失败: %s", err)
7373
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
7474
QueryID: update.CallbackQuery.QueryID,
7575
Alert: true,
@@ -81,7 +81,7 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
8181
if len(dirs) != 0 {
8282
markup, err := getSelectDirMarkup(fileChatID, fileMessageID, storageName, dirs)
8383
if err != nil {
84-
logger.L.Errorf("获取路径失败: %s", err)
84+
common.Log.Errorf("获取路径失败: %s", err)
8585
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
8686
QueryID: update.CallbackQuery.QueryID,
8787
Alert: true,
@@ -96,16 +96,16 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
9696
ReplyMarkup: markup,
9797
})
9898
if err != nil {
99-
logger.L.Errorf("编辑消息失败: %s", err)
99+
common.Log.Errorf("编辑消息失败: %s", err)
100100
}
101101
return dispatcher.EndGroups
102102
}
103103
}
104104

105-
logger.L.Tracef("Got add to queue: chatID: %d, messageID: %d, storage: %s", fileChatID, fileMessageID, storageName)
105+
common.Log.Tracef("Got add to queue: chatID: %d, messageID: %d, storage: %s", fileChatID, fileMessageID, storageName)
106106
record, err := dao.GetReceivedFileByChatAndMessageID(int64(fileChatID), fileMessageID)
107107
if err != nil {
108-
logger.L.Errorf("获取记录失败: %s", err)
108+
common.Log.Errorf("获取记录失败: %s", err)
109109
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
110110
QueryID: update.CallbackQuery.QueryID,
111111
Alert: true,
@@ -117,15 +117,15 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
117117
if update.CallbackQuery.MsgID != record.ReplyMessageID {
118118
record.ReplyMessageID = update.CallbackQuery.MsgID
119119
if err := dao.SaveReceivedFile(record); err != nil {
120-
logger.L.Errorf("更新接收的文件失败: %s", err)
120+
common.Log.Errorf("更新接收的文件失败: %s", err)
121121
}
122122
}
123123

124124
var dir *dao.Dir
125125
if addToDir && dirId != 0 {
126126
dir, err = dao.GetDirByID(dirId)
127127
if err != nil {
128-
logger.L.Errorf("获取路径失败: %s", err)
128+
common.Log.Errorf("获取路径失败: %s", err)
129129
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
130130
QueryID: update.CallbackQuery.QueryID,
131131
Alert: true,
@@ -138,7 +138,7 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
138138

139139
file, err := FileFromMessage(ctx, record.ChatID, record.MessageID, record.FileName)
140140
if err != nil {
141-
logger.L.Errorf("获取消息中的文件失败: %s", err)
141+
common.Log.Errorf("获取消息中的文件失败: %s", err)
142142
ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
143143
QueryID: update.CallbackQuery.QueryID,
144144
Alert: true,
@@ -174,7 +174,7 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
174174
styling.Plain("\n当前排队任务数: "),
175175
styling.Bold(strconv.Itoa(queue.Len())),
176176
); err != nil {
177-
logger.L.Errorf("Failed to build entity: %s", err)
177+
common.Log.Errorf("Failed to build entity: %s", err)
178178
} else {
179179
text, entities = entityBuilder.Complete()
180180
}

‎bot/handle_conversation.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ func (c *ConversationState) SetData(key string, value interface{}) {
6969
// func handleConversationState(ctx *ext.Context, update *ext.Update, state *ConversationState) error {
7070
// switch state.conversationType {
7171
// default:
72-
// logger.L.Errorf("Unknown conversation type: %s", state.conversationType)
72+
// common.Log.Errorf("Unknown conversation type: %s", state.conversationType)
7373
// }
7474
// return dispatcher.EndGroups
7575
// }

‎bot/handle_dir.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@ import (
66
"github.com/celestix/gotgproto/dispatcher"
77
"github.com/celestix/gotgproto/ext"
88
"github.com/gotd/td/telegram/message/styling"
9+
"github.com/krau/SaveAny-Bot/common"
910
"github.com/krau/SaveAny-Bot/dao"
10-
"github.com/krau/SaveAny-Bot/logger"
1111
"github.com/krau/SaveAny-Bot/storage"
1212
)
1313

@@ -16,7 +16,7 @@ func dirCmd(ctx *ext.Context, update *ext.Update) error {
1616
if len(args) < 3 {
1717
dirs, err := dao.GetUserDirsByChatID(update.GetUserChat().GetID())
1818
if err != nil {
19-
logger.L.Errorf("获取用户路径失败: %s", err)
19+
common.Log.Errorf("获取用户路径失败: %s", err)
2020
ctx.Reply(update, ext.ReplyTextString("获取用户路径失败"), nil)
2121
return dispatcher.EndGroups
2222
}
@@ -47,7 +47,7 @@ func dirCmd(ctx *ext.Context, update *ext.Update) error {
4747
}
4848
user, err := dao.GetUserByChatID(update.GetUserChat().GetID())
4949
if err != nil {
50-
logger.L.Errorf("获取用户失败: %s", err)
50+
common.Log.Errorf("获取用户失败: %s", err)
5151
ctx.Reply(update, ext.ReplyTextString("获取用户失败"), nil)
5252
return dispatcher.EndGroups
5353
}
@@ -69,7 +69,7 @@ func addDir(ctx *ext.Context, update *ext.Update, user *dao.User, storageName, p
6969
}
7070

7171
if err := dao.CreateDirForUser(user.ID, storageName, path); err != nil {
72-
logger.L.Errorf("创建路径失败: %s", err)
72+
common.Log.Errorf("创建路径失败: %s", err)
7373
ctx.Reply(update, ext.ReplyTextString("创建路径失败"), nil)
7474
return dispatcher.EndGroups
7575
}
@@ -79,7 +79,7 @@ func addDir(ctx *ext.Context, update *ext.Update, user *dao.User, storageName, p
7979

8080
func delDir(ctx *ext.Context, update *ext.Update, user *dao.User, storageName, path string) error {
8181
if err := dao.DeleteDirForUser(user.ID, storageName, path); err != nil {
82-
logger.L.Errorf("删除路径失败: %s", err)
82+
common.Log.Errorf("删除路径失败: %s", err)
8383
ctx.Reply(update, ext.ReplyTextString("删除路径失败"), nil)
8484
return dispatcher.EndGroups
8585
}

‎bot/handle_file.go

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,14 @@ import (
66
"github.com/celestix/gotgproto/dispatcher"
77
"github.com/celestix/gotgproto/ext"
88
"github.com/gotd/td/tg"
9+
"github.com/krau/SaveAny-Bot/common"
910
"github.com/krau/SaveAny-Bot/dao"
10-
"github.com/krau/SaveAny-Bot/logger"
1111
"github.com/krau/SaveAny-Bot/storage"
1212
"github.com/krau/SaveAny-Bot/types"
1313
)
1414

1515
func handleFileMessage(ctx *ext.Context, update *ext.Update) error {
16-
logger.L.Trace("Got media: ", update.EffectiveMessage.Media.TypeName())
16+
common.Log.Trace("Got media: ", update.EffectiveMessage.Media.TypeName())
1717
supported, err := supportedMediaFilter(update.EffectiveMessage.Message)
1818
if err != nil {
1919
return err
@@ -24,7 +24,7 @@ func handleFileMessage(ctx *ext.Context, update *ext.Update) error {
2424

2525
user, err := dao.GetUserByChatID(update.GetUserChat().GetID())
2626
if err != nil {
27-
logger.L.Errorf("获取用户失败: %s", err)
27+
common.Log.Errorf("获取用户失败: %s", err)
2828
ctx.Reply(update, ext.ReplyTextString("获取用户失败"), nil)
2929
return dispatcher.EndGroups
3030
}
@@ -36,13 +36,13 @@ func handleFileMessage(ctx *ext.Context, update *ext.Update) error {
3636

3737
msg, err := ctx.Reply(update, ext.ReplyTextString("正在获取文件信息..."), nil)
3838
if err != nil {
39-
logger.L.Errorf("回复失败: %s", err)
39+
common.Log.Errorf("回复失败: %s", err)
4040
return dispatcher.EndGroups
4141
}
4242
media := update.EffectiveMessage.Media
4343
file, err := FileFromMedia(media, "")
4444
if err != nil {
45-
logger.L.Errorf("获取文件失败: %s", err)
45+
common.Log.Errorf("获取文件失败: %s", err)
4646
ctx.Reply(update, ext.ReplyTextString(fmt.Sprintf("获取文件失败: %s", err)), nil)
4747
return dispatcher.EndGroups
4848
}
@@ -58,12 +58,12 @@ func handleFileMessage(ctx *ext.Context, update *ext.Update) error {
5858
ReplyMessageID: msg.ID,
5959
ReplyChatID: update.GetUserChat().GetID(),
6060
}); err != nil {
61-
logger.L.Errorf("添加接收的文件失败: %s", err)
61+
common.Log.Errorf("添加接收的文件失败: %s", err)
6262
if _, err := ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
6363
Message: fmt.Sprintf("添加接收的文件失败: %s", err),
6464
ID: msg.ID,
6565
}); err != nil {
66-
logger.L.Errorf("编辑消息失败: %s", err)
66+
common.Log.Errorf("编辑消息失败: %s", err)
6767
}
6868
return dispatcher.EndGroups
6969
}

‎bot/handle_link.go

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@ import (
99
"github.com/celestix/gotgproto/dispatcher"
1010
"github.com/celestix/gotgproto/ext"
1111
"github.com/gotd/td/tg"
12+
"github.com/krau/SaveAny-Bot/common"
1213
"github.com/krau/SaveAny-Bot/dao"
13-
"github.com/krau/SaveAny-Bot/logger"
1414
"github.com/krau/SaveAny-Bot/storage"
1515
"github.com/krau/SaveAny-Bot/types"
1616
)
@@ -21,7 +21,7 @@ var (
2121
)
2222

2323
func handleLinkMessage(ctx *ext.Context, update *ext.Update) error {
24-
logger.L.Trace("Got link message")
24+
common.Log.Trace("Got link message")
2525
link := linkRegex.FindString(update.EffectiveMessage.Text)
2626
if link == "" {
2727
return dispatcher.ContinueGroups
@@ -32,25 +32,25 @@ func handleLinkMessage(ctx *ext.Context, update *ext.Update) error {
3232
}
3333
messageID, err := strconv.Atoi(strSlice[2])
3434
if err != nil {
35-
logger.L.Errorf("解析消息 ID 失败: %s", err)
35+
common.Log.Errorf("解析消息 ID 失败: %s", err)
3636
ctx.Reply(update, ext.ReplyTextString("无法解析消息 ID"), nil)
3737
return dispatcher.EndGroups
3838
}
3939
chatUsername := strSlice[1]
4040
linkChat, err := ctx.ResolveUsername(chatUsername)
4141
if err != nil {
42-
logger.L.Errorf("解析 Chat ID 失败: %s", err)
42+
common.Log.Errorf("解析 Chat ID 失败: %s", err)
4343
ctx.Reply(update, ext.ReplyTextString("无法解析 Chat ID"), nil)
4444
return dispatcher.EndGroups
4545
}
4646
if linkChat == nil {
47-
logger.L.Errorf("无法找到聊天: %s", chatUsername)
47+
common.Log.Errorf("无法找到聊天: %s", chatUsername)
4848
ctx.Reply(update, ext.ReplyTextString("无法找到聊天"), nil)
4949
return dispatcher.EndGroups
5050
}
5151
user, err := dao.GetUserByChatID(update.GetUserChat().GetID())
5252
if err != nil {
53-
logger.L.Errorf("获取用户失败: %s", err)
53+
common.Log.Errorf("获取用户失败: %s", err)
5454
ctx.Reply(update, ext.ReplyTextString("获取用户失败"), nil)
5555
return dispatcher.EndGroups
5656
}
@@ -62,19 +62,19 @@ func handleLinkMessage(ctx *ext.Context, update *ext.Update) error {
6262
}
6363
replied, err := ctx.Reply(update, ext.ReplyTextString("正在获取文件..."), nil)
6464
if err != nil {
65-
logger.L.Errorf("回复失败: %s", err)
65+
common.Log.Errorf("回复失败: %s", err)
6666
return dispatcher.EndGroups
6767
}
6868

6969
file, err := FileFromMessage(ctx, linkChat.GetID(), messageID, "")
7070
if err != nil {
71-
logger.L.Errorf("获取文件失败: %s", err)
71+
common.Log.Errorf("获取文件失败: %s", err)
7272
ctx.Reply(update, ext.ReplyTextString("获取文件失败: "+err.Error()), nil)
7373
return dispatcher.EndGroups
7474
}
7575
// TODO: Better file name
7676
if file.FileName == "" {
77-
logger.L.Warnf("文件名为空，使用生成的名称")
77+
common.Log.Warnf("文件名为空，使用生成的名称")
7878
file.FileName = fmt.Sprintf("%d_%d_%s", linkChat.GetID(), messageID, file.Hash())
7979
}
8080

@@ -87,7 +87,7 @@ func handleLinkMessage(ctx *ext.Context, update *ext.Update) error {
8787
ReplyChatID: update.GetUserChat().GetID(),
8888
}
8989
if err := dao.SaveReceivedFile(receivedFile); err != nil {
90-
logger.L.Errorf("保存接收的文件失败: %s", err)
90+
common.Log.Errorf("保存接收的文件失败: %s", err)
9191
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
9292
Message: "无法保存文件: " + err.Error(),
9393
ID: replied.ID,

‎bot/handle_save.go

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@ import (
77
"github.com/celestix/gotgproto/dispatcher"
88
"github.com/celestix/gotgproto/ext"
99
"github.com/gotd/td/tg"
10+
"github.com/krau/SaveAny-Bot/common"
1011
"github.com/krau/SaveAny-Bot/dao"
11-
"github.com/krau/SaveAny-Bot/logger"
1212
"github.com/krau/SaveAny-Bot/storage"
1313
"github.com/krau/SaveAny-Bot/types"
1414
)
@@ -32,7 +32,7 @@ func saveCmd(ctx *ext.Context, update *ext.Update) error {
3232

3333
user, err := dao.GetUserByChatID(update.GetUserChat().GetID())
3434
if err != nil {
35-
logger.L.Errorf("获取用户失败: %s", err)
35+
common.Log.Errorf("获取用户失败: %s", err)
3636
ctx.Reply(update, ext.ReplyTextString("获取用户失败"), nil)
3737
return dispatcher.EndGroups
3838
}
@@ -46,7 +46,7 @@ func saveCmd(ctx *ext.Context, update *ext.Update) error {
4646

4747
msg, err := GetTGMessage(ctx, update.EffectiveChat().GetID(), replyToMsgID)
4848
if err != nil {
49-
logger.L.Errorf("获取消息失败: %s", err)
49+
common.Log.Errorf("获取消息失败: %s", err)
5050
ctx.Reply(update, ext.ReplyTextString("无法获取消息"), nil)
5151
return dispatcher.EndGroups
5252
}
@@ -59,7 +59,7 @@ func saveCmd(ctx *ext.Context, update *ext.Update) error {
5959

6060
replied, err := ctx.Reply(update, ext.ReplyTextString("正在获取文件信息..."), nil)
6161
if err != nil {
62-
logger.L.Errorf("回复失败: %s", err)
62+
common.Log.Errorf("回复失败: %s", err)
6363
return dispatcher.EndGroups
6464
}
6565

@@ -68,7 +68,7 @@ func saveCmd(ctx *ext.Context, update *ext.Update) error {
6868

6969
file, err := FileFromMessage(ctx, update.EffectiveChat().GetID(), msg.ID, customFileName)
7070
if err != nil {
71-
logger.L.Errorf("获取文件失败: %s", err)
71+
common.Log.Errorf("获取文件失败: %s", err)
7272
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
7373
Message: fmt.Sprintf("获取文件失败: %s", err),
7474
ID: replied.ID,
@@ -90,12 +90,12 @@ func saveCmd(ctx *ext.Context, update *ext.Update) error {
9090
}
9191

9292
if err := dao.SaveReceivedFile(receivedFile); err != nil {
93-
logger.L.Errorf("保存接收的文件失败: %s", err)
93+
common.Log.Errorf("保存接收的文件失败: %s", err)
9494
if _, err := ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
9595
Message: fmt.Sprintf("保存接收的文件失败: %s", err),
9696
ID: replied.ID,
9797
}); err != nil {
98-
logger.L.Errorf("编辑消息失败: %s", err)
98+
common.Log.Errorf("编辑消息失败: %s", err)
9999
}
100100
return dispatcher.EndGroups
101101
}

0 commit comments

Comments
 (0)