// Copyright (C) 2024 The Android Open Source Project
//
// 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 {SqlTableDescription} from '../../components/widgets/sql/table/table_description';
import {
  ArgSetIdColumn,
  DurationColumn,
  ProcessIdColumn,
  SchedIdColumn,
  SliceIdColumn,
  StandardColumn,
  ThreadIdColumn,
  ThreadStateIdColumn,
  TimestampColumn,
} from '../../components/widgets/sql/table/columns';

export function getThreadTable(): SqlTableDescription {
  return {
    name: 'thread',
    columns: [
      new ThreadIdColumn('utid', {type: 'id'}),
      new StandardColumn('tid'),
      new StandardColumn('name'),
      new TimestampColumn('start_ts'),
      new TimestampColumn('end_ts'),
      new ProcessIdColumn('upid', {notNull: true}),
      new StandardColumn('is_main_thread'),
    ],
  };
}

export function getProcessTable(): SqlTableDescription {
  return {
    name: 'process',
    columns: [
      new ProcessIdColumn('upid', {type: 'id'}),
      new StandardColumn('pid'),
      new StandardColumn('name'),
      new TimestampColumn('start_ts'),
      new TimestampColumn('end_ts'),
      new ProcessIdColumn('parent_upid'),
      new StandardColumn('uid'),
      new StandardColumn('android_appid'),
      new StandardColumn('cmdline', {startsHidden: true}),
      new StandardColumn('machine_id'),
      new ArgSetIdColumn('arg_set_id'),
    ],
  };
}

export function getSliceTable(): SqlTableDescription {
  return {
    imports: ['slices.with_context'],
    name: 'thread_or_process_slice',
    displayName: 'thread_or_process_slice',
    columns: [
      new SliceIdColumn('id', {notNull: true, type: 'id'}),
      new TimestampColumn('ts'),
      new DurationColumn('dur'),
      new StandardColumn('category'),
      new StandardColumn('name'),
      new StandardColumn('track_id', {startsHidden: true}),
      new ThreadIdColumn('utid'),
      new ProcessIdColumn('upid'),
      new StandardColumn('depth', {startsHidden: true}),
      new SliceIdColumn('parent_id'),
      new ArgSetIdColumn('arg_set_id'),
    ],
  };
}

export function getAndroidLogsTable(): SqlTableDescription {
  return {
    name: 'android_logs',
    columns: [
      new StandardColumn('id'),
      new TimestampColumn('ts'),
      new StandardColumn('tag'),
      new StandardColumn('prio'),
      new ThreadIdColumn('utid'),
      new ProcessIdColumn({
        column: 'upid',
        source: {
          table: 'thread',
          joinOn: {utid: 'utid'},
        },
      }),
      new StandardColumn('msg'),
    ],
  };
}

export function getSchedTable(): SqlTableDescription {
  return {
    name: 'sched',
    columns: [
      new SchedIdColumn('id'),
      new TimestampColumn('ts'),
      new DurationColumn('dur'),
      new StandardColumn('cpu'),
      new StandardColumn('priority'),
      new ThreadIdColumn('utid'),
      new ProcessIdColumn({
        column: 'upid',
        source: {
          table: 'thread',
          joinOn: {utid: 'utid'},
        },
      }),
      new StandardColumn('end_state'),
      new StandardColumn('ucpu', {startsHidden: true}),
    ],
  };
}

export function getThreadStateTable(): SqlTableDescription {
  return {
    name: 'thread_state',
    columns: [
      new ThreadStateIdColumn('id'),
      new TimestampColumn('ts'),
      new DurationColumn('dur'),
      new StandardColumn('state'),
      new StandardColumn('cpu'),
      new ThreadIdColumn('utid'),
      new ProcessIdColumn({
        column: 'upid',
        source: {
          table: 'thread',
          joinOn: {utid: 'utid'},
        },
      }),
      new StandardColumn('io_wait'),
      new StandardColumn('blocked_function'),
      new ThreadIdColumn('waker_utid'),
      new ThreadStateIdColumn('waker_id'),
      new StandardColumn('irq_context'),
      new StandardColumn('ucpu', {startsHidden: true}),
    ],
  };
}