package androidx.activity;

public class ComponentActivity extends androidx.core.app.ComponentActivity implements
  ...,
  MenuHost {
  // ...
  // MenuHost를 구현하기 위한 Helper 클래스가 내부에 선언되어 있습니다.
  private final MenuHostHelper mMenuHostHelper = new MenuHostHelper(this::invalidateMenu);

  // 아래 override된 함수는 MenuHost에서 선언된 함수를 구현한 부분입니다.
  // 대부분의 구현이 MenuHostHelper로 기능을 이관한 함수를 호출하는 형태로 되어 있습니다.
  @Override
  public boolean onCreateOptionsMenu(@NonNull Menu menu) {
    super.onCreateOptionsMenu(menu);
    mMenuHostHelper.onCreateMenu(menu, getMenuInflater());
    return true;
  }

  @Override
  public boolean onOptionsItemSelected(@NonNull MenuItem item) {
    if (super.onOptionsItemSelected(item)) {
      return true;
    }
    return mMenuHostHelper.onMenuItemSelected(item);
  }

  @Override
  public void addMenuProvider(@NonNull MenuProvider provider) {
    mMenuHostHelper.addMenuProvider(provider);
  }

  @Override
  public void addMenuProvider(@NonNull MenuProvider provider, @NonNull LifecycleOwner owner) {
    mMenuHostHelper.addMenuProvider(provider, owner);
  }

  @Override
  @SuppressLint("LambdaLast")
  public void addMenuProvider(@NonNull MenuProvider provider, @NonNull LifecycleOwner owner,
      @NonNull Lifecycle.State state) {
    mMenuHostHelper.addMenuProvider(provider, owner, state);
  }

  @Override
  public void removeMenuProvider(@NonNull MenuProvider provider) {
    mMenuHostHelper.removeMenuProvider(provider);
  }

  @Override
  public void invalidateMenu() {
    invalidateOptionsMenu();
  }
}

public interface MenuHost {
    // 전달된 MenuProvider를 MenuHost에 추가한다
    // 필요한 경우 MenuProvider를 수동으로 제거해야 한다
    void addMenuProvider(@NonNull MenuProvider provider);

    // MenuProvider를 MenuHost에 추가하며,
    // MenuProvider는 전달된 LifecycleOwner가 Lifecycle.Event.ON_DESTROY 이벤트를 수신하면 제거된다. 
    void addMenuProvider(@NonNull MenuProvider provider, @NonNull LifecycleOwner owner);

    // MenuProvider를 MenuHost에 추가한다.
    // LifecycleOwner가 전달된 Lifecycle.State에 도달하면 MenuProvider를 추가하며,
    // Lifecycle.State보다 down 되면 제거한다.
    @SuppressLint("LambdaLast")
    void addMenuProvider(@NonNull MenuProvider provider, @NonNull LifecycleOwner owner,
            @NonNull Lifecycle.State state);

    // MenuProvider를 MenuHost에서 제거한다.
    void removeMenuProvider(@NonNull MenuProvider provider);

    // Menu 정보를 갱신한다.
    void invalidateMenu();
}

public interface MenuProvider {
    // MenuProvider가 MenuItem을 메뉴로 확장하기 위해 MenuHost에 의해 호출된다.
    void onCreateMenu(@NonNull Menu menu, @NonNull MenuInflater menuInflater);

    // 메뉴에서 MenuItem이 선택되면 MenuHost에 의해 호출된다.
    boolean onMenuItemSelected(@NonNull MenuItem menuItem);
}

public class MenuHostHelper {

    private final Runnable mOnInvalidateMenuCallback;
    // 복수의 MenuProvider를 사용해 메뉴 노출
    private final CopyOnWriteArrayList<MenuProvider> mMenuProviders = new CopyOnWriteArrayList<>();

    // MenuHostHelper 초기화 시 InvalidateMenu Callback을 주입
    public MenuHostHelper(@NonNull Runnable onInvalidateMenuCallback) {
        mOnInvalidateMenuCallback = onInvalidateMenuCallback;
    }

    // 등록된 MenuProvider를 통해 menu를 구성
    public void onCreateMenu(@NonNull Menu menu, @NonNull MenuInflater menuInflater) {
        for (MenuProvider menuProvider : mMenuProviders) {
            menuProvider.onCreateMenu(menu, menuInflater);
        }
    }

    // 메뉴 선택시, MenuProvider에 처리되는지 체크 
    public boolean onMenuItemSelected(@NonNull MenuItem item) {
        for (MenuProvider menuProvider : mMenuProviders) {
            if (menuProvider.onMenuItemSelected(item)) {
                return true;
            }
        }
        return false;
    }

    // 전달된 MenuProvider를 MenuHost에 추가
    // InvalidateMenu로 갱신 처리
    public void addMenuProvider(@NonNull MenuProvider provider) {
        mMenuProviders.add(provider);
        mOnInvalidateMenuCallback.run();
    }

    // MenuProvider를 MenuHost에 추가하며,
    // MenuProvider는 전달된 LifecycleOwner가 Lifecycle.Event.ON_DESTROY 이벤트를 수신하면 제거된다. 
    public void addMenuProvider(@NonNull MenuProvider provider, @NonNull LifecycleOwner owner) {
        addMenuProvider(provider);
        Lifecycle lifecycle = owner.getLifecycle();
        lifecycle.addObserver((LifecycleEventObserver) (source, event) -> {
            if (event == Lifecycle.Event.ON_DESTROY) {
                removeMenuProvider(provider);
            }
        });
    }

    // MenuProvider를 MenuHost에 추가한다.
    // LifecycleOwner가 전달된 Lifecycle.State에 도달하면 MenuProvider를 추가하며,
    // Lifecycle.State보다 down 되면 제거한다.
    @SuppressLint("LambdaLast")
    public void addMenuProvider(@NonNull MenuProvider provider, @NonNull LifecycleOwner owner,
            @NonNull Lifecycle.State state) {
        Lifecycle lifecycle = owner.getLifecycle();
        lifecycle.addObserver((LifecycleEventObserver) (source, event) -> {
            if (event == Lifecycle.Event.upTo(state)) {
                addMenuProvider(provider);
            } else if (event == Lifecycle.Event.ON_DESTROY) {
                removeMenuProvider(provider);
            } else if (event == Lifecycle.Event.downFrom(state)) {
                removeMenuProvider(provider);
            }
        });
    }

    // MenuProvider를 MenuHost에서 제거
    public void removeMenuProvider(@NonNull MenuProvider provider) {
        mMenuProviders.remove(provider);
        mOnInvalidateMenuCallback.run();
    }
}

class ExampleActivity : ComponentActivity(R.layout.activity_example) {
  override fun onCreate(savedInstanceState: Bundle?) {
    super.onCreate(savedInstanceState)

    // "onCreateContextMenu"를 오버라이딩하지 않고 메뉴 추가
    addMenuProvider(object : MenuProvider {
      override fun onCreateMenu(menu: Menu, menuInflater: MenuInflater) {
        // 메뉴 inflate
        menuInflater.inflate(R.menu.example_menu, menu)
      }

      override fun onMenuItemSelected(menuItem: MenuItem): Boolean {
        // 메뉴 선택시 핸들링
        return true
      }
    })
  }
}

class ExampleFragment : Fragment(R.layout.fragment_example) {

  override fun onViewCreated(view: View, savedInstanceState: Bundle?) {
    // Activity에 정의된 MenuHost 인터페이스를 참조
    val menuHost: MenuHost = requireActivity()

    // "onCreateContextMenu"를 오버라이딩하지 않고 메뉴 추가
    // MenuProvider를 viewLifecycleOwner과 Lifecycle.State.RESUMED을 사용해 메뉴가 표시될 시기를 정의
    menuHost.addMenuProvider(object : MenuProvider {
      override fun onCreateMenu(menu: Menu, menuInflater: MenuInflater) {
        // 메뉴 inflate
        menuInflater.inflate(R.menu.example_menu, menu)
      }

      override fun onMenuItemSelected(menuItem: MenuItem): Boolean {
        // 메뉴 선택시 핸들링
        return true
      }
    }, viewLifecycleOwner, Lifecycle.State.RESUMED)
  }
}