From 4f50aa98190e39a83704cdca7e5074b421531f8e Mon Sep 17 00:00:00 2001
From: Idhibhat Pankam <marginpankam@gmail.com>
Date: Wed, 17 Jul 2024 00:08:27 +0700
Subject: [PATCH] fix: cdn url

---
 internal/object/object.repository.go      | 10 +++++-----
 internal/object/object.repository_test.go |  8 ++++----
 internal/object/object.service.go         |  4 ++--
 mocks/object/object.repository.go         |  8 ++++----
 4 files changed, 15 insertions(+), 15 deletions(-)

diff --git a/internal/object/object.repository.go b/internal/object/object.repository.go
index 1c2f886..55b217b 100644
--- a/internal/object/object.repository.go
+++ b/internal/object/object.repository.go
@@ -17,7 +17,7 @@ type Repository interface {
 	Upload(file []byte, bucketName string, objectKey string) (url string, key string, err error)
 	Delete(bucketName string, objectKey string) (err error)
 	Get(bucketName string, objectKey string) (url string, err error)
-	GetURL(bucketName string, objectKey string) string
+	GetURL(objectKey string) string
 }
 
 type repositoryImpl struct {
@@ -46,7 +46,7 @@ func (r *repositoryImpl) Upload(file []byte, bucketName string, objectKey string
 		return "", "", errors.Wrap(err, fmt.Sprintf("Couldn't upload object to %v/%v.", bucketName, objectKey))
 	}
 
-	return r.GetURL(bucketName, objectKey), objectKey, nil
+	return r.GetURL(bucketName), objectKey, nil
 }
 
 func (r *repositoryImpl) Delete(bucketName string, objectKey string) (err error) {
@@ -67,7 +67,7 @@ func (r *repositoryImpl) Get(bucketName string, objectKey string) (url string, e
 	_, cancel := context.WithTimeout(ctx, 50*time.Second)
 	defer cancel()
 
-	url = r.GetURL(bucketName, objectKey)
+	url = r.GetURL(bucketName)
 
 	resp, err := r.httpClient.Get(url)
 	if err != nil {
@@ -80,6 +80,6 @@ func (r *repositoryImpl) Get(bucketName string, objectKey string) (url string, e
 	return url, nil
 }
 
-func (r *repositoryImpl) GetURL(bucketName string, objectKey string) string {
-	return r.conf.CDNEndpoint + "/" + bucketName + "/" + objectKey
+func (r *repositoryImpl) GetURL(objectKey string) string {
+	return r.conf.CDNEndpoint + "/" + objectKey
 }
diff --git a/internal/object/object.repository_test.go b/internal/object/object.repository_test.go
index 67d86a4..45c478e 100644
--- a/internal/object/object.repository_test.go
+++ b/internal/object/object.repository_test.go
@@ -45,7 +45,7 @@ func (t *ObjectRepositoryTest) TestCreateObjectSuccess() {
 	url, key, err := repo.Upload([]byte{}, "mock-bucket", "mock-key")
 	t.Nil(err)
 	t.Equal("mock-key", key)
-	t.Equal(repo.GetURL("mock-bucket", "mock-key"), url)
+	t.Equal(repo.GetURL("mock-key"), url)
 }
 
 func (t *ObjectRepositoryTest) TestUploadSuccess() {
@@ -57,7 +57,7 @@ func (t *ObjectRepositoryTest) TestUploadSuccess() {
 	url, key, err := repo.Upload([]byte{}, "bucket", "object")
 	t.Nil(err)
 	t.Equal("object", key)
-	t.Equal(repo.GetURL("bucket", "object"), url)
+	t.Equal(repo.GetURL("object"), url)
 }
 
 func (t *ObjectRepositoryTest) TestUploadError() {
@@ -101,7 +101,7 @@ func (t *ObjectRepositoryTest) TestGetSuccess() {
 
 	url, err := repo.Get("bucket", "object")
 	t.Nil(err)
-	t.Equal(repo.GetURL("bucket", "object"), url)
+	t.Equal(repo.GetURL("object"), url)
 }
 
 func (t *ObjectRepositoryTest) TestGetError() {
@@ -130,6 +130,6 @@ func (t *ObjectRepositoryTest) TestGetStatusNotOK() {
 
 func (t *ObjectRepositoryTest) TestGetURL() {
 	repo := object.NewRepository(t.conf, nil, nil)
-	url := repo.GetURL("bucket", "object")
+	url := repo.GetURL("object")
 	t.Equal(t.mockEndpoint, url)
 }
diff --git a/internal/object/object.service.go b/internal/object/object.service.go
index d80f0b3..99dce3b 100644
--- a/internal/object/object.service.go
+++ b/internal/object/object.service.go
@@ -106,6 +106,6 @@ func (s *serviceImpl) DeleteByKey(_ context.Context, req *proto.DeleteByKeyObjec
 	}, nil
 }
 
-func (s *serviceImpl) GetURL(bucketName string, objectKey string) string {
-	return "https://" + s.conf.Endpoint + "/" + bucketName + "/" + objectKey
+func (s *serviceImpl) GetURL(objectKey string) string {
+	return s.conf.CDNEndpoint + "/" + objectKey
 }
diff --git a/mocks/object/object.repository.go b/mocks/object/object.repository.go
index 994b752..b68f327 100644
--- a/mocks/object/object.repository.go
+++ b/mocks/object/object.repository.go
@@ -63,17 +63,17 @@ func (mr *MockRepositoryMockRecorder) Get(bucketName, objectKey interface{}) *go
 }
 
 // GetURL mocks base method.
-func (m *MockRepository) GetURL(bucketName, objectKey string) string {
+func (m *MockRepository) GetURL(objectKey string) string {
 	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "GetURL", bucketName, objectKey)
+	ret := m.ctrl.Call(m, "GetURL", objectKey)
 	ret0, _ := ret[0].(string)
 	return ret0
 }
 
 // GetURL indicates an expected call of GetURL.
-func (mr *MockRepositoryMockRecorder) GetURL(bucketName, objectKey interface{}) *gomock.Call {
+func (mr *MockRepositoryMockRecorder) GetURL(objectKey interface{}) *gomock.Call {
 	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetURL", reflect.TypeOf((*MockRepository)(nil).GetURL), bucketName, objectKey)
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetURL", reflect.TypeOf((*MockRepository)(nil).GetURL), objectKey)
 }
 
 // Upload mocks base method.