Click button multiple times and have a different action each time

Never mind, for anybody wondering here it is.

alias: test wait_for_trigger
description: ""
trigger:
  - platform: state
    entity_id:
      - input_button.test_multi
    to: null
condition: []
action:
  - wait_for_trigger:
      - platform: state
        entity_id:
          - input_button.test_multi
        to: null
    timeout:
      hours: 0
      minutes: 0
      seconds: 2
      milliseconds: 0
  - if:
      - condition: template
        value_template: "{{ not wait.trigger }}"
    then:
      - data:
          message: action 1
        action: notify.mobile_app_matts_iphone
    else:
      - wait_for_trigger:
          - platform: state
            entity_id:
              - input_button.test_multi
            to: null
        timeout:
          hours: 0
          minutes: 0
          seconds: 2
          milliseconds: 0
      - if:
          - condition: template
            value_template: "{{ not wait.trigger }}"
        then:
          - data:
              message: action 2
            action: notify.mobile_app_matts_iphone
        else:
          - wait_for_trigger:
              - platform: state
                entity_id:
                  - input_button.test_multi
                to: null
            timeout:
              hours: 0
              minutes: 0
              seconds: 2
              milliseconds: 0
          - if:
              - condition: template
                value_template: "{{ not wait.trigger }}"
            then:
              - data:
                  message: action 3
                action: notify.mobile_app_matts_iphone
            else:
              - wait_for_trigger:
                  - platform: state
                    entity_id:
                      - input_button.test_multi
                    to: null
                timeout:
                  hours: 0
                  minutes: 0
                  seconds: 2
                  milliseconds: 0
              - if:
                  - condition: template
                    value_template: "{{ not wait.trigger }}"
                then:
                  - data:
                      message: action 4
                    action: notify.mobile_app_matts_iphone
                else:
                  - wait_for_trigger:
                      - platform: state
                        entity_id:
                          - input_button.test_multi
                        to: null
                    timeout:
                      hours: 0
                      minutes: 0
                      seconds: 2
                      milliseconds: 0
                  - if:
                      - condition: template
                        value_template: "{{ not wait.trigger }}"
                    then:
                      - data:
                          message: action 5
                        action: notify.mobile_app_matts_iphone
                    else:
                      - wait_for_trigger:
                          - platform: state
                            entity_id:
                              - input_button.test_multi
                            to: null
                        timeout:
                          hours: 0
                          minutes: 0
                          seconds: 2
                          milliseconds: 0
                      - if:
                          - condition: template
                            value_template: "{{ not wait.trigger }}"
                        then:
                          - data:
                              message: action 6
                            action: notify.mobile_app_matts_iphone
                        else:
                          - wait_for_trigger:
                              - platform: state
                                entity_id:
                                  - input_button.test_multi
                                to: null
                            timeout:
                              hours: 0
                              minutes: 0
                              seconds: 2
                              milliseconds: 0
                          - if:
                              - condition: template
                                value_template: "{{ not wait.trigger }}"
                            then:
                              - data:
                                  message: action 7
                                action: notify.mobile_app_matts_iphone
                            else:
                              - wait_for_trigger:
                                  - platform: state
                                    entity_id:
                                      - input_button.test_multi
                                    to: null
                                timeout:
                                  hours: 0
                                  minutes: 0
                                  seconds: 2
                                  milliseconds: 0
                              - if:
                                  - condition: template
                                    value_template: "{{ not wait.trigger }}"
                                then:
                                  - data:
                                      message: action 8
                                    action: notify.mobile_app_matts_iphone
                                else:
                                  - data:
                                      message: action default
                                    action: notify.mobile_app_matts_iphone
mode: single