wikimedia/mediawiki-extensions-Wikibase

import { HookHandler } from '@/HookHandler';
import { Store } from 'vuex';
import Application from '@/store/Application';
import { STATEMENT_TAINTED_STATE_UNTAINT, STATEMENT_TAINTED_STATE_TAINT } from '@/store/actionTypes';

export default class CSRHookHandler implements HookHandler {
    public addStore( store: Store<Application> ): void {
        document.querySelectorAll( '.wikibase-statementview' ).forEach( ( element ) => {
            const guid = element.getAttribute( 'id' );

            const untaintLink: HTMLElement | null = element.querySelector( '.wikibase-toolbar-button-untaint' );
            if ( untaintLink ) {
                untaintLink.onclick = () => {
                    store.dispatch( STATEMENT_TAINTED_STATE_UNTAINT, guid );
                    return false;
                };
            }
            const taintLink: HTMLElement | null = element.querySelector( '.wikibase-toolbar-button-taint' );
            if ( taintLink ) {
                taintLink.onclick = () => {
                    store.dispatch( STATEMENT_TAINTED_STATE_TAINT, guid );
                    return false;
                };
            }
        } );
    }
}