From ed149be84b75f03147f8c05e6a40a5daf45570d9 Mon Sep 17 00:00:00 2001
From: Jealous <CooLanfei@163.com>
Date: Mon, 30 Dec 2024 22:52:55 +0800
Subject: [PATCH] feat(index): add `disable index` option for storages (#7730)

---
 internal/model/storage.go | 5 ++++-
 internal/op/driver.go     | 6 ++++++
 internal/search/build.go  | 5 +++++
 3 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/internal/model/storage.go b/internal/model/storage.go
index 14bcf45f..e3c7e1f9 100644
--- a/internal/model/storage.go
+++ b/internal/model/storage.go
@@ -1,6 +1,8 @@
 package model
 
-import "time"
+import (
+	"time"
+)
 
 type Storage struct {
 	ID              uint      `json:"id" gorm:"primaryKey"`                        // unique key
@@ -13,6 +15,7 @@ type Storage struct {
 	Remark          string    `json:"remark"`
 	Modified        time.Time `json:"modified"`
 	Disabled        bool      `json:"disabled"` // if disabled
+	DisableIndex    bool      `json:"disable_index"`
 	EnableSign      bool      `json:"enable_sign"`
 	Sort
 	Proxy
diff --git a/internal/op/driver.go b/internal/op/driver.go
index 4f10e8e2..41b6f6d4 100644
--- a/internal/op/driver.go
+++ b/internal/op/driver.go
@@ -133,6 +133,12 @@ func getMainItems(config driver.Config) []driver.Item {
 		Type:    conf.TypeSelect,
 		Options: "front,back",
 	})
+	items = append(items, driver.Item{
+		Name:     "disable_index",
+		Type:     conf.TypeBool,
+		Default:  "false",
+		Required: true,
+	})
 	items = append(items, driver.Item{
 		Name:     "enable_sign",
 		Type:     conf.TypeBool,
diff --git a/internal/search/build.go b/internal/search/build.go
index 9865b298..2888c1f4 100644
--- a/internal/search/build.go
+++ b/internal/search/build.go
@@ -157,6 +157,11 @@ func BuildIndex(ctx context.Context, indexPaths, ignorePaths []string, maxDepth
 					return filepath.SkipDir
 				}
 			}
+			if storage, _, err := op.GetStorageAndActualPath(indexPath); err == nil {
+				if storage.GetStorage().DisableIndex {
+					return filepath.SkipDir
+				}
+			}
 			// ignore root
 			if indexPath == "/" {
 				return nil