@kinvolk/headlamp-plugin/official-plugins/knative/src/components/revisions/detail/hooks/useRevisionActions.ts

The needed infrastructure for building Headlamp plugins.

/*
 * Copyright 2025 The Kubernetes Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { useSnackbar } from 'notistack';
import { KRevision, KService } from '../../../../resources/knative';

export function useRevisionActions() {
  const { enqueueSnackbar } = useSnackbar();

  const isSafeToDelete = (revision: KRevision, kservice: KService | null) => {
    if (!kservice) return { safe: true };

    const traffic = revision.getTrafficInService(kservice);
    const hasActiveTraffic = traffic.some(t => (t.percent || 0) > 0 || !!t.tag);

    if (hasActiveTraffic) {
      return {
        safe: false,
        reason: 'Revision is currently receiving traffic or has an assigned tag.',
      };
    }

    return { safe: true };
  };

  const deleteRevision = async (revision: KRevision, kservice: KService | null) => {
    const { safe, reason } = isSafeToDelete(revision, kservice);
    if (!safe) {
      enqueueSnackbar(reason, { variant: 'error' });
      return;
    }

    try {
      await revision.delete();
      enqueueSnackbar(`Successfully deleted Revision ${revision.metadata.name}!`, {
        variant: 'success',
      });
    } catch (err) {
      const errMessage = err instanceof Error ? err.message : String(err);
      enqueueSnackbar(`Failed to delete Revision ${revision.metadata.name}: ${errMessage}`, {
        variant: 'error',
      });
    }
  };

  return { deleteRevision, isSafeToDelete };
}