// Copyright 2019 Google Inc. All rights reserved.
//
// 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.

package status

import (
 "android/soong/ui/metrics"

 soong_metrics_proto "android/soong/ui/metrics/metrics_proto"
 "time"

 "google.golang.org/protobuf/proto"
)

func NewCriticalPath() *CriticalPath {
 return &CriticalPath{
  running: make(map[*Action]time.Time),
  nodes:   make(map[string]*node),
  clock:   osClock{},
 }
}

type CriticalPath struct {
 nodes   map[string]*node
 running map[*Action]time.Time

 start, end time.Time

 clock clock
}

type clock interface {
 Now() time.Time
}

type osClock struct{}

func (osClock) Now() time.Time { return time.Now() }

// A critical path node stores the critical path (the minimum time to build the node and all of its dependencies given
// perfect parallelism) for an node.
type node struct {
 action             *Action
 cumulativeDuration time.Duration
 duration           time.Duration
 input              *node
}

func (cp *CriticalPath) StartAction(action *Action) {
 start := cp.clock.Now()
 if cp.start.IsZero() {
  cp.start = start
 }
 cp.running[action] = start
}

func (cp *CriticalPath) FinishAction(action *Action) {
 if start, ok := cp.running[action]; ok {
  delete(cp.running, action)

  // Determine the input to this edge with the longest cumulative duration
  var criticalPathInput *node
  for _, input := range action.Inputs {
   if x := cp.nodes[input]; x != nil {
    if criticalPathInput == nil || x.cumulativeDuration > criticalPathInput.cumulativeDuration {
     criticalPathInput = x
    }
   }
  }

  end := cp.clock.Now()
  duration := end.Sub(start)

  cumulativeDuration := duration
  if criticalPathInput != nil {
   cumulativeDuration += criticalPathInput.cumulativeDuration
  }

  node := &node{
   action:             action,
   cumulativeDuration: cumulativeDuration,
   duration:           duration,
   input:              criticalPathInput,
  }

  for _, output := range action.Outputs {
   cp.nodes[output] = node
  }

  cp.end = end
 }
}

func (cp *CriticalPath) criticalPath() (path []*node, elapsedTime time.Duration, criticalTime time.Duration) {
 var max *node

 // Find the node with the longest critical path
 for _, node := range cp.nodes {
  if max == nil || node.cumulativeDuration > max.cumulativeDuration {
   max = node
  }
 }

 node := max
 for node != nil {
  path = append(path, node)
  node = node.input
 }
 if len(path) > 0 {
  // Log the critical path to the verbose log
  criticalTime = path[0].cumulativeDuration
  if !cp.start.IsZero() {
   elapsedTime = cp.end.Sub(cp.start)
  }
 }
 return
}

func (cp *CriticalPath) longRunningJobs() (nodes []*node) {
 threshold := time.Second * 30
 for _, node := range cp.nodes {
  if node != nil && node.duration > threshold {
   nodes = append(nodes, node)
  }
 }
 return
}

func addJobInfos(jobInfos *[]*soong_metrics_proto.JobInfo, sources []*node) {
 for _, job := range sources {
  jobInfo := soong_metrics_proto.JobInfo{}
  jobInfo.ElapsedTimeMicros = proto.Uint64(uint64(job.duration.Microseconds()))
  jobInfo.JobDescription = &job.action.Description
  *jobInfos = append(*jobInfos, &jobInfo)
 }
}

func (cp *CriticalPath) WriteToMetrics(met *metrics.Metrics) {
 criticalPathInfo := soong_metrics_proto.CriticalPathInfo{}
 path, elapsedTime, criticalTime := cp.criticalPath()
 criticalPathInfo.ElapsedTimeMicros = proto.Uint64(uint64(elapsedTime.Microseconds()))
 criticalPathInfo.CriticalPathTimeMicros = proto.Uint64(uint64(criticalTime.Microseconds()))
 addJobInfos(&criticalPathInfo.LongRunningJobs, cp.longRunningJobs())
 addJobInfos(&criticalPathInfo.CriticalPath, path)
 met.SetCriticalPathInfo(&criticalPathInfo)
}