add repository provider getter

MRtecno98 · Jul 31, 2024 · d7a991c · d7a991c
1 parent 6a9e461
commit d7a991c
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/bucket/repositories/modrinth.go b/bucket/repositories/modrinth.go
@@ -183,6 +183,10 @@ func NewModrinthRepository(lock context.Context, context *bucket.OpenContext) *M
 	}
 }
 
+func (r *Modrinth) Provider() string {
+	return MODRINTH_REPOSITORY
+}
+
 func (r *Modrinth) makreq() *resty.Request {
 	return r.HttpClient.R().SetContext(r.Lock)
 }

diff --git a/bucket/repositories/spigotmc.go b/bucket/repositories/spigotmc.go
@@ -66,6 +66,10 @@ func NewSpigotRepository(ctx context.Context, context *bucket.OpenContext) *Spig
 	}
 }
 
+func (r *SpigotMC) Provider() string {
+	return SPIGOTMC_REPOSITORY
+}
+
 func (r *SpigotMC) Resolve(plugin bucket.Plugin) (bucket.RemotePlugin, []bucket.RemotePlugin, error) {
 	var tot int
 	var res []bucket.RemotePlugin

diff --git a/bucket/repository.go b/bucket/repository.go
@@ -8,6 +8,8 @@ import (
 )
 
 type Repository interface {
+	Provider() string
+
 	Search(query string, max int) ([]RemotePlugin, int, error)
 	SearchAll(query string, max int) ([]RemotePlugin, int, error)