some major updates

internal/commands/start.go

@@ -5,7 +5,9 @@ package commands
 import (
 	"EverythingSuckz/fsb/config"
 	"EverythingSuckz/fsb/internal/utils"
+	"fmt" // Import fmt
 	"math/rand"
+	"strings" // Import strings
 
 	"github.com/celestix/gotgproto/dispatcher"
 	"github.com/celestix/gotgproto/dispatcher/handlers"
@@ -47,6 +49,19 @@ func start(ctx *ext.Context, u *ext.Update) error {
 		return dispatcher.EndGroups
 	}
 
+	args := strings.Split(u.EffectiveMessage.GetMessage(), " ")
+	if len(args) > 1 && args[1] == "connect" {
+		userID := u.EffectiveUser().ID
+		message := fmt.Sprintf(
+			"Welcome! You're connecting your account.\n\n" +
+			"Here is your unique Telegram User ID. Copy this ID and paste it on the website to complete the process.\n\n" +
+			"Your ID: `%d`", // Using backticks for easy copy-paste
+			userID,
+		)
+		ctx.Reply(u, message, nil)
+		return dispatcher.EndGroups // Stop further processing
+	}
+	
 	welcomeText := "Hi! I am the **File Stream Bot**.\n\nSend me any file and I will give you a direct, streamable link. Use the buttons below to get started or learn more."
 	markup := getStartKeyboard()
 	ctx.Reply(u, welcomeText, &ext.ReplyOpts{Markup: markup})

internal/db/database.go

@@ -16,6 +16,7 @@ var (
 	Links  *mongo.Collection
 	Users  *mongo.Collection 
     OTPs   *mongo.Collection 
+	Files  *mongo.Collection
 )
 
 // InitDatabase initializes the MongoDB connection.
@@ -46,6 +47,7 @@ func InitDatabase(log *zap.Logger) error {
 	Links = database.Collection("links")
 	Users = database.Collection("users") 
 	OTPs = database.Collection("otps")   
+	Files = database.Collection("files")
 
 
 	log.Info("MongoDB connection established successfully.")

internal/db/models.go

@@ -35,4 +35,25 @@ type OTPRequest struct {
 	OTPHash   string             `bson:"otp_hash"`
 	ExpiresAt time.Time          `bson:"expires_at"`
 	CreatedAt time.Time          `bson:"created_at"`
+}
+
+// UserRecord represents a user from the 'users' collection.
+// We only need the fields Go needs to interact with.
+type UserRecord struct {
+    ID          primitive.ObjectID `bson:"_id,omitempty"`
+    TelegramID  int64              `bson:"telegram_id"`
+}
+
+type FileRecord struct {
+    ID          primitive.ObjectID `bson:"_id,omitempty"`
+    UserID      primitive.ObjectID `bson:"userId"`
+    FolderID    *primitive.ObjectID `bson:"folderId,omitempty"` // Pointer to allow null
+    MessageID   int                `bson:"messageId"`
+    FileName    string             `bson:"fileName"`
+    FileSize    int64              `bson:"fileSize"`
+    MimeType    string             `bson:"mimeType"`
+    ShortHash   string             `bson:"shortHash"`
+    FullHash    string             `bson:"fullHash"`
+    CreatedAt   time.Time          `bson:"createdAt"`
+    UpdatedAt   time.Time          `bson:"updatedAt"`
 }

internal/utils/helpers.go

@@ -3,19 +3,21 @@ package utils
 import (
 	"EverythingSuckz/fsb/config"
 	"EverythingSuckz/fsb/internal/cache"
-	"EverythingSuckz/fsb/internal/types"
 	"EverythingSuckz/fsb/internal/db"
-	"time"
+	"EverythingSuckz/fsb/internal/types"
 	"context"
 	"errors"
 	"fmt"
 	"math/rand"
+	"time"
 
 	"github.com/celestix/gotgproto"
 	"github.com/celestix/gotgproto/ext"
 	"github.com/celestix/gotgproto/storage"
 	"github.com/gotd/td/tg"
 	"go.uber.org/zap"
+
+	"go.mongodb.org/mongo-driver/bson"
 )
 
 // https://stackoverflow.com/a/70802740/15807350
@@ -221,12 +223,14 @@ func RefreshFileReference(ctx context.Context, client *gotgproto.Client, message
 
 
 
+// GenerateStreamLink takes a message context and a source message ID, processes it,
+// and returns the generated streamable link and the file properties.
 // GenerateStreamLink takes a message context and a source message ID, processes it,
 // and returns the generated streamable link and the file properties.
 func GenerateStreamLink(ctx *ext.Context, u *ext.Update, messageID int) (string, *types.File, error) {
 	log := Logger.Named("GenerateStreamLink")
 
-	// 1. Forward the message to the log channel
+	// 1. Forward the message to the log channel (no change here)
 	chatId := u.EffectiveChat().GetID()
 	update, err := ForwardMessages(ctx, chatId, config.ValueOf.LogChannelID, messageID)
 	if err != nil {
@@ -234,7 +238,7 @@ func GenerateStreamLink(ctx *ext.Context, u *ext.Update, messageID int) (string,
 		return "", nil, fmt.Errorf("could not forward message: %w", err)
 	}
 
-	// 2. Extract file properties from the forwarded message
+	// 2. Extract file properties from the forwarded message (no change here)
 	var newMsg *tg.Message
 	var newMsgID int
 	for _, up := range update.Updates {
@@ -257,26 +261,52 @@ func GenerateStreamLink(ctx *ext.Context, u *ext.Update, messageID int) (string,
 		return "", nil, fmt.Errorf("could not get file data: %w", err)
 	}
 
-	// 3. Generate the secure link
+	// 3. Generate the secure link (no change here)
 	fullHash := PackFile(file.FileName, file.FileSize, file.MimeType, file.ID)
 	hash := GetShortHash(fullHash)
 	link := fmt.Sprintf("%s/stream/%d?hash=%s", config.ValueOf.Host, newMsgID, hash)
 
-	if db.Links != nil {
-		trackedLink := db.TrackedLink{
-			MessageID:      newMsgID,
-			UserID:         u.EffectiveUser().ID,
-			ShortHash:      hash,
-			FullHash:       fullHash,
-			LastRefreshedAt: time.Now(),
-			CreatedAt:      time.Now(),
-		}
-		_, err := db.Links.InsertOne(context.Background(), trackedLink)
-		if err != nil {
-			log.Error("Failed to save tracked link to MongoDB", zap.Error(err))
-		}
+	// *** NEW LOGIC: SAVE THE FILE RECORD FOR THE DASHBOARD ***
+	if db.Files != nil && db.Users != nil {
+		go func() {
+			dbCtx := context.Background()
+			
+			// a. Find the user in the 'users' collection by their telegram_id
+			var userRecord db.UserRecord
+			err := db.Users.FindOne(dbCtx, bson.M{"telegram_id": u.EffectiveUser().ID}).Decode(&userRecord)
+			if err != nil {
+				// User hasn't logged into the website yet. Don't save the file to the dashboard.
+				// We can just log this and continue.
+				log.Warn("User not found in dashboard users collection. File will not appear on dashboard.", zap.Int64("telegramID", u.EffectiveUser().ID))
+				return
+			}
+			
+			// b. Create the new file record
+			now := time.Now()
+			fileRecord := db.FileRecord{
+				UserID:      userRecord.ID,
+				FolderID:    nil, // Uploads to root by default
+				MessageID:   newMsgID,
+				FileName:    file.FileName,
+				FileSize:    file.FileSize,
+				MimeType:    file.MimeType,
+				ShortHash:   hash,
+				FullHash:    fullHash,
+				CreatedAt:   now,
+				UpdatedAt:   now,
+			}
+			
+			// c. Insert the record into the 'files' collection
+			_, err = db.Files.InsertOne(dbCtx, fileRecord)
+			if err != nil {
+				log.Error("Failed to save file record to MongoDB for dashboard", zap.Error(err))
+			} else {
+				log.Info("Successfully saved file to dashboard.", zap.String("fileName", file.FileName), zap.String("userID", userRecord.ID.Hex()))
+			}
+		}()
 	}
 
+
 	log.Info("Successfully generated link", zap.String("link", link))
 	return link, file, nil
-}
+}