summaryrefslogtreecommitdiff
path: root/server.rb
blob: 11b6411735d1845d09ca17ff9d5f0629b0bbca6b (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
# frozen_string_literal: true

# server-server
post '/inbox' do
  request.body.rewind # in case someone already read it
  @body = request.body.read
  halt 400 if @body.empty?
  begin
    @activity = JSON.parse @body
  rescue StandardError
    p @body
    halt 400
  end
  type = @activity['type'].downcase.to_sym
  p type
  halt 501 unless respond_to?(type)
  @object = @activity['object']
  @object = fetch(@object) if @object.is_a?(String) && @object.match(/^http/)
  halt 400 unless @object
  verify!
  send(type)
end

# public
get '/.well-known/webfinger' do
  if request['resource'] == "acct:#{MENTION}"
    send_file('./public/webfinger',
              type: 'application/jrd+json')
  else
    halt 404
  end
end

get '/outbox' do
  ordered_collection(OUTBOX_DIR).to_json
end

['/following', '/followers'].each do |path|
  get path do
    ordered_collection(File.join(PUBLIC_DIR, path)).to_json
  end
end

get '/tags/:tag' do |tag|
  ordered_collection(File.join(TAGS, tag)).to_json
end

helpers do
  # https://github.com/mastodon/mastodon/blob/main/app/controllers/concerns/signature_verification.rb
  def verify!
    # digest
    sha256 = OpenSSL::Digest.new('SHA256')
    digest = "SHA-256=#{sha256.base64digest(@body)}"
    halt 403 unless digest == request.env['HTTP_DIGEST']

    # signature
    signature_params = {}
    request.env['HTTP_SIGNATURE'].split(',').each do |pair|
      k, v = pair.split('=')
      signature_params[k] = v.gsub('"', '')
    end

    key_id = signature_params['keyId']
    headers = signature_params['headers']
    signature = Base64.decode64(signature_params['signature'])

    actor = fetch key_id
    halt 403 unless actor

    key = OpenSSL::PKey::RSA.new(actor['publicKey']['publicKeyPem'])

    comparison = headers.split(' ').map do |signed_params_name|
      if signed_params_name == '(request-target)'
        '(request-target): post /inbox'
      elsif signed_params_name == 'content-type'
        "#{signed_params_name}: #{request.env['CONTENT_TYPE']}"
      else
        "#{signed_params_name}: #{request.env["HTTP_#{signed_params_name.upcase}"]}"
      end
    end.join("\n")

    halt 403 unless key.verify(OpenSSL::Digest.new('SHA256'), signature, comparison)
  end

  def create
    return unless @object
    return if object_exists?

    File.open(object_file, 'w+') { |f| f.puts @object.to_json }
    return unless @object['inReplyTo']

    @object = fetch @object['inReplyTo']
    create if @object
  end

  def announce
    create
  end

  def follow
    File.open(File.join(FOLLOWERS, "#{mention(@activity['actor'])}.json"), 'w+') { |f| f.puts @body }
    outbox 'Accept', @activity, [@activity['actor']]
  end

  def accept
    return unless @object['type'] == 'Follow'

    File.open(File.join(FOLLOWING_DIR, "#{mention(@object['object'])}.json"), 'w+') { |f| f.puts @object.to_json }
  end

  def undo
    return unless @object['type'] == 'Follow'

    Dir[File.join(FOLLOWERS, '*.json')].each do |follower|
      FileUtils.rm follower if JSON.parse(File.read(follower))['actor'] == @object['actor']
    end
  end

  def inbox
    Dir[File.join(INBOX_DIR, 'note', '*.json')].collect do |file|
      JSON.parse(File.read(file))
    end.sort_by { |o| o['published'] }
  end

  def object_exists?
    !inbox.select { |o| o['id'] == @object['id'] }.empty?
  end

  def object_file
    dir = File.join 'inbox', @object['type'].downcase
    FileUtils.mkdir_p dir
    File.join dir, "#{Time.now.strftime('%Y-%m-%dT%H:%M:%S.%N')}.json"
  end

  def ordered_collection(dir)
    posts = Dir[File.join(dir, '*.json')].collect { |f| JSON.parse(File.read(f)) }.sort_by { |o| o['published'] }
    {
      '@context' => 'https://www.w3.org/ns/activitystreams',
      'summary' => "#{USER} #{dir}",
      'type' => 'OrderedCollection',
      'totalItems' => posts.size,
      'orderedItems' => posts
    }
  end
end