mirror of https://github.com/lukechilds/damus.git
William Casarin
2 years ago
4 changed files with 183 additions and 0 deletions
@ -0,0 +1,91 @@ |
|||||
|
// |
||||
|
// Mute.swift |
||||
|
// damus |
||||
|
// |
||||
|
// Created by William Casarin on 2023-01-25. |
||||
|
// |
||||
|
|
||||
|
import Foundation |
||||
|
|
||||
|
func create_or_update_mutelist(keypair: FullKeypair, mprev: NostrEvent?, to_add: String) -> NostrEvent? { |
||||
|
return create_or_update_list_event(keypair: keypair, mprev: mprev, to_add: to_add, list_name: "mute", list_type: "p") |
||||
|
} |
||||
|
|
||||
|
func remove_from_mutelist(keypair: FullKeypair, prev: NostrEvent, to_remove: String) -> NostrEvent? { |
||||
|
return remove_from_list_event(keypair: keypair, prev: prev, to_remove: to_remove, tag_type: "p") |
||||
|
} |
||||
|
|
||||
|
func create_or_update_list_event(keypair: FullKeypair, mprev: NostrEvent?, to_add: String, list_name: String, list_type: String) -> NostrEvent? { |
||||
|
let pubkey = keypair.pubkey |
||||
|
|
||||
|
if let prev = mprev { |
||||
|
if let okprev = ensure_list_name(list: prev, name: list_name), prev.pubkey == keypair.pubkey { |
||||
|
return add_to_list_event(keypair: keypair, prev: okprev, to_add: to_add, tag_type: list_type) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
var tags = [["d", list_name], [list_type, to_add]] |
||||
|
let ev = NostrEvent(content: "", pubkey: pubkey, kind: 30000, tags: tags) |
||||
|
|
||||
|
ev.tags = tags |
||||
|
ev.id = calculate_event_id(ev: ev) |
||||
|
ev.sig = sign_event(privkey: keypair.privkey, ev: ev) |
||||
|
|
||||
|
return ev |
||||
|
} |
||||
|
|
||||
|
func remove_from_list_event(keypair: FullKeypair, prev: NostrEvent, to_remove: String, tag_type: String) -> NostrEvent? { |
||||
|
var exists = false |
||||
|
for tag in prev.tags { |
||||
|
if tag.count >= 2 && tag[0] == tag_type && tag[1] == to_remove { |
||||
|
exists = true |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// make sure we actually have the pubkey to remove |
||||
|
guard exists else { |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
let new_tags = prev.tags.filter { tag in |
||||
|
!(tag.count >= 2 && tag[0] == tag_type && tag[1] == to_remove) |
||||
|
} |
||||
|
|
||||
|
let ev = NostrEvent(content: prev.content, pubkey: keypair.pubkey, kind: 30000, tags: new_tags) |
||||
|
ev.id = calculate_event_id(ev: ev) |
||||
|
ev.sig = sign_event(privkey: keypair.privkey, ev: ev) |
||||
|
|
||||
|
return ev |
||||
|
} |
||||
|
|
||||
|
func add_to_list_event(keypair: FullKeypair, prev: NostrEvent, to_add: String, tag_type: String) -> NostrEvent? { |
||||
|
for tag in prev.tags { |
||||
|
// we are already muting this user |
||||
|
if tag.count >= 2 && tag[0] == tag_type && tag[1] == to_add { |
||||
|
return nil |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
let new = NostrEvent(content: prev.content, pubkey: keypair.pubkey, kind: 30000, tags: prev.tags) |
||||
|
new.tags.append([tag_type, to_add]) |
||||
|
new.id = calculate_event_id(ev: new) |
||||
|
new.sig = sign_event(privkey: keypair.privkey, ev: new) |
||||
|
|
||||
|
return new |
||||
|
} |
||||
|
|
||||
|
func ensure_list_name(list: NostrEvent, name: String) -> NostrEvent? { |
||||
|
for tag in list.tags { |
||||
|
if tag.count >= 2 && tag[0] == "d" { |
||||
|
if tag[1] != name { |
||||
|
return nil |
||||
|
} else { |
||||
|
return list |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
list.tags.insert(["d", name], at: 0) |
||||
|
|
||||
|
return list |
||||
|
} |
||||
@ -0,0 +1,71 @@ |
|||||
|
// |
||||
|
// ListTests.swift |
||||
|
// damusTests |
||||
|
// |
||||
|
// Created by William Casarin on 2023-01-25. |
||||
|
// |
||||
|
|
||||
|
import XCTest |
||||
|
@testable import damus |
||||
|
|
||||
|
final class ListTests: XCTestCase { |
||||
|
|
||||
|
override func setUpWithError() throws { |
||||
|
// Put setup code here. This method is called before the invocation of each test method in the class. |
||||
|
} |
||||
|
|
||||
|
override func tearDownWithError() throws { |
||||
|
// Put teardown code here. This method is called after the invocation of each test method in the class. |
||||
|
} |
||||
|
|
||||
|
func testCreateMuteList() throws { |
||||
|
let privkey = "87f313b03f2548e6eaf1c188db47078e08e894252949779b639b28db0891937a" |
||||
|
let pubkey = "4b0c29bf96496130c1253102f6870c0eee05db38a257315858272aa43fd19685" |
||||
|
let to_mute = "2fa2630fea3d2c188c49f2799fcd92f0e9879ea6a36ae60770a5428ed6c19edd" |
||||
|
let keypair = FullKeypair(pubkey: pubkey, privkey: privkey) |
||||
|
let mutelist = create_or_update_mutelist(keypair: keypair, mprev: nil, to_add: to_mute)! |
||||
|
|
||||
|
XCTAssertEqual(mutelist.pubkey, pubkey) |
||||
|
XCTAssertEqual(mutelist.content, "") |
||||
|
XCTAssertEqual(mutelist.tags.count, 2) |
||||
|
XCTAssertEqual(mutelist.tags[0][0], "d") |
||||
|
XCTAssertEqual(mutelist.tags[0][1], "mute") |
||||
|
XCTAssertEqual(mutelist.tags[1][0], "p") |
||||
|
XCTAssertEqual(mutelist.tags[1][1], to_mute) |
||||
|
} |
||||
|
|
||||
|
func testCreateAndRemoveMuteList() throws { |
||||
|
let privkey = "87f313b03f2548e6eaf1c188db47078e08e894252949779b639b28db0891937a" |
||||
|
let pubkey = "4b0c29bf96496130c1253102f6870c0eee05db38a257315858272aa43fd19685" |
||||
|
let to_mute = "2fa2630fea3d2c188c49f2799fcd92f0e9879ea6a36ae60770a5428ed6c19edd" |
||||
|
let keypair = FullKeypair(pubkey: pubkey, privkey: privkey) |
||||
|
let mutelist = create_or_update_mutelist(keypair: keypair, mprev: nil, to_add: to_mute)! |
||||
|
let new = remove_from_mutelist(keypair: keypair, prev: mutelist, to_remove: to_mute)! |
||||
|
|
||||
|
XCTAssertEqual(new.pubkey, pubkey) |
||||
|
XCTAssertEqual(new.content, "") |
||||
|
XCTAssertEqual(new.tags.count, 1) |
||||
|
XCTAssertEqual(new.tags[0][0], "d") |
||||
|
XCTAssertEqual(new.tags[0][1], "mute") |
||||
|
} |
||||
|
|
||||
|
func testAddToExistingMutelist() throws { |
||||
|
let privkey = "87f313b03f2548e6eaf1c188db47078e08e894252949779b639b28db0891937a" |
||||
|
let pubkey = "4b0c29bf96496130c1253102f6870c0eee05db38a257315858272aa43fd19685" |
||||
|
let to_mute = "2fa2630fea3d2c188c49f2799fcd92f0e9879ea6a36ae60770a5428ed6c19edd" |
||||
|
let to_mute_2 = "976b4ab41f8634119b4f21f57ef5836a4bef65d0bf72c7ced67b8b170ba4a38d" |
||||
|
let keypair = FullKeypair(pubkey: pubkey, privkey: privkey) |
||||
|
let mutelist = create_or_update_mutelist(keypair: keypair, mprev: nil, to_add: to_mute)! |
||||
|
let new = create_or_update_mutelist(keypair: keypair, mprev: mutelist, to_add: to_mute_2)! |
||||
|
|
||||
|
XCTAssertEqual(new.pubkey, pubkey) |
||||
|
XCTAssertEqual(new.content, "") |
||||
|
XCTAssertEqual(new.tags.count, 3) |
||||
|
XCTAssertEqual(new.tags[0][0], "d") |
||||
|
XCTAssertEqual(new.tags[0][1], "mute") |
||||
|
XCTAssertEqual(new.tags[1][0], "p") |
||||
|
XCTAssertEqual(new.tags[1][1], to_mute) |
||||
|
XCTAssertEqual(new.tags[2][0], "p") |
||||
|
XCTAssertEqual(new.tags[2][1], to_mute_2) |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue