RequiredUpdate.php
6.04 KB
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
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
<?php if ( ! defined( 'ABSPATH' ) ) exit;
/**
* Class NF_Abstracts_RequiredUpdate
*/
abstract class NF_Abstracts_RequiredUpdate
{
protected $_slug = '';
protected $_requires = array();
protected $_class_name = '';
protected $db;
public $response = array();
public $debug = false;
public $lock_process = false;
/**
* Constructor
*
* @since 3.4.0
*/
public function __construct( $data = array() )
{
// Save a reference to wpdb.
global $wpdb;
$this->db = $wpdb;
//Bail if we aren't in the admin.
if ( ! is_admin() ) return false;
// If we weren't provided with a slug or a class name...
if ( ! isset( $data[ 'slug' ] ) || ! isset( $data[ 'class_name' ] ) ) {
// Bail.
return false;
}
$this->_slug = $data[ 'slug' ];
$this->_class_name = $data[ 'class_name' ];
// Record debug settings if provided.
if ( isset( $data[ 'debug' ] ) ) $this->debug = $data[ 'debug' ];
}
/**
* Function to loop over the batch.
*
* @since 3.4.0
*/
public function process()
{
/**
* This function intentionlly left empty.
*/
}
/**
* Function to run any setup steps necessary to begin processing.
*
* @since 3.4.0
*/
public function startup()
{
/**
* This function intentionally left empty.
*/
}
/**
* Function to cleanup any lingering temporary elements of required updates after completion.
*
* @since 3.4.0
*/
public function cleanup()
{
// Delete our required updates data.
delete_option( 'ninja_forms_doing_required_updates' );
// Flag that updates are done.
update_option( 'ninja_forms_needs_updates', 0 );
// Set our new db version.
update_option( 'ninja_forms_db_version', Ninja_Forms::DB_VERSION );
// Fetch our list of completed updates.
$updates = get_option( 'ninja_forms_required_updates', array() );
// If we got something back...
if ( ! empty( $updates ) ) {
// Send out a call to telemetry.
Ninja_Forms()->dispatcher()->send( 'required_updates_complete', $updates );
}
// Output that we're done.
$this->response[ 'updatesRemaining' ] = 0;
$this->respond();
}
/**
* Function to dump our JSON response and kill processing.
*
* @since 3.4.0
*/
public function respond()
{
// Dump the response.
echo( json_encode( $this->response ) );
// Terminate processing.
die();
}
/**
* Function to run our table migrations.
*
* @param $callback (String) The callback function in the migration file.
*
* @since 3.4.0
*/
protected function migrate( $callback )
{
$migrations = new NF_Database_Migrations();
$migrations->do_upgrade( $callback );
}
/**
* Function to prepare our query values for insert.
*
* @param $value (Mixed) The value to be escaped for SQL.
* @return (String) The escaped (and possibly serialized) value of the string.
*
* @since 3.4.0
*/
public function prepare( $value )
{
// If our value is a number...
if ( is_float( $value ) ) {
// Exit early and return the value.
return $value;
}
// Serialize the value if necessary.
$escaped = maybe_serialize( $value );
// Escape it.
$escaped = $this->db->_real_escape( $escaped );
return $escaped;
}
/**
* Function used to call queries that are gated by debug.
*
* @param $sql (String) The query to be run.
* @return (Object) The response to the wpdb query call.
*
* @since 3.4.0
*/
protected function query( $sql )
{
// If we're not debugging...
if ( ! $this->debug ) {
// Run the query.
return $this->db->query( $sql );
} // Otherwise...
// Append the query to the response object.
$this->response[ 'queries' ][] = $sql;
// Return false.
return false;
}
/**
* Function to record the completion of our update in the DB.
*
* @since 3.4.0
*/
protected function confirm_complete()
{
// If we're not debugging...
if ( ! $this->debug ) {
// Fetch our required updates array.
$updates = get_option( 'ninja_forms_required_updates', array() );
// Get a timestamp.
date_default_timezone_set( 'UTC' );
$now = date( "Y-m-d H:i:s" );
// Append the current update to the array.
$updates[ $this->_slug ] = $now;
// Save it.
update_option( 'ninja_forms_required_updates', $updates );
}
}
/**
* Enable Maintenance mode
* Enables maintenance mode so the form will not render on the front end while updates are running.
*
* @since 3.4.0
*
* @param $prefix - the db prefix.
* @param $form_id - The id of the form.
*/
public function enable_maintenance_mode( $prefix, $form_id )
{
// Change maintenance column value to 1, run the query and then return.
$sql = $this->db->prepare( 'UPDATE `' . $prefix . 'nf3_upgrades` SET `maintenance` = 1 WHERE `id` = %d', $form_id );
$this->db->query( $sql );
return;
}
/**
* Disable Maintenance Mode
* Disables maintenance mode, so the form will be displayed on the front end..
*
* @since 3.4.0
*
* @param $prefix - the db prefix.
* @param $form_id - The id of the form.
*/
public function disable_maintenance_mode( $prefix, $form_id )
{
// Change maintenance column value to 0, run the query and then return.
$sql = $this->db->prepare( 'UPDATE `' . $prefix . 'nf3_upgrades` SET `maintenance` = 0 WHERE `id` = %d', $form_id );
$this->db->query( $sql );
return;
}
}