feat: set r18 when posting

Author: krau

bot/handlers_admin.go

@@ -216,7 +216,9 @@ func postArtwork(ctx context.Context, bot *telego.Bot, query telego.CallbackQuer
 		})
 		return
 	}
-	dataID := strings.Split(query.Data, " ")[2]
+	queryDataSlice := strings.Split(query.Data, " ")
+	asR18 := queryDataSlice[0] == "post_artwork_r18"
+	dataID := queryDataSlice[1]
 	sourceURL, err := service.GetCallbackDataByID(ctx, dataID)
 	if err != nil {
 		bot.AnswerCallbackQuery(&telego.AnswerCallbackQueryParams{
@@ -279,6 +281,9 @@ func postArtwork(ctx context.Context, bot *telego.Bot, query telego.CallbackQuer
 			return
 		}
 	}
+	if asR18 {
+		artwork.R18 = true
+	}
 	if err := fetcher.PostAndCreateArtwork(ctx, artwork, bot, storage.GetStorage(), query.From.ID); err != nil {
 		Logger.Errorf("发布失败: %s", err)
 		go bot.EditMessageCaption(&telego.EditMessageCaptionParams{
@@ -305,7 +310,7 @@ func postArtwork(ctx context.Context, bot *telego.Bot, query telego.CallbackQuer
 		Caption:   "发布成功: " + artwork.Title + "\n\n发布时间: " + artwork.CreatedAt.Format("2006-01-02 15:04:05"),
 		ReplyMarkup: telegoutil.InlineKeyboard(
 			[]telego.InlineKeyboardButton{
-				telegoutil.InlineKeyboardButton("去查看").WithURL(telegram.GetArtworkPostMessageURL(artwork.Pictures[0].TelegramInfo.MessageID)),
+				telegoutil.InlineKeyboardButton("查看").WithURL(telegram.GetArtworkPostMessageURL(artwork.Pictures[0].TelegramInfo.MessageID)),
 				telegoutil.InlineKeyboardButton("原图").WithURL(telegram.GetDeepLinkForFile(artwork.Pictures[0].TelegramInfo.MessageID)),
 			},
 		),

bot/handlers_common.go

@@ -249,7 +249,8 @@ func getArtworkInfo(ctx context.Context, bot *telego.Bot, message telego.Message
 		}
 		photo.WithReplyMarkup(telegoutil.InlineKeyboard(
 			[]telego.InlineKeyboardButton{
-				telegoutil.InlineKeyboardButton("发布到频道").WithCallbackData("admin post_artwork " + id),
+				telegoutil.InlineKeyboardButton("发布").WithCallbackData("post_artwork " + id),
+				telegoutil.InlineKeyboardButton("设为R18并发布").WithCallbackData("post_artwork_r18 " + id),
 			},
 		))
 	}

cmd/version.go

@@ -7,7 +7,7 @@ import (
 )
 
 const (
-	Version string = "0.6.6"
+	Version string = "0.6.7"
 )
 
 var VersionCmd = &cobra.Command{