# Helper script used for generating history and working copy files and content.

# The file's name corresponds to its history. The number of changesets can

# be specified on the command line. With 2 changesets, files with names like

# content1_content2_content1-untracked are generated. The first two filename

# segments describe the contents in the two changesets. The third segment

# ("content1-untracked") describes the state in the working copy, i.e.

# the file has content "content1" and is untracked (since it was previously

# tracked, it has been forgotten).

#

# This script generates the filenames and their content, but it's up to the

# caller to tell hg about the state.

#

# There are two subcommands:

#   filelist <numchangesets>

#   state <numchangesets> (<changeset>|wc)

#

# Typical usage:

#

# $ python $TESTDIR/generate-working-copy-states.py state 2 1

# $ hg addremove --similarity 0

# $ hg commit -m 'first'

#

# $ python $TESTDIR/generate-working-copy-states.py state 2 1

# $ hg addremove --similarity 0

# $ hg commit -m 'second'

#

# $ python $TESTDIR/generate-working-copy-states.py state 2 wc

# $ hg addremove --similarity 0

# $ hg forget *_*_*-untracked

# $ rm *_*_missing-*

import os

import sys

# Generates pairs of (filename, contents), where 'contents' is a list

# describing the file's content at each revision (or in the working copy).

# At each revision, it is either None or the file's actual content. When not

# None, it may be either new content or the same content as an earlier

# revisions, so all of (modified,clean,added,removed) can be tested.

def generatestates(maxchangesets, parentcontents):

    depth = len(parentcontents)

    if depth == maxchangesets + 1:

        for tracked in (b'untracked', b'tracked'):

            filename = (

                b"_".join(

                    [

                        (content is None and b'missing' or content)

                        for content in parentcontents

                    ]

                )

                + b"-"

                + tracked

            )

            yield (filename, parentcontents)

    else:

        for content in {None, b'content' + (b"%d" % (depth + 1))} | set(

            parentcontents

        ):

            for combination in generatestates(

                maxchangesets, parentcontents + [content]

            ):

                yield combination

# retrieve the command line arguments

target = sys.argv[1]

maxchangesets = int(sys.argv[2])

if target == 'state':

    depth = sys.argv[3]

# sort to make sure we have stable output

combinations = sorted(generatestates(maxchangesets, []))

# compute file content

content = []

for filename, states in combinations:

    if target == 'filelist':

        print(filename.decode('ascii'))

    elif target == 'state':

        if depth == 'wc':

            # Make sure there is content so the file gets written and can be

            # tracked. It will be deleted outside of this script.

            content.append((filename, states[maxchangesets] or b'TOBEDELETED'))

        else:

            content.append((filename, states[int(depth) - 1]))

    else:

        print("unknown target:", target, file=sys.stderr)

        sys.exit(1)

# write actual content

for filename, data in content:

    if data is not None:

        f = open(filename, 'wb')

        f.write(data + b'\n')

        f.close()

    elif os.path.exists(filename):

        os.remove(filename)