From b4eaa34ffb08b679125d6bdd2c10ec514e4cd1f6 Mon Sep 17 00:00:00 2001
From: Qu Xuan <quxuan@yunionyun.com>
Date: Wed, 27 Oct 2021 17:00:03 +0800
Subject: [PATCH] fix(region): add waf and app tags sync

---
 pkg/multicloud/azure/waf.go     | 15 +++++++++++++++
 pkg/multicloud/azure/web_app.go | 15 +++++++++++++++
 2 files changed, 30 insertions(+)

diff --git a/pkg/multicloud/azure/waf.go b/pkg/multicloud/azure/waf.go
index 82d1e3f030..3eb8c9b1ae 100644
--- a/pkg/multicloud/azure/waf.go
+++ b/pkg/multicloud/azure/waf.go
@@ -639,3 +639,18 @@ func (self *SAppGatewayWaf) GetCloudResources() ([]cloudprovider.SCloudResource,
 	}
 	return ret, nil
 }
+
+func (self *SAppGatewayWaf) SetTags(tags map[string]string, replace bool) error {
+	if !replace {
+		for k, v := range self.Tags {
+			if _, ok := tags[k]; !ok {
+				tags[k] = v
+			}
+		}
+	}
+	_, err := self.region.client.SetTags(self.ID, tags)
+	if err != nil {
+		return errors.Wrapf(err, "SetTags")
+	}
+	return nil
+}
diff --git a/pkg/multicloud/azure/web_app.go b/pkg/multicloud/azure/web_app.go
index a31e6ab766..e2603b9238 100644
--- a/pkg/multicloud/azure/web_app.go
+++ b/pkg/multicloud/azure/web_app.go
@@ -300,3 +300,18 @@ func (a *SApp) GetOsType() cloudprovider.TOsType {
 	}
 	return cloudprovider.OsTypeWindows
 }
+
+func (self *SApp) SetTags(tags map[string]string, replace bool) error {
+	if !replace {
+		for k, v := range self.Tags {
+			if _, ok := tags[k]; !ok {
+				tags[k] = v
+			}
+		}
+	}
+	_, err := self.region.client.SetTags(self.Id, tags)
+	if err != nil {
+		return errors.Wrapf(err, "SetTags")
+	}
+	return nil
+}