summaryrefslogtreecommitdiff
path: root/helpers.rb
blob: a43966befdc9b0bd4f8f6af3ce1a8d77fae93832 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
# frozen_string_literal: true

require 'English'

helpers do
  # add date and id, save
  def save_activity(activity, box)
    date = Time.now.utc.iso8601
    activity['published'] ||= date # if box == OUTBOX
    basename = "#{activity['published']}_#{mention(activity['actor'])}.json"
    activity_rel_path = File.join(activity['type'].downcase, basename)
    activity_path = File.join(box[:dir], activity_rel_path)
    if box == OUTBOX
      # return unless activity['to'].include? 'https://www.w3.org/ns/activitystreams#Public' # save only public messages

      activity['id'] = File.join(box[:url], activity_rel_path)
      activity['object']['published'] = date unless activity['object'].is_a? String
    end
    # save object
    save_object activity['object'], box if %w[Create Announce Update].include? activity['type']
    # save activity
    FileUtils.mkdir_p File.dirname(activity_path)
    File.open(activity_path, 'w+') { |f| f.puts activity.to_json }
    activity_path
  end

  def save_object(object, box)
    object = fetch(object) if object.is_a? String and object.match(/^http/)
    return unless object # and object['type'] != 'Person'

    unless object['attributedTo']
      jj object
      return
    end

    object['@context'] = 'https://www.w3.org/ns/activitystreams'
    basename = "#{object['published']}_#{mention(object['attributedTo'])}.json"
    object_rel_path = File.join 'object', object['type'].downcase, basename
    object['id'] ||= File.join box[:url], object_rel_path # if box == OUTBOX
    object_path = File.join box[:dir], object_rel_path
    FileUtils.mkdir_p File.dirname(object_path)
    File.open(object_path, 'w+') { |f| f.puts object.to_json }
    if box == OUTBOX and object['tag']
      object['tag'].each do |tag|
        next unless tag['type'] == 'Hashtag'

        tag_path = File.join(TAGS[:dir], tag['name'].sub('#', '')) + '.json'
        tag_collection = if File.exist? tag_path
                           JSON.load_file(tag_path)
                         else
                           {
                             '@context' => 'https://www.w3.org/ns/activitystreams',
                             'id' => tag['href'],
                             'type' => 'OrderedCollection',
                             'totalItems' => 0,
                             'orderedItems' => []
                           }
                         end
        tag_collection['orderedItems'] << object['id']
        tag_collection['totalItems'] = tag_collection['orderedItems'].size
        File.open(tag_path, 'w+') do |f|
          f.puts tag_collection.to_json
        end
      end
    end
    object
  end

  def update_collection(path, objects, delete = false)
    objects = [objects] unless objects.is_a? Array
    File.open(path, 'r+') do |f|
      f.flock(File::LOCK_EX)
      json = f.read
      collection = JSON.parse(json)
      objects.each do |object|
        id = object['id'] || object
        if delete
          collection['orderedItems'].delete_if { |o| o['id'] == id or o == id }
        else
          ids = collection['orderedItems'].collect { |i| i['id'] }
          collection['orderedItems'] << object unless ids.include?(id) or collection['orderedItems'].include?(id)
        end
      end
      collection['totalItems'] = collection['orderedItems'].size
      f.rewind
      f.puts collection.to_json
      f.truncate(f.pos)
    end
  end

  def fetch(url, accept = 'application/activity+json')
    begin
      uri = URI(url)
    rescue StandardError => e
      p url, e
      return nil
    end
    httpdate = Time.now.utc.httpdate
    keypair = OpenSSL::PKey::RSA.new(File.read('private.pem'))
    string = "(request-target): get #{uri.request_uri}\nhost: #{uri.host}\ndate: #{httpdate}"
    signature = Base64.strict_encode64(keypair.sign(OpenSSL::Digest.new('SHA256'), string))
    signed_header = "keyId=\"#{ACTOR}#main-key\",algorithm=\"rsa-sha256\",headers=\"(request-target) host date\",signature=\"#{signature}\""
    response = curl(
      "-H 'Accept: #{accept}' -H 'Host: #{uri.host}' -H 'Date: #{httpdate}' -H 'Signature: #{signed_header}' ", url
    )
    response ? JSON.parse(response) : nil
  end

  def curl(ext, url)
    response = `/run/current-system/sw/bin/curl -H 'Content-Type: #{CONTENT_TYPE}' -H 'Accept: #{CONTENT_TYPE}' --fail-with-body -sSL #{ext} #{url}`
    if $CHILD_STATUS.success?
      response
    else
      p url, response
      nil
    end
  end

  def mention(actor)
    person = people.select { |p| p[1] == actor }
    if person.empty?
      a = fetch(actor)
      return nil unless a

      mention = "#{a['preferredUsername']}@#{URI(actor).host}"
      cache mention, actor, a
      mention
    else
      person[0][0]
    end
  end

  def actor(mention)
    mention = mention.sub(/^@/, '').chomp
    actors = people.select { |p| p[0] == mention }
    if actors.empty?
      _, server = mention.split('@')
      a = fetch("https://#{server}/.well-known/webfinger?resource=acct:#{mention}", 'application/jrd+json')
      return nil unless a

      actor = a['links'].select do |l|
        l['rel'] == 'self'
      end[0]['href']
      cache mention, actor, a
      actor
    else
      actors[0][1]
    end
  end

  def people
    File.read('public/people.tsv').split("\n").collect { |l| l.chomp.split("\t") }
  end

  def cache(mention, actor, a)
    sharedInbox = a['endpoints']['sharedInbox'] if a['endpoints'] && a['endpoints']['sharedInbox']
    File.open('public/people.tsv', 'a') { |f| f.puts "#{mention}\t#{actor}\t#{sharedInbox}" }
  end

  def media_type(url) # TODO: extend extensions
    extensions = {
      image: %w[jpeg jpg png tiff],
      audio: %w[flac wav mp3 ogg aiff],
      video: %w[mp4 webm]
    }
    ext = File.extname(url).sub('.', '').downcase
    type = extensions.find { |_k, v| v.include? ext }
    "#{type[0]}/#{ext}"
  end

  def find_file(id)
    Dir[File.join('*', 'object', '*', '*.json')].find do |f|
      JSON.load_file(f)['id'] == id
    end
  end
end