/*
 *
 *  Licensed to the Apache Software Foundation (ASF) under one or more
 *  contributor license agreements.  See the NOTICE file distributed with
 *  this work for additional information regarding copyright ownership.
 *  The ASF licenses this file to You 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 {CAParam, StudioMetaStoreParam, StudioParam} from "@/components/Studio/StudioEdit/data";
import {request2} from "@/components/Common/crud";
import {request} from "umi";

export async function executeSql(params: StudioParam) {
  return request<API.Result>('/api/studio/executeSql', {
    method: 'POST',
    data: {
      ...params,
    },
  });
}

export async function executeDDL(params: StudioParam) {
  return request<API.Result>('/api/studio/executeDDL', {
    method: 'POST',
    data: {
      ...params,
    },
  });
}

export async function explainSql(params: StudioParam) {
  return request<API.Result>('/api/studio/explainSql', {
    method: 'POST',
    data: {
      ...params,
    },
  });
}

export async function getStreamGraph(params: StudioParam) {
  return request<API.Result>('/api/studio/getStreamGraph', {
    method: 'POST',
    data: {
      ...params,
    },
  });
}

export async function getJobPlan(params: StudioParam) {
  return request<API.Result>('/api/studio/getJobPlan', {
    method: 'POST',
    data: {
      ...params,
    },
  });
}

export async function getJobData(jobId: string) {
  return request<API.Result>('/api/studio/getJobData', {
    method: 'GET',
    params: {
      jobId,
    },
  });
}

export async function getCatalogueTreeData(params?: StudioParam) {
  return request2<API.Result>('/api/catalogue/getCatalogueTreeData', {
    method: 'POST',
    data: {
      ...params,
    },
  });
}

export async function getLineage(params: CAParam) {
  return request<API.Result>('/api/studio/getLineage', {
    method: 'POST',
    data: {
      ...params,
    },
  });
}

export async function getMSCatalogs(params: StudioMetaStoreParam) {
  return request<API.Result>('/api/studio/getMSCatalogs', {
    method: 'POST',
    data: {
      ...params,
    },
  });
}

export async function getMSSchemaInfo(params: StudioMetaStoreParam) {
  return request<API.Result>('/api/studio/getMSSchemaInfo', {
    method: 'POST',
    data: {
      ...params,
    },
  });
}

//海豚 获取任务定义
export async function getTaskDefinition(dinkyTaskId: string) {
  return request<API.Result>('/api/scheduler/task', {
    method: 'GET',
    params: {
      dinkyTaskId,
    },
  });
}

//海豚 获取任务定义集合
export async function getTaskMainInfos(dinkyTaskId: string) {
  return request<API.Result>('/api/scheduler/upstream/tasks', {
    method: 'GET',
    params: {
      dinkyTaskId,
    },
  });
}

//海豚 创建任务定义
export async function createTaskDefinition(dinkyTaskId: string, upstreamCodes: string, params: object) {
  return request<API.Result>('/api/scheduler/task', {
    method: 'POST',
    params: {
      dinkyTaskId,
      upstreamCodes
    },
    data: {
      ...params,
    },
  });
}

//海豚 更新任务定义
export async function updateTaskDefinition(processCode: string, projectCode: string, taskCode: string, upstreamCodes: string, params: object) {
  return request<API.Result>('/api/scheduler/task', {
    method: 'PUT',
    params: {
      processCode,
      projectCode,
      taskCode,
      upstreamCodes
    },
    data: {
      ...params,
    },
  });
}
