Mercurial > zeropaste

annotate app/models/paste.rb @ 384:7ba798dee39d

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
No need to assign to variable when it can be just chained.
author nanaya <me@myconan.net>
date Sun, 07 Jun 2015 02:59:29 +0900
parents 2ad092e60975
children 9369ad2f2ce8
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
2
42de15334db1 Added the pastes.
Edho Arief <edho@myconan.net>
parents:
diff changeset
1 class Paste < ActiveRecord::Base
210
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
2 attr_accessor :is_private
379
2ad092e60975 [mq]: new-design
nanaya <me@myconan.net>
parents: 377
diff changeset
3 after_initialize :set_privacy, :if => :new_record?
2ad092e60975 [mq]: new-design
nanaya <me@myconan.net>
parents: 377
diff changeset
4
70
8f0fb869e770 Limit pastes to 100/hour per IP address.
Edho Arief <edho@myconan.net>
parents: 51
diff changeset
5 before_validation :paste_limit
22
032686a0c995 Added newline converter, integrity fixups.
Edho Arief <edho@myconan.net>
parents: 2
diff changeset
6 before_validation :convert_newlines
2
42de15334db1 Added the pastes.
Edho Arief <edho@myconan.net>
parents:
diff changeset
7 before_validation :set_paste_hash
176
a9dba6a3008b Initial work to add paste deletion.
Edho Arief <edho@myconan.net>
parents: 71
diff changeset
8 before_validation :set_paste_key
210
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
9 before_validation :set_paste_secret
176
a9dba6a3008b Initial work to add paste deletion.
Edho Arief <edho@myconan.net>
parents: 71
diff changeset
10 validates :paste, :paste_hash, :key, :ip, :presence => true
267
0bf1d6f75baa Accidentally limited pastes to 0 characters.
edogawaconan <me@myconan.net>
parents: 265
diff changeset
11 validates :paste, :length => { :maximum => 1_000_000 }
2
42de15334db1 Added the pastes.
Edho Arief <edho@myconan.net>
parents:
diff changeset
12
210
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
13 def to_param
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
14 path
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
15 end
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
16
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
17 def self.safe_find(raw_id)
247
9aa67da381b0 Make sure to split a string.
edogawaconan <me@myconan.net>
parents: 230
diff changeset
18 id, secret = raw_id.to_s.split("-")
210
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
19 return unless id.to_i.to_s == id
229
388504e43bcf Properly return 404 when requesting invalid paste.
Edho Arief <edho@myconan.net>
parents: 210
diff changeset
20 begin
330
96631607785f Even more coding style.
edogawaconan <me@myconan.net>
parents: 316
diff changeset
21 where(:secret => secret).find(id)
229
388504e43bcf Properly return 404 when requesting invalid paste.
Edho Arief <edho@myconan.net>
parents: 210
diff changeset
22 rescue ActiveRecord::RecordNotFound
388504e43bcf Properly return 404 when requesting invalid paste.
Edho Arief <edho@myconan.net>
parents: 210
diff changeset
23 nil
388504e43bcf Properly return 404 when requesting invalid paste.
Edho Arief <edho@myconan.net>
parents: 210
diff changeset
24 end
210
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
25 end
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
26
373
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
27 def self.graceful_create(params)
376
02903dc8214d Fix code styles.
nanaya <me@myconan.net>
parents: 373
diff changeset
28 paste = new(params)
373
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
29 fresh = true
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
30 created = true
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
31
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
32 begin
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
33 created = paste.save
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
34 rescue ActiveRecord::RecordNotUnique
376
02903dc8214d Fix code styles.
nanaya <me@myconan.net>
parents: 373
diff changeset
35 paste = find_by(:ip => paste.ip, :paste_hash => paste.paste_hash)
373
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
36 fresh = false
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
37 end
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
38
376
02903dc8214d Fix code styles.
nanaya <me@myconan.net>
parents: 373
diff changeset
39 [created, paste, fresh]
373
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
40 end
6e3e1e7b0212 Handle unique error in model instead of controller.
nanaya <me@myconan.net>
parents: 330
diff changeset
41
316
61f7f258a6fb Move from-gzip paste parsing to model.
edogawaconan <me@myconan.net>
parents: 290
diff changeset
42 def paste_gzip=(paste)
61f7f258a6fb Move from-gzip paste parsing to model.
edogawaconan <me@myconan.net>
parents: 290
diff changeset
43 self.paste = ActiveSupport::Gzip.decompress paste
61f7f258a6fb Move from-gzip paste parsing to model.
edogawaconan <me@myconan.net>
parents: 290
diff changeset
44 end
61f7f258a6fb Move from-gzip paste parsing to model.
edogawaconan <me@myconan.net>
parents: 290
diff changeset
45
61f7f258a6fb Move from-gzip paste parsing to model.
edogawaconan <me@myconan.net>
parents: 290
diff changeset
46 def paste_gzip_base64=(paste)
61f7f258a6fb Move from-gzip paste parsing to model.
edogawaconan <me@myconan.net>
parents: 290
diff changeset
47 self.paste_gzip = Base64.decode64(paste)
61f7f258a6fb Move from-gzip paste parsing to model.
edogawaconan <me@myconan.net>
parents: 290
diff changeset
48 end
61f7f258a6fb Move from-gzip paste parsing to model.
edogawaconan <me@myconan.net>
parents: 290
diff changeset
49
265
6cca1ab53337 Infinitely better error messages and notice.
edogawaconan <me@myconan.net>
parents: 255
diff changeset
50 def safe_destroy(param_key)
6cca1ab53337 Infinitely better error messages and notice.
edogawaconan <me@myconan.net>
parents: 255
diff changeset
51 if key == param_key
6cca1ab53337 Infinitely better error messages and notice.
edogawaconan <me@myconan.net>
parents: 255
diff changeset
52 destroy
6cca1ab53337 Infinitely better error messages and notice.
edogawaconan <me@myconan.net>
parents: 255
diff changeset
53 else
6cca1ab53337 Infinitely better error messages and notice.
edogawaconan <me@myconan.net>
parents: 255
diff changeset
54 errors.add(:key, "is invalid")
6cca1ab53337 Infinitely better error messages and notice.
edogawaconan <me@myconan.net>
parents: 255
diff changeset
55 false
6cca1ab53337 Infinitely better error messages and notice.
edogawaconan <me@myconan.net>
parents: 255
diff changeset
56 end
6cca1ab53337 Infinitely better error messages and notice.
edogawaconan <me@myconan.net>
parents: 255
diff changeset
57 end
6cca1ab53337 Infinitely better error messages and notice.
edogawaconan <me@myconan.net>
parents: 255
diff changeset
58
210
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
59 def path
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
60 [id, secret.presence].compact.join("-")
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
61 end
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
62
2
42de15334db1 Added the pastes.
Edho Arief <edho@myconan.net>
parents:
diff changeset
63 def set_paste_hash
51
36d07f047ec2 Or maybe not. Still too long with b62. Backed out changeset ba29d6394863
Edho Arief <edho@myconan.net>
parents: 46
diff changeset
64 self.paste_hash = Digest::SHA512.hexdigest("#{paste}\n")
2
42de15334db1 Added the pastes.
Edho Arief <edho@myconan.net>
parents:
diff changeset
65 end
22
032686a0c995 Added newline converter, integrity fixups.
Edho Arief <edho@myconan.net>
parents: 2
diff changeset
66
176
a9dba6a3008b Initial work to add paste deletion.
Edho Arief <edho@myconan.net>
parents: 71
diff changeset
67 def set_paste_key
a9dba6a3008b Initial work to add paste deletion.
Edho Arief <edho@myconan.net>
parents: 71
diff changeset
68 self.key ||= SecureRandom.hex(4)
a9dba6a3008b Initial work to add paste deletion.
Edho Arief <edho@myconan.net>
parents: 71
diff changeset
69 end
a9dba6a3008b Initial work to add paste deletion.
Edho Arief <edho@myconan.net>
parents: 71
diff changeset
70
210
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
71 def set_paste_secret
330
96631607785f Even more coding style.
edogawaconan <me@myconan.net>
parents: 316
diff changeset
72 self.secret = SecureRandom.hex(4) if is_private?
230
1c750d3cde1b Correct way to test is_private flag.
Edho Arief <edho@myconan.net>
parents: 229
diff changeset
73 end
1c750d3cde1b Correct way to test is_private flag.
Edho Arief <edho@myconan.net>
parents: 229
diff changeset
74
1c750d3cde1b Correct way to test is_private flag.
Edho Arief <edho@myconan.net>
parents: 229
diff changeset
75 def is_private?
330
96631607785f Even more coding style.
edogawaconan <me@myconan.net>
parents: 316
diff changeset
76 is_private == "1"
210
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
77 end
d59731c3c7bf Add support for is_private flag
Edho Arief <edho@myconan.net>
parents: 197
diff changeset
78
22
032686a0c995 Added newline converter, integrity fixups.
Edho Arief <edho@myconan.net>
parents: 2
diff changeset
79 def convert_newlines
330
96631607785f Even more coding style.
edogawaconan <me@myconan.net>
parents: 316
diff changeset
80 self.paste = paste.to_s.gsub("\r\n", "\n").gsub("\r", "\n")
22
032686a0c995 Added newline converter, integrity fixups.
Edho Arief <edho@myconan.net>
parents: 2
diff changeset
81 end
032686a0c995 Added newline converter, integrity fixups.
Edho Arief <edho@myconan.net>
parents: 2
diff changeset
82
70
8f0fb869e770 Limit pastes to 100/hour per IP address.
Edho Arief <edho@myconan.net>
parents: 51
diff changeset
83 def paste_limit
377
4e6afc8140fb Use Time.zone as per rubocop recommendation.
nanaya <me@myconan.net>
parents: 376
diff changeset
84 ip_post_recent_count = self.class.where(:ip => ip).where("created_at > ?", Time.zone.now - 1.hour).count
70
8f0fb869e770 Limit pastes to 100/hour per IP address.
Edho Arief <edho@myconan.net>
parents: 51
diff changeset
85 errors.add :base, :limit if ip_post_recent_count > 100
8f0fb869e770 Limit pastes to 100/hour per IP address.
Edho Arief <edho@myconan.net>
parents: 51
diff changeset
86 end
8f0fb869e770 Limit pastes to 100/hour per IP address.
Edho Arief <edho@myconan.net>
parents: 51
diff changeset
87
379
2ad092e60975 [mq]: new-design
nanaya <me@myconan.net>
parents: 377
diff changeset
88 def set_privacy
2ad092e60975 [mq]: new-design
nanaya <me@myconan.net>
parents: 377
diff changeset
89 self.is_private ||= "0"
2ad092e60975 [mq]: new-design
nanaya <me@myconan.net>
parents: 377
diff changeset
90 end
2ad092e60975 [mq]: new-design
nanaya <me@myconan.net>
parents: 377
diff changeset
91
22
032686a0c995 Added newline converter, integrity fixups.
Edho Arief <edho@myconan.net>
parents: 2
diff changeset
92 def self.fix_all
255
a894d7696b7e More useful return of Paste.fix_all.
edogawaconan <me@myconan.net>
parents: 247
diff changeset
93 stats = Hash.new(0)
376
02903dc8214d Fix code styles.
nanaya <me@myconan.net>
parents: 373
diff changeset
94 all.find_each do |p|
22
032686a0c995 Added newline converter, integrity fixups.
Edho Arief <edho@myconan.net>
parents: 2
diff changeset
95 p.save
255
a894d7696b7e More useful return of Paste.fix_all.
edogawaconan <me@myconan.net>
parents: 247
diff changeset
96 stats[:count] += 1
a894d7696b7e More useful return of Paste.fix_all.
edogawaconan <me@myconan.net>
parents: 247
diff changeset
97 stats[:private] += 1 if p.secret
22
032686a0c995 Added newline converter, integrity fixups.
Edho Arief <edho@myconan.net>
parents: 2
diff changeset
98 end
330
96631607785f Even more coding style.
edogawaconan <me@myconan.net>
parents: 316
diff changeset
99 stats
22
032686a0c995 Added newline converter, integrity fixups.
Edho Arief <edho@myconan.net>
parents: 2
diff changeset
100 end
2
42de15334db1 Added the pastes.
Edho Arief <edho@myconan.net>
parents:
diff changeset
101 end