// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <tuple>

struct A{
  int a;
  int b;
};

A aggregate_with_very_very_very_long_name = {1, 2};

int UnsafeIndex();

void Test() {
  // Expected rewrite:
  // std::array<A, 1> a_1 = {{{1, 2}}};
  A a_1[1] = {{1, 2}};
  std::ignore = a_1[UnsafeIndex()];

  // Expected rewrite:
  // std::array<A, 2> a_2 = {{{1, 2}, {3, 4}}};
  A a_2[2] = {{1, 2}, {3, 4}};
  std::ignore = a_2[UnsafeIndex()];

  // Expected rewrite:
  // std::array<A, 2> a_2_long = {aggregate_with_very_very_very_long_name,
  //                              aggregate_with_very_very_very_long_name};
  A a_2_long[2] = {aggregate_with_very_very_very_long_name,
                   aggregate_with_very_very_very_long_name};
  std::ignore = a_2_long[UnsafeIndex()];

  // Expected rewrite:
  // std::array<A, 5> a_5 = {{
  //     {1, 2},
  //     {3, 4},
  //     {5, 6},
  //     {7, 8},
  //     {9, 10},
  // }};
  A a_5[5] = {{1, 2}, {3, 4}, {5, 6}, {7, 8}, {9, 10}};
  std::ignore = a_5[UnsafeIndex()];

  // Expected rewrite:
  // std::array<A, 5> a_5_long = {{
  //     {1, 2},
  //     {3, 4},
  //     aggregate_with_very_very_very_long_name,
  //     {7, 8},
  //     {9, 10},
  // }};
  A a_5_long[5] = {
      {1, 2}, {3, 4}, aggregate_with_very_very_very_long_name, {7, 8}, {9, 10}};
  std::ignore = a_5_long[UnsafeIndex()];

  // Expected rewrite:
  // std::array<A, 6> a_6 = {{
  //     {1, 2},
  //     {3, 4},
  //     {5, 6},
  //     {7, 8},
  //     {9, 10},
  //     {11, 12},
  // }};
  A a_6[6] = {{1, 2}, {3, 4}, {5, 6}, {7, 8}, {9, 10}, {11, 12}};
  std::ignore = a_6[UnsafeIndex()];

  // Expected rewrite:
  // std::array<A, 6> a_6_with_new_line = {{
  //     {1, 2},
  //     {3, 4},
  //     {5, 6},
  //     {7, 8},
  //     {9, 10},
  //     {11, 12},
  // }};
  A a_6_with_new_line[6] = {
    {1, 2}, {3, 4}, {5, 6},
    {7, 8}, {9, 10}, {11, 12}
  };
  std::ignore = a_6_with_new_line[UnsafeIndex()];

  // Expected rewrite:
  // std::array<A, 5> a_5_with_trailing_comma = {{
  //     {1, 2},
  //     {3, 4},
  //     {5, 6},
  //     {7, 8},
  //     {9, 10},
  // }};
  A a_5_with_trailing_comma[5] = {
      {1, 2}, {3, 4}, {5, 6}, {7, 8}, {9, 10},
  };
  std::ignore = a_5_with_trailing_comma[UnsafeIndex()];

  // Expected rewrite:
  // std::array<A, 5> a_5_with_trailing_comma_2 = {{
  //     {1, 2},
  //     {3, 4},
  //     {5, 6},
  //     {7, 8},
  //     {9, 10},
  // }};
  A a_5_with_trailing_comma_2[5] = {
       {1, 2},
       {3, 4},
       {5, 6},
       {7, 8},
       {9, 10},
  };
  std::ignore = a_5_with_trailing_comma_2[UnsafeIndex()];
}