From 9ff839ee4536242f52dd128d42865e3d4b422902 Mon Sep 17 00:00:00 2001
From: icleitoncosta <cleitoncosta83@gmail.com>
Date: Wed, 24 Jul 2024 21:06:17 -0300
Subject: [PATCH] feat: Added WPP.privacy.setAddGroup function

---
 src/privacy/functions/setAddGroup.ts | 83 ++++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)
 create mode 100644 src/privacy/functions/setAddGroup.ts

diff --git a/src/privacy/functions/setAddGroup.ts b/src/privacy/functions/setAddGroup.ts
new file mode 100644
index 0000000000..1928141b50
--- /dev/null
+++ b/src/privacy/functions/setAddGroup.ts
@@ -0,0 +1,83 @@
+/*!
+ * Copyright 2024 WPPConnect Team
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/**
+ * Set who can add you to a group.
+ *
+ * @example
+ * ```javascript
+ * Set value for who can add to a group like 'all'
+ * await WPP.privacy.setAddGroup('all');
+ *
+ * Set value for who can add to a group like 'only your contacts'
+ * await WPP.privacy.setAddGroup('contacts');
+ *
+ * Set value for who can add to a group like 'your contacts', but with excepts
+ * await WPP.privacy.setAddGroup('contact_blacklist', [
+ *   { id: '[number]@c.us', action: 'add' },
+ *   { id: '[number]@c.us', action: 'remove' }
+ * ]);
+ * ```
+ *
+ * @category Privacy
+ */
+
+import { WPPError } from '../../util';
+import {
+  getUserPrivacySettings,
+  PrivacyDisallowedListType,
+  setPrivacyForOneCategory,
+} from '../../whatsapp/functions';
+import { prepareDisallowedList } from './prepareDisallowedList';
+
+export enum setAddGroupTypes {
+  all = 'all',
+  contacts = 'contacts',
+  contact_blacklist = 'contact_blacklist',
+}
+export async function setAddGroup(
+  value: setAddGroupTypes,
+  disallowedList?: { id: string; action: 'add' | 'remove' }[]
+): Promise<setAddGroupTypes> {
+  if (
+    typeof value !== 'string' ||
+    !Object.values(setAddGroupTypes).includes(value)
+  ) {
+    throw new WPPError(
+      'incorrect_type',
+      `Incorrect type ${value || '<empty>'} for set add group privacy`,
+      {
+        value,
+      }
+    );
+  }
+  const disallowed = await prepareDisallowedList(
+    PrivacyDisallowedListType.GroupAdd,
+    value,
+    disallowedList
+  );
+  await setPrivacyForOneCategory(
+    {
+      name: PrivacyDisallowedListType.GroupAdd,
+      value: value,
+      dhash: value === 'contact_blacklist' ? disallowed.dhash : null,
+      users:
+        value === 'contact_blacklist' ? disallowed.idsFormatted : undefined,
+    },
+    value === 'contact_blacklist' ? disallowed.allUsers : undefined
+  );
+  return getUserPrivacySettings().groupAdd as any;
+}