@@ -18,115 +18,115 @@ |
||
18 | 18 | */ |
19 | 19 | class ShareDraftContentSiteTreeExtension extends DataExtension |
20 | 20 | { |
21 | - /** |
|
22 | - * The number of days a shared link should be valid for, before expiring. |
|
23 | - * |
|
24 | - * @config |
|
25 | - * |
|
26 | - * @var int |
|
27 | - */ |
|
28 | - private static $valid_for_days = 30; |
|
29 | - |
|
30 | - /** |
|
31 | - * @var array |
|
32 | - */ |
|
33 | - private static $db = array( |
|
34 | - 'ShareTokenSalt' => 'Varchar(16)', |
|
35 | - ); |
|
36 | - |
|
37 | - /** |
|
38 | - * @var array |
|
39 | - */ |
|
40 | - private static $has_many = array( |
|
41 | - 'ShareTokens' => ShareToken::class, |
|
42 | - ); |
|
43 | - |
|
44 | - /** |
|
45 | - * @var array |
|
46 | - */ |
|
47 | - private static $allowed_actions = array( |
|
48 | - 'MakeShareDraftLink' |
|
49 | - ); |
|
50 | - |
|
51 | - /** |
|
52 | - * @return string |
|
53 | - */ |
|
54 | - public function ShareTokenLink() |
|
55 | - { |
|
56 | - $shareToken = $this->getNewShareToken(); |
|
57 | - |
|
58 | - return Controller::join_links( |
|
59 | - Director::absoluteBaseURL(), |
|
60 | - 'preview', |
|
61 | - $this->generateKey($shareToken->Token), |
|
62 | - $shareToken->Token |
|
63 | - ); |
|
64 | - } |
|
65 | - |
|
66 | - /** |
|
67 | - * @return ShareToken |
|
68 | - */ |
|
69 | - protected function getNewShareToken() |
|
70 | - { |
|
71 | - if (!$this->owner->ShareTokenSalt) { |
|
72 | - $this->owner->ShareTokenSalt = $this->getNewToken(); |
|
73 | - $this->owner->writeWithoutVersion(); |
|
74 | - } |
|
75 | - |
|
76 | - $found = null; |
|
77 | - $token = null; |
|
78 | - $tries = 1; |
|
79 | - $limit = 5; |
|
80 | - |
|
81 | - while (!$found && ($tries++ < $limit)) { |
|
82 | - $token = $this->getNewToken(); |
|
83 | - |
|
84 | - $found = ShareToken::get()->filter(array( |
|
85 | - "Token" => $token, |
|
86 | - "PageID" => $this->owner->ID, |
|
87 | - ))->first(); |
|
88 | - } |
|
89 | - |
|
90 | - $config = Config::forClass(__CLASS__); |
|
91 | - |
|
92 | - $validForDays = $config->get('valid_for_days'); |
|
93 | - |
|
94 | - $token = ShareToken::create(array( |
|
95 | - "Token" => $token, |
|
96 | - "ValidForDays" => $validForDays, |
|
97 | - "PageID" => $this->owner->ID, |
|
98 | - )); |
|
99 | - |
|
100 | - $token->write(); |
|
101 | - |
|
102 | - return $token; |
|
103 | - } |
|
104 | - |
|
105 | - /** |
|
106 | - * @return string |
|
107 | - */ |
|
108 | - protected function getNewToken() |
|
109 | - { |
|
110 | - $generator = new RandomGenerator(); |
|
111 | - |
|
112 | - return substr($generator->randomToken('sha256'), 0, 16); |
|
113 | - } |
|
114 | - |
|
115 | - /** |
|
116 | - * @param string $salt |
|
117 | - * |
|
118 | - * @return string |
|
119 | - */ |
|
120 | - public function generateKey($salt) |
|
121 | - { |
|
122 | - return hash_pbkdf2('sha256', $salt, $this->owner->ShareTokenSalt, 1000, 16); |
|
123 | - } |
|
124 | - |
|
125 | - /** |
|
126 | - * @return string |
|
127 | - */ |
|
128 | - public function getShareDraftLinkAction() |
|
129 | - { |
|
130 | - return $this->owner->Link('MakeShareDraftLink'); |
|
131 | - } |
|
21 | + /** |
|
22 | + * The number of days a shared link should be valid for, before expiring. |
|
23 | + * |
|
24 | + * @config |
|
25 | + * |
|
26 | + * @var int |
|
27 | + */ |
|
28 | + private static $valid_for_days = 30; |
|
29 | + |
|
30 | + /** |
|
31 | + * @var array |
|
32 | + */ |
|
33 | + private static $db = array( |
|
34 | + 'ShareTokenSalt' => 'Varchar(16)', |
|
35 | + ); |
|
36 | + |
|
37 | + /** |
|
38 | + * @var array |
|
39 | + */ |
|
40 | + private static $has_many = array( |
|
41 | + 'ShareTokens' => ShareToken::class, |
|
42 | + ); |
|
43 | + |
|
44 | + /** |
|
45 | + * @var array |
|
46 | + */ |
|
47 | + private static $allowed_actions = array( |
|
48 | + 'MakeShareDraftLink' |
|
49 | + ); |
|
50 | + |
|
51 | + /** |
|
52 | + * @return string |
|
53 | + */ |
|
54 | + public function ShareTokenLink() |
|
55 | + { |
|
56 | + $shareToken = $this->getNewShareToken(); |
|
57 | + |
|
58 | + return Controller::join_links( |
|
59 | + Director::absoluteBaseURL(), |
|
60 | + 'preview', |
|
61 | + $this->generateKey($shareToken->Token), |
|
62 | + $shareToken->Token |
|
63 | + ); |
|
64 | + } |
|
65 | + |
|
66 | + /** |
|
67 | + * @return ShareToken |
|
68 | + */ |
|
69 | + protected function getNewShareToken() |
|
70 | + { |
|
71 | + if (!$this->owner->ShareTokenSalt) { |
|
72 | + $this->owner->ShareTokenSalt = $this->getNewToken(); |
|
73 | + $this->owner->writeWithoutVersion(); |
|
74 | + } |
|
75 | + |
|
76 | + $found = null; |
|
77 | + $token = null; |
|
78 | + $tries = 1; |
|
79 | + $limit = 5; |
|
80 | + |
|
81 | + while (!$found && ($tries++ < $limit)) { |
|
82 | + $token = $this->getNewToken(); |
|
83 | + |
|
84 | + $found = ShareToken::get()->filter(array( |
|
85 | + "Token" => $token, |
|
86 | + "PageID" => $this->owner->ID, |
|
87 | + ))->first(); |
|
88 | + } |
|
89 | + |
|
90 | + $config = Config::forClass(__CLASS__); |
|
91 | + |
|
92 | + $validForDays = $config->get('valid_for_days'); |
|
93 | + |
|
94 | + $token = ShareToken::create(array( |
|
95 | + "Token" => $token, |
|
96 | + "ValidForDays" => $validForDays, |
|
97 | + "PageID" => $this->owner->ID, |
|
98 | + )); |
|
99 | + |
|
100 | + $token->write(); |
|
101 | + |
|
102 | + return $token; |
|
103 | + } |
|
104 | + |
|
105 | + /** |
|
106 | + * @return string |
|
107 | + */ |
|
108 | + protected function getNewToken() |
|
109 | + { |
|
110 | + $generator = new RandomGenerator(); |
|
111 | + |
|
112 | + return substr($generator->randomToken('sha256'), 0, 16); |
|
113 | + } |
|
114 | + |
|
115 | + /** |
|
116 | + * @param string $salt |
|
117 | + * |
|
118 | + * @return string |
|
119 | + */ |
|
120 | + public function generateKey($salt) |
|
121 | + { |
|
122 | + return hash_pbkdf2('sha256', $salt, $this->owner->ShareTokenSalt, 1000, 16); |
|
123 | + } |
|
124 | + |
|
125 | + /** |
|
126 | + * @return string |
|
127 | + */ |
|
128 | + public function getShareDraftLinkAction() |
|
129 | + { |
|
130 | + return $this->owner->Link('MakeShareDraftLink'); |
|
131 | + } |
|
132 | 132 | } |