From ae7426c36a4458af5a62aac84c3e9269d42b10c5 Mon Sep 17 00:00:00 2001
From: ScuroNeko <nix1304@gmail.com>
Date: Sun, 1 Mar 2026 23:08:22 +0300
Subject: [PATCH] 1.0.0 beta 4

---
 drafts.go        | 15 +++++++++++----
 msg_context.go   |  7 ++++++-
 utils/version.go |  4 ++--
 3 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/drafts.go b/drafts.go
index f170f64..3bd2aa6 100644
--- a/drafts.go
+++ b/drafts.go
@@ -53,10 +53,17 @@ type Draft struct {
 }
 
 func NewRandomDraftProvider(api *tgapi.API) *DraftProvider {
-	return &DraftProvider{api: api, generator: &RandomDraftIdGenerator{}}
+	return &DraftProvider{
+		api: api, generator: &RandomDraftIdGenerator{},
+		drafts: make(map[uint64]*Draft),
+	}
 }
 func NewLinearDraftProvider(api *tgapi.API, startValue uint64) *DraftProvider {
-	return &DraftProvider{api: api, generator: &LinearDraftIdGenerator{lastId: startValue}}
+	return &DraftProvider{
+		api:       api,
+		generator: &LinearDraftIdGenerator{lastId: startValue},
+		drafts:    make(map[uint64]*Draft),
+	}
 }
 func (d *DraftProvider) NewDraft() *Draft {
 	id := d.generator.Next()
@@ -65,11 +72,11 @@ func (d *DraftProvider) NewDraft() *Draft {
 	return draft
 }
 
-func (d *Draft) Push(draftId uint64, newText string) error {
+func (d *Draft) Push(newText string) error {
 	d.Message += newText
 	params := tgapi.SendMessageDraftP{
 		ChatID:    d.chatID,
-		DraftID:   draftId,
+		DraftID:   d.ID,
 		Text:      d.Message,
 		ParseMode: d.parseMode,
 		Entities:  d.entities,
diff --git a/msg_context.go b/msg_context.go
index 400e3b7..7c353a5 100644
--- a/msg_context.go
+++ b/msg_context.go
@@ -216,7 +216,12 @@ func (ctx *MsgContext) error(err error) {
 }
 func (ctx *MsgContext) Error(err error) { ctx.error(err) }
 
-func (ctx *MsgContext) NewDraft() *Draft { return ctx.draftProvider.NewDraft() }
+func (ctx *MsgContext) NewDraft() *Draft {
+	draft := ctx.draftProvider.NewDraft()
+	draft.chatID = ctx.Msg.Chat.ID
+	draft.messageThreadID = ctx.Msg.MessageThreadID
+	return draft
+}
 func (ctx *MsgContext) Translate(key string) string {
 	if ctx.From == nil {
 		return key
diff --git a/utils/version.go b/utils/version.go
index 5cf975f..27bbd6f 100644
--- a/utils/version.go
+++ b/utils/version.go
@@ -1,9 +1,9 @@
 package utils
 
 const (
-	VersionString = "1.0.0-beta.3"
+	VersionString = "1.0.0-beta.4"
 	VersionMajor  = 1
 	VersionMinor  = 0
 	VersionPatch  = 0
-	Beta          = 3
+	Beta          = 4
 )