GroupsAndLabels.php
82 lines
| 1 | <?php |
| 2 | |
| 3 | namespace WCML\Compatibility\WcBookings\TranslationEditor; |
| 4 | |
| 5 | use WCML_Bookings; |
| 6 | use WCML\Compatibility\WcBookings\SharedHooks; |
| 7 | use WCML\TranslationJob\Hooks as TranslationJobHooks; |
| 8 | use WPML\FP\Obj; |
| 9 | use WPML\FP\Fns; |
| 10 | use WPML\FP\Str; |
| 11 | |
| 12 | class GroupsAndLabels implements \IWPML_Action { |
| 13 | |
| 14 | const PERSON_GROUP_SUFFIX = '/person'; |
| 15 | const PERSON_GROUP_LABEL = 'Person'; |
| 16 | const RESOURCE_GROUP_SUFFIX = '/resource'; |
| 17 | const RESOURCE_GROUP_LABEL = 'Resource'; |
| 18 | |
| 19 | public function add_hooks() { |
| 20 | add_filter( 'wpml_tm_adjust_translation_fields', [ $this, 'adjustFields' ], 10, 2 ); |
| 21 | } |
| 22 | |
| 23 | public function adjustFields( $fields, $job ) { |
| 24 | if ( ! SharedHooks::isBooking( $job->original_doc_id ) ) { |
| 25 | return $fields; |
| 26 | } |
| 27 | |
| 28 | foreach ( $fields as &$field ) { |
| 29 | $field = $this->adjustField( $field ); |
| 30 | } |
| 31 | |
| 32 | return $fields; |
| 33 | } |
| 34 | |
| 35 | private function adjustField( $field ) { |
| 36 | $typeStartsWith = Str::startsWith( Fns::__, Obj::prop( 'field_type', $field ) ); |
| 37 | |
| 38 | $getGroup = function( $item, $prefix ) { |
| 39 | switch ( $prefix ) { |
| 40 | case WCML_Bookings::PERSON_FIELD_PREFIX: |
| 41 | return $item . self::PERSON_GROUP_SUFFIX; |
| 42 | case WCML_Bookings::RESOURCE_FIELD_PREFIX: |
| 43 | return $item . self::RESOURCE_GROUP_SUFFIX; |
| 44 | } |
| 45 | return $item; |
| 46 | }; |
| 47 | |
| 48 | $getGroupLabel = function( $prefix, $default ) { |
| 49 | switch ( $prefix ) { |
| 50 | case WCML_Bookings::PERSON_FIELD_PREFIX: |
| 51 | return self::PERSON_GROUP_LABEL; |
| 52 | case WCML_Bookings::RESOURCE_FIELD_PREFIX: |
| 53 | return self::RESOURCE_GROUP_LABEL; |
| 54 | } |
| 55 | return $default; |
| 56 | }; |
| 57 | |
| 58 | $addTitlesAndGroups = function( $item, $prefix ) use ( $getGroup, $getGroupLabel ) { |
| 59 | $fieldData = Str::replace( $prefix, '', $item['field_type'] ); |
| 60 | $fieldParts = explode( ':', $fieldData ); |
| 61 | $title = end( $fieldParts ); |
| 62 | $group = reset( $fieldParts ); |
| 63 | $group = $getGroup( $group, $prefix ); |
| 64 | |
| 65 | $item['title'] = apply_filters( 'wpml_labelize_string', $title, 'TranslationJob' ); |
| 66 | $item['group'] = TranslationJobHooks::getTopLevelGroup(); |
| 67 | $item['group'][ $group ] = $getGroupLabel( $prefix, $group ); |
| 68 | |
| 69 | return $item; |
| 70 | }; |
| 71 | |
| 72 | if ( $typeStartsWith( WCML_Bookings::PERSON_FIELD_PREFIX ) ) { |
| 73 | $field = $addTitlesAndGroups( $field, WCML_Bookings::PERSON_FIELD_PREFIX ); |
| 74 | } elseif ( $typeStartsWith( WCML_Bookings::RESOURCE_FIELD_PREFIX ) ) { |
| 75 | $field = $addTitlesAndGroups( $field, WCML_Bookings::RESOURCE_FIELD_PREFIX ); |
| 76 | } |
| 77 | |
| 78 | return $field; |
| 79 | } |
| 80 | |
| 81 | } |
| 82 |