From 3ff020d23d70d873a1b8a392c11cede7200c0430 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jakub=20Soko=C5=82owski?= <jakub@status.im>
Date: Mon, 13 Jan 2020 19:11:50 +0100
Subject: [PATCH] create a utility for pinning sicker packs
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Jakub Sokołowski <jakub@status.im>
---
 ipfs/pin_sticker_packs.py | 53 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100755 ipfs/pin_sticker_packs.py

diff --git a/ipfs/pin_sticker_packs.py b/ipfs/pin_sticker_packs.py
new file mode 100755
index 0000000..03c3bc2
--- /dev/null
+++ b/ipfs/pin_sticker_packs.py
@@ -0,0 +1,53 @@
+#!/usr/bin/env python3
+
+import re
+import json
+import requests
+import content_hash # https://pypi.org/project/content-hash
+import ipfscluster  # https://pypi.org/project/ipfscluster
+from multiaddr import Multiaddr
+
+IPFS_CLUSTER_ADDR = 'http://localhost:9094'
+STICKER_PACKS_META_URLS = [
+  "https://cloudflare-ipfs.com/ipfs/QmWVVLwVKCwkVNjYJrRzQWREVvEk917PhbHYAUhA1gECTM",
+  "https://cloudflare-ipfs.com/ipfs/QmWpG2Q5NB472KLgFysdCjB8D1Qf9hxR2KNJvtCJQJufDj",
+]
+
+# https://github.com/ethereum/EIPs/blob/master/EIPS/eip-1577.md
+content_hash_rgx = r'e30101701220\w+'
+
+content_hashes = []
+
+for url in STICKER_PACKS_META_URLS:
+    resp = requests.get(url)
+    resp.raise_for_status()
+    matches = re.findall(content_hash_rgx, resp.text)
+    content_hashes.extend(matches)
+
+# IPFS can't handle EIP-1577 content hashes
+decoded_hashes = [content_hash.decode(ch) for ch in content_hashes]
+
+class IpfsPinner:
+
+    def __init__(self, addr=ipfscluster.DEFAULT_ADDR):
+        self.client = ipfscluster.connect(addr)
+
+    def is_pinned(self, chash):
+        resp = self.client.pins.ls(chash)
+        statuses = [peer['status'] for peer in resp['peer_map'].values()]
+        return all(s == 'pinned' for s in statuses), statuses
+
+    def pin(self, chash):
+        return self.client.pins.add(chash)
+
+ip = IpfsPinner()
+
+for chash in decoded_hashes:
+    pinned, statuses = ip.is_pinned(chash)
+    print('{} - {}'.format(chash, statuses))
+
+    if pinned:
+        continue
+
+    ip.pin(chash)
+    print('{} - PINNED'.format(chash))