#!/usr/bin/env python
"""Provides conversion functions to be used during RDFProtoStruct migration."""

from grr_response_proto import flows_pb2
from grr_response_server.flows.general import filesystem


def ToProtoListDirectoryArgs(
    rdf: filesystem.ListDirectoryArgs,
) -> flows_pb2.ListDirectoryArgs:
  return rdf.AsPrimitiveProto()


def ToRDFListDirectoryArgs(
    proto: flows_pb2.ListDirectoryArgs,
) -> filesystem.ListDirectoryArgs:
  return filesystem.ListDirectoryArgs.FromSerializedBytes(
      proto.SerializeToString()
  )


def ToProtoRecursiveListDirectoryArgs(
    rdf: filesystem.RecursiveListDirectoryArgs,
) -> flows_pb2.RecursiveListDirectoryArgs:
  return rdf.AsPrimitiveProto()


def ToRDFRecursiveListDirectoryArgs(
    proto: flows_pb2.RecursiveListDirectoryArgs,
) -> filesystem.RecursiveListDirectoryArgs:
  return filesystem.RecursiveListDirectoryArgs.FromSerializedBytes(
      proto.SerializeToString()
  )